|
@@ -2,7 +2,6 @@ package init_serve
|
|
|
|
|
|
import (
|
|
|
"context"
|
|
|
- "fmt"
|
|
|
"github.com/qiniu/qmgo"
|
|
|
"github.com/qiniu/qmgo/options"
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
@@ -76,20 +75,16 @@ func MgoNewClient() *qmgo.Client {
|
|
|
|
|
|
//mongodb+srv://myDatabaseUser:D1fficultP%40ssw0rd@mongodb0.example.com/?authSource=admin&replicaSet=myRepl
|
|
|
|
|
|
- var mongoUrl string
|
|
|
- if mgoConfig.Password != "" {
|
|
|
- mongoUrl = fmt.Sprintf("mongodb://%s:%s@%s:%d", mgoConfig.Username, mgoConfig.Password, mgoConfig.Host, mgoConfig.Port)
|
|
|
-
|
|
|
- if mgoConfig.AuthSource != `` {
|
|
|
- mongoUrl += mongoUrl + "?&authSource=" + mgoConfig.AuthSource
|
|
|
- }
|
|
|
- } else {
|
|
|
- mongoUrl = fmt.Sprintf("mongodb://%s:%d", mgoConfig.Host, mgoConfig.Port)
|
|
|
- }
|
|
|
-
|
|
|
// 创建一个数据库链接
|
|
|
client, err := qmgo.NewClient(ctx, &qmgo.Config{
|
|
|
- Uri: mongoUrl,
|
|
|
+ Uri: mgoConfig.Url,
|
|
|
+ Auth: &qmgo.Credential{
|
|
|
+ AuthMechanism: mgoConfig.AuthMechanism,
|
|
|
+ AuthSource: mgoConfig.AuthSource,
|
|
|
+ Username: mgoConfig.Username,
|
|
|
+ Password: mgoConfig.Password,
|
|
|
+ //PasswordSet: false,
|
|
|
+ },
|
|
|
Database: mgoConfig.Database,
|
|
|
ConnectTimeoutMS: &ConnectTimeoutMS,
|
|
|
SocketTimeoutMS: &SocketTimeoutMS,
|