blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-03 02:42:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix scripts functions upload

Browse Source 
Signed-off-by: Xinwei Xiong (cubxxw) <3293172751nss@gmail.com>
This commit is contained in:
Xinwei Xiong (cubxxw) 2024-01-12 17:45:50 +08:00
parent a11653d5e7
commit 4e4564c9cc
1 changed files with 11 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tools/component/component.go
View File

@ -141,35 +141,24 @@ func getEnv(key, fallback string) string {
return fallback
}
// checkMongo checks the MongoDB connection with retries and timeout
// checkMongo checks the MongoDB connection without retries
func checkMongo() (string, error) {
uri := getEnv("MONGO_URI", buildMongoURI())
var client *mongo.Client
var err error
for attempt := 0; attempt <= maxRetry; attempt++ {
ctx, cancel := context.WithTimeout(context.Background(), mongoConnTimeout)
defer cancel()
client, err = mongo.Connect(ctx, options.Client().ApplyURI(uri))
if err == nil {
break
}
if attempt < maxRetry {
fmt.Printf("Failed to connect to MongoDB, retrying: %s\n", err)
time.Sleep(time.Second * time.Duration(attempt+1)) // Exponential backoff
}
}
if err != nil {
return "", err // Wrap or handle the error as needed
}
defer client.Disconnect(context.Background())
ctx, cancel := context.WithTimeout(context.Background(), mongoConnTimeout)
defer cancel()
client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
if err != nil {
return "", err // Handle the error as needed
}
defer client.Disconnect(context.Background())
ctx, cancel = context.WithTimeout(context.Background(), mongoConnTimeout)
defer cancel()
if err = client.Ping(ctx, nil); err != nil {
return "", err // Wrap or handle the error as needed
return "", err // Handle the error as needed
}
str := "The addr is: " + strings.Join(config.Config.Mongo.Address, ",")