fix: 初验针对修改

internal/models/schema_version.go

@@ -0,0 +1,53 @@
+package models
+
+import (
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+const requiredSchemaScript = "0001_baseline.up.sql"
+
+var requiredSchema = struct {
+	checksum string
+	tables   []string
+}{
+	checksum: "092328c2cbd4e6316ae0e7fe2dd0f986ba30daf712945b8fa14be7f54c01f286",
+	tables:   []string{"logs_events", "logs_syslog_rules", "logs_trap_rules", "logs_trace_spans"},
+}
+
+// RequireSchemaVersion 确认当前数据库已经由受控迁移脚本初始化。
+func RequireSchemaVersion(db *gorm.DB, service string) error {
+	if db == nil {
+		return fmt.Errorf("%s 数据库连接未初始化", service)
+	}
+
+	var count int64
+	err := db.Raw(`
+SELECT COUNT(*)
+FROM public.ops_schema_migrations
+WHERE service = ? AND script_name = ? AND checksum = ? AND status = 'success'`,
+		service, requiredSchemaScript, requiredSchema.checksum).Scan(&count).Error
+	if err != nil {
+		return fmt.Errorf("%s 数据库版本检查失败，请先执行 migrate-all.sh up --service %s: %w", service, service, err)
+	}
+	if count != 1 {
+		return fmt.Errorf("%s 数据库迁移版本不匹配，请执行 migrate-all.sh status --service %s", service, service)
+	}
+
+	for _, table := range requiredSchema.tables {
+		var valid bool
+		err = db.Raw(`
+SELECT to_regclass(?) IS NOT NULL AND EXISTS (
+  SELECT 1
+  FROM pg_constraint c
+  JOIN pg_class t ON t.oid = c.conrelid
+  JOIN pg_namespace n ON n.oid = t.relnamespace
+  WHERE n.nspname = 'public' AND t.relname = ? AND c.contype = 'p'
+)`, "public."+table, table).Scan(&valid).Error
+		if err != nil || !valid {
+			return fmt.Errorf("%s 数据库关键表结构不完整：%s", service, table)
+		}
+	}
+	return nil
+}