Create & Init Project...

app/job/main/relation-cache/dao/relation.go

@@ -0,0 +1,164 @@
+package dao
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+
+	relation "go-common/app/service/main/relation/model"
+	"go-common/library/cache/memcache"
+	"go-common/library/cache/redis"
+	"go-common/library/log"
+	xtime "go-common/library/time"
+)
+
+const (
+	_prefixFollowings = "at_"
+	_prefixTags       = "tags_" // user tag info.
+	_prefixFollowing  = "pb_a_"
+	_prefixStat       = "c_"         // key of stat
+	_prefixTagCount   = "rs_tmtc_%d" // key of relation tag by mid & tag's count
+)
+
+func statKey(mid int64) string {
+	return _prefixStat + strconv.FormatInt(mid, 10)
+}
+
+func tagsKey(mid int64) string {
+	return _prefixTags + strconv.FormatInt(mid, 10)
+}
+
+func followingsKey(mid int64) string {
+	return _prefixFollowings + strconv.FormatInt(mid, 10)
+}
+
+func followingKey(mid int64) string {
+	return _prefixFollowing + strconv.FormatInt(mid, 10)
+}
+
+func tagCountKey(mid int64) string {
+	return fmt.Sprintf(_prefixTagCount, mid)
+}
+
+// DelStatCache is
+func (d *Dao) DelStatCache(ctx context.Context, mid int64) error {
+	conn := d.mc.Get(ctx)
+	defer conn.Close()
+	if err := conn.Delete(statKey(mid)); err != nil {
+		if err == memcache.ErrNotFound {
+			return nil
+		}
+		log.Error("Failed to delete stat cache: mid: %d: %+v", mid, err)
+		return err
+	}
+	return nil
+}
+
+// DelFollowerCache del follower cache
+func (d *Dao) DelFollowerCache(ctx context.Context, fid int64) error {
+	key := followingKey(fid)
+	conn := d.mc.Get(ctx)
+	defer conn.Close()
+	if err := conn.Delete(key); err != nil {
+		if err == memcache.ErrNotFound {
+			err = nil
+		} else {
+			log.Error("conn.Delete(%s) error(%v)", key, err)
+			return err
+		}
+	}
+	return nil
+}
+
+// DelFollowing del following cache.
+func (d *Dao) DelFollowing(c context.Context, mid int64, following *relation.Following) (err error) {
+	var (
+		ok  bool
+		key = followingsKey(mid)
+	)
+	conn := d.redis.Get(c)
+	if ok, err = redis.Bool(conn.Do("EXPIRE", key, d.RelationTTL)); err != nil {
+		log.Error("redis.Bool(conn.Do(EXPIRE, %s)) error(%v)", key, err)
+	} else if ok {
+		if _, err = conn.Do("HDEL", key, following.Mid); err != nil {
+			log.Error("conn.Do(HDEL, %s, %d) error(%v)", key, following.Mid, err)
+		}
+	}
+	conn.Close()
+	return
+}
+
+// DelTagsCache is
+func (d *Dao) DelTagsCache(ctx context.Context, mid int64) (err error) {
+	conn := d.mc.Get(ctx)
+	if err = conn.Delete(tagsKey(mid)); err != nil {
+		if err == memcache.ErrNotFound {
+			err = nil
+		} else {
+			log.Error("conn.Delete(%s) error(%v)", tagCountKey(mid), err)
+		}
+	}
+	conn.Close()
+	return
+}
+
+// AddFollowingCache is
+func (d *Dao) AddFollowingCache(c context.Context, mid int64, following *relation.Following) (err error) {
+	var (
+		ok  bool
+		key = followingsKey(mid)
+	)
+	conn := d.redis.Get(c)
+	if ok, err = redis.Bool(conn.Do("EXPIRE", key, d.RelationTTL)); err != nil {
+		log.Error("redis.Bool(conn.Do(EXPIRE, %s)) error(%v)", key, err)
+	} else if ok {
+		var ef []byte
+		if ef, err = d.encode(following.Attribute, following.MTime, following.Tag, following.Special); err != nil {
+			return
+		}
+		if _, err = conn.Do("HSET", key, following.Mid, ef); err != nil {
+			log.Error("conn.Do(HSET, %s, %d) error(%v)", key, following.Mid, err)
+		}
+	}
+	conn.Close()
+	return
+}
+
+// encode
+func (d *Dao) encode(attribute uint32, mtime xtime.Time, tagids []int64, special int32) (res []byte, err error) {
+	ft := &relation.FollowingTags{Attr: attribute, Ts: mtime, TagIds: tagids, Special: special}
+	return ft.Marshal()
+}
+
+// DelFollowingCache delete following cache.
+func (d *Dao) DelFollowingCache(c context.Context, mid int64) (err error) {
+	return d.delFollowingCache(c, followingKey(mid))
+}
+
+// delFollowingCache delete following cache.
+func (d *Dao) delFollowingCache(c context.Context, key string) (err error) {
+	conn := d.mc.Get(c)
+	if err = conn.Delete(key); err != nil {
+		if err == memcache.ErrNotFound {
+			err = nil
+		} else {
+			log.Error("conn.Delete(%s) error(%v)", key, err)
+		}
+	}
+	conn.Close()
+	return
+}
+
+// DelTagCountCache del tag count cache.
+func (d *Dao) DelTagCountCache(c context.Context, mid int64) (err error) {
+	conn := d.mc.Get(c)
+	if err = conn.Delete(tagCountKey(mid)); err != nil {
+		if err == memcache.ErrNotFound {
+			err = nil
+		} else {
+			log.Error("conn.Delete(%s) error(%v)", tagCountKey(mid), err)
+		}
+	}
+	conn.Close()
+	return
+}