Create & Init Project...

app/interface/main/creative/service/whitelist/BUILD

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

@@ -0,0 +1,187 @@
+package whitelist
+
+import (
+	"context"
+	"go-common/app/interface/main/creative/conf"
+	"go-common/app/interface/main/creative/dao/whitelist"
+	accmdl "go-common/app/interface/main/creative/model/account"
+	"go-common/app/interface/main/creative/service"
+	"go-common/library/log"
+	"time"
+)
+
+const (
+	_creator         = 0
+	_playerAttention = 2
+	_taskMiddleUp    = 3
+	_taskSmallUp     = 4
+	_taskNotUp       = 5
+)
+
+//Service struct
+type Service struct {
+	c               *conf.Config
+	Creator         map[int64]int64
+	PlayerAttention map[int64]int64
+	list            map[int]map[int64]int64
+	wl              *whitelist.Dao
+}
+
+//New get service
+func New(c *conf.Config, rpcdaos *service.RPCDaos) *Service {
+	s := &Service{
+		c:  c,
+		wl: whitelist.New(c),
+	}
+	s.loadAll()
+	s.loadCreator()
+	s.loadSwitch()
+	go s.loadproc()
+	return s
+}
+
+func (s *Service) loadCreator() {
+	s.Creator = s.list[_creator]
+}
+
+func (s *Service) loadSwitch() {
+	s.PlayerAttention = s.list[_playerAttention]
+}
+
+func (s *Service) loadAll() {
+	wlList, err := s.wl.List(context.TODO())
+	if err != nil {
+		return
+	}
+	temp := make(map[int]map[int64]int64)
+	for _, v := range wlList {
+		if _, ok := temp[v.Tp]; !ok {
+			temp[v.Tp] = make(map[int64]int64)
+		}
+		temp[v.Tp][v.Mid] = v.Mid
+	}
+	s.list = temp
+}
+
+// loadproc
+func (s *Service) loadproc() {
+	for {
+		time.Sleep(5 * time.Minute)
+		s.loadAll()
+		s.loadCreator()
+		s.loadSwitch()
+	}
+}
+
+// UploadInfoForCreator fn, 判断创作姬的能否进入app投稿的权限
+func (s *Service) UploadInfoForCreator(mf *accmdl.MyInfo, mid int64) (uploadinfo map[string]interface{}) {
+	uploadinfo = make(map[string]interface{})
+	uploadinfo["info"] = 1
+	uploadinfo["reason"] = "账号已经过校验，可以投稿。"
+	if mf == nil {
+		log.Error("accmdl.MyInfo is nil mid(%d)", mid)
+		return
+	}
+	if mf.Banned {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "您的账号已被禁用，无法投稿。"
+	}
+	if !mf.Activated {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "您的账号尚未激活，无法投稿。"
+	}
+	if mf.IdentifyInfo.Code != 0 {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "投稿需要进行实名制登记，请先在PC上进行登记。"
+	}
+	var ok bool
+	if _, ok = s.Creator[mf.Mid]; !ok && (mf.Level < 4) {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "亲爱的用户，应用内测期间仅开放给部分用户。敬请谅解。"
+	}
+	return
+}
+
+// Viewinfo fn, 判断创作姬的能否进入app查看的权限
+func (s *Service) Viewinfo(mf *accmdl.MyInfo) (uploadinfo map[string]interface{}) {
+	uploadinfo = make(map[string]interface{})
+	var ok bool
+	uploadinfo["info"] = 0
+	uploadinfo["reason"] = "亲爱的用户，应用内测期间仅开放给部分用户。敬请谅解。"
+	if _, ok = s.Creator[mf.Mid]; ok {
+		uploadinfo["info"] = 1
+		uploadinfo["reason"] = ""
+	}
+	return
+}
+
+// UploadInfoForMainApp fn, 判断主APP的能否进入app投稿的权限
+func (s *Service) UploadInfoForMainApp(mf *accmdl.MyInfo, plat string, mid int64) (uploadinfo map[string]interface{}, white int) {
+	uploadinfo = make(map[string]interface{})
+	uploadinfo["info"] = 1
+	uploadinfo["reason"] = "账号已经过校验，可以投稿。"
+	uploadinfo["url"] = ""
+	if mf == nil {
+		log.Error("accmdl.MyInfo is nil mid(%d)", mid)
+		return
+	}
+	if mf.Banned {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "您的账号已被禁用，无法投稿。"
+	}
+	if !mf.Activated {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "您的账号尚未激活，无法投稿。"
+	}
+	if mf.IdentifyInfo.Code != 0 {
+		uploadinfo["info"] = 0
+		if plat == "android" {
+			uploadinfo["reason"] = "投稿需要进行实名认证，请前往“设置”-“账号与隐私”中进行认证"
+		} else if plat == "ios" {
+			uploadinfo["reason"] = "投稿需要进行实名认证，请前往“设置”-“安全隐私”中进行认证"
+		} else {
+			uploadinfo["reason"] = "投稿需要进行实名制登记，请先在PC上进行登记。"
+		}
+		uploadinfo["url"] = s.c.H5Page.Passport
+	}
+	if mf.Level < 1 {
+		uploadinfo["info"] = 0
+		uploadinfo["reason"] = "LV1以上用户才能投稿哦，请在头像下方进入答题升级吧～"
+	}
+	white, _ = uploadinfo["info"].(int)
+	return
+}
+
+//TaskWhiteList 任务系统白名单 0-关闭 1-开启
+func (s *Service) TaskWhiteList(mid int64) (res int8) {
+	blackList := []int64{208259, 2, 9099524}
+	for _, v := range blackList {
+		if v == mid {
+			return 0
+		}
+	}
+	// internal whiteList
+	if s.c.Whitelist.DataMids != nil {
+		for _, m := range s.c.Whitelist.DataMids {
+			if m == mid {
+				return 1
+			}
+		}
+	}
+	if s.c.TaskCondition.WhiteSwitch {
+		if s.list != nil {
+			if _, ok := s.list[_taskNotUp][mid]; ok {
+				return 1
+			}
+			if _, ok := s.list[_taskSmallUp][mid]; ok {
+				return 1
+			}
+			if _, ok := s.list[_taskMiddleUp][mid]; ok {
+				return 1
+			}
+		}
+	} else {
+		return 0
+	}
+	return 0
+}

app/interface/main/creative/service/whitelist/service_test.go

@@ -0,0 +1,54 @@
+package whitelist
+
+import (
+	"flag"
+	"go-common/app/interface/main/creative/conf"
+	accmdl "go-common/app/interface/main/creative/model/account"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"go-common/app/interface/main/creative/service"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	s *Service
+)
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/creative.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	rpcdaos := service.NewRPCDaos(conf.Conf)
+	s = New(conf.Conf, rpcdaos)
+	time.Sleep(time.Second)
+}
+
+func WithService(f func(s *Service)) func() {
+	return func() {
+		Reset(func() {})
+		f(s)
+	}
+}
+func Test_whitelist(t *testing.T) {
+	mf := &accmdl.MyInfo{}
+	var (
+		uploadinfo map[string]interface{}
+		white      int
+	)
+	Convey("UploadInfoForMainApp", t, WithService(func(s *Service) {
+		uploadinfo, white = s.UploadInfoForMainApp(mf, "ios", 111)
+		So(uploadinfo, ShouldNotBeNil)
+		So(white, ShouldNotBeNil)
+	}))
+	Convey("UploadInfoForCreator", t, WithService(func(s *Service) {
+		uploadinfo = s.UploadInfoForCreator(mf, 111)
+		So(s.Creator, ShouldNotBeNil)
+	}))
+	Convey("Viewinfo", t, WithService(func(s *Service) {
+		s.Viewinfo(mf)
+		So(uploadinfo, ShouldNotBeNil)
+	}))
+}