package model

import (
	"errors"
	"time"

	"gitee.com/linqwen/momo/app/conf/dao"
	"gitee.com/linqwen/momo/base"
	"gitee.com/linqwen/momo/cache"
	"gorm.io/gorm"
)

func init() {
	dao.ConfDb.AutoMigrate(&ClientEntity{})
	initClientCache()
}

type ClientEntity struct {
	base.SnowflakeIDModel
	Status  int    `gorm:"column:status;type:int" json:"Status"`
	Access  int    `gorm:"column:access;default:1" json:"Access" form:"Access" comment:"Access"`
	Task    int    `gorm:"column:task;type:int" json:"Task"`
	Cpu     string `gorm:"column:cpu;size:255" json:"Cpu"`
	Mem     string `gorm:"column:mem;size:255" json:"Mem"`
	Account string `gorm:"column:account;size:255" json:"Account"`
	Name    string `gorm:"column:name;size:255" json:"Name"`
	Ip      string `gorm:"column:ip;size:39;unique" json:"Ip"` // 对于MySQL和PostgreSQL使用varchar长度控制，SQLite仍不支持inet类型

}

func (ClientEntity) TableName() string { return "conf_client" }

func (s *ClientEntity) AfterSave(tx *gorm.DB) (err error) {
	go initClientCache()
	return nil
}

func (s *ClientEntity) AfterDelete(tx *gorm.DB) (err error) {
	go initClientCache()
	return nil
}

func initClientCache() {
	time.Sleep(100 * time.Millisecond)
	var objs []ClientEntity
	var objmap = make(map[string]int)
	if err := dao.ConfDb.Where("status = ?", 1).Find(&objs).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return
		}
		return
	}
	if len(objs) > 0 {
		for _, a := range objs {
			objmap[a.Ip] = a.Access
		}
		cache.SetCache("clientMap", objmap, time.Now().Add(365*24*time.Hour))
	}
}