Create & Init Project...

app/admin/main/block/service/BUILD

@@ -0,0 +1,59 @@
+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/admin/main/block/conf:go_default_library",
+        "//app/admin/main/block/model:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "audit_log.go",
+        "block.go",
+        "msg.go",
+        "notify.go",
+        "service.go",
+    ],
+    importpath = "go-common/app/admin/main/block/service",
+    tags = ["automanaged"],
+    deps = [
+        "//app/admin/main/block/conf:go_default_library",
+        "//app/admin/main/block/dao:go_default_library",
+        "//app/admin/main/block/model:go_default_library",
+        "//library/cache:go_default_library",
+        "//library/database/sql:go_default_library",
+        "//library/log:go_default_library",
+        "//library/queue/databus:go_default_library",
+        "//library/queue/databus/report:go_default_library",
+        "//library/sync/errgroup: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/admin/main/block/service/audit_log.go

@@ -0,0 +1,48 @@
+package service
+
+import (
+	"context"
+	"strconv"
+	"time"
+
+	"go-common/app/admin/main/block/model"
+	"go-common/library/log"
+	manager "go-common/library/queue/databus/report"
+)
+
+// AddAuditLog .
+func (s *Service) AddAuditLog(c context.Context, tp model.BlockAction, uid int64, uname string, oids []int64, duration time.Duration, source model.BlockSource, area model.BlockArea, reason, comment string, notify bool, stime time.Time) error {
+	var (
+		err       error
+		dur       = int64(duration / time.Second)
+		notifyStr = strconv.FormatBool(notify)
+	)
+	for _, oid := range oids {
+		managerInfo := &manager.ManagerInfo{
+			UID:      uid,
+			Uname:    uname,
+			Business: model.BlockLogBizID,
+			Type:     int(tp),
+			Action:   tp.String(),
+			Oid:      oid,
+			Ctime:    time.Now(),
+			Index:    []interface{}{dur, uint8(source), uint8(area), reason, comment, notifyStr},
+			Content: map[string]interface{}{
+				"duration":    dur,
+				"source":      source,
+				"area":        area,
+				"reason":      reason,
+				"comment":     comment,
+				"notify":      notifyStr,
+				"action_time": stime.Unix(),
+				"remove_time": stime.Add(time.Second * time.Duration(dur)).Unix(),
+			},
+		}
+		if err = manager.Manager(managerInfo); err != nil {
+			log.Error("manager.Manager(%+v) error(%+v)", managerInfo, err)
+			continue
+		}
+		log.Info("s.managerSendLog(%+v)", managerInfo)
+	}
+	return err
+}

app/admin/main/block/service/block.go

@@ -0,0 +1,318 @@
+package service
+
+import (
+	"context"
+	"sync"
+	"time"
+
+	"go-common/app/admin/main/block/model"
+	xsql "go-common/library/database/sql"
+	"go-common/library/log"
+	"go-common/library/sync/errgroup"
+
+	"github.com/pkg/errors"
+)
+
+// Search .
+func (s *Service) Search(c context.Context, mids []int64) (infos []*model.BlockInfo, err error) {
+	var (
+		users       []*model.DBUser
+		userDetails []*model.DBUserDetail
+		eg          errgroup.Group
+		mapMu       sync.Mutex
+		userMap     = make(map[int64]*model.BlockInfo)
+	)
+	if users, err = s.dao.Users(c, mids); err != nil {
+		return
+	}
+	if userDetails, err = s.dao.UserDetails(c, mids); err != nil {
+		return
+	}
+	infos = make([]*model.BlockInfo, 0, len(mids))
+	for _, m := range mids {
+		mid := m
+		eg.Go(func() (err error) {
+			info := &model.BlockInfo{
+				MID: mid,
+			}
+			// 1. account 数据
+			if info.Nickname, info.TelStatus, info.Level, info.RegTime, err = s.dao.AccountInfo(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+				err = nil
+			}
+			if info.Nickname == "" {
+				log.Info("user mid(%d) not found", info.MID)
+				return
+			}
+			// 2. 封禁状态
+			for i := range users {
+				if users[i].MID == mid {
+					info.ParseStatus(users[i])
+					break
+				}
+			}
+			// 3. 封禁次数
+			for i := range userDetails {
+				if userDetails[i].MID == mid {
+					info.BlockCount = userDetails[i].BlockCount
+					break
+				}
+			}
+			// 4. spy 分值
+			if info.SpyScore, err = s.dao.SpyScore(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+				err = nil
+				info.SpyScore = -1
+			}
+			// 5. figure 排名
+			if info.FigureRank, err = s.dao.FigureRank(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+				err = nil
+				info.FigureRank = -1
+			}
+			// 6. extra 额外账号信息
+			if info.Tel, err = s.telInfo(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+				err = nil
+				info.Tel = "N/A"
+			}
+			if info.Mail, err = s.mailInfo(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+				err = nil
+				info.Mail = "N/A"
+			}
+			if info.Username, err = s.userID(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+				err = nil
+				info.Username = "N/A"
+			}
+
+			mapMu.Lock()
+			userMap[info.MID] = info
+			mapMu.Unlock()
+			return
+		})
+	}
+	eg.Wait()
+	for _, mid := range mids {
+		if info, ok := userMap[mid]; ok {
+			infos = append(infos, info)
+		}
+	}
+	return
+}
+
+// History .
+func (s *Service) History(c context.Context, mid int64, ps, pn int) (status model.BlockStatus, total int, history []*model.BlockHistory, err error) {
+	var (
+		start     = (pn - 1) * ps
+		limit     = ps
+		dbHistory []*model.DBHistory
+		dbUser    *model.DBUser
+	)
+	if dbUser, err = s.dao.User(c, mid); err != nil {
+		return
+	}
+	if dbUser != nil {
+		status = dbUser.Status
+	}
+	if total, err = s.dao.HistoryCount(c, mid); err != nil {
+		return
+	}
+	if dbHistory, err = s.dao.History(c, mid, start, limit); err != nil {
+		return
+	}
+	for i := range dbHistory {
+		his := &model.BlockHistory{}
+		his.ParseDB(dbHistory[i])
+		history = append(history, his)
+	}
+	return
+}
+
+// BatchBlock .
+func (s *Service) BatchBlock(c context.Context, p *model.ParamBatchBlock) (err error) {
+	var (
+		tx       *xsql.Tx
+		duration = time.Duration(p.Duration) * time.Hour * 24
+		source   model.BlockSource
+		mids     []int64
+		stime    = time.Now()
+	)
+	if tx, err = s.dao.BeginTX(c); err != nil {
+		return
+	}
+	if p.Source == 1 {
+		// 系统封禁
+		source = model.BlockSourceSys
+	} else if p.Source == 2 {
+		// 小黑屋封禁
+		source = model.BlockSourceBlackHouse
+		if err = s.dao.BlackhouseBlock(context.TODO(), p); err != nil {
+			return
+		}
+	}
+	for _, mid := range p.MIDs {
+		mids = append(mids, mid)
+		theMID := mid
+		s.cache.Save(func() {
+			if cacheErr := s.dao.DeleteUserCache(context.TODO(), theMID); cacheErr != nil {
+				log.Error("%+v", cacheErr)
+			}
+			if databusErr := s.accountNotify(context.TODO(), theMID); databusErr != nil {
+				log.Error("%+v", databusErr)
+			}
+		})
+		if err = s.action(c, tx, mid, p.AdminID, p.AdminName, source, p.Area, p.Reason, p.Comment, p.Action, duration, p.Notify, stime); err != nil {
+			tx.Rollback()
+			return
+		}
+	}
+	// 发送站内信
+	if p.Notify {
+		s.mission(func() {
+			if notifyErr := s.notifyMSG(context.TODO(), p.MIDs, source, p.Action, p.Area, p.Reason, p.Duration); notifyErr != nil {
+				log.Error("%+v", notifyErr)
+				return
+			}
+		})
+	}
+	s.mission(func() {
+		s.AddAuditLog(context.TODO(), p.Action, p.AdminID, p.AdminName, mids, duration, source, p.Area, p.Reason, p.Comment, p.Notify, stime)
+	})
+	err = tx.Commit()
+	return
+}
+
+// BatchRemove .
+func (s *Service) BatchRemove(c context.Context, p *model.ParamBatchRemove) (err error) {
+	var (
+		tx    *xsql.Tx
+		mids  []int64
+		stime = time.Now()
+	)
+	if tx, err = s.dao.BeginTX(c); err != nil {
+		return
+	}
+
+	for _, mid := range p.MIDs {
+		mids = append(mids, mid)
+		theMID := mid
+		s.cache.Save(func() {
+			if cacheErr := s.dao.DeleteUserCache(context.TODO(), theMID); cacheErr != nil {
+				log.Error("%+v", cacheErr)
+			}
+			if databusErr := s.accountNotify(context.TODO(), theMID); databusErr != nil {
+				log.Error("%+v", databusErr)
+			}
+		})
+		if err = s.action(c, tx, mid, p.AdminID, p.AdminName, model.BlockSourceRemove, model.BlockAreaNone, "", p.Comment, model.BlockActionAdminRemove, 0, p.Notify, stime); err != nil {
+			tx.Rollback()
+			return
+		}
+	}
+	// 发送站内信
+	if p.Notify {
+		s.mission(func() {
+			if notifyErr := s.notifyMSG(context.TODO(), p.MIDs, model.BlockSourceRemove, model.BlockActionAdminRemove, model.BlockAreaNone, "", 0); notifyErr != nil {
+				log.Error("%+v", notifyErr)
+				return
+			}
+		})
+	}
+	s.mission(func() {
+		s.AddAuditLog(context.TODO(), model.BlockActionAdminRemove, p.AdminID, p.AdminName, mids, 0, model.BlockSourceRemove, model.BlockAreaNone, "", p.Comment, p.Notify, stime)
+	})
+	err = tx.Commit()
+	return
+}
+
+// notifyMSG .
+func (s *Service) notifyMSG(c context.Context, mids []int64, source model.BlockSource, action model.BlockAction, area model.BlockArea, reason string, days int64) (err error) {
+	code, title, content := s.MSGInfo(source, action, area, reason, days)
+	log.Info("block admin title : %s , content : %s , mids : %+v", title, content, mids)
+	if err = s.dao.SendSysMsg(context.TODO(), code, mids, title, content, ""); err != nil {
+		return
+	}
+	return
+}
+
+func (s *Service) action(c context.Context, tx *xsql.Tx, mid int64, adminID int64, adminName string, source model.BlockSource, area model.BlockArea, reason, comment string, action model.BlockAction, duration time.Duration, notify bool, stime time.Time) (err error) {
+	var (
+		db = &model.DBHistory{
+			MID:       mid,
+			AdminID:   adminID,
+			AdminName: adminName,
+			Source:    source,
+			Area:      area,
+			Reason:    reason,
+			Comment:   comment,
+			Action:    action,
+			StartTime: stime,
+			Duration:  int64(duration / time.Second),
+			Notify:    notify,
+		}
+		blockStatus model.BlockStatus
+	)
+	if err = s.dao.TxInsertHistory(c, tx, db); err != nil {
+		return
+	}
+	switch action {
+	case model.BlockActionAdminRemove, model.BlockActionSelfRemove:
+		blockStatus = model.BlockStatusFalse
+	case model.BlockActionLimit:
+		switch source {
+		case model.BlockSourceBlackHouse:
+			blockStatus = model.BlockStatusCredit
+		default:
+			blockStatus = model.BlockStatusLimit
+		}
+		s.mission(func() {
+			if err = s.dao.UpdateAddBlockCount(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+			}
+		})
+	case model.BlockActionForever:
+		blockStatus = model.BlockStatusForever
+		s.mission(func() {
+			if err = s.dao.UpdateAddBlockCount(context.TODO(), mid); err != nil {
+				log.Error("%+v", err)
+			}
+		})
+	default:
+		err = errors.Errorf("unknown block action [%d]", action)
+		return
+	}
+	if err = s.dao.TxUpdateUser(c, tx, mid, blockStatus); err != nil {
+		return
+	}
+	return
+}
+
+func (s *Service) userID(c context.Context, mid int64) (id string, err error) {
+	return "N/A", nil
+}
+
+func (s *Service) mailInfo(c context.Context, mid int64) (mail string, err error) {
+	if mail, err = s.dao.MailInfo(c, mid); mail == "" {
+		mail = "N/A"
+	}
+	return
+}
+
+// TelInfo .
+func (s *Service) telInfo(c context.Context, mid int64) (tel string, err error) {
+	if tel, err = s.dao.TelInfo(c, mid); err != nil {
+		return
+	}
+	if len(tel) == 0 {
+		tel = "N/A"
+		return
+	}
+	if len(tel) < 4 {
+		tel = tel[:1] + "****"
+		return
+	}
+	tel = tel[:3] + "****" + tel[len(tel)-4:]
+	return
+}

app/admin/main/block/service/msg.go

@@ -0,0 +1,74 @@
+package service
+
+import (
+	"fmt"
+
+	"go-common/app/admin/main/block/conf"
+	"go-common/app/admin/main/block/model"
+	"go-common/library/log"
+)
+
+// MSGInfo get msg info
+func (s *Service) MSGInfo(source model.BlockSource, action model.BlockAction, area model.BlockArea, reason string, days int64) (code string, title, content string) {
+	if source == model.BlockSourceBlackHouse {
+		areaStr := area.String()
+		if areaStr != "" {
+			areaStr = fmt.Sprintf("在%s中", areaStr)
+		}
+		if action == model.BlockActionLimit {
+			code = conf.Conf.Property.MSG.BlackHouseLimit.Code
+			title = conf.Conf.Property.MSG.BlackHouseLimit.Title
+			content = fmt.Sprintf(conf.Conf.Property.MSG.BlackHouseLimit.Content, areaStr, s.convertReason(reason), days)
+			return
+		}
+		if action == model.BlockActionForever {
+			code = conf.Conf.Property.MSG.BlackHouseForever.Code
+			title = conf.Conf.Property.MSG.BlackHouseForever.Title
+			content = fmt.Sprintf(conf.Conf.Property.MSG.BlackHouseForever.Content, areaStr, s.convertReason(reason))
+			return
+		}
+	}
+	if source == model.BlockSourceSys {
+		if action == model.BlockActionLimit {
+			code = conf.Conf.Property.MSG.SysLimit.Code
+			title = conf.Conf.Property.MSG.SysLimit.Title
+			content = fmt.Sprintf(conf.Conf.Property.MSG.SysLimit.Content, s.convertReason(reason), days)
+			return
+		}
+		if action == model.BlockActionForever {
+			code = conf.Conf.Property.MSG.SysForever.Code
+			title = conf.Conf.Property.MSG.SysForever.Title
+			content = fmt.Sprintf(conf.Conf.Property.MSG.SysForever.Content, s.convertReason(reason))
+			return
+		}
+	}
+	if action == model.BlockActionAdminRemove || action == model.BlockActionSelfRemove {
+		code = conf.Conf.Property.MSG.BlockRemove.Code
+		title = conf.Conf.Property.MSG.BlockRemove.Title
+		content = conf.Conf.Property.MSG.BlockRemove.Content
+		return
+	}
+	log.Error("s.MSGInfo unkown source[%v] action[%v] area[%v] reason[%s] days[%d]", source, action, area, reason, days)
+	return
+}
+
+func (s *Service) convertReason(reason string) string {
+	switch reason {
+	case "账号资料相关违规":
+		return "账号资料违规"
+	case "作品投稿违规":
+		return "作品投稿违规"
+	case "异常注册账号":
+		return "异常注册"
+	case "异常答题账号":
+		return "异常答题"
+	case "异常数据行为":
+		return "异常数据行为"
+	case "发布违规信息":
+		return "发布违规信息"
+	case "其他自动封禁", "手动封禁":
+		return "违反社区规则"
+	default:
+		return reason
+	}
+}

app/admin/main/block/service/notify.go

@@ -0,0 +1,18 @@
+package service
+
+import (
+	"context"
+	"strconv"
+
+	"go-common/app/admin/main/block/model"
+
+	"github.com/pkg/errors"
+)
+
+func (s *Service) accountNotify(c context.Context, mid int64) (err error) {
+	msg := &model.AccountNotify{UID: mid, Action: "blockUser"}
+	if err = s.accountNotifyPub.Send(c, strconv.FormatInt(msg.UID, 10), msg); err != nil {
+		err = errors.Errorf("mid(%d) s.accountNotify.Send(%+v) error(%+v)", msg.UID, msg, err)
+	}
+	return
+}

app/admin/main/block/service/service.go

@@ -0,0 +1,63 @@
+package service
+
+import (
+	"context"
+
+	"go-common/app/admin/main/block/conf"
+	"go-common/app/admin/main/block/dao"
+	"go-common/library/cache"
+	"go-common/library/log"
+	"go-common/library/queue/databus"
+)
+
+// Service struct
+type Service struct {
+	dao              *dao.Dao
+	cache            *cache.Cache
+	missch           chan func()
+	accountNotifyPub *databus.Databus
+}
+
+// New init
+func New() (s *Service) {
+	s = &Service{
+		dao:              dao.New(),
+		cache:            cache.New(1, 10240),
+		missch:           make(chan func(), 10240),
+		accountNotifyPub: databus.New(conf.Conf.AccountNotify),
+	}
+	go s.missproc()
+	return s
+}
+
+func (s *Service) missproc() {
+	defer func() {
+		if x := recover(); x != nil {
+			log.Error("service.missproc panic(%v)", x)
+			go s.missproc()
+			log.Info("service.missproc recover")
+		}
+	}()
+	for {
+		f := <-s.missch
+		f()
+	}
+}
+
+func (s *Service) mission(f func()) {
+	select {
+	case s.missch <- f:
+	default:
+		log.Error("s.missch full")
+	}
+}
+
+// Ping Service
+func (s *Service) Ping(c context.Context) (err error) {
+	return s.dao.Ping(c)
+}
+
+// Close Service
+func (s *Service) Close() {
+	s.dao.Close()
+}

app/admin/main/block/service/service_test.go

@@ -0,0 +1,68 @@
+package service
+
+import (
+	"context"
+	"flag"
+	"os"
+	"testing"
+
+	"go-common/app/admin/main/block/conf"
+	"go-common/app/admin/main/block/model"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	s *Service
+	c = context.TODO()
+)
+
+func TestMain(m *testing.M) {
+	defer os.Exit(0)
+	flag.Set("conf", "../cmd/block-admin-test.toml")
+	var err error
+	if err = conf.Init(); err != nil {
+		panic(err)
+	}
+	s = New()
+	defer s.Close()
+
+	m.Run()
+}
+
+func TestService(t *testing.T) {
+	Convey("", t, func() {
+		s.Ping(c)
+		s.Close()
+	})
+}
+
+func TestBlock(t *testing.T) {
+	Convey("block", t, func() {
+		var (
+			p = &model.ParamBatchBlock{
+				MIDs:      []int64{1, 2, 3, 4},
+				AdminID:   233,
+				AdminName: "233",
+				Source:    1,
+				Area:      model.BlockAreaNone,
+				Reason:    "test",
+				Comment:   "test",
+				Action:    model.BlockActionLimit,
+				Duration:  1,
+				Notify:    false,
+			}
+			pm = &model.ParamBatchRemove{
+				MIDs:      []int64{1, 2, 3, 4},
+				AdminID:   233,
+				AdminName: "233",
+				Comment:   "test",
+				Notify:    false,
+			}
+		)
+		err := s.BatchBlock(c, p)
+		So(err, ShouldBeNil)
+		err = s.BatchRemove(c, pm)
+		So(err, ShouldBeNil)
+	})
+}