gorm/main.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      sql_common
	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.(sql_db); 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()
}