diff --git a/modules/agent/agent.go b/modules/agent/agent.go index c4800736..1a9241e4 100644 --- a/modules/agent/agent.go +++ b/modules/agent/agent.go @@ -118,7 +118,7 @@ func (module *AgentModule) Start() error { return } 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 { log.Error(err) } diff --git a/modules/agent/model/const.go b/modules/agent/model/const.go index 6115f841..3cea10ea 100644 --- a/modules/agent/model/const.go +++ b/modules/agent/model/const.go @@ -8,3 +8,8 @@ const ( StatusOnline string = "online" StatusOffline = "offline" ) + +const ( + KVAgentIngestConfigChanged = "agent_ingest_config_changed" + KVSyncDynamicTaskSettings = "agent_sync_dynamic_task_settings" +) \ No newline at end of file diff --git a/modules/agent/state/state.go b/modules/agent/state/state.go index eb88a17d..7b65effd 100644 --- a/modules/agent/state/state.go +++ b/modules/agent/state/state.go @@ -140,9 +140,8 @@ func (sm *StateManager) checkAgentStatus() { } } -const KVSyncSettings = "agent_sync_settings" 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 { log.Error(err) } @@ -219,13 +218,13 @@ func (sm *StateManager) syncSettings(agentID string) { err = agClient.SaveDynamicConfig(context.Background(), ag.GetEndpoint(), "dynamic_task", string(cfgBytes)) 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 { log.Error(err) } } 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 { log.Error(err) } @@ -241,7 +240,7 @@ func (sm *StateManager) syncIngestSettings(agentID string) { } err = sm.agentClient.SaveIngestConfig(context.Background(), ag.GetEndpoint()) if err == nil { - kv.AddValue("agent_ingest_config_change",[]byte(agentID), []byte("0")) + kv.AddValue(model.KVAgentIngestConfigChanged,[]byte(agentID), []byte("0")) } }