package migrator import ( "fmt" "strconv" "strings" ) type Postgres struct { BaseDialect } func NewPostgresDialect() *Postgres { d := Postgres{} d.BaseDialect.dialect = &d d.BaseDialect.driverName = POSTGRES return &d } func (db *Postgres) SupportEngine() bool { return false } func (db *Postgres) Quote(name string) string { return "\"" + name + "\"" } func (db *Postgres) QuoteStr() string { return "\"" } func (b *Postgres) LikeStr() string { return "ILIKE" } func (db *Postgres) AutoIncrStr() string { return "" } func (db *Postgres) BooleanStr(value bool) string { return strconv.FormatBool(value) } func (b *Postgres) Default(col *Column) string { if col.Type == DB_Bool { if col.Default == "0" { return "FALSE" } else { return "TRUE" } } return col.Default } 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, DB_Int, DB_Integer: if c.IsAutoIncrement { return DB_Serial } return DB_Integer case DB_Serial, DB_BigSerial: c.IsAutoIncrement = true c.Nullable = false res = t case DB_Binary, DB_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, DB_MediumText, DB_LongText: res = DB_Text case DB_NVarchar: res = DB_Varchar case DB_Uuid: res = DB_Uuid case DB_Blob, DB_TinyBlob, DB_MediumBlob, DB_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 } func (db *Postgres) DropIndexSql(tableName string, index *Index) string { quote := db.Quote idxName := index.XName(tableName) return fmt.Sprintf("DROP INDEX %v", quote(idxName)) } func (db *Postgres) UpdateTableSql(tableName string, columns []*Column) string { var statements = []string{} for _, col := range columns { statements = append(statements, "ALTER "+db.QuoteStr()+col.Name+db.QuoteStr()+" TYPE "+db.SqlType(col)) } return "ALTER TABLE " + db.Quote(tableName) + " " + strings.Join(statements, ", ") + ";" }