[feat] 新增 表结构操作

schema/dialect/mysql/mysql_test.go

@@ -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)
+}