From 0d27f524edb15eb03df3dd5758e87dbab28ccd2e Mon Sep 17 00:00:00 2001 From: a74589669 <290198252@qq.com> Date: Sun, 6 Oct 2019 01:25:44 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dsql=E8=AF=AD=E5=8F=A5?= =?UTF-8?q?=E5=8D=95=E5=BC=95=E5=8F=B7=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- db/sqlManager.go | 10 ++++++++-- main.go | 3 ++- model/blog.go | 4 +++- 3 files changed, 13 insertions(+), 4 deletions(-) diff --git a/db/sqlManager.go b/db/sqlManager.go index 5cd4238..ba22746 100644 --- a/db/sqlManager.go +++ b/db/sqlManager.go @@ -83,7 +83,14 @@ func initMysqlTLS(mysqlconf *config.MysqlConfig) *sql.DB { return _db } func initMongoDb(conf *config.MongoConfig) *mgo.Session{ - url := fmt.Sprintf("mongodb://%s:%s@%s:%d/%s",conf.UserName,conf.Password,conf.Addr,conf.Port,conf.Db) + var url string + if conf.UserName == ""{ + url = fmt.Sprintf("mongodb://%s:%d/%s",conf.Addr,conf.Port,conf.Db) + + }else{ + url = fmt.Sprintf("mongodb://%s:%s@%s:%d/%s",conf.UserName,conf.Password,conf.Addr,conf.Port,conf.Db) + + } log.Print("connect to url " + url) logs.Debug("connect to url " + url) @@ -111,5 +118,4 @@ func InitMongoDb() { log.Print("error mongo initial") os.Exit(25) } - } \ No newline at end of file diff --git a/main.go b/main.go index 0b6dfb2..1748ca6 100644 --- a/main.go +++ b/main.go @@ -44,7 +44,8 @@ func InitLogs() { } func CORSMiddleware(c *gin.Context) { c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE") - c.Writer.Header().Set("Access-Control-Allow-Origin", "http://localhost:8080") + c.Writer.Header().Set("Access-Control-Allow-Origin", + "http://172.21.102.193:8080") c.Writer.Header().Set("Access-Control-Max-Age", "86400") c.Writer.Header().Set("Access-Control-Allow-Headers", "X-Requested-With," + diff --git a/model/blog.go b/model/blog.go index a51604c..374f422 100644 --- a/model/blog.go +++ b/model/blog.go @@ -4,6 +4,7 @@ import ( "background/db" "background/logs" "fmt" + "strings" ) type Doc struct { @@ -50,7 +51,8 @@ WHERE func UpdateDoc(doc Doc) error{ sql := fmt.Sprintf(`update doc set doc.author = '%s' ,doc.title = '%s',doc.type = '%d',doc.content = '%s' where doc.id = '%d'; `, - doc.Author, doc.Title, doc.Type, doc.Content,doc.ID) + doc.Author, strings.Replace(doc.Title, "'", "\\'", -1), doc.Type, + strings.Replace(doc.Content, "'", "\\'", -1),doc.ID) _, e := db.GetMysqlClient().Query(sql) if nil != e { logs.Error(e.Error())