Skip to content

Commit c4a5bbe

Browse files
committed
feat: genesis add attribute isMaster
1 parent 42a2a81 commit c4a5bbe

File tree

5 files changed

+19
-17
lines changed

5 files changed

+19
-17
lines changed

server/controller/cloud/kubernetes_gather/kubernetes_gather_test.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ func TestKubernetes(t *testing.T) {
8585
})
8686
defer k8sInfoPatch.Reset()
8787

88-
g := genesis.NewGenesis(context.Background(), &config.ControllerConfig{})
88+
g := genesis.NewGenesis(context.Background(), true, &config.ControllerConfig{})
8989
vJsonData, _ := os.ReadFile("./testfiles/vinterfaces.json")
9090
var vData gcommon.GenesisSyncDataResponse
9191
json.Unmarshal(vJsonData, &vData)

server/controller/controller/controller.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ func Start(ctx context.Context, configPath, serverLogFile string, shared *server
107107
}
108108

109109
// 初始化Redis
110-
if cfg.RedisCfg.Enabled && cfg.TrisolarisCfg.NodeType == "master" {
110+
if cfg.RedisCfg.Enabled {
111111
router.SetInitStageForHealthChecker("Redis init")
112112

113113
err := redis.Init(ctx, cfg.RedisCfg)
@@ -124,7 +124,7 @@ func Start(ctx context.Context, configPath, serverLogFile string, shared *server
124124

125125
router.SetInitStageForHealthChecker("Genesis init")
126126
// 启动genesis
127-
g := genesis.NewGenesis(ctx, cfg)
127+
g := genesis.NewGenesis(ctx, isMasterController, cfg)
128128

129129
// start tagrecorder before manager to prevent recorder from publishing message when tagrecorder is not ready
130130
router.SetInitStageForHealthChecker("TagRecorder init")

server/controller/genesis/common/const.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
package common
1818

1919
const (
20-
SYNC_TYPE_FORMAT = "%v-%v-%v-%v" // region-orgID-type-vtapID
20+
SYNC_TYPE_FORMAT = "%v-%v-%v" // orgID-type-vtapID
2121
)
2222

2323
const (

server/controller/genesis/genesis.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,11 @@ type Genesis struct {
4343
Synchronizer *grpc.SynchronizerServer
4444
}
4545

46-
func NewGenesis(ctx context.Context, config *config.ControllerConfig) *Genesis {
46+
func NewGenesis(ctx context.Context, isMaster bool, config *config.ControllerConfig) *Genesis {
4747
syncQueue := queue.NewOverwriteQueue("genesis-sync-data", config.GenesisCfg.QueueLengths)
4848
kubernetesQueue := queue.NewOverwriteQueue("genesis-k8s-data", config.GenesisCfg.QueueLengths)
4949

50-
genesisSync := sstore.NewGenesisSync(ctx, syncQueue, config)
50+
genesisSync := sstore.NewGenesisSync(ctx, isMaster, syncQueue, config)
5151
genesisSync.Start()
5252

5353
genesisK8S := kstore.NewGenesisKubernetes(ctx, kubernetesQueue, config)

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

+13-11
Original file line numberDiff line numberDiff line change
@@ -42,23 +42,25 @@ import (
4242
var log = logger.MustGetLogger("genesis.store.sync")
4343

4444
type GenesisSync struct {
45-
data atomic.Value
46-
ctx context.Context
47-
cancel context.CancelFunc
48-
queue queue.QueueReader
49-
config *config.ControllerConfig
45+
isMaster bool
46+
data atomic.Value
47+
ctx context.Context
48+
cancel context.CancelFunc
49+
queue queue.QueueReader
50+
config *config.ControllerConfig
5051
}
5152

52-
func NewGenesisSync(ctx context.Context, queue queue.QueueReader, config *config.ControllerConfig) *GenesisSync {
53+
func NewGenesisSync(ctx context.Context, isMaster bool, queue queue.QueueReader, config *config.ControllerConfig) *GenesisSync {
5354
var data atomic.Value
5455
data.Store(common.GenesisSyncData{})
5556
ctx, cancel := context.WithCancel(ctx)
5657
return &GenesisSync{
57-
ctx: ctx,
58-
cancel: cancel,
59-
data: data,
60-
queue: queue,
61-
config: config,
58+
isMaster: isMaster,
59+
ctx: ctx,
60+
cancel: cancel,
61+
data: data,
62+
queue: queue,
63+
config: config,
6264
}
6365
}
6466

0 commit comments

Comments
 (0)