feat: genesis add attribute isMaster

Author: askyrie

server/controller/cloud/kubernetes_gather/kubernetes_gather_test.go

@@ -85,7 +85,7 @@ func TestKubernetes(t *testing.T) {
 		})
 		defer k8sInfoPatch.Reset()
 
-		g := genesis.NewGenesis(context.Background(), &config.ControllerConfig{})
+		g := genesis.NewGenesis(context.Background(), true, &config.ControllerConfig{})
 		vJsonData, _ := os.ReadFile("./testfiles/vinterfaces.json")
 		var vData gcommon.GenesisSyncDataResponse
 		json.Unmarshal(vJsonData, &vData)

server/controller/controller/controller.go

@@ -125,7 +125,7 @@ func Start(ctx context.Context, configPath, serverLogFile string, shared *server
 
 	router.SetInitStageForHealthChecker("Genesis init")
 	// 启动genesis
-	g := genesis.NewGenesis(ctx, cfg)
+	g := genesis.NewGenesis(ctx, isMasterController, cfg)
 
 	// start tagrecorder before manager to prevent recorder from publishing message when tagrecorder is not ready
 	router.SetInitStageForHealthChecker("TagRecorder init")

server/controller/genesis/genesis.go

@@ -43,11 +43,11 @@ type Genesis struct {
 	Synchronizer *grpc.SynchronizerServer
 }
 
-func NewGenesis(ctx context.Context, config *config.ControllerConfig) *Genesis {
+func NewGenesis(ctx context.Context, isMaster bool, config *config.ControllerConfig) *Genesis {
 	syncQueue := queue.NewOverwriteQueue("genesis-sync-data", config.GenesisCfg.QueueLengths)
 	kubernetesQueue := queue.NewOverwriteQueue("genesis-k8s-data", config.GenesisCfg.QueueLengths)
 
-	genesisSync := sstore.NewGenesisSync(ctx, syncQueue, config)
+	genesisSync := sstore.NewGenesisSync(ctx, isMaster, syncQueue, config)
 	genesisSync.Start()
 
 	genesisK8S := kstore.NewGenesisKubernetes(ctx, kubernetesQueue, config)

server/controller/genesis/store/sync/run.go

@@ -43,23 +43,25 @@ import (
 var log = logger.MustGetLogger("genesis.store.sync")
 
 type GenesisSync struct {
-	data   atomic.Value
-	ctx    context.Context
-	cancel context.CancelFunc
-	queue  queue.QueueReader
-	config *config.ControllerConfig
+	isMaster bool
+	data     atomic.Value
+	ctx      context.Context
+	cancel   context.CancelFunc
+	queue    queue.QueueReader
+	config   *config.ControllerConfig
 }
 
-func NewGenesisSync(ctx context.Context, queue queue.QueueReader, config *config.ControllerConfig) *GenesisSync {
+func NewGenesisSync(ctx context.Context, isMaster bool, queue queue.QueueReader, config *config.ControllerConfig) *GenesisSync {
 	var data atomic.Value
 	data.Store(common.GenesisSyncData{})
 	ctx, cancel := context.WithCancel(ctx)
 	return &GenesisSync{
-		ctx:    ctx,
-		cancel: cancel,
-		data:   data,
-		queue:  queue,
-		config: config,
+		isMaster: isMaster,
+		ctx:      ctx,
+		cancel:   cancel,
+		data:     data,
+		queue:    queue,
+		config:   config,
 	}
 }