Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in
Toggle navigation
B
beego
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Commits
Issue Boards
Open sidebar
go
beego
Commits
b5c29d61
Commit
b5c29d61
authored
Nov 08, 2016
by
Back
Committed by
GitHub
Nov 08, 2016
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Fix #2263
Update db_mysql.go instead of db.go, in order to avoid affect other database.
parent
3baac140
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
79 additions
and
0 deletions
+79
-0
db_mysql.go
orm/db_mysql.go
+79
-0
No files found.
orm/db_mysql.go
View file @
b5c29d61
...
...
@@ -16,6 +16,8 @@ package orm
import
(
"fmt"
"reflect"
"strings"
)
// mysql operators.
...
...
@@ -96,6 +98,83 @@ func (d *dbBaseMysql) IndexExists(db dbQuerier, table string, name string) bool
return
cnt
>
0
}
// InsertOrUpdate a row
// If your primary key or unique column conflict will update
// If no will insert
// Add "`" for mysql sql building
func
(
d
*
dbBaseMysql
)
InsertOrUpdate
(
q
dbQuerier
,
mi
*
modelInfo
,
ind
reflect
.
Value
,
a
*
alias
,
args
...
string
)
(
int64
,
error
)
{
iouStr
:=
""
argsMap
:=
map
[
string
]
string
{}
iouStr
=
"ON DUPLICATE KEY UPDATE"
//Get on the key-value pairs
for
_
,
v
:=
range
args
{
kv
:=
strings
.
Split
(
v
,
"="
)
if
len
(
kv
)
==
2
{
argsMap
[
strings
.
ToLower
(
kv
[
0
])]
=
kv
[
1
]
}
}
isMulti
:=
false
names
:=
make
([]
string
,
0
,
len
(
mi
.
fields
.
dbcols
)
-
1
)
Q
:=
d
.
ins
.
TableQuote
()
values
,
_
,
err
:=
d
.
collectValues
(
mi
,
ind
,
mi
.
fields
.
dbcols
,
true
,
true
,
&
names
,
a
.
TZ
)
if
err
!=
nil
{
return
0
,
err
}
marks
:=
make
([]
string
,
len
(
names
))
updateValues
:=
make
([]
interface
{},
0
)
updates
:=
make
([]
string
,
len
(
names
))
for
i
,
v
:=
range
names
{
marks
[
i
]
=
"?"
valueStr
:=
argsMap
[
strings
.
ToLower
(
v
)]
if
valueStr
!=
""
{
updates
[
i
]
=
"`"
+
v
+
"`"
+
"="
+
valueStr
}
else
{
updates
[
i
]
=
"`"
+
v
+
"`"
+
"=?"
updateValues
=
append
(
updateValues
,
values
[
i
])
}
}
values
=
append
(
values
,
updateValues
...
)
sep
:=
fmt
.
Sprintf
(
"%s, %s"
,
Q
,
Q
)
qmarks
:=
strings
.
Join
(
marks
,
", "
)
qupdates
:=
strings
.
Join
(
updates
,
", "
)
columns
:=
strings
.
Join
(
names
,
sep
)
multi
:=
len
(
values
)
/
len
(
names
)
if
isMulti
{
qmarks
=
strings
.
Repeat
(
qmarks
+
"), ("
,
multi
-
1
)
+
qmarks
}
//conflitValue maybe is a int,can`t use fmt.Sprintf
query
:=
fmt
.
Sprintf
(
"INSERT INTO %s%s%s (%s%s%s) VALUES (%s) %s "
+
qupdates
,
Q
,
mi
.
table
,
Q
,
Q
,
columns
,
Q
,
qmarks
,
iouStr
)
d
.
ins
.
ReplaceMarks
(
&
query
)
if
isMulti
||
!
d
.
ins
.
HasReturningID
(
mi
,
&
query
)
{
res
,
err
:=
q
.
Exec
(
query
,
values
...
)
if
err
==
nil
{
if
isMulti
{
return
res
.
RowsAffected
()
}
return
res
.
LastInsertId
()
}
return
0
,
err
}
row
:=
q
.
QueryRow
(
query
,
values
...
)
var
id
int64
err
=
row
.
Scan
(
&
id
)
return
id
,
err
}
// create new mysql dbBaser.
func
newdbBaseMysql
()
dbBaser
{
b
:=
new
(
dbBaseMysql
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment