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
}
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)
}

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

@ -8,3 +8,8 @@ const (
StatusOnline string = "online"
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{
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"))
}
}