re
/
gorm
forked from mirror/gorm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change internal API

This commit is contained in:
Jinzhu 2013-10-27 07:52:04 +08:00
parent 9d2f7650be
commit 284cf3f649
2 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
orm.go
View File

@ -14,7 +14,7 @@ type Orm struct {
Error error
Sql string
SqlVars []interface{}
Model *Model
model *Model
db *sql.DB
driver string
@ -26,10 +26,10 @@ type Orm struct {
operation string
}
func (s *Orm) setModel(model interface{}) (err error) {
s.Model = s.toModel(model)
s.TableName = s.Model.TableName()
s.PrimaryKey = s.Model.PrimaryKeyDb()
func (s *Orm) Model(model interface{}) (err error) {
s.model = s.toModel(model)
s.TableName = s.model.TableName()
s.PrimaryKey = s.model.PrimaryKeyDb()
return
}
@ -86,8 +86,8 @@ func (s *Orm) Select(value interface{}) *Orm {
}
func (s *Orm) Save(value interface{}) *Orm {
s.setModel(value)
if s.Model.PrimaryKeyIsEmpty() {
s.Model(value)
if s.model.PrimaryKeyIsEmpty() {
s.explain(value, "Create").create(value)
} else {
s.explain(value, "Update").update(value)

18
sql.go
View File

@ -8,7 +8,7 @@ import (
)
func (s *Orm) explain(value interface{}, operation string) *Orm {
s.setModel(value)
s.Model(value)
switch operation {
case "Create":
s.createSql(value)
@ -19,7 +19,7 @@ func (s *Orm) explain(value interface{}, operation string) *Orm {
case "Query":
s.querySql(value)
case "CreateTable":
s.Sql = s.Model.CreateTable()
s.Sql = s.model.CreateTable()
}
return s
}
@ -76,7 +76,7 @@ func (s *Orm) query(out interface{}) {
}
func (s *Orm) createSql(value interface{}) {
columns, values := s.Model.ColumnsAndValues()
columns, values := s.model.ColumnsAndValues()
var sqls []string
for _, value := range values {
@ -88,7 +88,7 @@ func (s *Orm) createSql(value interface{}) {
s.TableName,
strings.Join(s.quoteMap(columns), ","),
strings.Join(sqls, ","),
s.Model.ReturningStr(),
s.model.ReturningStr(),
)
return
}
@ -102,12 +102,12 @@ func (s *Orm) create(value interface{}) {
id, s.Error = s.SqlResult.LastInsertId()
}
result := reflect.ValueOf(s.Model.Data).Elem()
result.FieldByName(s.Model.PrimaryKey()).SetInt(id)
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()
columns, values := s.model.ColumnsAndValues()
var sets []string
for index, column := range columns {
sets = append(sets, fmt.Sprintf("%v = %v", s.quote(column), s.addToVars(values[index])))
@ -135,8 +135,8 @@ func (s *Orm) deleteSql(value interface{}) {
func (s *Orm) whereSql() (sql string) {
var conditions []string
if !s.Model.PrimaryKeyIsEmpty() {
conditions = append(conditions, fmt.Sprintf("(%v = %v)", s.quote(s.Model.PrimaryKeyDb()), s.addToVars(s.Model.PrimaryKeyValue())))
if !s.model.PrimaryKeyIsEmpty() {
conditions = append(conditions, fmt.Sprintf("(%v = %v)", s.quote(s.model.PrimaryKeyDb()), s.addToVars(s.model.PrimaryKeyValue())))
}
if len(s.whereClause) > 0 {