forked from mirror/gorm
Update DB struct for plugin system
This commit is contained in:
parent
68bb65d942
commit
ccae2f0345
28
main.go
28
main.go
|
@ -7,10 +7,10 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
type DB struct {
|
type DB struct {
|
||||||
|
Value interface{}
|
||||||
db sqlCommon
|
db sqlCommon
|
||||||
parent *DB
|
parent *DB
|
||||||
search *search
|
search *search
|
||||||
data interface{}
|
|
||||||
Error error
|
Error error
|
||||||
dialect dialect.Dialect
|
dialect dialect.Dialect
|
||||||
tagIdentifier string
|
tagIdentifier string
|
||||||
|
@ -121,15 +121,15 @@ func (s *DB) Find(out interface{}, where ...interface{}) *DB {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Row() *sql.Row {
|
func (s *DB) Row() *sql.Row {
|
||||||
return s.do(s.data).row()
|
return s.do(s.Value).row()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Rows() (*sql.Rows, error) {
|
func (s *DB) Rows() (*sql.Rows, error) {
|
||||||
return s.do(s.data).rows()
|
return s.do(s.Value).rows()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Scan(dest interface{}) *DB {
|
func (s *DB) Scan(dest interface{}) *DB {
|
||||||
return s.do(s.data).query(dest).db
|
return s.do(s.Value).query(dest).db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Attrs(attrs ...interface{}) *DB {
|
func (s *DB) Attrs(attrs ...interface{}) *DB {
|
||||||
|
@ -165,7 +165,7 @@ func (s *DB) Update(attrs ...interface{}) *DB {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Updates(values interface{}, ignore_protected_attrs ...bool) *DB {
|
func (s *DB) Updates(values interface{}, ignore_protected_attrs ...bool) *DB {
|
||||||
return s.clone().do(s.data).begin().updateAttrs(values, ignore_protected_attrs...).update().commit_or_rollback().db
|
return s.clone().do(s.Value).begin().updateAttrs(values, ignore_protected_attrs...).update().commit_or_rollback().db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) UpdateColumn(attrs ...interface{}) *DB {
|
func (s *DB) UpdateColumn(attrs ...interface{}) *DB {
|
||||||
|
@ -173,7 +173,7 @@ func (s *DB) UpdateColumn(attrs ...interface{}) *DB {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) UpdateColumns(values interface{}, ignore_protected_attrs ...bool) *DB {
|
func (s *DB) UpdateColumns(values interface{}, ignore_protected_attrs ...bool) *DB {
|
||||||
return s.clone().do(s.data).begin().updateColumns(values).commit_or_rollback().db
|
return s.clone().do(s.Value).begin().updateColumns(values).commit_or_rollback().db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Save(value interface{}) *DB {
|
func (s *DB) Save(value interface{}) *DB {
|
||||||
|
@ -194,21 +194,21 @@ func (s *DB) Exec(sql string, values ...interface{}) *DB {
|
||||||
|
|
||||||
func (s *DB) Model(value interface{}) *DB {
|
func (s *DB) Model(value interface{}) *DB {
|
||||||
c := s.clone()
|
c := s.clone()
|
||||||
c.data = value
|
c.Value = value
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Related(value interface{}, foreign_keys ...string) *DB {
|
func (s *DB) Related(value interface{}, foreign_keys ...string) *DB {
|
||||||
old_data := s.data
|
old_data := s.Value
|
||||||
return s.do(value).related(old_data, foreign_keys...).db
|
return s.do(value).related(old_data, foreign_keys...).db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Pluck(column string, value interface{}) *DB {
|
func (s *DB) Pluck(column string, value interface{}) *DB {
|
||||||
return s.do(s.data).pluck(column, value).db
|
return s.do(s.Value).pluck(column, value).db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Count(value interface{}) *DB {
|
func (s *DB) Count(value interface{}) *DB {
|
||||||
return s.do(s.data).count(value).db
|
return s.do(s.Value).count(value).db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) Table(name string) *DB {
|
func (s *DB) Table(name string) *DB {
|
||||||
|
@ -271,21 +271,21 @@ func (s *DB) AutoMigrate(value interface{}) *DB {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) ModifyColumn(column string, typ string) *DB {
|
func (s *DB) ModifyColumn(column string, typ string) *DB {
|
||||||
s.clone().do(s.data).modifyColumn(column, typ)
|
s.clone().do(s.Value).modifyColumn(column, typ)
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) DropColumn(column string) *DB {
|
func (s *DB) DropColumn(column string) *DB {
|
||||||
s.do(s.data).dropColumn(column)
|
s.do(s.Value).dropColumn(column)
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) AddIndex(column string, index_name ...string) *DB {
|
func (s *DB) AddIndex(column string, index_name ...string) *DB {
|
||||||
s.clone().do(s.data).addIndex(column, index_name...)
|
s.clone().do(s.Value).addIndex(column, index_name...)
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) RemoveIndex(column string) *DB {
|
func (s *DB) RemoveIndex(column string) *DB {
|
||||||
s.clone().do(s.data).removeIndex(column)
|
s.clone().do(s.Value).removeIndex(column)
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,7 +10,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func (s *DB) clone() *DB {
|
func (s *DB) clone() *DB {
|
||||||
db := DB{db: s.db, parent: s.parent, logMode: s.logMode, data: s.data, Error: s.Error}
|
db := DB{db: s.db, parent: s.parent, logMode: s.logMode, Value: s.Value, Error: s.Error}
|
||||||
|
|
||||||
if s.search == nil {
|
if s.search == nil {
|
||||||
db.search = &search{}
|
db.search = &search{}
|
||||||
|
@ -23,13 +23,13 @@ func (s *DB) clone() *DB {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) new() *DB {
|
func (s *DB) new() *DB {
|
||||||
db := DB{db: s.db, parent: s.parent, logMode: s.logMode, data: s.data, Error: s.Error, search: &search{}}
|
db := DB{db: s.db, parent: s.parent, logMode: s.logMode, Value: s.Value, Error: s.Error, search: &search{}}
|
||||||
db.search.db = &db
|
db.search.db = &db
|
||||||
return &db
|
return &db
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *DB) do(data interface{}) *Do {
|
func (s *DB) do(data interface{}) *Do {
|
||||||
s.data = data
|
s.Value = data
|
||||||
do := Do{db: s}
|
do := Do{db: s}
|
||||||
do.setModel(data)
|
do.setModel(data)
|
||||||
return &do
|
return &do
|
||||||
|
|
Loading…
Reference in New Issue