修复截取mongo数据库名称截取不全

2020-03-11 18:52:15 +08:00
parent 5e61291703
commit 9f27939471
1 changed files with 5 additions and 0 deletions
--- a/app/db/Mgo.go
+++ b/app/db/Mgo.go
@ -76,6 +76,11 @@ func Init(url, dbname string) {
 			// get dbname from urlEnv
 			urls := strings.Split(url, "/")
 			dbname = urls[len(urls)-1]
+
+			if strings.Contains(dbname, "?") {
+				urls = strings.Split(dbname, "?")
+				dbname = urls[0]
+			}
 		}
 	}
 	if dbname == "" {