Przeglądaj źródła

Merge pull request #6161 from sinsharat/master

etcdserver: stats/server - refactored
Xiang Li 9 lat temu
rodzic
commit
c33ea20fef
1 zmienionych plików z 5 dodań i 8 usunięć
  1. 5 8
      etcdserver/stats/server.go

+ 5 - 8
etcdserver/stats/server.go

@@ -123,17 +123,11 @@ func (ss *ServerStats) SendAppendReq(reqSize int) {
 	ss.Lock()
 	defer ss.Unlock()
 
-	now := time.Now()
-
-	if ss.State != raft.StateLeader {
-		ss.State = raft.StateLeader
-		ss.LeaderInfo.Name = ss.ID
-		ss.LeaderInfo.StartTime = now
-	}
+	ss.becomeLeader()
 
 	ss.sendRateQueue.Insert(
 		&RequestStats{
-			SendingTime: now,
+			SendingTime: time.Now(),
 			Size:        reqSize,
 		},
 	)
@@ -144,7 +138,10 @@ func (ss *ServerStats) SendAppendReq(reqSize int) {
 func (ss *ServerStats) BecomeLeader() {
 	ss.Lock()
 	defer ss.Unlock()
+	ss.becomeLeader()
+}
 
+func (ss *ServerStats) becomeLeader() {
 	if ss.State != raft.StateLeader {
 		ss.State = raft.StateLeader
 		ss.LeaderInfo.Name = ss.ID