Create & Init Project...

app/interface/main/dm2/dao/mc_limiter.go

@@ -0,0 +1,61 @@
+package dao
+
+import (
+	"context"
+	"strconv"
+
+	"go-common/app/interface/main/dm2/model"
+	"go-common/library/cache/memcache"
+	"go-common/library/log"
+)
+
+const (
+	_prefixDmDailyLimit = "dm_daily_limit_"
+)
+
+func keyDmDailyLimit(mid int64) string {
+	return _prefixDmDailyLimit + strconv.FormatInt(mid, 10)
+}
+
+// SetDmDailyLimitCache .
+func (d *Dao) SetDmDailyLimitCache(c context.Context, mid int64, limiter *model.DailyLimiter) (err error) {
+	var (
+		conn = d.dmSegMC.Get(c)
+		key  = keyDmDailyLimit(mid)
+	)
+	defer conn.Close()
+	item := &memcache.Item{
+		Key:        key,
+		Object:     limiter,
+		Flags:      memcache.FlagJSON,
+		Expiration: d.dmLimiterMCExpire,
+	}
+	if err = conn.Set(item); err != nil {
+		log.Error("conn.Set(%v) error(%v)", item, err)
+	}
+	return
+}
+
+// GetDmDailyLimitCache .
+func (d *Dao) GetDmDailyLimitCache(c context.Context, mid int64) (limiter *model.DailyLimiter, err error) {
+	var (
+		conn = d.dmSegMC.Get(c)
+		key  = keyDmDailyLimit(mid)
+		rp   *memcache.Item
+	)
+	defer conn.Close()
+	if rp, err = conn.Get(key); err != nil {
+		if err == memcache.ErrNotFound {
+			limiter = nil
+			err = nil
+		} else {
+			log.Error("mc.Get(%s) error(%v)", key, err)
+		}
+		return
+	}
+	limiter = &model.DailyLimiter{}
+	if err = conn.Scan(rp, &limiter); err != nil {
+		log.Error("mc.Scan() error(%v)", err)
+	}
+	return
+}