mirror of
https://gitee.com/samwaf/SamWaf.git
synced 2026-11-02 19:21:44 +08:00
86 lines
3.0 KiB
Go
86 lines
3.0 KiB
Go
package wafdb
|
|
|
|
import (
|
|
"SamWaf/common/zlog"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
/*
|
|
*
|
|
一些后续补丁
|
|
*/
|
|
func pathLogSql(db *gorm.DB) {
|
|
// 20241018 创建联合索引 weblog
|
|
startTime := time.Now()
|
|
|
|
zlog.Info("ready create index maybe use a few minutes ")
|
|
err := db.Exec("CREATE INDEX IF NOT EXISTS idx_web_logs_task_flag_time ON web_logs (task_flag, unix_add_time)").Error
|
|
if err != nil {
|
|
panic("failed to create index: idx_web_logs_task_flag_time " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "idx_web_logs_task_flag_time created")
|
|
}
|
|
// 创建联合索引
|
|
err = db.Exec("CREATE INDEX IF NOT EXISTS idx_web_time_tenant_user_code ON web_logs (unix_add_time, tenant_id,user_code)").Error
|
|
if err != nil {
|
|
panic("failed to create index: idx_web_time_tenant_user_code " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "idx_web_time_tenant_user_code created")
|
|
}
|
|
// 详情索引
|
|
err = db.Exec("CREATE INDEX IF NOT EXISTS idx_req_uuid_web_logs ON web_logs (REQ_UUID, tenant_id, user_code)").Error
|
|
if err != nil {
|
|
panic("failed to create index:idx_req_uuid_web_logs " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "idx_unique_req_uuid_web_logs created")
|
|
}
|
|
// 整体索引
|
|
err = db.Exec("CREATE INDEX IF NOT EXISTS idx_tenant_usercode_web_logs ON web_logs ( tenant_id, user_code)").Error
|
|
if err != nil {
|
|
panic("failed to create index: idx_tenant_usercode_web_logs " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "idx_tenant_usercode_web_logs created")
|
|
}
|
|
// 记录结束时间并计算耗时
|
|
duration := time.Since(startTime)
|
|
zlog.Info("create index completely", "duration", duration.String())
|
|
}
|
|
func pathCoreSql(db *gorm.DB) {
|
|
// 20241018 创建联合索引 weblog
|
|
err := db.Exec("UPDATE system_configs SET item_class = 'system' WHERE item_class IS NULL or item_class='' ").Error
|
|
if err != nil {
|
|
panic("failed to system_config :item_class " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "system_config :item_class init successfully")
|
|
}
|
|
// 20241026 是否自动跳转https站点
|
|
err = db.Exec("UPDATE hosts SET auto_jump_http_s=0 WHERE auto_jump_http_s IS NULL ").Error
|
|
if err != nil {
|
|
panic("failed to hosts :auto_jump_https " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "hosts :auto_jump_https init successfully")
|
|
}
|
|
//20241030 初始化一次状态信息
|
|
err = db.Exec("UPDATE hosts SET start_status=0 WHERE start_status IS NULL ").Error
|
|
if err != nil {
|
|
panic("failed to hosts :start_status " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "hosts :start_status init successfully")
|
|
}
|
|
//20241106 创建iptagtag索引
|
|
err = db.Exec("CREATE UNIQUE INDEX IF NOT EXISTS uni_iptags_full ON ip_tags (user_code, tenant_id, ip, ip_tag)").Error
|
|
if err != nil {
|
|
panic("failed to create index : uni_iptags_full " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "uni_iptags_full created")
|
|
}
|
|
// 创建iptag ip索引
|
|
err = db.Exec("CREATE INDEX IF NOT EXISTS idx_iptag_ip ON ip_tags ( user_code, tenant_id, ip)").Error
|
|
if err != nil {
|
|
panic("failed to create index: idx_iptag_ip " + err.Error())
|
|
} else {
|
|
zlog.Info("db", "idx_iptag_ip created")
|
|
}
|
|
}
|