openbilibili/app/job/main/dm2/dao/mc_special.go

package dao

import (
	"context"
	"fmt"

	"go-common/app/job/main/dm2/model"
	"go-common/library/cache/memcache"
	"go-common/library/log"
)

const (
	_fmtSpecialDm = "s_special_%d_%d"
)

func (d *Dao) specialDmKey(oid int64, tp int32) string {
	return fmt.Sprintf(_fmtSpecialDm, oid, tp)
}

// DelSpecialDmCache .
func (d *Dao) DelSpecialDmCache(c context.Context, oid int64, tp int32) (err error) {
	var (
		key  = d.specialDmKey(oid, tp)
		conn = d.dmSegMC.Get(c)
	)
	defer conn.Close()
	if err = conn.Delete(key); err != nil {
		if err == memcache.ErrNotFound {
			err = nil
		} else {
			log.Error("memcache.Delete(%s) error(%v)", key, err)
		}
	}
	return
}

// AddSpecialDmCache add special content to memcache.
func (d *Dao) AddSpecialDmCache(c context.Context, ds *model.DmSpecial) (err error) {
	conn := d.dmSegMC.Get(c)
	key := d.specialDmKey(ds.Oid, ds.Type)
	defer conn.Close()
	item := &memcache.Item{
		Key:        key,
		Object:     ds,
		Flags:      memcache.FlagJSON,
		Expiration: d.dmSegMCExpire,
	}
	if err = conn.Set(item); err != nil {
		log.Error("conn.Set(%s) error(%v)", key, err)
	}
	return
}