mirror of https://github.com/go-gorm/gorm.git
147 lines
3.0 KiB
Go
147 lines
3.0 KiB
Go
package gorm
|
|
|
|
import "database/sql"
|
|
import "github.com/jinzhu/gorm/dialect"
|
|
|
|
var singularTableName bool
|
|
var tagIdentifier string
|
|
|
|
func init() {
|
|
tagIdentifier = "sql"
|
|
}
|
|
|
|
type DB struct {
|
|
db sqlCommon
|
|
dialect dialect.Dialect
|
|
logger Logger
|
|
logMode bool
|
|
}
|
|
|
|
func Open(driver, source string) (db DB, err error) {
|
|
db.db, err = sql.Open(driver, source)
|
|
db.dialect = dialect.New(driver)
|
|
return
|
|
}
|
|
|
|
func (s *DB) SetPool(n int) {
|
|
if db, ok := s.db.(sqlDb); ok {
|
|
db.SetMaxIdleConns(n)
|
|
}
|
|
}
|
|
|
|
func (s *DB) SetTagIdentifier(str string) {
|
|
tagIdentifier = str
|
|
}
|
|
|
|
func (s *DB) SetLogger(l Logger) {
|
|
s.logger = l
|
|
}
|
|
|
|
func (s *DB) LogMode(b bool) {
|
|
s.logMode = b
|
|
}
|
|
|
|
func (s *DB) SingularTable(result bool) {
|
|
singularTableName = result
|
|
}
|
|
|
|
func (s *DB) buildChain() *Chain {
|
|
return &Chain{db: s.db, d: s}
|
|
}
|
|
|
|
func (s *DB) Where(querystring interface{}, args ...interface{}) *Chain {
|
|
return s.buildChain().Where(querystring, args...)
|
|
}
|
|
|
|
func (s *DB) Not(querystring interface{}, args ...interface{}) *Chain {
|
|
return s.buildChain().Not(querystring, args...)
|
|
}
|
|
|
|
func (s *DB) First(out interface{}, where ...interface{}) *Chain {
|
|
return s.buildChain().First(out, where...)
|
|
}
|
|
|
|
func (s *DB) Last(out interface{}, where ...interface{}) *Chain {
|
|
return s.buildChain().Last(out, where...)
|
|
}
|
|
|
|
func (s *DB) Attrs(attrs ...interface{}) *Chain {
|
|
return s.buildChain().Attrs(attrs...)
|
|
}
|
|
|
|
func (s *DB) Assign(attrs ...interface{}) *Chain {
|
|
return s.buildChain().Assign(attrs...)
|
|
}
|
|
|
|
func (s *DB) FirstOrInit(out interface{}, where ...interface{}) *Chain {
|
|
return s.buildChain().FirstOrInit(out, where...)
|
|
}
|
|
|
|
func (s *DB) FirstOrCreate(out interface{}, where ...interface{}) *Chain {
|
|
return s.buildChain().FirstOrCreate(out, where...)
|
|
}
|
|
|
|
func (s *DB) Find(out interface{}, where ...interface{}) *Chain {
|
|
return s.buildChain().Find(out, where...)
|
|
}
|
|
|
|
func (s *DB) Limit(value interface{}) *Chain {
|
|
return s.buildChain().Limit(value)
|
|
}
|
|
|
|
func (s *DB) Offset(value interface{}) *Chain {
|
|
return s.buildChain().Offset(value)
|
|
}
|
|
|
|
func (s *DB) Order(value string, reorder ...bool) *Chain {
|
|
return s.buildChain().Order(value, reorder...)
|
|
}
|
|
|
|
func (s *DB) Select(value interface{}) *Chain {
|
|
return s.buildChain().Select(value)
|
|
}
|
|
|
|
func (s *DB) Save(value interface{}) *Chain {
|
|
return s.buildChain().Save(value)
|
|
}
|
|
|
|
func (s *DB) Delete(value interface{}) *Chain {
|
|
return s.buildChain().Delete(value)
|
|
}
|
|
|
|
func (s *DB) Unscoped() *Chain {
|
|
return s.buildChain().Unscoped()
|
|
}
|
|
|
|
func (s *DB) Exec(sql string) *Chain {
|
|
return s.buildChain().Exec(sql)
|
|
}
|
|
|
|
func (s *DB) Model(value interface{}) *Chain {
|
|
return s.buildChain().Model(value)
|
|
}
|
|
|
|
func (s *DB) Table(name string) *Chain {
|
|
return s.buildChain().Table(name)
|
|
}
|
|
|
|
func (s *DB) CreateTable(value interface{}) *Chain {
|
|
return s.buildChain().CreateTable(value)
|
|
}
|
|
|
|
func (s *DB) DropTable(value interface{}) *Chain {
|
|
return s.buildChain().DropTable(value)
|
|
}
|
|
|
|
func (s *DB) AutoMigrate(value interface{}) *Chain {
|
|
return s.buildChain().AutoMigrate(value)
|
|
}
|
|
|
|
func (s *DB) Debug() *Chain {
|
|
return s.buildChain().Debug()
|
|
}
|
|
|
|
func (s *DB) Begin() *Chain {
|
|
return s.buildChain().Begin()
|
|
}
|