floraachy
/
console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

move kv constant variable to model

This commit is contained in:
liugq 2023-06-06 19:12:44 +08:00
parent 196956f9ef
commit ca361b6a50
3 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/agent/agent.go
View File

@ -118,7 +118,7 @@ func (module *AgentModule) Start() error {
return return
} }
for _, ag := range agents { for _, ag := range agents {
err = kv.AddValue("agent_ingest_config_change", []byte(ag.ID), []byte("1")) err = kv.AddValue(model.KVAgentIngestConfigChanged, []byte(ag.ID), []byte("1"))
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }

5
modules/agent/model/const.go
View File

@ -8,3 +8,8 @@ const (
StatusOnline string = "online" StatusOnline string = "online"
StatusOffline = "offline" StatusOffline = "offline"
) )
const (
KVAgentIngestConfigChanged = "agent_ingest_config_changed"
KVSyncDynamicTaskSettings = "agent_sync_dynamic_task_settings"
)

9
modules/agent/state/state.go
View File

@ -140,9 +140,8 @@ func (sm *StateManager) checkAgentStatus() {
} }
} }
const KVSyncSettings = "agent_sync_settings"
func (sm *StateManager) getLastSyncSettingsTimestamp(agentID string) int64{ func (sm *StateManager) getLastSyncSettingsTimestamp(agentID string) int64{
vbytes, err := kv.GetValue(KVSyncSettings, []byte(agentID)) vbytes, err := kv.GetValue(model.KVSyncDynamicTaskSettings, []byte(agentID))
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }
@ -219,13 +218,13 @@ func (sm *StateManager) syncSettings(agentID string) {
err = agClient.SaveDynamicConfig(context.Background(), ag.GetEndpoint(), "dynamic_task", string(cfgBytes)) err = agClient.SaveDynamicConfig(context.Background(), ag.GetEndpoint(), "dynamic_task", string(cfgBytes))
newTimestampStr := strconv.FormatInt(newTimestamp, 10) newTimestampStr := strconv.FormatInt(newTimestamp, 10)
err = kv.AddValue(KVSyncSettings, []byte(agentID), []byte(newTimestampStr)) err = kv.AddValue(model.KVSyncDynamicTaskSettings, []byte(agentID), []byte(newTimestampStr))
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }
} }
func (sm *StateManager) syncIngestSettings(agentID string) { func (sm *StateManager) syncIngestSettings(agentID string) {
v, err := kv.GetValue("agent_ingest_config_change", []byte(agentID)) v, err := kv.GetValue(model.KVAgentIngestConfigChanged, []byte(agentID))
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }
@ -241,7 +240,7 @@ func (sm *StateManager) syncIngestSettings(agentID string) {
} }
err = sm.agentClient.SaveIngestConfig(context.Background(), ag.GetEndpoint()) err = sm.agentClient.SaveIngestConfig(context.Background(), ag.GetEndpoint())
if err == nil { if err == nil {
kv.AddValue("agent_ingest_config_change",[]byte(agentID), []byte("0")) kv.AddValue(model.KVAgentIngestConfigChanged,[]byte(agentID), []byte("0"))
} }
} }