mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
added postgres support for db migrations
This commit is contained in:
@@ -35,6 +35,7 @@ func TestMigrations(t *testing.T) {
|
||||
log.NewLogger(0, "console", `{"level": 0}`)
|
||||
|
||||
testDBs := [][]string{
|
||||
[]string{"postgres", "user=grafanatest password=grafanatest host=localhost port=5432 dbname=grafanatest sslmode=disable"},
|
||||
[]string{"mysql", "grafana:password@tcp(localhost:3306)/grafana_tests?charset=utf8"},
|
||||
[]string{"sqlite3", ":memory:"},
|
||||
}
|
||||
|
||||
@@ -38,6 +38,8 @@ func NewMigrator(engine *xorm.Engine) *Migrator {
|
||||
mg.dialect = NewMysqlDialect()
|
||||
case SQLITE:
|
||||
mg.dialect = NewSqlite3Dialect()
|
||||
case POSTGRES:
|
||||
mg.dialect = NewPostgresDialect()
|
||||
}
|
||||
|
||||
return mg
|
||||
|
||||
@@ -1,9 +1,6 @@
|
||||
package migrations
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
)
|
||||
import "strconv"
|
||||
|
||||
type Mysql struct {
|
||||
BaseDialect
|
||||
@@ -71,15 +68,6 @@ func (db *Mysql) SqlType(c *Column) string {
|
||||
return res
|
||||
}
|
||||
|
||||
func (db *Mysql) ToDBTypeSql(columnType ColumnType, length int) string {
|
||||
switch columnType {
|
||||
case DB_TYPE_STRING:
|
||||
return fmt.Sprintf("NVARCHAR(%d)", length)
|
||||
}
|
||||
|
||||
panic("Unsupported db type")
|
||||
}
|
||||
|
||||
func (db *Mysql) TableCheckSql(tableName string) (string, []interface{}) {
|
||||
args := []interface{}{"grafana", tableName}
|
||||
sql := "SELECT `TABLE_NAME` from `INFORMATION_SCHEMA`.`TABLES` WHERE `TABLE_SCHEMA`=? and `TABLE_NAME`=?"
|
||||
|
||||
86
pkg/services/sqlstore/migrations/postgres_dialect.go
Normal file
86
pkg/services/sqlstore/migrations/postgres_dialect.go
Normal file
@@ -0,0 +1,86 @@
|
||||
package migrations
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"github.com/go-xorm/core"
|
||||
)
|
||||
|
||||
type Postgres struct {
|
||||
BaseDialect
|
||||
}
|
||||
|
||||
func NewPostgresDialect() *Postgres {
|
||||
d := Postgres{}
|
||||
d.BaseDialect.dialect = &d
|
||||
d.BaseDialect.driverName = POSTGRES
|
||||
return &d
|
||||
}
|
||||
|
||||
func (db *Postgres) Quote(name string) string {
|
||||
return "\"" + name + "\""
|
||||
}
|
||||
|
||||
func (db *Postgres) QuoteStr() string {
|
||||
return "\""
|
||||
}
|
||||
|
||||
func (db *Postgres) AutoIncrStr() string {
|
||||
return ""
|
||||
}
|
||||
|
||||
func (db *Postgres) SqlType(c *Column) string {
|
||||
var res string
|
||||
switch t := c.Type; t {
|
||||
case DB_TinyInt:
|
||||
res = DB_SmallInt
|
||||
return res
|
||||
case DB_MediumInt, core.Int, core.Integer:
|
||||
if c.IsAutoIncrement {
|
||||
return DB_Serial
|
||||
}
|
||||
return DB_Integer
|
||||
case DB_Serial, core.BigSerial:
|
||||
c.IsAutoIncrement = true
|
||||
c.Nullable = false
|
||||
res = t
|
||||
case DB_Binary, core.VarBinary:
|
||||
return DB_Bytea
|
||||
case DB_DateTime:
|
||||
res = DB_TimeStamp
|
||||
case DB_TimeStampz:
|
||||
return "timestamp with time zone"
|
||||
case DB_Float:
|
||||
res = DB_Real
|
||||
case DB_TinyText, core.MediumText, core.LongText:
|
||||
res = DB_Text
|
||||
case DB_NVarchar:
|
||||
res = DB_Varchar
|
||||
case DB_Uuid:
|
||||
res = DB_Uuid
|
||||
case DB_Blob, core.TinyBlob, core.MediumBlob, core.LongBlob:
|
||||
return DB_Bytea
|
||||
case DB_Double:
|
||||
return "DOUBLE PRECISION"
|
||||
default:
|
||||
if c.IsAutoIncrement {
|
||||
return DB_Serial
|
||||
}
|
||||
res = t
|
||||
}
|
||||
|
||||
var hasLen1 bool = (c.Length > 0)
|
||||
var hasLen2 bool = (c.Length2 > 0)
|
||||
if hasLen2 {
|
||||
res += "(" + strconv.Itoa(c.Length) + "," + strconv.Itoa(c.Length2) + ")"
|
||||
} else if hasLen1 {
|
||||
res += "(" + strconv.Itoa(c.Length) + ")"
|
||||
}
|
||||
return res
|
||||
}
|
||||
|
||||
func (db *Postgres) TableCheckSql(tableName string) (string, []interface{}) {
|
||||
args := []interface{}{"grafana", tableName}
|
||||
sql := "SELECT `TABLE_NAME` from `INFORMATION_SCHEMA`.`TABLES` WHERE `TABLE_SCHEMA`=? and `TABLE_NAME`=?"
|
||||
return sql, args
|
||||
}
|
||||
Reference in New Issue
Block a user