Skip to content

Commit 70bb290

Browse files
slight refactor
1 parent dc04bb2 commit 70bb290

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

command_policy_resolver.go

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -118,17 +118,17 @@ var defaultPolicies = map[module]map[commandName]*routing.CommandPolicy{
118118
}
119119

120120
type CommandInfoResolver interface {
121-
getCommandPolicy(ctx context.Context, cmdName string) *routing.CommandPolicy
122-
setFallbackResolver(fallback CommandInfoResolver)
121+
GetCommandPolicy(ctx context.Context, cmdName string) *routing.CommandPolicy
122+
SetFallbackResolver(fallback CommandInfoResolver)
123123
}
124124

125-
type resolver struct {
125+
type internalCommandInfoResolver struct {
126126
resolve func(ctx context.Context, cmdName string) *routing.CommandPolicy
127127
fallBackResolver CommandInfoResolver
128128
}
129129

130-
func NewDefaultCommandPolicyResolver() *resolver {
131-
return &resolver{
130+
func NewDefaultCommandPolicyResolver() *internalCommandInfoResolver {
131+
return &internalCommandInfoResolver{
132132
resolve: func(ctx context.Context, cmdName string) *routing.CommandPolicy {
133133
module := "core"
134134
command := cmdName
@@ -147,19 +147,19 @@ func NewDefaultCommandPolicyResolver() *resolver {
147147
}
148148
}
149149

150-
func (r *resolver) getCommandPolicy(ctx context.Context, cmdName string) *routing.CommandPolicy {
150+
func (r *internalCommandInfoResolver) GetCommandPolicy(ctx context.Context, cmdName string) *routing.CommandPolicy {
151151
policy := r.resolve(ctx, cmdName)
152152
if policy != nil {
153153
return policy
154154
}
155155

156156
if r.fallBackResolver != nil {
157-
return r.fallBackResolver.getCommandPolicy(ctx, cmdName)
157+
return r.fallBackResolver.GetCommandPolicy(ctx, cmdName)
158158
}
159159

160160
return nil
161161
}
162162

163-
func (r *resolver) setFallbackResolver(fallbackResolver CommandInfoResolver) {
163+
func (r *internalCommandInfoResolver) SetFallbackResolver(fallbackResolver CommandInfoResolver) {
164164
r.fallBackResolver = fallbackResolver
165165
}

osscluster.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1998,7 +1998,7 @@ func (c *ClusterClient) extractCommandInfo(ctx context.Context, cmdName string)
19981998
}
19991999

20002000
func (c *ClusterClient) NewDynamicResolver() CommandInfoResolver {
2001-
return &resolver{
2001+
return &internalCommandInfoResolver{
20022002
resolve: c.extractCommandInfo,
20032003
}
20042004
}

osscluster_router.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ type slotResult struct {
2121

2222
// routeAndRun routes a command to the appropriate cluster nodes and executes it
2323
func (c *ClusterClient) routeAndRun(ctx context.Context, cmd Cmder, node *clusterNode) error {
24-
policy := c.cmdInfoResolver.getCommandPolicy(ctx, cmd.Name())
24+
policy := c.cmdInfoResolver.GetCommandPolicy(ctx, cmd.Name())
2525
if policy == nil {
2626
return c.executeDefault(ctx, cmd, node)
2727
}

0 commit comments

Comments
 (0)