[feat] 新增 表结构操作
This commit is contained in:
@@ -0,0 +1,279 @@
|
||||
package mysql
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/dialect/mysql"
|
||||
"git.fsdpf.net/go/db/v2/internal/sb"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"git.fsdpf.net/go/db/v2/sqlgen"
|
||||
)
|
||||
|
||||
type Mysql struct {
|
||||
db *db.Database
|
||||
esg sqlgen.ExpressionSQLGenerator
|
||||
}
|
||||
|
||||
var mysqlDefaultModifiers = []string{
|
||||
"Unsigned", "Charset", "Collate", "VirtualAs", "StoredAs",
|
||||
"Nullable", "Default", "Increment", "Comment", "After", "First",
|
||||
}
|
||||
var mysqlSerials = []string{
|
||||
"bigInteger", "integer", "mediumInteger", "smallInteger", "tinyInteger",
|
||||
}
|
||||
|
||||
func (this Mysql) GetColumnListing(table string) ([]string, error) {
|
||||
cols := []string{}
|
||||
|
||||
err := this.db.From(db.S("information_schema").Table("columns")).Where(
|
||||
db.C("table_name").Eq(table),
|
||||
db.C("table_schema").Eq(db.L("database()")),
|
||||
).Pluck(&cols, "column_name")
|
||||
|
||||
return cols, err
|
||||
}
|
||||
|
||||
func (this Mysql) TableExists(table string) (bool, error) {
|
||||
result, err := this.db.From(db.S("information_schema").Table("tables")).Where(db.Ex{
|
||||
"table_name": table,
|
||||
"table_type": "BASE TABLE",
|
||||
"table_schema": db.L("database()"),
|
||||
}).Count()
|
||||
|
||||
return result > 0, err
|
||||
}
|
||||
|
||||
// 创建表
|
||||
func (this Mysql) CompileCreate(bp *schema.Blueprint) []string {
|
||||
temporary := db.L("CREATE")
|
||||
if bp.IsTemporary() {
|
||||
temporary = db.L("CREATE TEMPORARY")
|
||||
}
|
||||
columns := strings.Join(this.getAddedColumns(bp), ",\n")
|
||||
|
||||
sql := this.GenerateSQL("? TABLE ? (\n?\n)", temporary, db.T(bp.GetTable()), db.L(columns))
|
||||
|
||||
charset := bp.GetCharset()
|
||||
if charset == "" {
|
||||
charset = "utf8mb4"
|
||||
}
|
||||
|
||||
collation := bp.GetCollation()
|
||||
if collation == "" {
|
||||
collation = "utf8mb4_general_ci"
|
||||
}
|
||||
|
||||
sql += this.GenerateSQL(" default charset=? collate=?", charset, collation)
|
||||
|
||||
if engine := bp.GetEngine(); engine != "" {
|
||||
sql += this.GenerateSQL(" engine=?", engine)
|
||||
}
|
||||
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
// 添加字段
|
||||
func (this Mysql) CompileAdd(bp *schema.Blueprint) []string {
|
||||
columns := strings.Join(schema.PrefixArray("ADD COLUMN", this.getAddedColumns(bp)), ",\n")
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
// 修改字段
|
||||
func (this Mysql) CompileChange(bp *schema.Blueprint) []string {
|
||||
columns := strings.Join(schema.PrefixArray("CHANGE COLUMN", this.getChangedColumns(bp)), ",\n")
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
// 删除列
|
||||
func (this Mysql) CompileDropColumn(bp *schema.Blueprint) []string {
|
||||
columns := []string{}
|
||||
|
||||
for _, item := range bp.GetCommands() {
|
||||
if item.Type != "dropColumn" {
|
||||
continue
|
||||
}
|
||||
for _, col := range item.Columns {
|
||||
columns = append(columns, this.GenerateSQL("DROP COLUMN ?", db.C(col)))
|
||||
}
|
||||
}
|
||||
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(strings.Join(columns, ",\n")))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (this Mysql) CompileDrop(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
// 删除表, 先判断再删除
|
||||
func (this Mysql) CompileDropIfExists(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE IF EXISTS ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
// 表重命名
|
||||
func (this Mysql) CompileRename(bp *schema.Blueprint) []string {
|
||||
toName := ""
|
||||
commands := bp.GetCommands()
|
||||
if len(commands) == 0 {
|
||||
panic("new table undefined")
|
||||
}
|
||||
toName = bp.GetCommands()[0].To
|
||||
if toName == "" {
|
||||
panic("new table undefined")
|
||||
}
|
||||
return []string{this.GenerateSQL("RENAME TABLE ? TO ?", db.T(bp.GetTable()), db.T(toName))}
|
||||
}
|
||||
|
||||
func (this Mysql) addModifiers(sql string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
for _, modifier := range mysqlDefaultModifiers {
|
||||
sql += this.GetColumnModifier(modifier, bp, column)
|
||||
}
|
||||
return sql
|
||||
}
|
||||
|
||||
func (this Mysql) GetColumnType(column *schema.ColumnDefinition) string {
|
||||
switch column.Type {
|
||||
case "char":
|
||||
return this.GenerateSQL("char(?)", column.Length)
|
||||
case "string":
|
||||
return this.GenerateSQL("varchar(?)", column.Length)
|
||||
case "text":
|
||||
return "text"
|
||||
case "integer":
|
||||
return this.GenerateSQL("int(?)", column.Length)
|
||||
case "bigInteger":
|
||||
return "bigint(20)"
|
||||
case "tinyInteger":
|
||||
return "tinyint(1)"
|
||||
case "smallInteger":
|
||||
return "smallint(4)"
|
||||
case "decimal":
|
||||
return this.GenerateSQL("decimal(?,?)", column.Total, column.Places)
|
||||
case "boolean":
|
||||
return "tinyint(1)"
|
||||
case "enum":
|
||||
return this.GenerateSQL("enum?", column.Allowed)
|
||||
case "json":
|
||||
return "json"
|
||||
case "binary":
|
||||
return "binary"
|
||||
case "datetime":
|
||||
return "datetime"
|
||||
case "timestamp":
|
||||
return "timestamp"
|
||||
case "date":
|
||||
return "date"
|
||||
case "time":
|
||||
return "time"
|
||||
case "year":
|
||||
return "year"
|
||||
case "uuid":
|
||||
return "char(36)"
|
||||
}
|
||||
panic("Unsupported data type: " + column.Type)
|
||||
}
|
||||
|
||||
func (this Mysql) GetColumnModifier(modifier string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
switch modifier {
|
||||
case "Unsigned":
|
||||
if column.IsUnsigned() {
|
||||
return " UNSIGNED"
|
||||
}
|
||||
case "Charset":
|
||||
if v := column.GetCharset(); v != "" {
|
||||
return this.GenerateSQL(" CHARACTER SET ?", v)
|
||||
}
|
||||
case "Collate":
|
||||
if v := column.GetCollaction(); v != "" {
|
||||
return this.GenerateSQL(" COLLATE ?", v)
|
||||
}
|
||||
case "VirtualAs":
|
||||
if v := column.GetVirtualAs(); v != "" {
|
||||
return this.GenerateSQL(" GENERATED ALWAYS AS (?) VIRTUAL", db.L(v))
|
||||
}
|
||||
case "StoredAs":
|
||||
if v := column.GetStoredAs(); v != "" {
|
||||
return this.GenerateSQL(" GENERATED ALWAYS AS (?) STORED", db.L(v))
|
||||
}
|
||||
case "Nullable":
|
||||
if column.GetVirtualAs() == "" && column.GetStoredAs() == "" {
|
||||
if column.IsNullable() {
|
||||
return " NULL"
|
||||
}
|
||||
return " NOT NULL"
|
||||
}
|
||||
case "Default":
|
||||
v := column.GetDefault()
|
||||
if v == nil {
|
||||
if column.IsUseCurrent() {
|
||||
return " DEFAULT CURRENT_TIMESTAMP"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
if column.IsUseCurrent() {
|
||||
return this.GenerateSQL(" DEFAULT CURRENT_TIMESTAMP ?", v)
|
||||
}
|
||||
return this.GenerateSQL(" DEFAULT ?", v)
|
||||
case "Increment":
|
||||
if column.IsAutoIncrement() {
|
||||
return " AUTO_INCREMENT PRIMARY KEY"
|
||||
}
|
||||
case "Comment":
|
||||
if v := column.GetComment(); v != "" {
|
||||
return this.GenerateSQL(" COMMENT ?", v)
|
||||
}
|
||||
case "After":
|
||||
if v := column.GetAfter(); v != "" {
|
||||
return this.GenerateSQL(" AFTER ?", v)
|
||||
}
|
||||
case "First":
|
||||
if column.IsFirst() {
|
||||
return " FIRST"
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// 获取新增表结构字段
|
||||
func (this Mysql) getAddedColumns(bp *schema.Blueprint) (columns []string) {
|
||||
for _, column := range bp.GetAddedColumns() {
|
||||
sql := this.GenerateSQL("? ?", db.C(column.Name), db.L(this.GetColumnType(column)))
|
||||
columns = append(columns, this.addModifiers(sql, bp, column))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 获取修改表结构字段
|
||||
func (this Mysql) getChangedColumns(bp *schema.Blueprint) (columns []string) {
|
||||
for _, column := range bp.GetChangedColumns() {
|
||||
name := column.Name
|
||||
rename := column.Name
|
||||
if column.GetRename() != "" {
|
||||
rename = column.GetRename()
|
||||
}
|
||||
sql := this.GenerateSQL("? ? ?", db.C(name), db.C(rename), db.L(this.GetColumnType(column)))
|
||||
columns = append(columns, this.addModifiers(sql, bp, column))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 生成SQL
|
||||
func (this Mysql) GenerateSQL(sql string, args ...any) string {
|
||||
sb := sb.NewSQLBuilder(false)
|
||||
this.esg.Generate(sb, db.L(sql, args...))
|
||||
result, _, _ := sb.ToSQL()
|
||||
return result
|
||||
}
|
||||
|
||||
func init() {
|
||||
schema.RegisterDialect("mysql", func(db *db.Database) schema.Schema {
|
||||
return &Mysql{
|
||||
db: db,
|
||||
esg: sqlgen.NewExpressionSQLGenerator("mysql", mysql.DialectOptions()),
|
||||
}
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,109 @@
|
||||
package mysql_test
|
||||
|
||||
import (
|
||||
"os"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/engine"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"github.com/stretchr/testify/suite"
|
||||
)
|
||||
|
||||
type mysqlBuilderTest struct {
|
||||
suite.Suite
|
||||
builder *schema.Builder
|
||||
}
|
||||
|
||||
func (t *mysqlBuilderTest) SetupSuite() {
|
||||
db := engine.Open(map[string]engine.DBConfig{
|
||||
"test-mysql": engine.NewDBConfig("mysql",
|
||||
engine.WithHost(os.Getenv("MYSQL_HOST")),
|
||||
engine.WithPort(os.Getenv("MYSQL_PORT")),
|
||||
engine.WithDatabase(os.Getenv("MYSQL_DB")),
|
||||
engine.WithUsername(os.Getenv("MYSQL_USER")),
|
||||
engine.WithPassword(os.Getenv("MYSQL_PASSWD")),
|
||||
engine.WithParseTime(true),
|
||||
),
|
||||
}).Connection("test-mysql")
|
||||
|
||||
t.builder = schema.New(db)
|
||||
}
|
||||
|
||||
func (t *mysqlBuilderTest) TestHasTable() {
|
||||
t.builder.HasTable("users")
|
||||
}
|
||||
|
||||
func (t *mysqlBuilderTest) TestGetColumnListing() {
|
||||
t.builder.GetColumnListing("users")
|
||||
}
|
||||
|
||||
// 创建表
|
||||
func (t *mysqlBuilderTest) TestCreateTable() {
|
||||
if err := t.builder.Create("users", func(table *schema.Blueprint) {
|
||||
table.Charset("utf8mb4")
|
||||
|
||||
table.BigIncrements("id").Comment("ID")
|
||||
table.Boolean("enabled").Default("1").Comment("是否有效")
|
||||
table.Char("created_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("创建者")
|
||||
table.Char("owned_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("拥有者")
|
||||
table.Timestamp("created_at").UseCurrent().Comment("创建时间")
|
||||
table.Timestamp("updated_at").UseCurrent().Default(db.L("ON UPDATE CURRENT_TIMESTAMP")).Comment("更新时间")
|
||||
table.DateTime("deleted_at").Nullable().Comment("删除时间")
|
||||
}); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 修改表, 添加字段
|
||||
func (t *mysqlBuilderTest) TestAddColumn() {
|
||||
if err := t.builder.Table("users", func(table *schema.Blueprint) {
|
||||
table.String("name", 50).Nullable().Comment("用户名")
|
||||
table.TinyInteger("age").Nullable().Comment("年龄")
|
||||
}); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 修改表, 添加/编辑字段
|
||||
func (t *mysqlBuilderTest) TestChangeColumn() {
|
||||
if err := t.builder.Table("users", func(table *schema.Blueprint) {
|
||||
table.String("name", 100).Nullable().Comment("用户名").Change("username")
|
||||
table.SmallInteger("age").Nullable().Comment("年龄").Change()
|
||||
|
||||
table.String("nickname", 100).Nullable().Comment("昵称")
|
||||
}); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 修改表, 添加/编辑/删除字段
|
||||
func (t *mysqlBuilderTest) TestDropColumn() {
|
||||
if err := t.builder.Table("users", func(table *schema.Blueprint) {
|
||||
table.String("username", 100).Default("").Comment("用户名").Change("name")
|
||||
table.Boolean("is_vip").Default(1).Comment("是否VIP")
|
||||
table.DropColumn("age")
|
||||
}); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 重命名表
|
||||
func (t *mysqlBuilderTest) TestRename() {
|
||||
if err := t.builder.Rename("users", "users_alias"); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (t *mysqlBuilderTest) TestDropTable() {
|
||||
if err := t.builder.Drop("users_alias"); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (t *mysqlBuilderTest) TestDropTableIfExists() {
|
||||
if err := t.builder.DropIfExists("users_alias"); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,190 @@
|
||||
package mysql_test
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/engine"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
)
|
||||
|
||||
type mysqlTest struct {
|
||||
suite.Suite
|
||||
schema schema.Schema
|
||||
}
|
||||
|
||||
var (
|
||||
tableName = "entry"
|
||||
|
||||
dropTable = "DROP TABLE IF EXISTS `entry`;"
|
||||
|
||||
createTable = "CREATE TABLE IF NOT EXISTS `entry` (" +
|
||||
"`id` INT NOT NULL AUTO_INCREMENT ," +
|
||||
"`int` INT NOT NULL UNIQUE," +
|
||||
"`float` FLOAT NOT NULL ," +
|
||||
"`string` VARCHAR(255) NOT NULL ," +
|
||||
"`time` DATETIME NOT NULL ," +
|
||||
"`bool` TINYINT NOT NULL ," +
|
||||
"`bytes` BLOB NOT NULL ," +
|
||||
"PRIMARY KEY (`id`) );"
|
||||
)
|
||||
|
||||
func TestMysqlSuite(t *testing.T) {
|
||||
suite.Run(t, new(mysqlTest))
|
||||
}
|
||||
|
||||
func (t *mysqlTest) SetupSuite() {
|
||||
db := engine.Open(map[string]engine.DBConfig{
|
||||
"test-mysql": engine.NewDBConfig("mysql",
|
||||
engine.WithHost(os.Getenv("MYSQL_HOST")),
|
||||
engine.WithPort(os.Getenv("MYSQL_PORT")),
|
||||
engine.WithDatabase(os.Getenv("MYSQL_DB")),
|
||||
engine.WithUsername(os.Getenv("MYSQL_USER")),
|
||||
engine.WithPassword(os.Getenv("MYSQL_PASSWD")),
|
||||
engine.WithParseTime(true),
|
||||
),
|
||||
}).Connection("test-mysql")
|
||||
|
||||
t.schema = schema.GetSchemaDialect(db)
|
||||
|
||||
// db.Exec(dropTable)
|
||||
db.Exec(createTable)
|
||||
}
|
||||
|
||||
func (t *mysqlTest) TestGetColumnListing() {
|
||||
result, err := t.schema.GetColumnListing(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().ElementsMatch([]string{"bool", "bytes", "float", "id", "int", "string", "time"}, result)
|
||||
}
|
||||
|
||||
func (t *mysqlTest) TestTableExists() {
|
||||
result, err := t.schema.TableExists(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().Equal(result, true)
|
||||
}
|
||||
|
||||
// 创建列表
|
||||
func (t *mysqlTest) TestCompileCreate() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
bp.Create()
|
||||
bp.Charset("utf8mb4")
|
||||
|
||||
bp.BigIncrements("id").AutoIncrement().Comment("ID")
|
||||
bp.Boolean("enabled").Default("1").Comment("是否有效")
|
||||
bp.Char("created_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("创建者")
|
||||
bp.Char("owned_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("拥有者")
|
||||
bp.Timestamp("created_at").UseCurrent().Comment("创建时间")
|
||||
bp.Timestamp("updated_at").UseCurrent().Default(db.L("ON UPDATE CURRENT_TIMESTAMP")).Comment("更新时间")
|
||||
bp.DateTime("deleted_at").Nullable().Comment("删除时间")
|
||||
|
||||
sql := t.schema.CompileCreate(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"CREATE TABLE `users` (\n" +
|
||||
"`id` bigint(20) UNSIGNED NOT NULL AUTO_INCREMENT PRIMARY KEY COMMENT 'ID',\n" +
|
||||
"`enabled` tinyint(1) NOT NULL DEFAULT '1' COMMENT '是否有效',\n" +
|
||||
"`created_user` char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '创建者',\n" +
|
||||
"`owned_user` char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '拥有者',\n" +
|
||||
"`created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" +
|
||||
"`updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',\n" +
|
||||
"`deleted_at` datetime NULL COMMENT '删除时间'\n" +
|
||||
") default charset='utf8mb4' collate='utf8mb4_general_ci'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 添加字段
|
||||
func (t *mysqlTest) TestCompileAdd() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Comment("用户名")
|
||||
bp.SmallInteger("age").Default("18").Comment("年龄")
|
||||
bp.Enum("sex", []string{"0", "1"}).Default("0").Comment("性别")
|
||||
|
||||
sql := t.schema.CompileAdd(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE `users`\n" +
|
||||
"ADD COLUMN `name` varchar(50) NOT NULL COMMENT '用户名',\n" +
|
||||
"ADD COLUMN `age` smallint(4) NOT NULL DEFAULT '18' COMMENT '年龄',\n" +
|
||||
"ADD COLUMN `sex` enum('0', '1') NOT NULL DEFAULT '0' COMMENT '性别'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 修改字段
|
||||
func (t *mysqlTest) TestCompileChange() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Change("username")
|
||||
bp.SmallInteger("age").Default("19").Comment("年龄").Change()
|
||||
|
||||
sql := "ALTER TABLE `users`\n" +
|
||||
"CHANGE COLUMN `name` `username` varchar(50) NOT NULL,\n" +
|
||||
"CHANGE COLUMN `age` `age` smallint(4) NOT NULL DEFAULT '19' COMMENT '年龄';"
|
||||
|
||||
t.Equal(t.schema.CompileChange(bp), sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 删除字段
|
||||
func (t *mysqlTest) TestCompileDropColumn() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.DropColumn("age", "sex")
|
||||
|
||||
sql := t.schema.CompileDropColumn(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE `users`\n" +
|
||||
"DROP COLUMN `age`,\n" +
|
||||
"DROP COLUMN `sex`",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (t *mysqlTest) TestCompileDrop() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDrop(bp)
|
||||
|
||||
t.Equal([]string{"DROP TABLE `users`"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (t *mysqlTest) TestCompileDropIfExists() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDropIfExists(bp)
|
||||
t.Equal([]string{"DROP TABLE IF EXISTS `users`"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 表重命名
|
||||
func (t *mysqlTest) TestCompileRename() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Rename("user")
|
||||
|
||||
sql := t.schema.CompileRename(bp)
|
||||
|
||||
t.Equal([]string{"RENAME TABLE `users` TO `user`"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
@@ -0,0 +1,213 @@
|
||||
package postgres
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/dialect/postgres"
|
||||
"git.fsdpf.net/go/db/v2/internal/sb"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"git.fsdpf.net/go/db/v2/sqlgen"
|
||||
)
|
||||
|
||||
type Postgres struct {
|
||||
db *db.Database
|
||||
esg sqlgen.ExpressionSQLGenerator
|
||||
}
|
||||
|
||||
var pgDefaultModifiers = []string{
|
||||
"Nullable", "Default", "Increment", "Comment",
|
||||
}
|
||||
var pgSerials = []string{
|
||||
"bigInteger", "integer", "smallInteger",
|
||||
}
|
||||
|
||||
func (this Postgres) GetColumnListing(table string) ([]string, error) {
|
||||
cols := []string{}
|
||||
err := this.db.From(db.S("information_schema").Table("columns")).Where(
|
||||
db.C("table_name").Eq(table),
|
||||
db.C("table_schema").Eq(db.L("current_database()")),
|
||||
).Pluck(&cols, "column_name")
|
||||
return cols, err
|
||||
}
|
||||
|
||||
func (this Postgres) TableExists(table string) (bool, error) {
|
||||
result, err := this.db.From(db.S("information_schema").Table("tables")).Where(db.Ex{
|
||||
"table_name": table,
|
||||
"table_schema": db.L("current_database()"),
|
||||
}).Count()
|
||||
return result > 0, err
|
||||
}
|
||||
|
||||
func (this Postgres) CompileCreate(bp *schema.Blueprint) []string {
|
||||
temporary := db.L("CREATE")
|
||||
if bp.IsTemporary() {
|
||||
temporary = db.L("CREATE TEMPORARY")
|
||||
}
|
||||
columns := strings.Join(this.getAddedColumns(bp), ",\n")
|
||||
sql := this.GenerateSQL("? TABLE ? (\n?\n)", temporary, db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this Postgres) CompileAdd(bp *schema.Blueprint) []string {
|
||||
columns := strings.Join(schema.PrefixArray("ADD COLUMN", this.getAddedColumns(bp)), ",\n")
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this Postgres) CompileChange(bp *schema.Blueprint) []string {
|
||||
columns := strings.Join(schema.PrefixArray("ALTER COLUMN", this.getChangedColumns(bp)), ",\n")
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this Postgres) CompileDropColumn(bp *schema.Blueprint) []string {
|
||||
columns := []string{}
|
||||
for _, item := range bp.GetCommands() {
|
||||
if item.Type != "dropColumn" {
|
||||
continue
|
||||
}
|
||||
for _, col := range item.Columns {
|
||||
columns = append(columns, this.GenerateSQL("DROP COLUMN ?", db.C(col)))
|
||||
}
|
||||
}
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(strings.Join(columns, ",\n")))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this Postgres) CompileDrop(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
func (this Postgres) CompileDropIfExists(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE IF EXISTS ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
func (this Postgres) CompileRename(bp *schema.Blueprint) []string {
|
||||
toName := ""
|
||||
commands := bp.GetCommands()
|
||||
if len(commands) == 0 {
|
||||
panic("new table undefined")
|
||||
}
|
||||
toName = bp.GetCommands()[0].To
|
||||
if toName == "" {
|
||||
panic("new table undefined")
|
||||
}
|
||||
return []string{this.GenerateSQL("ALTER TABLE ? RENAME TO ?", db.T(bp.GetTable()), db.T(toName))}
|
||||
}
|
||||
|
||||
func (this Postgres) addModifiers(sql string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
for _, modifier := range pgDefaultModifiers {
|
||||
sql += this.GetColumnModifier(modifier, bp, column)
|
||||
}
|
||||
return sql
|
||||
}
|
||||
|
||||
func (this Postgres) GetColumnType(column *schema.ColumnDefinition) string {
|
||||
switch column.Type {
|
||||
case "char":
|
||||
return this.GenerateSQL("char(?)", column.Length)
|
||||
case "string":
|
||||
return this.GenerateSQL("varchar(?)", column.Length)
|
||||
case "text":
|
||||
return "text"
|
||||
case "integer":
|
||||
return "integer"
|
||||
case "bigInteger":
|
||||
return "bigint"
|
||||
case "smallInteger":
|
||||
return "smallint"
|
||||
case "decimal":
|
||||
return this.GenerateSQL("numeric(?,?)", column.Total, column.Places)
|
||||
case "boolean":
|
||||
return "boolean"
|
||||
case "enum":
|
||||
return this.GenerateSQL("varchar(?)", column.Length) // PostgreSQL uses custom types or check constraints
|
||||
case "json":
|
||||
return "json"
|
||||
case "binary":
|
||||
return "bytea"
|
||||
case "datetime":
|
||||
return "timestamp without time zone"
|
||||
case "timestamp":
|
||||
return "timestamp with time zone"
|
||||
case "date":
|
||||
return "date"
|
||||
case "time":
|
||||
return "time"
|
||||
case "year":
|
||||
return "integer"
|
||||
case "uuid":
|
||||
return "uuid"
|
||||
}
|
||||
panic("Unsupported data type: " + column.Type)
|
||||
}
|
||||
|
||||
func (this Postgres) GetColumnModifier(modifier string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
switch modifier {
|
||||
case "Nullable":
|
||||
if column.IsNullable() {
|
||||
return " NULL"
|
||||
}
|
||||
return " NOT NULL"
|
||||
case "Default":
|
||||
v := column.GetDefault()
|
||||
if v == nil {
|
||||
if column.IsUseCurrent() {
|
||||
return " DEFAULT CURRENT_TIMESTAMP"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
if column.IsUseCurrent() {
|
||||
return this.GenerateSQL(" DEFAULT CURRENT_TIMESTAMP ?", v)
|
||||
}
|
||||
return this.GenerateSQL(" DEFAULT ?", v)
|
||||
case "Increment":
|
||||
if column.IsAutoIncrement() {
|
||||
switch column.Type {
|
||||
case "integer":
|
||||
return " SERIAL PRIMARY KEY"
|
||||
case "bigInteger":
|
||||
return " BIGSERIAL PRIMARY KEY"
|
||||
case "smallInteger":
|
||||
return " SMALLSERIAL PRIMARY KEY"
|
||||
}
|
||||
}
|
||||
case "Comment":
|
||||
// PostgreSQL handles comments separately via COMMENT ON
|
||||
return ""
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (this Postgres) getAddedColumns(bp *schema.Blueprint) (columns []string) {
|
||||
for _, column := range bp.GetAddedColumns() {
|
||||
sql := this.GenerateSQL("? ?", db.C(column.Name), db.L(this.GetColumnType(column)))
|
||||
columns = append(columns, this.addModifiers(sql, bp, column))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (this Postgres) getChangedColumns(bp *schema.Blueprint) (columns []string) {
|
||||
for _, column := range bp.GetChangedColumns() {
|
||||
sql := this.GenerateSQL("? TYPE ?", db.C(column.Name), db.L(this.GetColumnType(column)))
|
||||
columns = append(columns, this.addModifiers(sql, bp, column))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (this Postgres) GenerateSQL(sql string, args ...any) string {
|
||||
sb := sb.NewSQLBuilder(false)
|
||||
this.esg.Generate(sb, db.L(sql, args...))
|
||||
result, _, _ := sb.ToSQL()
|
||||
return result
|
||||
}
|
||||
|
||||
func init() {
|
||||
schema.RegisterDialect("postgres", func(db *db.Database) schema.Schema {
|
||||
return &Postgres{
|
||||
db: db,
|
||||
esg: sqlgen.NewExpressionSQLGenerator("postgres", postgres.DialectOptions()),
|
||||
}
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,182 @@
|
||||
package postgres_test
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"git.fsdpf.net/go/db/v2/engine"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
_ "github.com/lib/pq"
|
||||
)
|
||||
|
||||
type postgresTest struct {
|
||||
suite.Suite
|
||||
schema schema.Schema
|
||||
}
|
||||
|
||||
var (
|
||||
tableName = "entry"
|
||||
|
||||
dropTable = "DROP TABLE IF EXISTS entry;"
|
||||
|
||||
createTable = "CREATE TABLE IF NOT EXISTS entry (" +
|
||||
"id SERIAL PRIMARY KEY," +
|
||||
"int INTEGER NOT NULL UNIQUE," +
|
||||
"float REAL NOT NULL," +
|
||||
"string VARCHAR(255) NOT NULL," +
|
||||
"time TIMESTAMP NOT NULL," +
|
||||
"bool BOOLEAN NOT NULL," +
|
||||
"bytes BYTEA NOT NULL" +
|
||||
");"
|
||||
)
|
||||
|
||||
func TestPostgresSuite(t *testing.T) {
|
||||
suite.Run(t, new(postgresTest))
|
||||
}
|
||||
|
||||
func (t *postgresTest) SetupSuite() {
|
||||
db := engine.Open(map[string]engine.DBConfig{
|
||||
"test-postgres": engine.NewDBConfig("postgres",
|
||||
engine.WithHost(os.Getenv("POSTGRES_HOST")),
|
||||
engine.WithPort(os.Getenv("POSTGRES_PORT")),
|
||||
engine.WithDatabase(os.Getenv("POSTGRES_DB")),
|
||||
engine.WithUsername(os.Getenv("POSTGRES_USER")),
|
||||
engine.WithPassword(os.Getenv("POSTGRES_PASSWD")),
|
||||
),
|
||||
}).Connection("test-postgres")
|
||||
|
||||
t.schema = schema.GetSchemaDialect(db)
|
||||
|
||||
db.Exec(dropTable)
|
||||
db.Exec(createTable)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestGetColumnListing() {
|
||||
result, err := t.schema.GetColumnListing(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().ElementsMatch([]string{"id", "int", "float", "string", "time", "bool", "bytes"}, result)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestTableExists() {
|
||||
result, err := t.schema.TableExists(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().Equal(result, true)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileCreate() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
bp.Create()
|
||||
|
||||
bp.BigIncrements("id").AutoIncrement().Comment("ID")
|
||||
bp.Boolean("enabled").Default("true").Comment("是否有效")
|
||||
bp.Char("created_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("创建者")
|
||||
bp.Char("owned_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("拥有者")
|
||||
bp.Timestamp("created_at").UseCurrent().Comment("创建时间")
|
||||
bp.Timestamp("updated_at").UseCurrent().Comment("更新时间")
|
||||
bp.DateTime("deleted_at").Nullable().Comment("删除时间")
|
||||
|
||||
sql := t.schema.CompileCreate(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"CREATE TABLE \"users\" (\n" +
|
||||
"\"id\" bigint NOT NULL BIGSERIAL PRIMARY KEY COMMENT 'ID',\n" +
|
||||
"\"enabled\" boolean NOT NULL DEFAULT true COMMENT '是否有效',\n" +
|
||||
"\"created_user\" char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '创建者',\n" +
|
||||
"\"owned_user\" char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '拥有者',\n" +
|
||||
"\"created_at\" timestamp with time zone NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" +
|
||||
"\"updated_at\" timestamp with time zone NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '更新时间',\n" +
|
||||
"\"deleted_at\" timestamp without time zone NULL COMMENT '删除时间'\n" +
|
||||
")",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileAdd() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Comment("用户名")
|
||||
bp.SmallInteger("age").Default("18").Comment("年龄")
|
||||
bp.String("sex", 1).Default("0").Comment("性别") // Using string instead of enum
|
||||
|
||||
sql := t.schema.CompileAdd(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE \"users\"\n" +
|
||||
"ADD COLUMN \"name\" varchar(50) NOT NULL COMMENT '用户名',\n" +
|
||||
"ADD COLUMN \"age\" smallint NOT NULL DEFAULT 18 COMMENT '年龄',\n" +
|
||||
"ADD COLUMN \"sex\" varchar(1) NOT NULL DEFAULT '0' COMMENT '性别'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileChange() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Change("username")
|
||||
bp.SmallInteger("age").Default("19").Comment("年龄").Change()
|
||||
|
||||
sql := t.schema.CompileChange(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE \"users\"\n" +
|
||||
"ALTER COLUMN \"name\" TYPE varchar(50) NOT NULL,\n" +
|
||||
"ALTER COLUMN \"age\" TYPE smallint NOT NULL DEFAULT 19 COMMENT '年龄'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileDropColumn() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.DropColumn("age", "sex")
|
||||
|
||||
sql := t.schema.CompileDropColumn(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE \"users\"\n" +
|
||||
"DROP COLUMN \"age\",\n" +
|
||||
"DROP COLUMN \"sex\"",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileDrop() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDrop(bp)
|
||||
|
||||
t.Equal([]string{"DROP TABLE \"users\""}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileDropIfExists() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDropIfExists(bp)
|
||||
t.Equal([]string{"DROP TABLE IF EXISTS \"users\""}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *postgresTest) TestCompileRename() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Rename("user")
|
||||
|
||||
sql := t.schema.CompileRename(bp)
|
||||
|
||||
t.Equal([]string{"ALTER TABLE \"users\" RENAME TO \"user\""}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
@@ -0,0 +1,264 @@
|
||||
package sqlite3
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/dialect/sqlite3"
|
||||
"git.fsdpf.net/go/db/v2/internal/sb"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"git.fsdpf.net/go/db/v2/sqlgen"
|
||||
)
|
||||
|
||||
var (
|
||||
sqlite3DefaultModifiers = []string{"VirtualAs", "StoredAs", "Nullable", "Default", "Increment"}
|
||||
sqlite3Serials = []string{"bigInteger", "integer", "mediumInteger", "smallInteger", "tinyInteger"}
|
||||
)
|
||||
|
||||
type Sqlite3 struct {
|
||||
db *db.Database
|
||||
esg sqlgen.ExpressionSQLGenerator
|
||||
}
|
||||
|
||||
// 判断表是否存在
|
||||
func (this Sqlite3) TableExists(table string) (bool, error) {
|
||||
result, err := this.db.From(db.T("sqlite_master")).Where(db.Ex{
|
||||
"name": table,
|
||||
"type": "table",
|
||||
}).Count()
|
||||
return result > 0, err
|
||||
}
|
||||
|
||||
// 字段列
|
||||
func (this Sqlite3) GetColumnListing(table string) ([]string, error) {
|
||||
cols := []struct {
|
||||
Cid string `db:"cid"`
|
||||
Name string `db:"name"`
|
||||
Type string `db:"type"`
|
||||
Notnull int32 `db:"notnull"`
|
||||
DfltValue string `db:"dflt_value"`
|
||||
Pk string `db:"pk"`
|
||||
}{}
|
||||
|
||||
err := this.db.ScanStructs(&cols, "PRAGMA table_info(entry)")
|
||||
|
||||
columns := []string{}
|
||||
|
||||
for _, item := range cols {
|
||||
columns = append(columns, item.Name)
|
||||
}
|
||||
|
||||
return columns, err
|
||||
}
|
||||
|
||||
// 创建表
|
||||
func (this Sqlite3) CompileCreate(bp *schema.Blueprint) (sqls []string) {
|
||||
temporary := db.L("CREATE")
|
||||
columns := []string{}
|
||||
|
||||
count := len(bp.GetAddedColumns()) - 1
|
||||
for i, column := range bp.GetAddedColumns() {
|
||||
sql := this.GenerateSQL(this.addModifiers("? ? ", bp, column),
|
||||
db.C(column.Name),
|
||||
db.L(this.GetColumnType(column)),
|
||||
)
|
||||
if v := column.GetComment(); v != "" {
|
||||
if i != count {
|
||||
sql += ","
|
||||
}
|
||||
sql += " -- " + v
|
||||
}
|
||||
columns = append(columns, sql)
|
||||
}
|
||||
sqls = append(sqls,
|
||||
this.GenerateSQL("? TABLE ? (\n?\n)",
|
||||
temporary,
|
||||
db.T(bp.GetTable()),
|
||||
db.L(strings.Join(columns, "\n")),
|
||||
),
|
||||
)
|
||||
return sqls
|
||||
}
|
||||
|
||||
// 添加字段
|
||||
func (this Sqlite3) CompileAdd(bp *schema.Blueprint) (sqls []string) {
|
||||
for _, column := range bp.GetAddedColumns() {
|
||||
sqls = append(sqls, this.addModifiers(
|
||||
this.GenerateSQL("ALTER TABLE ? ADD COLUMN ? ? ",
|
||||
db.T(bp.GetTable()),
|
||||
db.C(column.Name),
|
||||
db.L(this.GetColumnType(column)),
|
||||
), bp, column),
|
||||
)
|
||||
}
|
||||
return sqls
|
||||
}
|
||||
|
||||
// 修改字段
|
||||
func (this Sqlite3) CompileChange(bp *schema.Blueprint) (sqls []string) {
|
||||
for _, column := range bp.GetChangedColumns() {
|
||||
name := column.Name
|
||||
rename := column.Name
|
||||
old_name := fmt.Sprintf("%s_%d", name, time.Now().UnixNano())
|
||||
|
||||
sql := this.GenerateSQL("ALTER TABLE ? ADD COLUMN ? ? ", db.T(bp.GetTable()), db.C(old_name), db.L(this.GetColumnType(column)))
|
||||
sqls = append(sqls, this.addModifiers(sql, bp, column))
|
||||
if name := column.GetRename(); name != "" {
|
||||
rename = name
|
||||
}
|
||||
sqls = append(sqls, this.GenerateSQL("UPDATE ? SET ?=?", db.T(bp.GetTable()), db.C(old_name), db.C(name)))
|
||||
sqls = append(sqls, this.GenerateSQL("ALTER TABLE ? DROP COLUMN ?", db.T(bp.GetTable()), db.C(name)))
|
||||
sqls = append(sqls, this.GenerateSQL("ALTER TABLE ? RENAME COLUMN ? TO ?", db.T(bp.GetTable()), db.T(old_name), db.T(rename)))
|
||||
}
|
||||
return sqls
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (this Sqlite3) CompileDrop(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
// 删除表, 先判断再删除
|
||||
func (this Sqlite3) CompileDropIfExists(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE IF EXISTS ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
// 删除表, 删除列
|
||||
func (this Sqlite3) CompileDropColumn(bp *schema.Blueprint) (sqls []string) {
|
||||
for _, item := range bp.GetCommands() {
|
||||
if item.Type != "dropColumn" {
|
||||
continue
|
||||
}
|
||||
for _, col := range item.Columns {
|
||||
sqls = append(sqls, this.GenerateSQL("ALTER TABLE ? DROP COLUMN ?", db.T(bp.GetTable()), db.C(col)))
|
||||
}
|
||||
}
|
||||
return sqls
|
||||
}
|
||||
|
||||
// 表重命名
|
||||
func (this Sqlite3) CompileRename(bp *schema.Blueprint) []string {
|
||||
toName := ""
|
||||
commands := bp.GetCommands()
|
||||
if len(commands) == 0 {
|
||||
panic("new table undefined")
|
||||
}
|
||||
toName = bp.GetCommands()[0].To
|
||||
if toName == "" {
|
||||
panic("new table undefined")
|
||||
}
|
||||
return []string{this.GenerateSQL("RENAME TABLE ? TO ?", db.T(bp.GetTable()), db.T(toName))}
|
||||
}
|
||||
|
||||
func (this Sqlite3) GetColumnModifier(modifier string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
switch modifier {
|
||||
case "Collate":
|
||||
if v := column.GetCollaction(); v != "" {
|
||||
return this.GenerateSQL(" COLLATE ?", v)
|
||||
}
|
||||
case "VirtualAs":
|
||||
if v := column.GetVirtualAs(); v != "" {
|
||||
return this.GenerateSQL(" GENERATED ALWAYS AS (?) VIRTUAL", db.L(v))
|
||||
}
|
||||
case "StoredAs":
|
||||
if v := column.GetStoredAs(); v != "" {
|
||||
return this.GenerateSQL(" GENERATED ALWAYS AS (?) STORED", db.L(v))
|
||||
}
|
||||
case "Nullable":
|
||||
if column.GetVirtualAs() == "" && column.GetStoredAs() == "" {
|
||||
if column.IsNullable() {
|
||||
return "NULL"
|
||||
}
|
||||
return "NOT NULL"
|
||||
}
|
||||
case "Default":
|
||||
if column.IsUseCurrent() {
|
||||
return " DEFAULT CURRENT_TIMESTAMP"
|
||||
}
|
||||
v := column.GetDefault()
|
||||
if v == nil {
|
||||
return ""
|
||||
}
|
||||
return this.GenerateSQL(" DEFAULT ?", v)
|
||||
case "Increment":
|
||||
if column.IsAutoIncrement() {
|
||||
return " PRIMARY KEY AUTOINCREMENT"
|
||||
}
|
||||
case "Comment":
|
||||
if v := column.GetComment(); v != "" {
|
||||
return this.GenerateSQL(" COMMENT ?", v)
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (this Sqlite3) GetColumnType(column *schema.ColumnDefinition) string {
|
||||
if column.IsAutoIncrement() {
|
||||
return "INTEGER"
|
||||
}
|
||||
switch column.Type {
|
||||
case "char":
|
||||
return this.GenerateSQL("char(?)", column.Length)
|
||||
case "string":
|
||||
return this.GenerateSQL("varchar(?)", column.Length)
|
||||
case "text":
|
||||
return "text"
|
||||
case "integer":
|
||||
return this.GenerateSQL("INTEGER(?)", column.Length)
|
||||
case "bigInteger":
|
||||
return "bigint(20)"
|
||||
case "tinyInteger":
|
||||
return "tinyint(1)"
|
||||
case "smallInteger":
|
||||
return "smallint(4)"
|
||||
case "decimal":
|
||||
return this.GenerateSQL("numeric(?,?)", column.Total, column.Places)
|
||||
case "boolean":
|
||||
return "tinyint(1)"
|
||||
case "enum":
|
||||
return this.GenerateSQL("varchar check(? in?)", column.Name, column.Allowed)
|
||||
case "json":
|
||||
return "json"
|
||||
case "binary":
|
||||
return "binary"
|
||||
case "datetime":
|
||||
return "datetime"
|
||||
case "timestamp":
|
||||
return "timestamp"
|
||||
case "date":
|
||||
return "date"
|
||||
case "time":
|
||||
return "time"
|
||||
case "year":
|
||||
return "year"
|
||||
case "uuid":
|
||||
return "char(36)"
|
||||
}
|
||||
panic("Unsupported data type: " + column.Type)
|
||||
}
|
||||
|
||||
func (this Sqlite3) addModifiers(sql string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
for _, modifier := range sqlite3DefaultModifiers {
|
||||
sql += this.GetColumnModifier(modifier, bp, column)
|
||||
}
|
||||
return sql
|
||||
}
|
||||
|
||||
// 生成SQL
|
||||
func (this Sqlite3) GenerateSQL(sql string, args ...any) string {
|
||||
sb := sb.NewSQLBuilder(false)
|
||||
this.esg.Generate(sb, db.L(sql, args...))
|
||||
result, _, _ := sb.ToSQL()
|
||||
return result
|
||||
}
|
||||
|
||||
func init() {
|
||||
schema.RegisterDialect("sqlite3", func(db *db.Database) schema.Schema {
|
||||
return &Sqlite3{
|
||||
db: db,
|
||||
esg: sqlgen.NewExpressionSQLGenerator("sqlite3", sqlite3.DialectOptions()),
|
||||
}
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,192 @@
|
||||
package sqlite3_test
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/engine"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
var (
|
||||
tableName = "entry"
|
||||
|
||||
dropTable = "DROP TABLE IF EXISTS `entry`;"
|
||||
|
||||
createTable = "CREATE TABLE IF NOT EXISTS `entry` (" +
|
||||
"`id` INTEGER PRIMARY KEY AUTOINCREMENT," +
|
||||
"`int` INT NOT NULL ," +
|
||||
"`float` FLOAT NOT NULL ," +
|
||||
"`string` VARCHAR(255) NOT NULL ," +
|
||||
"`time` DATETIME NOT NULL ," +
|
||||
"`bool` TINYINT NOT NULL ," +
|
||||
"`bytes` BLOB NOT NULL" +
|
||||
");"
|
||||
)
|
||||
|
||||
type sqlite3Test struct {
|
||||
suite.Suite
|
||||
schema schema.Schema
|
||||
}
|
||||
|
||||
func TestSqlite3Suite(t *testing.T) {
|
||||
suite.Run(t, new(sqlite3Test))
|
||||
}
|
||||
|
||||
func (t *sqlite3Test) SetupSuite() {
|
||||
db := engine.Open(map[string]engine.DBConfig{
|
||||
"test-sqlite3": engine.NewDBConfig("sqlite3",
|
||||
engine.WithSQLiteFile(os.Getenv("DB_FILE")),
|
||||
),
|
||||
}).Connection("test-sqlite3")
|
||||
|
||||
t.schema = schema.GetSchemaDialect(db)
|
||||
|
||||
// db.Exec(dropTable)
|
||||
if _, err := db.Exec(createTable); err != nil {
|
||||
t.Require().NoError(err)
|
||||
}
|
||||
}
|
||||
|
||||
func (t *sqlite3Test) TestTableExists() {
|
||||
result, err := t.schema.TableExists(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().Equal(result, true)
|
||||
}
|
||||
|
||||
func (t *sqlite3Test) TestGetColumnListing() {
|
||||
result, err := t.schema.GetColumnListing(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().ElementsMatch([]string{"bool", "bytes", "float", "id", "int", "string", "time"}, result)
|
||||
}
|
||||
|
||||
// 创建列表
|
||||
func (t *sqlite3Test) TestCompileCreate() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
bp.Create()
|
||||
bp.Charset("utf8mb4")
|
||||
|
||||
bp.BigIncrements("id").AutoIncrement().Comment("ID")
|
||||
bp.Boolean("enabled").Default("1").Comment("是否有效")
|
||||
bp.Char("created_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("创建者")
|
||||
bp.Char("owned_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("拥有者")
|
||||
bp.Timestamp("created_at").UseCurrent().Comment("创建时间")
|
||||
bp.Timestamp("updated_at").UseCurrent().Default(db.L("ON UPDATE CURRENT_TIMESTAMP")).Comment("更新时间")
|
||||
bp.DateTime("deleted_at").Nullable().Comment("删除时间")
|
||||
|
||||
sql := t.schema.CompileCreate(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"CREATE TABLE `users` (\n" +
|
||||
"`id` INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, -- ID\n" +
|
||||
"`enabled` tinyint(1) NOT NULL DEFAULT '1', -- 是否有效\n" +
|
||||
"`created_user` char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000', -- 创建者\n" +
|
||||
"`owned_user` char(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000', -- 拥有者\n" +
|
||||
"`created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP, -- 创建时间\n" +
|
||||
"`updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP, -- 更新时间\n" +
|
||||
"`deleted_at` datetime NULL -- 删除时间\n" +
|
||||
")",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 添加字段
|
||||
func (t *sqlite3Test) TestCompileAdd() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Comment("用户名")
|
||||
bp.SmallInteger("age").Default("18").Comment("年龄")
|
||||
bp.Enum("sex", []string{"0", "1"}).Default("0").Comment("性别")
|
||||
|
||||
sql := t.schema.CompileAdd(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE `users` ADD COLUMN `name` varchar(50) NOT NULL",
|
||||
"ALTER TABLE `users` ADD COLUMN `age` smallint(4) NOT NULL DEFAULT '18'",
|
||||
"ALTER TABLE `users` ADD COLUMN `sex` varchar check('sex' in('0', '1')) NOT NULL DEFAULT '0'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 修改字段
|
||||
func (t *sqlite3Test) TestCompileChange() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Change("username")
|
||||
bp.SmallInteger("age").Default("19").Comment("年龄").Change()
|
||||
|
||||
sql := t.schema.CompileChange(bp)
|
||||
t.Equal(len([]string{
|
||||
"ALTER TABLE `users` ADD COLUMN `name_1742738970482105700` varchar(50) NOT NULL",
|
||||
"UPDATE `users` SET `name_1742738970482105700`=`name`",
|
||||
"ALTER TABLE `users` DROP COLUMN `name`",
|
||||
"ALTER TABLE `users` RENAME COLUMN `name_1742738970482105700` TO `username`",
|
||||
"ALTER TABLE `users` ADD COLUMN `age_1742738970482151800` smallint(4) NOT NULL DEFAULT '19'",
|
||||
"UPDATE `users` SET `age_1742738970482151800`=`age`",
|
||||
"ALTER TABLE `users` DROP COLUMN `age`",
|
||||
"ALTER TABLE `users` RENAME COLUMN `age_1742738970482151800` TO `age`",
|
||||
}), len(sql))
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 删除字段
|
||||
func (t *sqlite3Test) TestCompileDropColumn() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.DropColumn("age", "sex")
|
||||
|
||||
sql := t.schema.CompileDropColumn(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE `users` DROP COLUMN `age`",
|
||||
"ALTER TABLE `users` DROP COLUMN `sex`",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (t *sqlite3Test) TestCompileDrop() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDrop(bp)
|
||||
|
||||
t.Equal([]string{"DROP TABLE `users`"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 删除表
|
||||
func (t *sqlite3Test) TestCompileDropIfExists() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDropIfExists(bp)
|
||||
|
||||
t.Equal([]string{"DROP TABLE IF EXISTS `users`"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
// 表重命名
|
||||
func (t *sqlite3Test) TestCompileRename() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Rename("user")
|
||||
|
||||
sql := t.schema.CompileRename(bp)
|
||||
|
||||
t.Equal([]string{"RENAME TABLE `users` TO `user`"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
@@ -0,0 +1,201 @@
|
||||
package sqlserver
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"git.fsdpf.net/go/db/v2"
|
||||
"git.fsdpf.net/go/db/v2/dialect/sqlserver"
|
||||
"git.fsdpf.net/go/db/v2/internal/sb"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"git.fsdpf.net/go/db/v2/sqlgen"
|
||||
)
|
||||
|
||||
type SQLServer struct {
|
||||
db *db.Database
|
||||
esg sqlgen.ExpressionSQLGenerator
|
||||
}
|
||||
|
||||
var sqlServerDefaultModifiers = []string{
|
||||
"Nullable", "Default", "Increment", "Comment",
|
||||
}
|
||||
|
||||
func (this SQLServer) GetColumnListing(table string) ([]string, error) {
|
||||
cols := []string{}
|
||||
err := this.db.From(db.S("INFORMATION_SCHEMA").Table("COLUMNS")).Where(
|
||||
db.C("TABLE_NAME").Eq(table),
|
||||
db.C("TABLE_SCHEMA").Eq(db.L("SCHEMA_NAME()")),
|
||||
).Pluck(&cols, "COLUMN_NAME")
|
||||
return cols, err
|
||||
}
|
||||
|
||||
func (this SQLServer) TableExists(table string) (bool, error) {
|
||||
result, err := this.db.From(db.S("INFORMATION_SCHEMA").Table("TABLES")).Where(db.Ex{
|
||||
"TABLE_NAME": table,
|
||||
"TABLE_TYPE": "BASE TABLE",
|
||||
"TABLE_SCHEMA": db.L("SCHEMA_NAME()"),
|
||||
}).Count()
|
||||
return result > 0, err
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileCreate(bp *schema.Blueprint) []string {
|
||||
temporary := db.L("CREATE")
|
||||
if bp.IsTemporary() {
|
||||
temporary = db.L("CREATE")
|
||||
}
|
||||
columns := strings.Join(this.getAddedColumns(bp), ",\n")
|
||||
sql := this.GenerateSQL("? TABLE ? (\n?\n)", temporary, db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileAdd(bp *schema.Blueprint) []string {
|
||||
columns := strings.Join(schema.PrefixArray("ADD", this.getAddedColumns(bp)), ",\n")
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileChange(bp *schema.Blueprint) []string {
|
||||
columns := strings.Join(schema.PrefixArray("ALTER COLUMN", this.getChangedColumns(bp)), ",\n")
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(columns))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileDropColumn(bp *schema.Blueprint) []string {
|
||||
columns := []string{}
|
||||
for _, item := range bp.GetCommands() {
|
||||
if item.Type != "dropColumn" {
|
||||
continue
|
||||
}
|
||||
for _, col := range item.Columns {
|
||||
columns = append(columns, this.GenerateSQL("DROP COLUMN ?", db.C(col)))
|
||||
}
|
||||
}
|
||||
sql := this.GenerateSQL("ALTER TABLE ?\n?", db.T(bp.GetTable()), db.L(strings.Join(columns, ",\n")))
|
||||
return []string{sql}
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileDrop(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileDropIfExists(bp *schema.Blueprint) []string {
|
||||
return []string{this.GenerateSQL("DROP TABLE IF EXISTS ?", db.T(bp.GetTable()))}
|
||||
}
|
||||
|
||||
func (this SQLServer) CompileRename(bp *schema.Blueprint) []string {
|
||||
toName := ""
|
||||
commands := bp.GetCommands()
|
||||
if len(commands) == 0 {
|
||||
panic("new table undefined")
|
||||
}
|
||||
toName = bp.GetCommands()[0].To
|
||||
if toName == "" {
|
||||
panic("new table undefined")
|
||||
}
|
||||
return []string{this.GenerateSQL("EXEC sp_rename ?, ?", db.T(bp.GetTable()), db.T(toName))}
|
||||
}
|
||||
|
||||
func (this SQLServer) addModifiers(sql string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
for _, modifier := range sqlServerDefaultModifiers {
|
||||
sql += this.GetColumnModifier(modifier, bp, column)
|
||||
}
|
||||
return sql
|
||||
}
|
||||
|
||||
func (this SQLServer) GetColumnType(column *schema.ColumnDefinition) string {
|
||||
switch column.Type {
|
||||
case "char":
|
||||
return this.GenerateSQL("CHAR(?)", column.Length)
|
||||
case "string":
|
||||
return this.GenerateSQL("NVARCHAR(?)", column.Length)
|
||||
case "text":
|
||||
return "NVARCHAR(MAX)"
|
||||
case "integer":
|
||||
return "INT"
|
||||
case "bigInteger":
|
||||
return "BIGINT"
|
||||
case "smallInteger":
|
||||
return "SMALLINT"
|
||||
case "decimal":
|
||||
return this.GenerateSQL("DECIMAL(?,?)", column.Total, column.Places)
|
||||
case "boolean":
|
||||
return "BIT"
|
||||
case "enum":
|
||||
return this.GenerateSQL("NVARCHAR(?)", column.Length) // SQL Server doesn't have native enum
|
||||
case "json":
|
||||
return "NVARCHAR(MAX)" // SQL Server 2016+ has JSON support
|
||||
case "binary":
|
||||
return "VARBINARY(MAX)"
|
||||
case "datetime":
|
||||
return "DATETIME2"
|
||||
case "timestamp":
|
||||
return "DATETIME2"
|
||||
case "date":
|
||||
return "DATE"
|
||||
case "time":
|
||||
return "TIME"
|
||||
case "year":
|
||||
return "SMALLINT"
|
||||
case "uuid":
|
||||
return "UNIQUEIDENTIFIER"
|
||||
}
|
||||
panic("Unsupported data type: " + column.Type)
|
||||
}
|
||||
|
||||
func (this SQLServer) GetColumnModifier(modifier string, bp *schema.Blueprint, column *schema.ColumnDefinition) string {
|
||||
switch modifier {
|
||||
case "Nullable":
|
||||
if column.IsNullable() {
|
||||
return " NULL"
|
||||
}
|
||||
return " NOT NULL"
|
||||
case "Default":
|
||||
v := column.GetDefault()
|
||||
if v == nil {
|
||||
if column.IsUseCurrent() {
|
||||
return " DEFAULT GETDATE()"
|
||||
}
|
||||
return ""
|
||||
}
|
||||
return this.GenerateSQL(" DEFAULT ?", v)
|
||||
case "Increment":
|
||||
if column.IsAutoIncrement() {
|
||||
return " IDENTITY(1,1) PRIMARY KEY"
|
||||
}
|
||||
case "Comment":
|
||||
// SQL Server handles comments via extended properties
|
||||
return ""
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (this SQLServer) getAddedColumns(bp *schema.Blueprint) (columns []string) {
|
||||
for _, column := range bp.GetAddedColumns() {
|
||||
sql := this.GenerateSQL("? ?", db.C(column.Name), db.L(this.GetColumnType(column)))
|
||||
columns = append(columns, this.addModifiers(sql, bp, column))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (this SQLServer) getChangedColumns(bp *schema.Blueprint) (columns []string) {
|
||||
for _, column := range bp.GetChangedColumns() {
|
||||
sql := this.GenerateSQL("? ?", db.C(column.Name), db.L(this.GetColumnType(column)))
|
||||
columns = append(columns, this.addModifiers(sql, bp, column))
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (this SQLServer) GenerateSQL(sql string, args ...any) string {
|
||||
sb := sb.NewSQLBuilder(false)
|
||||
this.esg.Generate(sb, db.L(sql, args...))
|
||||
result, _, _ := sb.ToSQL()
|
||||
return result
|
||||
}
|
||||
|
||||
func init() {
|
||||
schema.RegisterDialect("sqlserver", func(db *db.Database) schema.Schema {
|
||||
return &SQLServer{
|
||||
db: db,
|
||||
esg: sqlgen.NewExpressionSQLGenerator("sqlserver", sqlserver.DialectOptions()),
|
||||
}
|
||||
})
|
||||
}
|
||||
@@ -0,0 +1,182 @@
|
||||
package sqlserver_test
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"git.fsdpf.net/go/db/v2/engine"
|
||||
"git.fsdpf.net/go/db/v2/schema"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
_ "github.com/denisenkom/go-mssqldb"
|
||||
)
|
||||
|
||||
type sqlServerTest struct {
|
||||
suite.Suite
|
||||
schema schema.Schema
|
||||
}
|
||||
|
||||
var (
|
||||
tableName = "entry"
|
||||
|
||||
dropTable = "IF OBJECT_ID('entry', 'U') IS NOT NULL DROP TABLE entry;"
|
||||
|
||||
createTable = "CREATE TABLE entry (" +
|
||||
"id INT NOT NULL IDENTITY(1,1) PRIMARY KEY," +
|
||||
"int INT NOT NULL UNIQUE," +
|
||||
"float REAL NOT NULL," +
|
||||
"string NVARCHAR(255) NOT NULL," +
|
||||
"time DATETIME2 NOT NULL," +
|
||||
"bool BIT NOT NULL," +
|
||||
"bytes VARBINARY(MAX) NOT NULL" +
|
||||
");"
|
||||
)
|
||||
|
||||
func TestSQLServerSuite(t *testing.T) {
|
||||
suite.Run(t, new(sqlServerTest))
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) SetupSuite() {
|
||||
db := engine.Open(map[string]engine.DBConfig{
|
||||
"test-sqlserver": engine.NewDBConfig("sqlserver",
|
||||
engine.WithHost(os.Getenv("SQLSERVER_HOST")),
|
||||
engine.WithPort(os.Getenv("SQLSERVER_PORT")),
|
||||
engine.WithDatabase(os.Getenv("SQLSERVER_DB")),
|
||||
engine.WithUsername(os.Getenv("SQLSERVER_USER")),
|
||||
engine.WithPassword(os.Getenv("SQLSERVER_PASSWD")),
|
||||
),
|
||||
}).Connection("test-sqlserver")
|
||||
|
||||
t.schema = schema.GetSchemaDialect(db)
|
||||
|
||||
db.Exec(dropTable)
|
||||
db.Exec(createTable)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestGetColumnListing() {
|
||||
result, err := t.schema.GetColumnListing(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().ElementsMatch([]string{"id", "int", "float", "string", "time", "bool", "bytes"}, result)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestTableExists() {
|
||||
result, err := t.schema.TableExists(tableName)
|
||||
t.Require().NoError(err)
|
||||
t.Require().Equal(result, true)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileCreate() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
bp.Create()
|
||||
|
||||
bp.BigIncrements("id").AutoIncrement().Comment("ID")
|
||||
bp.Boolean("enabled").Default("1").Comment("是否有效")
|
||||
bp.Char("created_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("创建者")
|
||||
bp.Char("owned_user", 36).Default("00000000-0000-0000-0000-000000000000").Comment("拥有者")
|
||||
bp.Timestamp("created_at").UseCurrent().Comment("创建时间")
|
||||
bp.Timestamp("updated_at").UseCurrent().Comment("更新时间")
|
||||
bp.DateTime("deleted_at").Nullable().Comment("删除时间")
|
||||
|
||||
sql := t.schema.CompileCreate(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"CREATE TABLE [users] (\n" +
|
||||
"[id] BIGINT NOT NULL IDENTITY(1,1) PRIMARY KEY COMMENT 'ID',\n" +
|
||||
"[enabled] BIT NOT NULL DEFAULT 1 COMMENT '是否有效',\n" +
|
||||
"[created_user] CHAR(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '创建者',\n" +
|
||||
"[owned_user] CHAR(36) NOT NULL DEFAULT '00000000-0000-0000-0000-000000000000' COMMENT '拥有者',\n" +
|
||||
"[created_at] DATETIME2 NOT NULL DEFAULT GETDATE() COMMENT '创建时间',\n" +
|
||||
"[updated_at] DATETIME2 NOT NULL DEFAULT GETDATE() COMMENT '更新时间',\n" +
|
||||
"[deleted_at] DATETIME2 NULL COMMENT '删除时间'\n" +
|
||||
")",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileAdd() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Comment("用户名")
|
||||
bp.SmallInteger("age").Default("18").Comment("年龄")
|
||||
bp.String("sex", 1).Default("0").Comment("性别")
|
||||
|
||||
sql := t.schema.CompileAdd(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE [users]\n" +
|
||||
"ADD [name] NVARCHAR(50) NOT NULL COMMENT '用户名',\n" +
|
||||
"ADD [age] SMALLINT NOT NULL DEFAULT 18 COMMENT '年龄',\n" +
|
||||
"ADD [sex] NVARCHAR(1) NOT NULL DEFAULT '0' COMMENT '性别'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileChange() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.String("name", 50).Change("username")
|
||||
bp.SmallInteger("age").Default("19").Comment("年龄").Change()
|
||||
|
||||
sql := t.schema.CompileChange(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE [users]\n" +
|
||||
"ALTER COLUMN [name] NVARCHAR(50) NOT NULL,\n" +
|
||||
"ALTER COLUMN [age] SMALLINT NOT NULL DEFAULT 19 COMMENT '年龄'",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileDropColumn() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.DropColumn("age", "sex")
|
||||
|
||||
sql := t.schema.CompileDropColumn(bp)
|
||||
|
||||
t.Equal([]string{
|
||||
"ALTER TABLE [users]\n" +
|
||||
"DROP COLUMN [age],\n" +
|
||||
"DROP COLUMN [sex]",
|
||||
}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileDrop() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDrop(bp)
|
||||
|
||||
t.Equal([]string{"DROP TABLE [users]"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileDropIfExists() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Drop()
|
||||
|
||||
sql := t.schema.CompileDropIfExists(bp)
|
||||
t.Equal([]string{"DROP TABLE IF EXISTS [users]"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
|
||||
func (t *sqlServerTest) TestCompileRename() {
|
||||
bp := schema.NewBlueprint("users")
|
||||
|
||||
bp.Rename("user")
|
||||
|
||||
sql := t.schema.CompileRename(bp)
|
||||
|
||||
t.Equal([]string{"EXEC sp_rename [users], [user]"}, sql)
|
||||
|
||||
t.T().Log(sql)
|
||||
}
|
||||
Reference in New Issue
Block a user