mirror of https://github.com/go-gorm/gorm.git
143 lines
3.0 KiB
Go
143 lines
3.0 KiB
Go
package gorm
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func (s *Orm) explain(value interface{}, operation string) *Orm {
|
|
s.setModel(value)
|
|
switch operation {
|
|
case "Create":
|
|
s.createSql(value)
|
|
case "Update":
|
|
s.updateSql(value)
|
|
case "Delete":
|
|
s.deleteSql(value)
|
|
case "Query":
|
|
s.querySql(value)
|
|
case "CreateTable":
|
|
s.Sql = s.Model.CreateTable()
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (s *Orm) querySql(out interface{}) {
|
|
s.Sql = fmt.Sprintf("SELECT * FROM %v %v", s.TableName, s.whereSql())
|
|
return
|
|
}
|
|
|
|
func (s *Orm) query(out interface{}) {
|
|
var (
|
|
is_slice bool
|
|
dest_type reflect.Type
|
|
)
|
|
dest_out := reflect.Indirect(reflect.ValueOf(out))
|
|
|
|
if x := dest_out.Kind(); x == reflect.Slice {
|
|
is_slice = true
|
|
dest_type = dest_out.Type().Elem()
|
|
}
|
|
|
|
rows, err := s.db.Query(s.Sql, s.SqlVars...)
|
|
defer rows.Close()
|
|
|
|
s.Error = err
|
|
if rows.Err() != nil {
|
|
s.Error = rows.Err()
|
|
}
|
|
|
|
counts := 0
|
|
for rows.Next() {
|
|
counts += 1
|
|
var dest reflect.Value
|
|
if is_slice {
|
|
dest = reflect.New(dest_type).Elem()
|
|
} else {
|
|
dest = reflect.ValueOf(out).Elem()
|
|
}
|
|
|
|
columns, _ := rows.Columns()
|
|
var values []interface{}
|
|
for _, value := range columns {
|
|
values = append(values, dest.FieldByName(snakeToUpperCamel(value)).Addr().Interface())
|
|
}
|
|
s.Error = rows.Scan(values...)
|
|
}
|
|
if (counts == 0) && !is_slice {
|
|
s.Error = errors.New("Record not found!")
|
|
}
|
|
}
|
|
|
|
func (s *Orm) createSql(value interface{}) {
|
|
columns, values := s.Model.ColumnsAndValues()
|
|
s.Sql = fmt.Sprintf(
|
|
"INSERT INTO \"%v\" (%v) VALUES (%v) %v",
|
|
s.TableName,
|
|
strings.Join(s.quoteMap(columns), ","),
|
|
valuesToBinVar(values),
|
|
s.Model.ReturningStr(),
|
|
)
|
|
s.SqlVars = values
|
|
return
|
|
}
|
|
|
|
func (s *Orm) create(value interface{}) {
|
|
var id int64
|
|
if s.driver == "postgres" {
|
|
s.Error = s.db.QueryRow(s.Sql, s.SqlVars...).Scan(&id)
|
|
} else {
|
|
s.SqlResult, s.Error = s.db.Exec(s.Sql, s.SqlVars...)
|
|
id, s.Error = s.SqlResult.LastInsertId()
|
|
}
|
|
|
|
result := reflect.ValueOf(s.Model.Data).Elem()
|
|
result.FieldByName(s.Model.PrimaryKey()).SetInt(id)
|
|
}
|
|
|
|
func (s *Orm) updateSql(value interface{}) {
|
|
columns, values := s.Model.ColumnsAndValues()
|
|
var sets []string
|
|
for index, column := range columns {
|
|
s.SqlVars = append(s.SqlVars, values[index])
|
|
sets = append(sets, fmt.Sprintf("%v = $%d", s.quote(column), len(s.SqlVars)))
|
|
}
|
|
|
|
s.Sql = fmt.Sprintf(
|
|
"UPDATE %v SET %v",
|
|
s.TableName,
|
|
strings.Join(sets, ", "),
|
|
)
|
|
return
|
|
}
|
|
|
|
func (s *Orm) update(value interface{}) {
|
|
s.Exec()
|
|
return
|
|
}
|
|
|
|
func (s *Orm) deleteSql(value interface{}) {
|
|
s.Sql = fmt.Sprintf("DELETE FROM %v WHERE %v", s.TableName, s.whereSql)
|
|
return
|
|
}
|
|
|
|
func (s *Orm) whereSql() (sql string) {
|
|
if len(s.whereClause) == 0 {
|
|
return
|
|
} else {
|
|
sql = "WHERE "
|
|
for _, clause := range s.whereClause {
|
|
sql += clause["query"].(string)
|
|
args := clause["args"].([]interface{})
|
|
for _, arg := range args {
|
|
s.SqlVars = append(s.SqlVars, arg.([]interface{})...)
|
|
sql = strings.Replace(sql, "?", "$"+strconv.Itoa(len(s.SqlVars)), 1)
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|