forked from mirror/gorm
Add method PrimaryKeyIsEmpty for model
This commit is contained in:
parent
cef81cba11
commit
2f4945dc35
6
model.go
6
model.go
|
@ -4,6 +4,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"reflect"
|
"reflect"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -23,6 +24,11 @@ func (s *Orm) toModel(value interface{}) *Model {
|
||||||
return &Model{Data: value, driver: s.driver}
|
return &Model{Data: value, driver: s.driver}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (m *Model) PrimaryKeyIsEmpty() bool {
|
||||||
|
result := reflect.ValueOf(m.Data).Elem()
|
||||||
|
return result.FieldByName(m.PrimaryKey()).Interface().(int64) == int64(0)
|
||||||
|
}
|
||||||
|
|
||||||
func (m *Model) PrimaryKey() string {
|
func (m *Model) PrimaryKey() string {
|
||||||
return "Id"
|
return "Id"
|
||||||
}
|
}
|
||||||
|
|
9
orm.go
9
orm.go
|
@ -3,6 +3,7 @@ package gorm
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"errors"
|
"errors"
|
||||||
|
|
||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -86,8 +87,12 @@ func (s *Orm) Select(value interface{}) *Orm {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Orm) Save(value interface{}) *Orm {
|
func (s *Orm) Save(value interface{}) *Orm {
|
||||||
s.explain(value, "Create").create(value)
|
s.setModel(value)
|
||||||
// s.explain(value, "Update").update(value)
|
if s.Model.PrimaryKeyIsEmpty() {
|
||||||
|
s.explain(value, "Create").create(value)
|
||||||
|
} else {
|
||||||
|
s.explain(value, "Update").update(value)
|
||||||
|
}
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue