From 329b4a6470cfda7e82821ccbc2b336297c0aeb67 Mon Sep 17 00:00:00 2001
From: yaziming <Shao.Jiang-DAJ@yumchina.com>
Date: Tue, 1 Oct 2019 09:58:24 +0800
Subject: [PATCH] fix(backend): fix mongo connect error when password has
 special characters

---
 backend/database/mongo.go | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/backend/database/mongo.go b/backend/database/mongo.go
index 1c2d6433..e72baeaa 100644
--- a/backend/database/mongo.go
+++ b/backend/database/mongo.go
@@ -3,6 +3,7 @@ package database
 import (
 	"github.com/globalsign/mgo"
 	"github.com/spf13/viper"
+	"net"
 	"time"
 )
 
@@ -39,13 +40,28 @@ func InitMongo() error {
 	var mongoAuth = viper.GetString("mongo.authSource")
 
 	if Session == nil {
-		var uri string
-		if mongoUsername == "" {
-			uri = "mongodb://" + mongoHost + ":" + mongoPort + "/" + mongoDb
-		} else {
-			uri = "mongodb://" + mongoUsername + ":" + mongoPassword + "@" + mongoHost + ":" + mongoPort + "/" + mongoDb + "?authSource=" + mongoAuth
+		var dialInfo mgo.DialInfo
+		addr := net.JoinHostPort(mongoHost, mongoPort)
+		timeout := time.Second * 10
+		dialInfo = mgo.DialInfo{
+			Addrs:         []string{addr},
+			Timeout:       timeout,
+			Database:      mongoDb,
+			PoolLimit:     100,
+			PoolTimeout:   timeout,
+			ReadTimeout:   timeout,
+			WriteTimeout:  timeout,
+			AppName:       "crawlab",
+			FailFast:      true,
+			MinPoolSize:   10,
+			MaxIdleTimeMS: 1000 * 30,
 		}
-		sess, err := mgo.DialWithTimeout(uri, time.Second*5)
+		if mongoUsername != "" {
+			dialInfo.Username = mongoUsername
+			dialInfo.Password = mongoPassword
+			dialInfo.Source = mongoAuth
+		}
+		sess, err := mgo.DialWithInfo(&dialInfo)
 		if err != nil {
 			return err
 		}