STALENESS_CHECK option

productiveops · Feb 7, 2024 · b3af11c · b3af11c
1 parent a576cea
commit b3af11c
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 4 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -12,6 +12,7 @@ func main() {
 		os.Getenv("DATA_PATH"),
 		os.Getenv("LOG_LEVEL"),
 		os.Getenv("SSL_ENABLED"),
+		os.Getenv("STALENESS_CHECK"),
 	)
 	s.Run(os.Getenv("BIND_ADDRESS"))
 }
diff --git a/pkg/agent/agent.go b/pkg/agent/agent.go
@@ -21,19 +21,23 @@ var (
 	logLevel string
 	wsUrl string
 	token string
+	stalenessCheck string = "ON"
 )
 
 func Main() {
  	parseArgs()
 	setLogLevel(logLevel)
-	go dockerapi.ContainerScheduleRefreshStaleStatus()
+	if stalenessCheck != "OFF" {
+		go dockerapi.ContainerScheduleRefreshStaleStatus()
+	}
 	listen()
 }
 
 func parseArgs() {
 	logLevel = os.Getenv("LOG_LEVEL")
 	serverUrl := os.Getenv("SERVER_URL")
 	token = os.Getenv("TOKEN")
+	stalenessCheck = os.Getenv("STALENESS_CHECK")
 
 	serverScheme := "ws"
 	if strings.HasPrefix(serverUrl, "https") {

diff --git a/pkg/dockerapi/container_stale_check.go b/pkg/dockerapi/container_stale_check.go
@@ -43,7 +43,7 @@ func ContainerScheduleRefreshStaleStatus() {
 	for {
 		log.Info().Msg("Refreshing container stale status")
 		ContainerRefreshStaleStatus()
-		time.Sleep(1 * time.Hour)
+		time.Sleep(24 * time.Hour)
 	}
 }
 

diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -35,7 +35,7 @@ type Server struct {
 	sslEnabled bool
 }
 
-func NewServer(dbConnectionString string, dataPath string, logLevel string, sslEnabled string) (*Server) {
+func NewServer(dbConnectionString string, dataPath string, logLevel string, sslEnabled string, stalenessCheck string) (*Server) {
 	s := Server{}
 
 	setLogLevel(logLevel)
@@ -82,7 +82,9 @@ func NewServer(dbConnectionString string, dataPath string, logLevel string, sslE
 		log.Error().Err(err).Msg("Error while updating old version data")
 	}
 
-	go dockerapi.ContainerScheduleRefreshStaleStatus()
+	if stalenessCheck != "OFF" {
+		go dockerapi.ContainerScheduleRefreshStaleStatus()
+	}
 
 	// Web Server
 	s.handler = h