EdgeAPI/internal/db/models/tcp_firewall_policy_dao.go

package models

import (
	_ "github.com/go-sql-driver/mysql"
	"github.com/iwind/TeaGo/Tea"
	"github.com/iwind/TeaGo/dbs"
)

type TCPFirewallPolicyDAO dbs.DAO

func NewTCPFirewallPolicyDAO() *TCPFirewallPolicyDAO {
	return dbs.NewDAO(&TCPFirewallPolicyDAO{
		DAOObject: dbs.DAOObject{
			DB:     Tea.Env,
			Table:  "edgeTCPFirewallPolicies",
			Model:  new(TCPFirewallPolicy),
			PkName: "id",
		},
	}).(*TCPFirewallPolicyDAO)
}

var SharedTCPFirewallPolicyDAO *TCPFirewallPolicyDAO

func init() {
	dbs.OnReady(func() {
		SharedTCPFirewallPolicyDAO = NewTCPFirewallPolicyDAO()
	})
}

// 初始化
func (this *TCPFirewallPolicyDAO) Init() {
	this.DAOObject.Init()
	this.DAOObject.OnUpdate(func() error {
		return SharedSysEventDAO.CreateEvent(NewServerChangeEvent())
	})
	this.DAOObject.OnInsert(func() error {
		return SharedSysEventDAO.CreateEvent(NewServerChangeEvent())
	})
	this.DAOObject.OnDelete(func() error {
		return SharedSysEventDAO.CreateEvent(NewServerChangeEvent())
	})
}