package model

import (
	"errors"
	"time"

	"gitee.com/linqwen/momo/cache"

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

func init() {
	dao.ConfDb.AutoMigrate(&DictEntity{})
	initDictCache()
}

type DictEntity struct {
	base.SnowflakeIDModel
	ParentId int64  `gorm:"column:parent_id" json:"ParentId,string"`
	Code     string `gorm:"column:code" json:"Code"`
	Key      string `gorm:"column:key;unique" json:"Key"`
	Value    string `gorm:"column:value;size:255" json:"value"`
	Status   int    `gorm:"column:status;type:int" json:"Status"`
	Remark   string `gorm:"column:remark" json:"Remark"`
}

func (DictEntity) TableName() string { return "conf_dict" }

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

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

func initDictCache() {
	time.Sleep(100 * time.Millisecond)
	var objs []DictEntity
	var objmap = make(map[string]string)
	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.Key] = a.Value
		}
		cache.SetCache("dict", objmap, time.Now().Add(365*24*time.Hour))
	}
}