ahead init

Signed-off-by: husharp <[email protected]>

pkg/tso/keyspace_group_manager.go

@@ -688,6 +688,24 @@ func (kgm *KeyspaceGroupManager) updateKeyspaceGroup(group *endpoint.KeyspaceGro
 		}}
 	}
 
+	// If the keyspace group is not initialized, initialize it.
+	// The format of leader name is address-groupID.
+	uniqueName := fmt.Sprintf("%s-%05d", kgm.electionNamePrefix, group.ID)
+	uniqueID := memberutil.GenerateUniqueID(uniqueName)
+	log.Info("joining primary election",
+		zap.Uint32("keyspace-group-id", group.ID),
+		zap.String("participant-name", uniqueName),
+		zap.Uint64("participant-id", uniqueID))
+	// Initialize the participant info to join the primary election.
+	participant := member.NewParticipant(kgm.etcdClient, mcsutils.TSOServiceName)
+	p := &tsopb.Participant{
+		Name:       uniqueName,
+		Id:         uniqueID, // id is unique among all participants
+		ListenUrls: []string{kgm.cfg.GetAdvertiseListenAddr()},
+	}
+	participant.InitInfo(p, endpoint.KeyspaceGroupsElectionPath(kgm.tsoSvcRootPath, group.ID), mcsutils.PrimaryKey, "keyspace group primary election")
+	kgm.tsoServiceID.MemberValue = []byte(participant.MemberValue())
+
 	if !kgm.isAssignedToMe(group) {
 		// Not assigned to me. If this host/pod owns a replica of this keyspace group,
 		// it should resign the election membership now.
@@ -720,24 +738,6 @@ func (kgm *KeyspaceGroupManager) updateKeyspaceGroup(group *endpoint.KeyspaceGro
 		kgm.updateKeyspaceGroupMembership(oldGroup, group, true)
 		return
 	}
-
-	// If the keyspace group is not initialized, initialize it.
-	// The format of leader name is address-groupID.
-	uniqueName := fmt.Sprintf("%s-%05d", kgm.electionNamePrefix, group.ID)
-	uniqueID := memberutil.GenerateUniqueID(uniqueName)
-	log.Info("joining primary election",
-		zap.Uint32("keyspace-group-id", group.ID),
-		zap.String("participant-name", uniqueName),
-		zap.Uint64("participant-id", uniqueID))
-	// Initialize the participant info to join the primary election.
-	participant := member.NewParticipant(kgm.etcdClient, mcsutils.TSOServiceName)
-	p := &tsopb.Participant{
-		Name:       uniqueName,
-		Id:         uniqueID, // id is unique among all participants
-		ListenUrls: []string{kgm.cfg.GetAdvertiseListenAddr()},
-	}
-	participant.InitInfo(p, endpoint.KeyspaceGroupsElectionPath(kgm.tsoSvcRootPath, group.ID), mcsutils.PrimaryKey, "keyspace group primary election")
-	kgm.tsoServiceID.MemberValue = []byte(participant.MemberValue())
 	// If the keyspace group is in split, we should ensure that the primary elected by the new keyspace group
 	// is always on the same TSO Server node as the primary of the old keyspace group, and this constraint cannot
 	// be broken until the entire split process is completed.