Create & Init Project...

app/job/live/gift/internal/dao/BUILD

@@ -0,0 +1,41 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "dao.go",
+        "mysql.go",
+        "redis.go",
+    ],
+    importpath = "go-common/app/job/live/gift/internal/dao",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/job/live/gift/internal/conf:go_default_library",
+        "//app/job/live/gift/internal/model:go_default_library",
+        "//library/cache/memcache:go_default_library",
+        "//library/cache/redis:go_default_library",
+        "//library/database/sql:go_default_library",
+        "//library/log:go_default_library",
+        "//vendor/github.com/satori/go.uuid:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/job/live/gift/internal/dao/dao.go

@@ -0,0 +1,42 @@
+package dao
+
+import (
+	"context"
+
+	"go-common/app/job/live/gift/internal/conf"
+	"go-common/library/cache/memcache"
+	"go-common/library/cache/redis"
+	xsql "go-common/library/database/sql"
+)
+
+// Dao dao
+type Dao struct {
+	c     *conf.Config
+	mc    *memcache.Pool
+	redis *redis.Pool
+	db    *xsql.DB
+}
+
+// New init mysql db
+func New(c *conf.Config) (dao *Dao) {
+	dao = &Dao{
+		c:     c,
+		mc:    memcache.NewPool(c.Memcache),
+		redis: redis.NewPool(c.Redis),
+		db:    xsql.NewMySQL(c.MySQL),
+	}
+	return
+}
+
+// Close close the resource.
+func (d *Dao) Close() {
+	d.mc.Close()
+	d.redis.Close()
+	d.db.Close()
+}
+
+// Ping dao ping
+func (d *Dao) Ping(ctx context.Context) error {
+	// TODO: add mc,redis... if you use
+	return d.db.Ping(ctx)
+}

app/job/live/gift/internal/dao/mysql.go

@@ -0,0 +1,79 @@
+package dao
+
+import (
+	"context"
+	"crypto/md5"
+	"database/sql"
+	"encoding/hex"
+	"fmt"
+	"go-common/app/job/live/gift/internal/model"
+	"go-common/library/log"
+	"strconv"
+)
+
+var (
+	_getBag       = "SELECT id,gift_num FROM user_gift_%s WHERE uid = ? AND gift_id = ? AND expireat = ? LIMIT 1"
+	_getBagByID   = "SELECT id,gift_num FROM user_gift_%s WHERE id = ?"
+	_updateBagNum = "UPDATE user_gift_%s SET gift_num = gift_num + ? WHERE id = ?"
+	_insertBag    = "INSERT INTO user_gift_%s (uid,gift_id,gift_num,expireat) VALUES (?,?,?,?)"
+)
+
+// GetBag GetBag
+func (d *Dao) GetBag(ctx context.Context, uid, giftID, expireAt int64) (res *model.BagInfo, err error) {
+	log.Info("GetBag,uid:%d,giftID:%d,expireAt:%d", uid, giftID, expireAt)
+	row := d.db.QueryRow(ctx, fmt.Sprintf(_getBag, getPostFix(uid)), uid, giftID, expireAt)
+	res = &model.BagInfo{}
+	if err = row.Scan(&res.ID, &res.GiftNum); err != nil {
+		if err == sql.ErrNoRows {
+			err = nil
+			return
+		}
+		log.Error("GetBag row.Scan error(%v)", err)
+	}
+	return
+}
+
+// UpdateBagNum UpdateBagNum
+func (d *Dao) UpdateBagNum(ctx context.Context, uid, id, num int64) (affected int64, err error) {
+	log.Info("UpdateBagNum,uid:%d,id:%d,num:%d", uid, id, num)
+	res, err := d.db.Exec(ctx, fmt.Sprintf(_updateBagNum, getPostFix(uid)), num, id)
+	if err != nil {
+		log.Error("UpdateBagNum error(%v)", err)
+		return
+	}
+	return res.RowsAffected()
+}
+
+// AddBag AddBag
+func (d *Dao) AddBag(ctx context.Context, uid, giftID, giftNum, expireAt int64) (affected int64, err error) {
+	log.Info("AddBag,uid:%d,giftID:%d,giftNum:%d,expireAt:%d", uid, giftID, giftNum, expireAt)
+	res, err := d.db.Exec(ctx, fmt.Sprintf(_insertBag, getPostFix(uid)), uid, giftID, giftNum, expireAt)
+	if err != nil {
+		log.Error("AddBag error(%v)", err)
+		return
+	}
+	return res.LastInsertId()
+}
+
+// GetBagByID GetBagByID
+func (d *Dao) GetBagByID(ctx context.Context, uid, id int64) (res *model.BagInfo, err error) {
+	log.Info("GetBagByID,uid:%d,id:%d", uid, id)
+	row := d.db.QueryRow(ctx, fmt.Sprintf(_getBagByID, getPostFix(uid)), id)
+	res = &model.BagInfo{}
+	if err = row.Scan(&res.ID, &res.GiftNum); err != nil {
+		if err == sql.ErrNoRows {
+			err = nil
+			return
+		}
+		log.Error("GetBagByID row.Scan error(%v)", err)
+	}
+	return
+}
+
+func getPostFix(uid int64) string {
+	uidStr := strconv.Itoa(int(uid))
+	h := md5.New()
+	h.Write([]byte(uidStr))
+	md5Str := hex.EncodeToString(h.Sum(nil))
+	return md5Str[0:1]
+}

app/job/live/gift/internal/dao/redis.go

@@ -0,0 +1,110 @@
+package dao
+
+import (
+	"context"
+	"fmt"
+	"go-common/library/cache/redis"
+	"go-common/library/log"
+	"time"
+
+	"github.com/satori/go.uuid"
+)
+
+func bagIDCache(uid, giftID, expireAt int64) string {
+	return fmt.Sprintf("bag_id:%d:%d:%d", uid, giftID, expireAt)
+}
+
+// SetBagIDCache SetBagIDCache
+func (d *Dao) SetBagIDCache(ctx context.Context, uid, giftID, expireAt, bagID, expire int64) (err error) {
+	key := bagIDCache(uid, giftID, expireAt)
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+	_, err = conn.Do("SETEX", key, expire, bagID)
+	if err != nil {
+		log.Error("conn.Do(SETEX, %s) error(%v)", key, err)
+	}
+	return
+}
+
+func bagListKey(uid int64) string {
+	return fmt.Sprintf("bag_list:%d", uid)
+}
+
+// ClearBagListCache ClearBagListCache
+func (d *Dao) ClearBagListCache(ctx context.Context, uid int64) (err error) {
+	key := bagListKey(uid)
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+	_, err = conn.Do("DEL", key)
+	if err != nil {
+		log.Error("conn.Do(DEL, %s) error(%v)", key, err)
+	}
+	return
+}
+
+// GetBagIDCache GetBagIDCache
+func (d *Dao) GetBagIDCache(ctx context.Context, uid, giftID, expireAt int64) (bagID int64, err error) {
+	key := bagIDCache(uid, giftID, expireAt)
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+	bagID, err = redis.Int64(conn.Do("GET", key))
+	if err != nil {
+		if err == redis.ErrNil {
+			err = nil
+		} else {
+			log.Error("conn.Do(GET, %s) error(%v)", key, err)
+		}
+		return
+	}
+	return
+}
+
+func bagNumKey(uid, giftID, expireAt int64) string {
+	return fmt.Sprintf("bag_num:%d:%d:%d", uid, giftID, expireAt)
+}
+
+// SetBagNumCache SetBagNumCache
+func (d *Dao) SetBagNumCache(ctx context.Context, uid, giftID, expireAt, giftNum, expire int64) (err error) {
+	key := bagNumKey(uid, giftID, expireAt)
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+	_, err = conn.Do("SETEX", key, expire, giftNum)
+	if err != nil {
+		log.Error("conn.Do(SETEX, %s) error(%v)", key, err)
+	}
+	return
+}
+
+//Lock Lock
+func (d *Dao) Lock(ctx context.Context, key string, ttl int, retry int, retryDelay int) (gotLock bool, lockValue string, err error) {
+
+	if retry <= 0 {
+		retry = 1
+	}
+	lockValue = uuid.NewV4().String()
+	retryTimes := 0
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+
+	realKey := lockKey(key)
+
+	for ; retryTimes < retry; retryTimes++ {
+		var res interface{}
+		res, err = conn.Do("SET", realKey, lockValue, "PX", ttl, "NX")
+		if err != nil {
+			log.Error("redis_lock failed:%s:%v", realKey, err)
+			break
+		}
+
+		if res != nil {
+			gotLock = true
+			break
+		}
+		time.Sleep(time.Duration(retryDelay) * time.Millisecond)
+	}
+	return
+}
+
+func lockKey(key string) string {
+	return fmt.Sprintf("gift_job_lock:%s", key)
+}