fix mongodb 在有账号密码情况下连接错误的问题

This commit is contained in:
陈景阳
2019-08-01 14:08:13 +08:00
parent a09d2f8b50
commit 1f5e95b6a5
2 changed files with 3 additions and 1 deletions

View File

@@ -6,6 +6,7 @@ mongo:
db: crawlab_test
username: ""
password: ""
auth-source: "admin"
redis:
address: localhost
password: ""

View File

@@ -35,13 +35,14 @@ func InitMongo() error {
var mongoDb = viper.GetString("mongo.db")
var mongoUsername = viper.GetString("mongo.username")
var mongoPassword = viper.GetString("mongo.password")
var mongoAuth = viper.GetString("mongo.auth-source")
if Session == nil {
var uri string
if mongoUsername == "" {
uri = "mongodb://" + mongoHost + ":" + mongoPort + "/" + mongoDb
} else {
uri = "mongodb://" + mongoUsername + ":" + mongoPassword + "@" + mongoHost + ":" + mongoPort + "/" + mongoDb
uri = "mongodb://" + mongoUsername + ":" + mongoPassword + "@" + mongoHost + ":" + mongoPort + "/" + mongoDb + "?authSource=" + mongoAuth
}
sess, err := mgo.Dial(uri)
if err != nil {