lbsadmin/Beego
1
0
Fork 0
mirror of https://github.com/astaxie/beego.git synced 2024-11-26 12:01:29 +00:00
Code Issues Releases Activity

update

Browse Source
This commit is contained in:
“fudali113” 2016-07-20 16:26:02 +08:00
parent 530c32017c
commit 50c5df32b1
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
orm/db.go
View File

@ -503,7 +503,7 @@ func (d *dbBase) InsertOrUpdate(q dbQuerier, mi *modelInfo, ind reflect.Value, t
if len(args) == 0 || (len(strings.Split(args0, "=")) != 1) { if len(args) == 0 || (len(strings.Split(args0, "=")) != 1) {
return 0, fmt.Errorf("`%s` use insert or update must have a conflict column arg in first", dn) return 0, fmt.Errorf("`%s` use insert or update must have a conflict column arg in first", dn)
} else { } else {
args0 = args[0] args0 = strings.ToLower(args[0])
iouStr = fmt.Sprintf("ON CONFLICT (%s) DO UPDATE SET", args0) iouStr = fmt.Sprintf("ON CONFLICT (%s) DO UPDATE SET", args0)
} }
} else { } else {
@ -513,7 +513,8 @@ func (d *dbBase) InsertOrUpdate(q dbQuerier, mi *modelInfo, ind reflect.Value, t
for _, v := range args { for _, v := range args {
kv := strings.Split(v, "=") kv := strings.Split(v, "=")
if len(kv) == 2 { if len(kv) == 2 {
argsMap[kv[0]] = kv[1] k := strings.ToLower(kv[0])
argsMap[k] = kv[1]
} }
} }
@ -532,8 +533,9 @@ func (d *dbBase) InsertOrUpdate(q dbQuerier, mi *modelInfo, ind reflect.Value, t
var conflitValue interface{} var conflitValue interface{}
for i, v := range names { for i, v := range names {
marks[i] = "?" marks[i] = "?"
valueStr := argsMap[v] vtl := strings.ToLower(v)
if strings.ToLower(v) == strings.ToLower(args0) { valueStr := argsMap[vtl]
if vtl == args0 {
conflitValue = values[i] conflitValue = values[i]
} }
if valueStr != "" { if valueStr != "" {
@ -544,10 +546,10 @@ func (d *dbBase) InsertOrUpdate(q dbQuerier, mi *modelInfo, ind reflect.Value, t
case postgres: case postgres:
if conflitValue != nil { if conflitValue != nil {
//postgres ON CONFLICT DO UPDATE SET can`t use colu=colu+values //postgres ON CONFLICT DO UPDATE SET can`t use colu=colu+values
updates[i] = fmt.Sprintf("%s=(select %s from %s where %s = ? )", v, valueStr, mi.table, args[0]) updates[i] = fmt.Sprintf("%s=(select %s from %s where %s = ? )", v, valueStr, mi.table, args0)
updateValues = append(updateValues, conflitValue) updateValues = append(updateValues, conflitValue)
} else { } else {
return 0, fmt.Errorf("`%s` must be in front of `%s` in your struct", args[0], v) return 0, fmt.Errorf("`%s` must be in front of `%s` in your struct", args0, v)
} }
break break
} }