Skip to content

Commit

Permalink
Shorter the namings
Browse files Browse the repository at this point in the history
Signed-off-by: JmPotato <[email protected]>
  • Loading branch information
JmPotato committed Jan 22, 2025
1 parent a4b3ceb commit f0753d0
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
6 changes: 3 additions & 3 deletions client/servicediscovery/callbacks.go
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ func (c *serviceCallbacks) addServiceURLsSwitchedCallback(cb func()) {
c.membersChangedCbs = append(c.membersChangedCbs, cb)
}

func (c *serviceCallbacks) callServiceModeUpdateCallback(mode pdpb.ServiceMode) {
func (c *serviceCallbacks) onServiceModeUpdate(mode pdpb.ServiceMode) {
c.RLock()
cb := c.serviceModeUpdateCb
c.RUnlock()
Expand All @@ -70,7 +70,7 @@ func (c *serviceCallbacks) callServiceModeUpdateCallback(mode pdpb.ServiceMode)
cb(mode)
}

func (c *serviceCallbacks) callLeaderSwitchedCallbacks(leader string) error {
func (c *serviceCallbacks) onLeaderSwitched(leader string) error {
c.RLock()
cbs := make([]leaderSwitchedCallbackFunc, len(c.leaderSwitchedCbs))
copy(cbs, c.leaderSwitchedCbs)
Expand All @@ -89,7 +89,7 @@ func (c *serviceCallbacks) callLeaderSwitchedCallbacks(leader string) error {
return nil
}

func (c *serviceCallbacks) callMembersChangedCallbacks() {
func (c *serviceCallbacks) onMembersChanged() {
c.RLock()
cbs := make([]func(), len(c.membersChangedCbs))
copy(cbs, c.membersChangedCbs)
Expand Down
14 changes: 7 additions & 7 deletions client/servicediscovery/service_discovery.go
Original file line number Diff line number Diff line change
Expand Up @@ -555,7 +555,7 @@ func (c *serviceDiscovery) updateServiceModeLoop() {
failpoint.Return()
})
failpoint.Inject("usePDServiceMode", func() {
c.callbacks.callServiceModeUpdateCallback(pdpb.ServiceMode_PD_SVC_MODE)
c.callbacks.onServiceModeUpdate(pdpb.ServiceMode_PD_SVC_MODE)
failpoint.Return()
})

Expand Down Expand Up @@ -854,15 +854,15 @@ func (c *serviceDiscovery) checkServiceModeChanged() error {
// If the method is not supported, we set it to pd mode.
// TODO: it's a hack way to solve the compatibility issue.
// we need to remove this after all maintained version supports the method.
c.callbacks.callServiceModeUpdateCallback(pdpb.ServiceMode_PD_SVC_MODE)
c.callbacks.onServiceModeUpdate(pdpb.ServiceMode_PD_SVC_MODE)
return nil
}
return err
}
if clusterInfo == nil || len(clusterInfo.ServiceModes) == 0 {
return errors.WithStack(errs.ErrNoServiceModeReturned)
}
c.callbacks.callServiceModeUpdateCallback(clusterInfo.ServiceModes[0])
c.callbacks.onServiceModeUpdate(clusterInfo.ServiceModes[0])
return nil
}

Expand Down Expand Up @@ -951,7 +951,7 @@ func (c *serviceDiscovery) updateURLs(members []*pdpb.Member) {
}
c.urls.Store(urls)
// Run callbacks to reflect the membership changes in the leader and followers.
c.callbacks.callMembersChangedCallbacks()
c.callbacks.onMembersChanged()
log.Info("[pd] update member urls", zap.Strings("old-urls", oldURLs), zap.Strings("new-urls", urls))
}

Expand All @@ -961,18 +961,18 @@ func (c *serviceDiscovery) switchLeader(url string) (bool, error) {
return false, nil
}

newConn, _ := c.GetOrCreateGRPCConn(url)
newConn, err := c.GetOrCreateGRPCConn(url)
// If gRPC connect is created successfully or leader is new, still saves.
if url != oldLeader.GetURL() || newConn != nil {
leaderClient := newPDServiceClient(url, url, newConn, true)
c.leader.Store(leaderClient)
}
// Run callbacks
if err := c.callbacks.callLeaderSwitchedCallbacks(url); err != nil {
if err := c.callbacks.onLeaderSwitched(url); err != nil {
return true, err
}
log.Info("[pd] switch leader", zap.String("new-leader", url), zap.String("old-leader", oldLeader.GetURL()))
return true, nil
return true, err
}

func (c *serviceDiscovery) updateFollowers(members []*pdpb.Member, leaderID uint64, leaderURL string) (changed bool) {
Expand Down

0 comments on commit f0753d0

Please sign in to comment.