feat: refactor go.mod.

This commit is contained in:
tx7do
2024-11-14 13:04:30 +08:00
parent 3ef28da068
commit 719fa01c8d
18 changed files with 1007 additions and 469 deletions

View File

@@ -0,0 +1,36 @@
package mongodb
import (
"context"
"fmt"
"github.com/go-kratos/kratos/v2/log"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
)
// NewMongoClient 创建MongoDB客户端
func NewMongoClient(ctx context.Context, cfg *conf.Bootstrap, l *log.Helper) *mongo.Client {
if cfg.Data == nil || cfg.Data.Mongodb == nil {
l.Warn("Mongodb config is nil")
return nil
}
var opts []*options.ClientOptions
uri := fmt.Sprintf("mongodb://%s:%s@%s",
cfg.Data.Mongodb.Username, cfg.Data.Mongodb.Password, cfg.Data.Mongodb.Address,
)
opts = append(opts, options.Client().ApplyURI(uri))
cli, err := mongo.Connect(ctx, opts...)
if err != nil {
l.Fatalf("failed opening connection to mongodb: %v", err)
return nil
}
return cli
}