Create & Init Project...

app/interface/main/web-feed/service/BUILD

@@ -0,0 +1,57 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["feed_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/web-feed/conf:go_default_library",
+        "//app/interface/main/web-feed/model:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "feed.go",
+        "service.go",
+    ],
+    importpath = "go-common/app/interface/main/web-feed/service",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/web-feed/conf:go_default_library",
+        "//app/interface/main/web-feed/dao:go_default_library",
+        "//app/interface/main/web-feed/model:go_default_library",
+        "//app/interface/openplatform/article/model:go_default_library",
+        "//app/service/main/account/model:go_default_library",
+        "//app/service/main/account/rpc/client:go_default_library",
+        "//app/service/main/feed/model:go_default_library",
+        "//app/service/main/feed/rpc/client:go_default_library",
+        "//library/cache:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/metadata: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/interface/main/web-feed/service/feed.go

@@ -0,0 +1,171 @@
+package service
+
+import (
+	"context"
+
+	"go-common/app/interface/main/web-feed/dao"
+	"go-common/app/interface/main/web-feed/model"
+	artmdl "go-common/app/interface/openplatform/article/model"
+	account "go-common/app/service/main/account/model"
+	feedmdl "go-common/app/service/main/feed/model"
+	"go-common/library/log"
+	"go-common/library/net/metadata"
+)
+
+// Feed get feed of ups and bangumi.
+func (s *Service) Feed(c context.Context, mid int64, pn, ps int) (res []*model.Feed, err error) {
+	var (
+		feeds, newFeeds []*feedmdl.Feed
+		upAids          []int64
+		accRes          map[int64]*account.Card
+	)
+	arg := &feedmdl.ArgFeed{
+		Mid:    mid,
+		Pn:     pn,
+		Ps:     ps,
+		RealIP: metadata.String(c, metadata.RemoteIP),
+	}
+	if feeds, err = s.feedRPC.WebFeed(c, arg); err != nil || len(feeds) == 0 {
+		log.Error("s.feedRPC.WebFeed(%v) error(%v)", arg, err)
+		if pn == 1 {
+			res, err = s.dao.FeedCache(c, mid)
+			log.Info("s.dao.FeedCache(%d) len(%d) error(%v)", mid, len(res), err)
+		}
+		return
+	}
+	for _, item := range feeds {
+		if (item.Type == feedmdl.BangumiType) && (item.Bangumi == nil) {
+			dao.PromError("bangumi为空")
+			log.Error("s.feedRPC.WebFeed(%v) error(%v)", arg, err)
+			continue
+		}
+		if (item.Type == feedmdl.ArchiveType) && (item.Archive == nil) {
+			dao.PromError("archive为空")
+			log.Error("s.feedRPC.WebFeed(%v) error(%v)", arg, err)
+			continue
+		}
+		newFeeds = append(newFeeds, item)
+	}
+	feeds = newFeeds
+	for _, item := range feeds {
+		if item.Type != feedmdl.ArchiveType {
+			continue
+		}
+		if item.Archive != nil {
+			upAids = append(upAids, item.Archive.Author.Mid)
+		}
+	}
+	accArg := &account.ArgMids{Mids: upAids}
+	if accRes, err = s.accRPC.Cards3(c, accArg); err != nil {
+		dao.PromError("rpc:accRPC.Infos2")
+		log.Error("Feed s.accRPC.info(%v) error(%v)", arg, err)
+		err = nil
+	}
+	for _, item := range feeds {
+		tmp := model.Feed{Feed: item}
+		if tmp.Type == feedmdl.ArchiveType && accRes != nil {
+			if ai, ok := accRes[item.Archive.Author.Mid]; ok {
+				tmp.OfficialVerify = &ai.Official
+			}
+		}
+		res = append(res, &tmp)
+	}
+	if pn == 1 {
+		s.cache.Save(func() {
+			s.dao.SetFeedCache(context.TODO(), mid, res)
+		})
+	}
+	return
+}
+
+// UnreadCount get unread count of feed
+func (s *Service) UnreadCount(c context.Context, mid int64) (count int, err error) {
+	arg := &feedmdl.ArgMid{
+		Mid:    mid,
+		RealIP: metadata.String(c, metadata.RemoteIP),
+	}
+	if count, err = s.feedRPC.WebUnreadCount(c, arg); err != nil {
+		dao.PromError("rpc:feedRPC.WebUnreadCount")
+		log.Error("s.feedRPC.UnreadCount(%v) error(%v)", arg, err)
+	}
+	return
+}
+
+// ArticleUnreadCount get unread count of feed
+func (s *Service) ArticleUnreadCount(c context.Context, mid int64) (count int, err error) {
+	arg := &feedmdl.ArgMid{
+		Mid:    mid,
+		RealIP: metadata.String(c, metadata.RemoteIP),
+	}
+	if count, err = s.feedRPC.ArticleUnreadCount(c, arg); err != nil {
+		dao.PromError("feed:ArticleUnreadCount")
+		log.Error("s.feedRPC.ArticleUnreadCount(%v) error(%v)", arg, err)
+	}
+	return
+}
+
+// ArticleFeed get feed of ups and bangumi.
+func (s *Service) ArticleFeed(c context.Context, mid int64, pn, ps int) (res []*artmdl.Meta, err error) {
+	arg := &feedmdl.ArgFeed{
+		Mid:    mid,
+		Pn:     pn,
+		Ps:     ps,
+		RealIP: metadata.String(c, metadata.RemoteIP),
+	}
+	if res, err = s.feedRPC.ArticleFeed(c, arg); err != nil {
+		log.Error("s.feedRPC.ArticleFeed(%v) error(%v)", arg, err)
+		return
+	}
+	var mids []int64
+	var accRes map[int64]*account.Card
+	for _, meta := range res {
+		if meta.Author != nil {
+			mids = append(mids, meta.Author.Mid)
+		}
+	}
+	accArg := &account.ArgMids{
+		Mids: mids,
+	}
+	if accRes, err = s.accRPC.Cards3(c, accArg); err != nil {
+		dao.PromError("rpc:accRPC.Infos2")
+		log.Error("Feed s.accRPC.info(%v) error(%v)", arg, err)
+		err = nil
+		return
+	}
+	for _, item := range res {
+		if (item.Author == nil) || (accRes[item.Author.Mid] == nil) {
+			continue
+		}
+		info := accRes[item.Author.Mid]
+		item.Author = &artmdl.Author{
+			Mid:  item.Author.Mid,
+			Name: info.Name,
+			Face: info.Face,
+			Pendant: artmdl.Pendant{
+				Pid:    int32(info.Pendant.Pid),
+				Name:   info.Pendant.Name,
+				Image:  info.Pendant.Image,
+				Expire: int32(info.Pendant.Expire),
+			},
+			Nameplate: artmdl.Nameplate{
+				Nid:        info.Nameplate.Nid,
+				Name:       info.Nameplate.Name,
+				Image:      info.Nameplate.Image,
+				ImageSmall: info.Nameplate.ImageSmall,
+				Level:      info.Nameplate.Level,
+				Condition:  info.Nameplate.Condition,
+			},
+		}
+		if info.Official.Role == 0 {
+			item.Author.OfficialVerify.Type = -1
+		} else {
+			if info.Official.Role <= 2 {
+				item.Author.OfficialVerify.Type = 0
+			} else {
+				item.Author.OfficialVerify.Type = 1
+			}
+			item.Author.OfficialVerify.Desc = info.Official.Title
+		}
+	}
+	return
+}

app/interface/main/web-feed/service/feed_test.go

@@ -0,0 +1,52 @@
+package service
+
+import (
+	"context"
+	"testing"
+
+	"go-common/app/interface/main/web-feed/conf"
+	"go-common/app/interface/main/web-feed/model"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func initConf(t *testing.T) {
+	if err := conf.Init(); err != nil {
+		t.Errorf("conf.Init() error(%v)", err)
+		t.FailNow()
+	}
+}
+
+func TestFeed(t *testing.T) {
+	Convey("feed", t, func() {
+		var (
+			mid int64 = 27515256
+			pn        = 1
+			ps        = 20
+			c         = context.TODO()
+			res []*model.Feed
+			err error
+		)
+		initConf(t)
+		svr := New(conf.Conf)
+		if res, err = svr.Feed(c, mid, pn, ps); err != nil {
+			t.Error(err)
+		}
+		t.Logf("result length:%d", len(res))
+	})
+}
+
+func TestUnreadCount(t *testing.T) {
+	var (
+		mid   int64 = 27515256
+		c           = context.TODO()
+		count int
+		err   error
+	)
+	initConf(t)
+	svr := New(conf.Conf)
+	if count, err = svr.UnreadCount(c, mid); err != nil {
+		t.Error(err)
+	}
+	t.Logf("count:%d", count)
+}

app/interface/main/web-feed/service/service.go

@@ -0,0 +1,42 @@
+package service
+
+import (
+	"context"
+
+	"go-common/app/interface/main/web-feed/conf"
+	"go-common/app/interface/main/web-feed/dao"
+	accrpc "go-common/app/service/main/account/rpc/client"
+	feedrpc "go-common/app/service/main/feed/rpc/client"
+	"go-common/library/cache"
+)
+
+// Service service struct info
+type Service struct {
+	c       *conf.Config
+	dao     *dao.Dao
+	feedRPC *feedrpc.Service
+	accRPC  *accrpc.Service3
+	cache   *cache.Cache
+}
+
+// New .
+func New(c *conf.Config) (s *Service) {
+	s = &Service{
+		c:       c,
+		dao:     dao.New(c),
+		feedRPC: feedrpc.New(c.FeedRPC),
+		accRPC:  accrpc.New3(c.AccountRPC),
+		cache:   cache.New(1, 1024),
+	}
+	return s
+}
+
+// Close closes dao.
+func (s *Service) Close() {
+	s.dao.Close()
+}
+
+// Ping is check server ping.
+func (s *Service) Ping(c context.Context) (err error) {
+	return s.dao.Ping(c)
+}