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

Merge branch 'sunfmin-master' 

Conflicts:
	main.go
This commit is contained in:
Jinzhu 2014-04-25 07:23:26 +08:00
parent a46d149579 1ff3c79c75
commit 288c8e37af
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.go
View File

@ -16,11 +16,12 @@ type DB struct {
dialect Dialect dialect Dialect
tagIdentifier string tagIdentifier string
singularTable bool singularTable bool
source string
} }
func Open(driver, source string) (DB, error) { func Open(driver, source string) (DB, error) {
var err error var err error
db := DB{dialect: NewDialect(driver), tagIdentifier: "sql", logger: defaultLogger, callback: DefaultCallback} db := DB{dialect: NewDialect(driver), tagIdentifier: "sql", logger: defaultLogger, callback: DefaultCallback, source: source}
db.db, err = sql.Open(driver, source) db.db, err = sql.Open(driver, source)
db.parent = &db db.parent = &db
return db, err return db, err

5
scope_private.go
View File

@ -448,7 +448,8 @@ func (scope *Scope) removeIndex(indexName string) {
func (scope *Scope) autoMigrate() *Scope { func (scope *Scope) autoMigrate() *Scope {
var tableName string var tableName string
scope.Raw(fmt.Sprintf("SELECT table_name FROM INFORMATION_SCHEMA.tables where table_name = %v", scope.AddToVars(scope.TableName()))) scope.Raw(fmt.Sprintf("SELECT table_name FROM INFORMATION_SCHEMA.tables where table_name = %v",
scope.AddToVars(scope.TableName())))
scope.DB().QueryRow(scope.Sql, scope.SqlVars...).Scan(&tableName) scope.DB().QueryRow(scope.Sql, scope.SqlVars...).Scan(&tableName)
scope.SqlVars = []interface{}{} scope.SqlVars = []interface{}{}
@ -458,7 +459,7 @@ func (scope *Scope) autoMigrate() *Scope {
} else { } else {
for _, field := range scope.Fields() { for _, field := range scope.Fields() {
var column, data string var column, data string
scope.Raw(fmt.Sprintf("SELECT column_name, data_type FROM information_schema.columns WHERE table_name = %v and column_name = %v", scope.Raw(fmt.Sprintf("SELECT column_name, data_type FROM information_schema.columns WHERE table_name = %v AND column_name = %v",
scope.AddToVars(scope.TableName()), scope.AddToVars(scope.TableName()),
scope.AddToVars(field.DBName), scope.AddToVars(field.DBName),
)) ))