package cache

import (
	"context"
	"joybuywork/internal/app"
	"joybuywork/internal/plugins/jsonm"
	"joybuywork/internal/repository/dao"
	"joybuywork/internal/repository/dbcheck"
	"joybuywork/internal/repository/model"
	"fmt"
)

type CampaignReferral struct {
	Cache
}

func NewCampaignReferral(ctx context.Context) *CampaignReferral {
	c := new(CampaignReferral)
	c.init(ctx, &model.FissionCampaignReferral{})

	return c
}

// FindByCampaignId campaignId 获取数据
func (c CampaignReferral) FindByCampaignId(campaignId int64) (*model.FissionCampaignReferral, error) {
	data, err := c.get(c.campaignIdKey(campaignId), func() ([]byte, error) {
		m := dao.Q.FissionCampaignReferral
		data, err := m.WithContext(c.ctx).Where(m.CampaignID.Eq(campaignId)).Take()
		if err = dbcheck.DbError(err); err != nil {
			return nil, err
		}

		return jsonm.MarshalCkZero(data)
	})
	if err != nil {
		return nil, err
	}

	var conf *model.FissionCampaignReferral
	err = jsonm.Unmarshal(data, &conf)

	return conf, err
}

func (c CampaignReferral) campaignIdKey(campaignId int64) string {
	return fmt.Sprintf("%s:%d", c.Prefix(), campaignId)
}

// Clear 清理缓存
func (c CampaignReferral) Clear(vals ...*model.FissionCampaignReferral) error {
	var key []string
	for _, v := range vals {
		key = append(key, c.campaignIdKey(v.CampaignID))
	}

	if len(key) > 0 {
		return app.Redis.Cache.Del(c.ctx, key...).Err()
	}

	return nil
}