feat: set mongo pool size

dmzlingyin · May 17, 2024 · e838d3e · e838d3e
1 parent 50cb05e
commit e838d3e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/config/global.go b/config/global.go
@@ -16,6 +16,10 @@ func GetBool(field string) bool {
 	return config.Get(field).Bool()
 }
 
+func GetUint64(field string) uint64 {
+	return config.Get(field).Uint()
+}
+
 func SetProfile(profile string) {
 	config.SetProfile(profile)
 }
diff --git a/database/mongo/mongo.go b/database/mongo/mongo.go
@@ -15,7 +15,10 @@ func NewDatabase() *mongo.Database {
 	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
 
-	client, err := mongo.Connect(ctx, options.Client().ApplyURI(config.GetString("mongo.uri")))
+	opt := options.Client().ApplyURI(config.GetString("mongo.uri"))
+	opt.SetMaxPoolSize(config.GetUint64("mongo.max_pool_size"))
+	opt.SetMinPoolSize(config.GetUint64("mongo.min_pool_size"))
+	client, err := mongo.Connect(ctx, opt)
 	if err != nil {
 		panic(err)
 	}