build: update build mongo URI logic. (#152)

* build: implement kafka health check.

* build: update build mongo URI logic.

db/mongoutil/common.go

@@ -17,23 +17,23 @@ package mongoutil
 import (
 	"context"
 	"fmt"
-	"go.mongodb.org/mongo-driver/mongo"
 	"strings"
+
+	"go.mongodb.org/mongo-driver/mongo"
 )
 
 const (
 	defaultMaxPoolSize = 100
 	defaultMaxRetry    = 3
 )
 
-func buildMongoURI(config *Config) string {
+func buildMongoURI(config *Config, authSource string) string {
 	credentials := ""
-	authSource := "admin"
+
 	if config.Username != "" && config.Password != "" {
 		credentials = fmt.Sprintf("%s:%s", config.Username, config.Password)
 	}
 
-	// 构建 MongoDB URI
 	return fmt.Sprintf(
 		"mongodb://%s@%s/%s?authSource=%s&maxPoolSize=%d",
 		credentials,

db/mongoutil/mongo.go

@@ -38,6 +38,7 @@ type Config struct {
 	Database    string
 	Username    string
 	Password    string
+	AuthSource  string
 	MaxPoolSize int
 	MaxRetry    int
 }

db/mongoutil/verify.go

@@ -62,7 +62,12 @@ func (c *Config) ValidateAndSetDefaults() error {
 		c.MaxRetry = defaultMaxRetry
 	}
 	if c.Uri == "" {
-		c.Uri = buildMongoURI(c)
+		// if authSource is not provided, default to admin
+		if c.AuthSource == "" {
+			c.Uri = buildMongoURI(c, "admin")
+		} else {
+			c.Uri = buildMongoURI(c, c.AuthSource)
+		}
 	}
 	return nil
 }