HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Convert args to database time

This commit is contained in:
DrWrong 2018-06-05 15:59:16 +08:00
parent a8bd843a55
commit 6e70297bc8
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
session_raw.go
View File

@ -21,9 +21,28 @@ func (session *Session) queryPreprocess(sqlStr *string, paramStr ...interface{})
session.lastSQLArgs = paramStr session.lastSQLArgs = paramStr
} }
func (session *Session) argsConverter(args ...interface{}) []interface{} {
if session.engine.DatabaseTZ == nil {
return args
}
for i, arg := range args {
switch t := arg.(type) {
case time.Time:
args[i] = t.In(session.engine.DatabaseTZ)
case *time.Time:
if t != nil {
dbTime := t.In(session.engine.DatabaseTZ)
args[i] = &dbTime
}
}
}
return args
}
func (session *Session) queryRows(sqlStr string, args ...interface{}) (*core.Rows, error) { func (session *Session) queryRows(sqlStr string, args ...interface{}) (*core.Rows, error) {
defer session.resetStatement() defer session.resetStatement()
args = session.argsConverter(args)
session.queryPreprocess(&sqlStr, args...) session.queryPreprocess(&sqlStr, args...)
if session.engine.showSQL { if session.engine.showSQL {
@ -150,7 +169,7 @@ func (session *Session) queryBytes(sqlStr string, args ...interface{}) ([]map[st
func (session *Session) exec(sqlStr string, args ...interface{}) (sql.Result, error) { func (session *Session) exec(sqlStr string, args ...interface{}) (sql.Result, error) {
defer session.resetStatement() defer session.resetStatement()
args = session.argsConverter(args)
session.queryPreprocess(&sqlStr, args...) session.queryPreprocess(&sqlStr, args...)
if session.engine.showSQL { if session.engine.showSQL {