Create & Init Project...

app/interface/main/growup/service/newbie/BUILD

@@ -0,0 +1,56 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+    "go_test",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "letter.go",
+        "service.go",
+    ],
+    importpath = "go-common/app/interface/main/growup/service/newbie",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/growup/conf:go_default_library",
+        "//app/interface/main/growup/dao/newbiedao:go_default_library",
+        "//app/interface/main/growup/model:go_default_library",
+        "//app/service/main/account/api:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/log:go_default_library",
+        "//library/sync/errgroup.v2: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"],
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = [
+        "letter_test.go",
+        "service_test.go",
+    ],
+    embed = [":go_default_library"],
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/growup/conf:go_default_library",
+        "//app/interface/main/growup/model:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)

app/interface/main/growup/service/newbie/letter.go

@@ -0,0 +1,174 @@
+package newbie
+
+import (
+	"context"
+	"go-common/app/interface/main/growup/conf"
+	"go-common/app/interface/main/growup/dao/newbiedao"
+	"go-common/app/interface/main/growup/model"
+	accApi "go-common/app/service/main/account/api"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	"go-common/library/sync/errgroup.v2"
+
+	"strconv"
+	"time"
+)
+
+// Letter newbie letter
+func (s *Service) Letter(c context.Context, req *model.NewbieLetterReq) (*model.NewbieLetterRes, error) {
+	var (
+		group        *errgroup.Group
+		recUps       = make(map[int64]*model.RecommendUp)
+		NewbieConf   = conf.Conf.Newbie
+		category     *model.Category
+		recUpMidList []int64
+		activities   []*model.Activity
+		archive      *model.VideoUpArchive
+		accInfo      *accApi.InfoReply
+
+		i   = 0
+		ok  bool
+		err error
+	)
+	res := new(model.NewbieLetterRes)
+
+	log.Info("req: %+v", req)
+	group = errgroup.WithCancel(c)
+	// get up info
+	group.Go(func(ctx context.Context) error {
+		accInfo, err = s.dao.GetInfo(ctx, req.Mid)
+		return err
+	})
+
+	// get activities
+	group.Go(func(ctx context.Context) error {
+		activities, err = s.dao.GetActivities(ctx)
+		return err
+	})
+
+	// get video up, and set talent
+	group.Go(func(ctx context.Context) error {
+		archive, err = s.dao.GetVideoUp(ctx, req.Aid)
+		return err
+	})
+
+	err = group.Wait()
+	if err != nil {
+		return nil, err
+	}
+
+	// data validation, deal with default data
+	if req.Mid != archive.Mid {
+		log.Error("The archive is not yours, mid: %d, archive.Mid: %v", req.Mid, archive.Mid)
+		return nil, ecode.GrowupArchiveNotYours
+	}
+	if category, ok = newbiedao.Categories[archive.Tid]; !ok {
+		log.Error("not found the sub tid, sub tid: %d, Categories: %v", archive.Tid, newbiedao.Categories)
+		return nil, ecode.GrowupSubTidNotExist
+	}
+	if _, ok = newbiedao.Categories[category.Pid]; !ok {
+		log.Error("not found the tid, tid: %d, Categories: %v", archive.Tid, newbiedao.Categories)
+		return nil, ecode.GrowupTidNotExist
+	}
+	res.Area = newbiedao.Categories[category.Pid].Name
+	log.Info("sub tid: %d, tid: %d", archive.Tid, category.Pid)
+
+	sTid := strconv.FormatInt(archive.Tid, 10)
+	if res.Talent, ok = NewbieConf.Talents[sTid]; !ok {
+		res.Talent = NewbieConf.DefaultTalent
+	}
+	for _, activity := range activities {
+		if i >= NewbieConf.ActivityCount {
+			break
+		}
+		if activity.Type != NewbieConf.ActivityShotType {
+			continue
+		}
+		if activity.Cover == "" {
+			activity.Cover = NewbieConf.DefaultCover
+		}
+		res.Activities = append(res.Activities, activity)
+		i++
+	}
+	if len(res.Activities) < NewbieConf.ActivityCount {
+		log.Error("activity count is not enough %d", NewbieConf.ActivityCount)
+		return nil, ecode.GrowupActivityCountNotEnough
+	}
+
+	res.UperInfo = new(model.NewbieLetterUpInfo)
+	res.UperInfo.Mid = accInfo.Info.Mid
+	res.UperInfo.Name = accInfo.Info.Name
+
+	res.Archive = new(model.NewbieLetterArchive)
+	res.Archive.Title = archive.Title
+	res.Archive.PTime = time.Unix(archive.PTime, 0).Format(model.TimeLayout)
+	log.Info("after data validation: data(%+v)", res)
+
+	// get recommend up list
+	if _, ok := newbiedao.RecommendUpList[category.Pid]; !ok {
+		for _, lists := range newbiedao.RecommendUpList {
+			for recUpMid, recUp := range lists {
+				recUps[recUpMid] = recUp
+				break
+			}
+		}
+		log.Info("Not found recommend up list, system random get them : %+v", recUps)
+	} else {
+		recUps = newbiedao.RecommendUpList[category.Pid]
+		log.Info("found recommend up list : %+v", recUps)
+	}
+
+	// get relations
+	i = 0
+	for recUpMid := range recUps {
+		if i >= NewbieConf.RecommendUpPoolCount {
+			break
+		}
+		if recUpMid == req.Mid {
+			continue
+		}
+		recUpMidList = append(recUpMidList, recUpMid)
+		i++
+	}
+	log.Info("recUpMidList: %+v", recUpMidList)
+
+	relations, err := s.dao.GetRelations(c, req.Mid, recUpMidList)
+	if err != nil {
+		return nil, err
+	}
+	log.Info("relations: %+v", relations)
+
+	// get ups info
+	infosReply, err := s.dao.GetInfos(c, recUpMidList)
+	if err != nil {
+		err = ecode.GrowupRecommendUpNotExist
+		return nil, err
+	}
+	log.Info("recUpInfos: %+v", infosReply.Infos)
+
+	// select 3 ups
+	i = 0
+	for recUpMid := range recUps {
+		if i >= NewbieConf.RecommendUpCount {
+			break
+		}
+		if _, ok := infosReply.Infos[recUpMid]; !ok {
+			continue
+		}
+
+		if _, ok := relations[recUpMid]; !ok {
+			relations[recUpMid] = &model.Relation{
+				Mid:       recUpMid,
+				Attribute: -1,
+			}
+		}
+		relations[recUpMid].Face = infosReply.Infos[recUpMid].Face
+		relations[recUpMid].Name = infosReply.Infos[recUpMid].Name
+		res.Relations = append(res.Relations, relations[recUpMid])
+
+		i++
+	}
+
+	log.Info("res.Relations: %+v", res.Relations)
+	return res, nil
+}

app/interface/main/growup/service/newbie/letter_test.go

@@ -0,0 +1,25 @@
+package newbie
+
+import (
+	"context"
+	"go-common/app/interface/main/growup/model"
+	"testing"
+
+	"github.com/smartystreets/goconvey/convey"
+)
+
+func TestNewbieLetter(t *testing.T) {
+	convey.Convey("Letter", t, func(ctx convey.C) {
+		var (
+			c   = context.Background()
+			req = &model.NewbieLetterReq{Aid: 10110467, Mid: 27515398}
+		)
+		ctx.Convey("When everything goes positive", func(ctx convey.C) {
+			p1, err := s.Letter(c, req)
+			ctx.Convey("Then err should be nil.p1 should not be nil.", func(ctx convey.C) {
+				ctx.So(err, convey.ShouldBeNil)
+				ctx.So(p1, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}

app/interface/main/growup/service/newbie/service.go

@@ -0,0 +1,33 @@
+package newbie
+
+import (
+	"context"
+
+	"go-common/app/interface/main/growup/conf"
+	"go-common/app/interface/main/growup/dao/newbiedao"
+)
+
+// Service is growup service
+type Service struct {
+	conf *conf.Config
+	dao  *newbiedao.Dao
+}
+
+// New fn
+func New(c *conf.Config) (s *Service) {
+	s = &Service{
+		conf: c,
+		dao:  newbiedao.New(c),
+	}
+	return s
+}
+
+// Ping fn
+func (s *Service) Ping(c context.Context) (err error) {
+	return s.dao.Ping(c)
+}
+
+// Close dao
+func (s *Service) Close() {
+	s.dao.Close()
+}

app/interface/main/growup/service/newbie/service_test.go

@@ -0,0 +1,22 @@
+package newbie
+
+import (
+	"flag"
+	"go-common/app/interface/main/growup/conf"
+	"os"
+	"testing"
+)
+
+var (
+	s *Service
+)
+
+func TestMain(m *testing.M) {
+	flag.Set("conf", "../../cmd/growup-interface.toml")
+	flag.Parse()
+	if err := conf.Init(); err != nil {
+		panic(err)
+	}
+	s = New(conf.Conf)
+	os.Exit(m.Run())
+}