[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,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)
+}