yunzerwebsiteallinone/go/models/init.go

package models

import (
	"fmt"

	"github.com/beego/beego/v2/client/orm"
	beego "github.com/beego/beego/v2/server/web"
	_ "github.com/go-sql-driver/mysql"
)

// Orm 全局 ORM 对象，供业务层和控制器使用
var Orm orm.Ormer

// Init 初始化模型层资源（数据库连接、模型注册等）。
func Init(_ string) {
	// 从配置读取数据库连接信息
	user, _ := beego.AppConfig.String("mysqluser")
	pass, _ := beego.AppConfig.String("mysqlpass")
	urls, _ := beego.AppConfig.String("mysqlurls")
	dbname, _ := beego.AppConfig.String("mysqldb")

	if user == "" || urls == "" || dbname == "" {
		panic("数据库配置(mysqluser/mysqlurls/mysqldb) 未正确设置")
	}

	// 组装 DSN：user:pass@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local
	dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", user, pass, urls, dbname)

	// 注册默认数据库
	if err := orm.RegisterDataBase("default", "mysql", dsn); err != nil {
		panic("注册数据库失败: " + err.Error())
	}

	// 注册模型
	orm.RegisterModel(
		new(SystemTenant),
		new(SystemTenantUser),
		new(BackendOrganization),
		new(BackendEmployee),
		new(BackendPosition),
		new(BackendErpContact),
		new(BackendApprovalFlow),
		new(BackendApprovalRecord),
		new(BackendReimbursement),
		new(BackendReimbursementItem),
		new(BackendReimbursementExpenseType),
		new(SystemMenu),
		new(AdminUser),
		new(AdminRole),
		new(SystemFile),
		new(SystemFilesCategory),
		new(SystemSMSTask),
		new(SystemOperationLog),
		new(SystemLoginLog),
		new(SystemDomainPool),
		new(SystemTenantDomain),
		new(SystemModules),
		new(StorageConfig),
		new(TenantSiteSetting),
		new(ComplaintCategory),
		new(PlatformComplaint),
		new(SystemSoftwareUpgrade),
		new(PlatformCursorEquipment),
		new(PlatformCursorActivationCode),
		new(PlatformCursorEquipmentIpLog),
		new(PlatformAccountPoolKiro),
		new(PlatformAccountPoolWindsurf),
		new(PlatformAccountPoolCursor),
		new(PlatformAccountPoolCodex),
		new(PlatformNotebook),
		new(PlatformAgentApi),
		new(PlatformPasswordStore),
		new(BackendPasswordStore),
		new(BackendUserNotifyConfig),
		new(TenantCrmCustomer),
		new(TenantCrmSupplier),
		new(TenantCrmClue),
		new(TenantCrmBusiness),
		new(TenantCrmProject),
		new(TenantCrmFollow),
		new(TenantCrmAttach),
		new(TenantCrmEntityContact),
		new(TenantCrmOperateLog),
		new(ErpAccountSet),
		new(ErpNormalSetting),
		new(ErpCompanyContact),
		new(BackendAiProvider),
		new(BackendAiChatSession),
		new(BackendAiChatMessage),
		new(BackendAiChatPreset),
		new(BackendMcpTool),
		new(BackendMcpServer),

		new(CmsArticleCategory),
		new(CmsArticle),
		new(CmsProductCategory),
		new(CmsProduct),
		new(CmsSolutionCategory),
		new(CmsSolution),

		new(CmsFrontendTemplate),
		new(CmsFrontendBanner),
		new(CmsFrontendFriendlink),
		new(CmsFrontendOnepage),
		new(BackendMenuFront),

		new(SystemReminderList),

		new(SystemNormalSetting),
		new(PlatformNormalSetting),
		new(BackendNormalSetting),

		new(PlatformSchedule),
		new(PlatformScheduleReminder),
		new(PlatformScheduleReminderSendLog),

		new(BackendNotebook),
		new(BackendSchedule),
		new(BackendScheduleReminder),
		new(BackendScheduleReminderSendLog),

		new(OaSchedule),
		new(OaNotice),
		new(OaDocCategory),
		new(OaDoc),
		new(OaDocLink),
		new(OaDocShare),
		new(BackendOaCompensationScheme),
		new(BackendOaPayroll),
		new(BackendOaPayrollItem),
		new(BackendEmployeeFile),
		new(BackendEmployeeFileRecord),
	)

	// 创建全局 Ormer
	Orm = orm.NewOrm()

	// 补齐历史库缺失的列（已存在时 MySQL 报 duplicate column，统一忽略）
	EnsureAdminRoleTenantColumn()
	EnsureAdminRoleCustomColumn()
	EnsureTenantUserOrgColumn()
	EnsureCrmCustomerPoolColumns()
	EnsureCrmCreateUserColumn()
}


// EnsureCrmCreateUserColumn 补齐客户/供应商的创建人字段（用于删除权限校验）。
// 同样兼容历史表名差异，表不存在或列已存在时忽略错误。
func EnsureCrmCreateUserColumn() {
	tables := []string{
		new(TenantCrmCustomer).TableName(), "yz_tenant_crm_customer",
		new(TenantCrmSupplier).TableName(), "yz_tenant_crm_supplier",
	}
	for _, table := range tables {
		sql := "ALTER TABLE " + table +
			" ADD COLUMN create_user_id varchar(64) NOT NULL DEFAULT '' COMMENT '创建人用户ID，用于删除权限校验'"
		if _, err := Orm.Raw(sql).Exec(); err != nil {
			continue
		}
	}
}

// EnsureCrmCustomerPoolColumns 补齐客户公海相关字段。
// 兼容历史表名差异：模型 TableName 为 yz_backend_erp_customer，初始化 SQL 中为 yz_tenant_crm_customer，
// 对两个候选表名都执行 ALTER（表不存在或列已存在时忽略错误）。
func EnsureCrmCustomerPoolColumns() {
	candidates := []string{
		new(TenantCrmCustomer).TableName(),
		"yz_tenant_crm_customer",
	}
	cols := []string{
		"in_pool tinyint NOT NULL DEFAULT 0 COMMENT '是否在公海：0-否 1-是'",
		"owner_user_id varchar(64) NOT NULL DEFAULT '' COMMENT '负责人ID，公海中为空'",
		"owner_user_name varchar(128) NOT NULL DEFAULT '' COMMENT '负责人姓名'",
		"last_owner_name varchar(128) NOT NULL DEFAULT '' COMMENT '移入公海前的原负责人'",
		"pool_in_time datetime DEFAULT NULL COMMENT '进入公海时间'",
		"pool_reason varchar(200) NOT NULL DEFAULT '' COMMENT '移入公海原因'",
	}
	for _, table := range candidates {
		for _, col := range cols {
			sql := fmt.Sprintf("ALTER TABLE %s ADD COLUMN %s", table, col)
			if _, err := Orm.Raw(sql).Exec(); err != nil {
				// 列已存在或表不存在时忽略
				continue
			}
		}
	}
}