tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Postgres driver updated for FoundationDB auto-migration compatibility.

Browse Source
This commit is contained in:
Jay Taylor 2015-03-13 14:08:22 -07:00
parent 2989e42d4d
commit a1f1d1e42a
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
postgres.go
View File

@ -76,13 +76,13 @@ func (s *postgres) Quote(key string) string {
func (s *postgres) HasTable(scope *Scope, tableName string) bool { func (s *postgres) HasTable(scope *Scope, tableName string) bool {
var count int var count int
scope.NewDB().Raw("SELECT count(*) FROM INFORMATION_SCHEMA.tables WHERE table_name = ? AND table_type = 'BASE TABLE'", tableName).Row().Scan(&count) scope.NewDB().Raw("SELECT count(*) FROM INFORMATION_SCHEMA.tables WHERE table_schema = current_schema AND table_name = ? AND (table_type = 'BASE TABLE' OR table_type = 'TABLE')", tableName).Row().Scan(&count)
return count > 0 return count > 0
} }
func (s *postgres) HasColumn(scope *Scope, tableName string, columnName string) bool { func (s *postgres) HasColumn(scope *Scope, tableName string, columnName string) bool {
var count int var count int
scope.NewDB().Raw("SELECT count(*) FROM INFORMATION_SCHEMA.columns WHERE table_name = ? AND column_name = ?", tableName, columnName).Row().Scan(&count) scope.NewDB().Raw("SELECT count(*) FROM INFORMATION_SCHEMA.columns WHERE table_schema = current_schema AND table_name = ? AND column_name = ?", tableName, columnName).Row().Scan(&count)
return count > 0 return count > 0
} }
@ -91,8 +91,19 @@ func (s *postgres) RemoveIndex(scope *Scope, indexName string) {
} }
func (s *postgres) HasIndex(scope *Scope, tableName string, indexName string) bool { func (s *postgres) HasIndex(scope *Scope, tableName string, indexName string) bool {
// Determine whehter the pg_indexes talbe is
detect := []int{} //make([]int, 2) // If detect[0] == 1 then we have postgres, else if detect[1] == 1 then we have detectationDB.
scope.NewDB().Raw(`SELECT count(*) "detect" FROM INFORMATION_SCHEMA.tables WHERE table_name = 'pg_indexes' UNION SELECT count(*) "detect" FROM INFORMATION_SCHEMA.tables WHERE table_name = 'indexes'`).Pluck("detect", &detect)
var count int var count int
scope.NewDB().Raw("SELECT count(*) FROM pg_indexes WHERE tablename = ? AND indexname = ?", tableName, indexName).Row().Scan(&count) var indexQuery string
if detect[0] == 1 { // Then we have postgres.
indexQuery = "SELECT count(*) FROM pg_indexes WHERE tablename = ? AND indexname = ?"
} else if detect[1] == 1 { // Then we have FoundationDB.
indexQuery = "SELECT count(*) FROM INFORMATION_SCHEMA.indexes WHERE table_schema = current_schema AND table_name = ? AND index_name = ?"
} else {
panic("unable to query for indexes, unrecognized database schema layout")
}
scope.NewDB().Raw(indexQuery, tableName, indexName).Row().Scan(&count)
return count > 0 return count > 0
} }