fix: fix the cycle detection

This commit is contained in:
luhaoling 2024-02-02 16:52:40 +08:00
parent 633388b6d8
commit 7a8597fa85
2 changed files with 6 additions and 2 deletions

View File

@ -57,7 +57,6 @@ func NewMongo() (*Mongo, error) {
return &Mongo{db: mongoClient}, nil
}
if shouldRetry(err) {
fmt.Printf("Failed to connect to MongoDB, retrying: %s\n", err)
time.Sleep(time.Second) // exponential backoff could be implemented here
continue
}

View File

@ -56,6 +56,7 @@ func initCfg() error {
type checkFunc struct {
name string
function func() error
flag bool
}
func main() {
@ -87,11 +88,15 @@ func main() {
var err error
allSuccess := true
for _, check := range checks {
if !check.flag {
err = check.function()
continue
}
if err != nil {
component.ErrorPrint(fmt.Sprintf("Starting %s failed:%v.", check.name, err))
allSuccess = false
} else {
check.flag = true
component.SuccessPrint(fmt.Sprintf("%s connected successfully", check.name))
}
}