[feat] 新增表结构操作

2025-03-24 16:58:37 +08:00
parent f9ca4a49ce
commit 764eccfafd
14 changed files with 2606 additions and 0 deletions
@@ -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)
+}