Skip to content

Commit

Permalink
Merge pull request #181 from GeoNet/checkIfCheck
Browse files Browse the repository at this point in the history
feat: add helper function to check whether -check flag used
  • Loading branch information
sue-h-gns authored Nov 17, 2024
2 parents 9ad1489 + 7c048c2 commit f2c17e5
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 1 deletion.
12 changes: 12 additions & 0 deletions health/check.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package health

import (
"context"
"flag"
"fmt"
"io"
"net/http"
Expand Down Expand Up @@ -66,3 +67,14 @@ func CheckStatus(servicePath string, timeout time.Duration) int {

return 0
}

// RunningHealthCheck returns whether the -check flag was used when starting the program.
// This flag indicates that the program is being used to run a health check on another program.
func RunningHealthCheck() bool {

var isHealthCheck bool
flag.BoolVar(&isHealthCheck, "check", false, "Whether the program is being used to run a health check")
flag.Parse()

return isHealthCheck
}
2 changes: 1 addition & 1 deletion slogger/logger.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,8 +46,8 @@ func (sl *SmartLogger) Log(message ...any) int {
if repeated && now.Sub(sl.lastLogTime) <= sl.window {
sl.repeatCount++
} else {
log.Println(msgString)
if sl.repeatedPrefix != "" && !repeated { //this is a random message
log.Println(msgString)
sl.lastMessage = "" // Reset lastMessage to avoid tracking it as a repeated message
sl.lastLogTime = time.Time{} // Reset the time
return 0
Expand Down

0 comments on commit f2c17e5

Please sign in to comment.