package cache

import (
	"fmt"
	"log"
	"strings"
	"time"
)

type _CountRecord struct {
	Count int64
}

func assembleCountKey(entityName string, additional ...string) string {
	var keys = make([]string, 0)
	keys = append(keys, strings.ToUpper(entityName))
	keys = append(keys, additional...)
	var b strings.Builder
	b.WriteString(TAG_COUNT)
	for _, s := range keys {
		fmt.Fprintf(&b, ":%s", s)
	}
	return b.String()
}

// 向缓存中缓存模型名称明确的包含指定条件的实体记录数量
func CacheCount(relationNames []string, entityName string, count int64, conditions ...string) error {
	countKey := assembleCountKey(entityName, conditions...)
	cacheInstance := &_CountRecord{Count: count}
	err := Cache(countKey, cacheInstance, 5*time.Minute)
	for _, relationName := range relationNames {
		CacheRelation(relationName, STORE_TYPE_KEY, countKey)
	}
	log.Printf("[debug] [cache|save] count key: [%s], err:  %+v", countKey, err)
	return err
}

// 从缓存中获取模型名称明确的，包含指定条件的实体记录数量
func RetreiveCount(entityName string, condtions ...string) (int64, error) {
	countKey := assembleCountKey(entityName, condtions...)
	exist, err := Exists(countKey)
	if err != nil {
		return -1, err
	}
	if !exist {
		return -1, nil
	}
	instance, err := Retreive[_CountRecord](countKey)
	log.Printf("[debug] [cache|retreive] count key: [%s], instance: [%v], err: %+v", countKey, instance, err)
	if instance != nil {
		return instance.Count, nil
	} else {
		return -1, nil
	}
}

// 删除指定模型名称的数量缓存
func AbolishCountEntity(entityName string) error {
	pattern := fmt.Sprintf("%s:%s:*", TAG_COUNT, strings.ToUpper(entityName))
	return DeleteAll(pattern)
}