Create & Init Project...

app/job/live/xlottery/internal/dao/BUILD

@@ -0,0 +1,45 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "api.go",
+        "cache.go",
+        "capsule.go",
+        "dao.go",
+    ],
+    importpath = "go-common/app/job/live/xlottery/internal/dao",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/job/live/xlottery/internal/conf:go_default_library",
+        "//app/job/live/xlottery/internal/model:go_default_library",
+        "//app/service/live/rc/api/liverpc:go_default_library",
+        "//app/service/live/rc/api/liverpc/v1:go_default_library",
+        "//library/cache/redis:go_default_library",
+        "//library/database/sql:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/rpc/liverpc:go_default_library",
+        "//library/queue/databus/report:go_default_library",
+        "//vendor/github.com/pkg/errors: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/xlottery/internal/dao/api.go

@@ -0,0 +1,23 @@
+package dao
+
+import (
+	"go-common/app/job/live/xlottery/internal/conf"
+	rcApi "go-common/app/service/live/rc/api/liverpc"
+	"go-common/library/net/rpc/liverpc"
+)
+
+// RcApi liverpc reward-service api
+var RcApi *rcApi.Client
+
+// InitAPI init all service APIs
+func InitAPI() {
+	RcApi = rcApi.New(getConf("rc"))
+}
+
+func getConf(appName string) *liverpc.ClientConfig {
+	c := conf.Conf.LiveRpc
+	if c != nil {
+		return c[appName]
+	}
+	return nil
+}

app/job/live/xlottery/internal/dao/cache.go

@@ -0,0 +1,70 @@
+package dao
+
+import (
+	"context"
+	"math/rand"
+	"time"
+
+	"go-common/library/cache/redis"
+	"go-common/library/log"
+)
+
+func randomString(l int) string {
+	str := "0123456789abcdefghijklmnopqrstuvwxyz"
+	bytes := []byte(str)
+	result := []byte{}
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	for i := 0; i < l; i++ {
+		result = append(result, bytes[r.Intn(len(bytes))])
+	}
+	return string(result)
+}
+
+// Lock .
+func (d *Dao) Lock(ctx context.Context, realKey string, ttl int, retry int, retryDelay int) (gotLock bool, lockValue string, err error) {
+
+	if retry <= 0 {
+		retry = 1
+	}
+	lockValue = "locked:" + randomString(5)
+	retryTimes := 0
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+
+	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:%s", realKey, err.Error())
+			break
+		}
+
+		if res != nil {
+			gotLock = true
+			break
+		}
+		time.Sleep(time.Duration(retryDelay * 1000))
+	}
+	return
+}
+
+// UnLock .
+func (d *Dao) UnLock(ctx context.Context, realKey string, lockValue string) (err error) {
+	conn := d.redis.Get(ctx)
+	defer conn.Close()
+	res, err := redis.String(conn.Do("GET", realKey))
+	if err != nil {
+		log.Error("redis_unlock get error:%s:%v", realKey, err)
+		return
+	}
+	if res != lockValue {
+		err = ErrUnLockGet
+		return
+	}
+
+	_, err = conn.Do("DEL", realKey)
+	if err != nil {
+		log.Error("redis_unlock del error:%s:%v", realKey, err)
+	}
+	return
+}

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

@@ -0,0 +1,49 @@
+package dao
+
+import (
+	"context"
+
+	"go-common/library/log"
+
+	"go-common/app/job/live/xlottery/internal/conf"
+	"go-common/library/cache/redis"
+	xsql "go-common/library/database/sql"
+)
+
+// Dao dao
+type Dao struct {
+	c     *conf.Config
+	redis *redis.Pool
+	db    *xsql.DB
+}
+
+// New init mysql db
+func New(c *conf.Config) (dao *Dao) {
+	dao = &Dao{
+		c:     c,
+		redis: redis.NewPool(c.Redis.Lottery),
+		db:    xsql.NewMySQL(c.Database.Lottery),
+	}
+	return
+}
+
+// Close close the resource.
+func (d *Dao) 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)
+}
+
+func (d *Dao) execSqlWithBindParams(c context.Context, sql *string, bindParams ...interface{}) (affect int64, err error) {
+	res, err := d.db.Exec(c, *sql, bindParams...)
+	if err != nil {
+		log.Error("db.Exec(%s) error(%v)", *sql, err)
+		return
+	}
+	return res.RowsAffected()
+}