go_gameshield/internal/model/waflog.go

package model

import (
	"encoding/json"
	"time"
)

type WafLog struct {
	Id         int    `json:"id" form:"id" gorm:"column:id;primary_key;AUTO_INCREMENT;not null"`
	Uid        int    `json:"uid" form:"uid" gorm:"column:uid;default:0;not null"`
	Name       string `json:"name" form:"name" gorm:"column:name"`
	RequestIp  string `json:"requestIp" form:"requestIp" gorm:"column:request_ip"`
	RuleId     int    `json:"ruleId" form:"ruleId" gorm:"column:rule_id;default:0"`
	HostId     int    `json:"hostId" form:"hostId" gorm:"column:host_id;default:0"`
	UserAgent  string `json:"userAgent" form:"userAgent" gorm:"column:user_agent"`
	Api        string `json:"api" form:"api" gorm:"column:api"`
	ApiName    string `json:"apiName" form:"apiName" gorm:"column:api_name"`
	ApiType    string `json:"apiType" form:"apiType" gorm:"column:api_type"`
	ExtraData  json.RawMessage `json:"extraData" form:"extraData" gorm:"column:extra_data"`
	CreatedAt  time.Time `json:"createdAt" form:"createdAt" gorm:"column:created_at"`
	UpdatedAt  time.Time `json:"updatedAt" form:"updatedAt" gorm:"column:updated_at"`
	
	// 分表相关字段
	tableName string `gorm:"-"` // 动态表名，不存储到数据库
}

func (m *WafLog) TableName() string {
	if m.tableName != "" {
		return m.tableName
	}
	return "waf_log"
}

// SetTableName 设置动态表名(用于分表)
func (m *WafLog) SetTableName(tableName string) {
	m.tableName = tableName
}

// GetBaseTableName 实现TableModel接口
func (m *WafLog) GetBaseTableName() string {
	return "waf_log"
}

// GetCreatedAt 实现TableModel接口
func (m *WafLog) GetCreatedAt() time.Time {
	return m.CreatedAt
}

// WafLogWithGatewayIP 包含WAF日志及其在特定时间点的网关IP数据
type WafLogWithGatewayIP struct {
	WafLog
	GatewayIpData json.RawMessage `json:"gatewayIpData" gorm:"column:gateway_ip_data"`
}