feat(db): 添加对达梦数据库的支持

- 在应用程序中新增对达梦数据库（DM）的配置和连接支持 - 实现 SetDmDB 函数以配置达梦数据库连接 - 更新数据库操作逻辑，支持达梦特有的 SQL 语法和功能 - 在相关文件中添加达梦数据库的处理逻辑，包括表创建、数据插入和查询 - 更新 go.mod 和 go.sum 文件以引入达梦数据库驱动 - 增强文档，详细说明达梦数据库的配置和使用方法
2026-03-20 10:46:51 +08:00
parent 1546967918
commit 7f7b585ffb
194 changed files with 211502 additions and 2328 deletions
@@ -0,0 +1,126 @@
+package app
+
+import (
+	"fmt"
+
+	. "code.hoteas.com/golang/hotime/db"
+)
+
+// SetupDMDatabase 为达梦数据库创建测试表并灌入测试数据
+func SetupDMDatabase(db *HoTimeDB) {
+	if db.Type != "dm" && db.Type != "dameng" {
+		return
+	}
+
+	fmt.Println("[DM Setup] 开始初始化达梦数据库测试环境...")
+
+	prefix := db.GetPrefix()
+	createDMTables(db, prefix)
+	seedTestData(db, "DM")
+
+	fmt.Println("[DM Setup] 达梦数据库测试环境初始化完成")
+}
+
+func dmTableExists(db *HoTimeDB, tableName string) bool {
+	// COUNT(*) 即使表为空也会返回一行；若表不存在则报错返回空切片。
+	// 相比 USER_TABLES，这里直接检测当前 schema 下表的可访问性，避免
+	// 连接 schema（?schema=TEST）与 USER_TABLES（SYSDBA 自身 schema）错配。
+	res := db.Query(`SELECT COUNT(*) as cnt FROM "` + tableName + `"`)
+	return len(res) > 0
+}
+
+func createDMTables(db *HoTimeDB, prefix string) {
+	// admin 表
+	tbl := prefix + "admin"
+	if !dmTableExists(db, tbl) {
+		fmt.Println("[DM Setup] 创建表:", tbl)
+		db.Exec(`CREATE TABLE "` + tbl + `" (
+			"id" INT IDENTITY(1,1) PRIMARY KEY,
+			"name" VARCHAR(100),
+			"phone" VARCHAR(20),
+			"state" INT DEFAULT 0,
+			"password" VARCHAR(100),
+			"role_id" INT DEFAULT 0,
+			"title" VARCHAR(100),
+			"create_time" TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+			"modify_time" TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+		)`)
+		db.Exec(`CREATE UNIQUE INDEX "uk_` + tbl + `_phone" ON "` + tbl + `" ("phone")`)
+	}
+
+	// ctg 分类表
+	tbl = prefix + "ctg"
+	if !dmTableExists(db, tbl) {
+		fmt.Println("[DM Setup] 创建表:", tbl)
+		db.Exec(`CREATE TABLE "` + tbl + `" (
+			"id" INT IDENTITY(1,1) PRIMARY KEY,
+			"name" VARCHAR(100),
+			"state" INT DEFAULT 0,
+			"create_time" TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+			"modify_time" TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+		)`)
+	}
+
+	// article 文章表
+	tbl = prefix + "article"
+	if !dmTableExists(db, tbl) {
+		fmt.Println("[DM Setup] 创建表:", tbl)
+		db.Exec(`CREATE TABLE "` + tbl + `" (
+			"id" INT IDENTITY(1,1) PRIMARY KEY,
+			"title" VARCHAR(200),
+			"author" VARCHAR(100),
+			"content" CLOB,
+			"state" INT DEFAULT 0,
+			"click_num" INT DEFAULT 0,
+			"sort" INT DEFAULT 0,
+			"img" VARCHAR(500),
+			"ctg_id" INT DEFAULT 0,
+			"admin_id" INT DEFAULT 0,
+			"create_time" TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+			"modify_time" TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+		)`)
+	}
+
+	// test_batch 测试批量表
+	tbl = prefix + "test_batch"
+	if !dmTableExists(db, tbl) {
+		fmt.Println("[DM Setup] 创建表:", tbl)
+		db.Exec(`CREATE TABLE "` + tbl + `" (
+			"id" INT IDENTITY(1,1) PRIMARY KEY,
+			"name" VARCHAR(100),
+			"title" VARCHAR(200),
+			"state" INT DEFAULT 0,
+			"create_time" TIMESTAMP
+		)`)
+	}
+
+	// cached 老版本缓存表（兼容模式测试需要）
+	tbl = prefix + "cached"
+	if !dmTableExists(db, tbl) {
+		fmt.Println("[DM Setup] 创建表:", tbl)
+		db.Exec(`CREATE TABLE "` + tbl + `" (
+			"id" INT IDENTITY(1,1) PRIMARY KEY,
+			"key" VARCHAR(64),
+			"value" CLOB,
+			"endtime" BIGINT,
+			"time" BIGINT
+		)`)
+	}
+
+	// hotime_cache 新版缓存表
+	tbl = prefix + "hotime_cache"
+	if !dmTableExists(db, tbl) {
+		fmt.Println("[DM Setup] 创建表:", tbl)
+		db.Exec(`CREATE TABLE "` + tbl + `" (
+			"id" INT IDENTITY(1,1) PRIMARY KEY,
+			"key" VARCHAR(64) NOT NULL,
+			"value" TEXT,
+			"end_time" TIMESTAMP,
+			"state" INT DEFAULT 0,
+			"create_time" TIMESTAMP,
+			"modify_time" TIMESTAMP,
+			CONSTRAINT "uk_` + tbl + `_key" UNIQUE ("key")
+		)`)
+		db.Exec(`CREATE INDEX "idx_` + tbl + `_end_time" ON "` + tbl + `" ("end_time")`)
+	}
+}