Create & Init Project...

app/job/live/gift/internal/service/BUILD

@@ -0,0 +1,60 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+    "go_test",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "infoc.go",
+        "package.go",
+        "service.go",
+    ],
+    importpath = "go-common/app/job/live/gift/internal/service",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/job/live/gift/api:go_default_library",
+        "//app/job/live/gift/internal/conf:go_default_library",
+        "//app/job/live/gift/internal/dao:go_default_library",
+        "//app/job/live/gift/internal/model:go_default_library",
+        "//library/log:go_default_library",
+        "//library/log/infoc:go_default_library",
+        "//library/queue/databus:go_default_library",
+        "//library/sync/errgroup:go_default_library",
+        "@io_bazel_rules_go//proto/wkt:empty_go_proto",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = [
+        "infoc_test.go",
+        "package_test.go",
+        "service_test.go",
+    ],
+    embed = [":go_default_library"],
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/live/gift/internal/conf:go_default_library",
+        "//app/job/live/gift/internal/model:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)

app/job/live/gift/internal/service/infoc.go

@@ -0,0 +1,121 @@
+package service
+
+import (
+	"go-common/library/log"
+	"go-common/library/log/infoc"
+	"strconv"
+	"time"
+)
+
+var inCh = make(chan interface{}, 10240)
+
+const maxInt = int(^uint(0) >> 1)
+
+type bagLogInfoc struct {
+	id       string
+	uid      string
+	bagID    string
+	giftID   string
+	num      string
+	afterNum string
+	source   string
+	infoType string
+	ctime    string
+}
+
+type giftActionInfoc struct {
+	uid       int64
+	roomid    int64
+	item      int64
+	value     int64
+	change    int64
+	describe  string
+	extra     string
+	ts        int64
+	platform  string
+	clientver string
+	buvid     string
+	ua        string
+	referer   string
+}
+
+// bagLogInfoc 包裹日志打点
+func (s *Service) bagLogInfoc(uid, bagID, giftID, num, afterNum int64, source string) {
+	s.infoc(bagLogInfoc{
+		id:       MakeID(uid),
+		uid:      strconv.FormatInt(uid, 10),
+		bagID:    strconv.FormatInt(bagID, 10),
+		giftID:   strconv.FormatInt(giftID, 10),
+		num:      strconv.FormatInt(num, 10),
+		afterNum: strconv.FormatInt(afterNum, 10),
+		source:   source,
+		infoType: "1",
+		ctime:    time.Now().Format("2006-01-02 15:04:05"),
+	})
+}
+
+//giftActionInfoc 道具打点
+func (s *Service) giftActionInfoc(uid, roomid, item, value, change int64, describe, platform string) {
+	s.infoc(giftActionInfoc{
+		uid:       uid,
+		roomid:    roomid,
+		item:      item,
+		value:     value,
+		change:    change,
+		describe:  describe,
+		extra:     "",
+		ts:        time.Now().Unix(),
+		platform:  platform,
+		clientver: "",
+		buvid:     "",
+		ua:        "",
+		referer:   "",
+	})
+}
+
+// MakeID MakeID
+func MakeID(uid int64) string {
+	prefix := strconv.FormatInt(uid%10, 10)
+	postfix := strconv.Itoa(maxInt - int(time.Now().Unix()*10000))
+	uidStr := strconv.FormatInt(uid, 10)
+	l := len(uidStr)
+	var middle string
+	if l >= 10 {
+		middle = uidStr
+	} else {
+		var s string
+		for i := 0; i < (10 - l); i++ {
+			s += "0"
+		}
+		middle = s + uidStr
+	}
+	return prefix + middle + postfix
+}
+
+//infoc
+func (s *Service) infoc(i interface{}) {
+	select {
+	case inCh <- i:
+	default:
+		log.Warn("infocproc chan full")
+	}
+}
+
+// infocproc
+func (s *Service) infocproc() {
+	var bl = infoc.New(s.c.Infoc["bagLog"])
+	var ga = infoc.New(s.c.Infoc["giftAction"])
+	for {
+		i := <-inCh
+		switch v := i.(type) {
+		case bagLogInfoc:
+			err := bl.Info(v.id, v.uid, v.bagID, v.giftID, v.num, v.afterNum, v.source, v.infoType, v.ctime)
+			log.Info("bagLogInfoc info %v,ret:%v", v, err)
+		case giftActionInfoc:
+			err := ga.Info(v.uid, v.roomid, v.item, v.value, v.change, v.describe, v.extra, v.ts, v.platform, v.clientver, v.buvid, v.ua, v.referer)
+			log.Info("giftActionInfoc info %v,ret:%v", v, err)
+		default:
+			log.Warn("infocproc can't process the type")
+		}
+	}
+}

app/job/live/gift/internal/service/infoc_test.go

@@ -0,0 +1,58 @@
+package service
+
+import (
+	"testing"
+
+	"github.com/smartystreets/goconvey/convey"
+)
+
+func TestServicebagLogInfoc(t *testing.T) {
+	convey.Convey("bagLogInfoc", t, func(ctx convey.C) {
+		var (
+			uid      = int64(0)
+			bagID    = int64(0)
+			giftID   = int64(0)
+			num      = int64(0)
+			afterNum = int64(0)
+			source   = ""
+		)
+		ctx.Convey("When everything gose positive", func(ctx convey.C) {
+			s.bagLogInfoc(uid, bagID, giftID, num, afterNum, source)
+			ctx.Convey("No return values", func(ctx convey.C) {
+			})
+		})
+	})
+}
+
+func TestServicegiftActionInfoc(t *testing.T) {
+	convey.Convey("giftActionInfoc", t, func(ctx convey.C) {
+		var (
+			uid      = int64(0)
+			roomid   = int64(0)
+			item     = int64(0)
+			value    = int64(0)
+			change   = int64(0)
+			describe = ""
+			platform = ""
+		)
+		ctx.Convey("When everything gose positive", func(ctx convey.C) {
+			s.giftActionInfoc(uid, roomid, item, value, change, describe, platform)
+			ctx.Convey("No return values", func(ctx convey.C) {
+			})
+		})
+	})
+}
+
+func TestServiceMakeID(t *testing.T) {
+	convey.Convey("MakeID", t, func(ctx convey.C) {
+		var (
+			uid = int64(0)
+		)
+		ctx.Convey("When everything gose positive", func(ctx convey.C) {
+			p1 := MakeID(uid)
+			ctx.Convey("Then p1 should not be nil.", func(ctx convey.C) {
+				ctx.So(p1, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}

app/job/live/gift/internal/service/package.go

@@ -0,0 +1,99 @@
+package service
+
+import (
+	"context"
+	"errors"
+	"go-common/app/job/live/gift/internal/model"
+	"go-common/library/log"
+	"go-common/library/sync/errgroup"
+)
+
+// AddGift AddGift
+func (s *Service) AddGift(ctx context.Context, m *model.AddFreeGift) (bagId int64, err error) {
+	uid := m.UID
+	giftID := m.GiftID
+	giftNum := m.GiftNum
+	expireAt := m.ExpireAt
+	source := m.Source
+	if uid == 0 || giftID == 0 || giftNum == 0 {
+		log.Error("add gift params error,uid:%d,giftID:%d,giftNum:%d", uid, giftID, giftNum)
+		err = errors.New("params error")
+		return
+	}
+	bagID, err := s.GetBagID(ctx, uid, giftID, expireAt)
+	if err != nil {
+		return
+	}
+	var (
+		affectNum int64
+		isUpdate  = false
+		eg, _     = errgroup.WithContext(ctx)
+	)
+	if bagID != 0 {
+		isUpdate = true
+		affectNum, _ = s.dao.UpdateBagNum(ctx, uid, bagID, giftNum)
+	} else {
+		affectNum, _ = s.dao.AddBag(ctx, uid, giftID, giftNum, expireAt)
+		bagID = affectNum
+		eg.Go(
+			func() error {
+				s.dao.SetBagIDCache(ctx, uid, giftID, expireAt, bagID, 14400)
+				return nil
+			})
+
+	}
+	newNum := giftNum
+	if affectNum > 0 {
+		eg.Go(
+			func() error {
+				s.dao.ClearBagListCache(ctx, uid)
+				return nil
+			})
+		if isUpdate {
+			res, _ := s.dao.GetBagByID(ctx, uid, bagID)
+			newNum = res.GiftNum
+			//上报lancer TODO
+			s.bagLogInfoc(uid, bagID, giftID, giftNum, newNum, source)
+		}
+	}
+	// 更新免费礼物数量缓存
+	eg.Go(
+		func() error {
+			s.UpdateFreeGiftCache(ctx, uid, giftID, expireAt, newNum)
+			return nil
+		})
+	eg.Wait()
+	return
+}
+
+// GetBagID GetBagID
+func (s *Service) GetBagID(ctx context.Context, uid, giftID, expireAt int64) (id int64, err error) {
+	id, err = s.dao.GetBagIDCache(ctx, uid, giftID, expireAt)
+	if err != nil {
+		return
+	}
+	if id == 0 {
+		//queryDB
+		var r *model.BagInfo
+		r, err = s.dao.GetBag(ctx, uid, giftID, expireAt)
+		if err != nil {
+			return
+		}
+		id = r.ID
+	}
+
+	// 缓存或数据库本身有，再更新缓存
+	if id != 0 {
+		s.dao.SetBagIDCache(ctx, uid, giftID, expireAt, id, 14400)
+	}
+	return
+}
+
+// UpdateFreeGiftCache UpdateFreeGiftCache
+func (s *Service) UpdateFreeGiftCache(ctx context.Context, uid, giftID, expireAt, num int64) {
+	//giftInfo := s.GetGiftInfoByID(ctx, giftID)
+	//if giftInfo.Id == 0 || giftInfo.Type != 3 {
+	//	return
+	//}
+	//s.dao.SetBagNumCache(ctx, uid, giftID, expireAt, num, 14400)
+}

app/job/live/gift/internal/service/package_test.go

@@ -0,0 +1,66 @@
+package service
+
+import (
+	"context"
+	"go-common/app/job/live/gift/internal/model"
+	"testing"
+
+	"github.com/smartystreets/goconvey/convey"
+)
+
+func TestServiceAddGift(t *testing.T) {
+	convey.Convey("AddGift", t, func(c convey.C) {
+		var (
+			ctx = context.Background()
+			m   = &model.AddFreeGift{
+				UID:      1,
+				GiftID:   1,
+				GiftNum:  1,
+				ExpireAt: 0,
+				Source:   "test",
+			}
+		)
+		c.Convey("When everything gose positive", func(c convey.C) {
+			bagId, err := s.AddGift(ctx, m)
+			c.Convey("Then err should be nil.bagId should not be nil.", func(c convey.C) {
+				c.So(err, convey.ShouldBeNil)
+				c.So(bagId, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}
+
+func TestServiceGetBagID(t *testing.T) {
+	convey.Convey("GetBagID", t, func(c convey.C) {
+		var (
+			ctx      = context.Background()
+			uid      = int64(1)
+			giftID   = int64(1)
+			expireAt = int64(0)
+		)
+		c.Convey("When everything gose positive", func(c convey.C) {
+			id, err := s.GetBagID(ctx, uid, giftID, expireAt)
+			c.Convey("Then err should be nil.id should not be nil.", func(c convey.C) {
+				c.So(err, convey.ShouldBeNil)
+				c.So(id, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}
+
+func TestServiceUpdateFreeGiftCache(t *testing.T) {
+	convey.Convey("UpdateFreeGiftCache", t, func(c convey.C) {
+		var (
+			ctx      = context.Background()
+			uid      = int64(0)
+			giftID   = int64(0)
+			expireAt = int64(0)
+			num      = int64(0)
+		)
+		c.Convey("When everything gose positive", func(c convey.C) {
+			s.UpdateFreeGiftCache(ctx, uid, giftID, expireAt, num)
+			c.Convey("No return values", func(ctx convey.C) {
+			})
+		})
+	})
+}

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

@@ -0,0 +1,95 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"sync"
+
+	"go-common/app/job/live/gift/internal/model"
+	"go-common/library/log"
+	"go-common/library/queue/databus"
+
+	pb "go-common/app/job/live/gift/api"
+	"go-common/app/job/live/gift/internal/conf"
+	"go-common/app/job/live/gift/internal/dao"
+
+	"github.com/golang/protobuf/ptypes/empty"
+)
+
+// Service struct
+type Service struct {
+	c           *conf.Config
+	dao         *dao.Dao
+	addFreeGift *databus.Databus
+	waiter      sync.WaitGroup
+}
+
+// New init
+func New(c *conf.Config) (s *Service) {
+	s = &Service{
+		c:           c,
+		dao:         dao.New(c),
+		addFreeGift: databus.New(c.Databus.AddGift),
+	}
+	go s.infocproc()
+	for i := 0; i < c.Consumer.AddGift.Num; i++ {
+		s.waiter.Add(1)
+		go s.addGiftConsumeProc()
+	}
+	return s
+}
+
+func (s *Service) addGiftConsumeProc() {
+	defer s.waiter.Done()
+	var err error
+	for {
+		msg, ok := <-s.addFreeGift.Messages()
+		if !ok {
+			log.Error("s.addFreeGift.Messages channel closed")
+			return
+		}
+		m := &model.AddFreeGift{}
+		if err = json.Unmarshal(msg.Value, &m); err != nil {
+			log.Error("json.Unmarshal(%v) error(%v)", msg, err)
+			continue
+		}
+		ctx := context.Background()
+		// 消息幂等
+		if m.MsgID != "" {
+			key := m.MsgID + m.Source
+			gotLock, _, errLock := s.dao.Lock(ctx, key, 3600000, 0, 0)
+			if errLock != nil {
+				continue
+			}
+			if !gotLock {
+				log.Error("msg has been processed,%v", m)
+				continue
+			}
+		}
+		s.AddGift(ctx, m)
+		// 打点上报
+		s.giftActionInfoc(m.UID, 0, m.GiftID, 0, m.GiftNum, m.Source, "")
+		log.Info("consume addFreeGift topic:%s, Key:%s, Value:%s ", msg.Topic, msg.Key, msg.Value)
+		if err = msg.Commit(); err != nil {
+			log.Error("commit msg(%v) error(%v)", msg, err)
+		}
+	}
+}
+
+// SayHello grpc demo func
+func (s *Service) SayHello(ctx context.Context, req *pb.HelloReq) (reply *empty.Empty, err error) {
+	reply = new(empty.Empty)
+	fmt.Printf("hello %s", req.Name)
+	return
+}
+
+// Ping Service
+func (s *Service) Ping(ctx context.Context) (err error) {
+	return s.dao.Ping(ctx)
+}
+
+// Close Service
+func (s *Service) Close() {
+	s.dao.Close()
+}

app/job/live/gift/internal/service/service_test.go

@@ -0,0 +1,34 @@
+package service
+
+import (
+	"flag"
+	"go-common/app/job/live/gift/internal/conf"
+	"os"
+	"testing"
+)
+
+var (
+	s *Service
+)
+
+func TestMain(m *testing.M) {
+	if os.Getenv("DEPLOY_ENV") != "" {
+		flag.Set("app_id", "")
+		flag.Set("conf_token", "")
+		flag.Set("tree_id", "")
+		flag.Set("conf_version", "docker-1")
+		flag.Set("deploy_env", "uat")
+		flag.Set("conf_host", "config.bilibili.co")
+		flag.Set("conf_path", "/tmp")
+		flag.Set("region", "sh")
+		flag.Set("zone", "sh001")
+	} else {
+		flag.Set("conf", "../../cmd/test.toml")
+	}
+	flag.Parse()
+	if err := conf.Init(); err != nil {
+		panic(err)
+	}
+	s = New(conf.Conf)
+	os.Exit(m.Run())
+}