76 lines
3.1 KiB
Go
76 lines
3.1 KiB
Go
package business
|
|
|
|
import (
|
|
"context"
|
|
"gorm.io/gorm"
|
|
datasystem "kra/internal/data/system"
|
|
datatask "kra/internal/data/task"
|
|
"kra/internal/modules/surface"
|
|
"kra/pkg/module"
|
|
)
|
|
|
|
var alertExemptionMenu = module.Menu{Name: "sw_ExemptChannels", Path: "ExemptChannels", ParentName: "sw_AlertCenter", Component: "view/sw/ExemptChannels.vue", Title: "免预警渠道", Icon: "CircleCheck", Sort: 3}
|
|
var alertDetectionTask = module.TimedTask{Name: "AlertTrigger", Description: "每日检测预警并清理流失渠道豁免", Spec: "CRON_TZ=Asia/Shanghai 0 0 * * *", MethodName: "alertTriggerTask", Enabled: true}
|
|
|
|
func migrateAlertSurface(db *gorm.DB) error {
|
|
if db.Migrator().HasTable("sys_timed_tasks") {
|
|
var count int64
|
|
q := db.Table("sys_timed_tasks").Where("method_name=? AND deleted_at IS NULL", alertDetectionTask.MethodName)
|
|
if err := q.Count(&count).Error; err != nil {
|
|
return err
|
|
}
|
|
if count == 0 {
|
|
if err := datatask.SeedDefaults(context.Background(), db, []module.TimedTask{alertDetectionTask}); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
// 旧检测任务调整到上海零点；保留管理员主动停用的状态。
|
|
if err := db.Table("sys_timed_tasks").Where("method_name=? AND deleted_at IS NULL", alertDetectionTask.MethodName).Updates(map[string]any{"spec": alertDetectionTask.Spec, "with_seconds": false}).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
if !db.Migrator().HasTable("sys_authorities") {
|
|
return nil
|
|
}
|
|
var count int64
|
|
if err := db.Table("sys_authorities").Where("authority_id=888 AND deleted_at IS NULL").Count(&count).Error; err != nil {
|
|
return err
|
|
}
|
|
if count == 0 {
|
|
return nil
|
|
}
|
|
apis := []module.API{}
|
|
for _, api := range surface.APIsForPrefix("/sw/", 0) {
|
|
switch api.Path {
|
|
case "/sw/alertRecord/platforms", "/sw/alertRecord/ignore", "/sw/alertRule/gap", "/sw/alertExemption/list", "/sw/alertExemption/candidates", "/sw/alertExemption/add", "/sw/alertExemption/remove/:id":
|
|
apis = append(apis, api)
|
|
}
|
|
}
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|
if err := datasystem.EnsureAdminSurface(tx, module.Surface{Menus: []module.Menu{alertExemptionMenu}, APIs: apis}, 888); err != nil {
|
|
return err
|
|
}
|
|
if err := datasystem.GrantMenuAPIs(tx, "sw_AlertList", apis); err != nil {
|
|
return err
|
|
}
|
|
if err := datasystem.GrantMenuAPIs(tx, "sw_AlertRules", apis); err != nil {
|
|
return err
|
|
}
|
|
var menuID uint
|
|
if err := tx.Table("sys_base_menus").Select("id").Where("name=? AND deleted_at IS NULL", alertExemptionMenu.Name).Take(&menuID).Error; err != nil {
|
|
return err
|
|
}
|
|
var roles []uint
|
|
if err := tx.Table("sys_users").Where("level=1 AND deleted_at IS NULL").Distinct().Pluck("authority_id", &roles).Error; err != nil {
|
|
return err
|
|
}
|
|
for _, role := range roles {
|
|
if err := tx.Exec("INSERT INTO sys_authority_menus(sys_authority_authority_id,sys_base_menu_id) SELECT ?,? WHERE NOT EXISTS (SELECT 1 FROM sys_authority_menus WHERE sys_authority_authority_id=? AND sys_base_menu_id=?)", role, menuID, role, menuID).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return datasystem.GrantMenuAPIs(tx, alertExemptionMenu.Name, apis)
|
|
})
|
|
}
|