bilibili-go-common/app/admin/main/reply/service/subject.go

package service

import (
	"context"
	"encoding/json"
	"strconv"
	"time"

	"go-common/library/queue/databus/report"

	"go-common/app/admin/main/reply/model"
	"go-common/library/ecode"
	"go-common/library/log"
)

func (s *Service) subject(c context.Context, oid int64, typ int32) (sub *model.Subject, err error) {
	if sub, err = s.dao.Subject(c, oid, typ); err != nil {
		return
	}
	if sub == nil {
		err = ecode.NothingFound
	}
	return
}

func (s *Service) subjects(c context.Context, oids []int64, typ int32) (res map[int64]*model.Subject, err error) {
	return s.dao.Subjects(c, oids, typ)
}

// Subject get subject info.
func (s *Service) Subject(c context.Context, oid int64, typ int32) (sub *model.Subject, err error) {
	if sub, err = s.dao.Subject(c, oid, typ); err != nil {
		return
	}
	// NOTE nothing found return to normal
	return
}

// ModifySubState modify subject state
func (s *Service) ModifySubState(c context.Context, adid int64, adName string, oids []int64, typ int32, state int32, remark string) (fails map[int64]string, err error) {
	var (
		action string
		sub    *model.Subject
		now    = time.Now()
	)
	switch state {
	case model.SubStateNormal:
		action = model.SujectAllow
	case model.SubStateForbid:
		action = model.SujectForbid
	default:
		err = ecode.ReplyIllegalSubState
		return
	}
	fails = make(map[int64]string)
	for _, oid := range oids {
		sub, err = s.subject(c, oid, typ)
		if err != nil {
			log.Error("rpSvr.subject(oid,%d,type,%d)error(%v)", oid, typ, err)
			fails[oid] = ecode.Cause(err).Message()
			err = nil
			continue
		}
		// subject frozen
		if sub.AttrVal(model.SubAttrFrozen) == model.AttrYes {
			fails[oid] = ecode.ReplySubjectFrozen.Message()
			continue
		}
		if _, err = s.dao.UpSubjectState(c, oid, typ, state, now); err != nil {
			log.Error("s.UpSubjectState(%d,%d) error(%v)", oid, typ, err)
			return
		}
		if err = s.dao.DelSubjectCache(c, oid, typ); err != nil {
			log.Error("s.dao.DeleteSubjectCache(%d,%d) state:%d error(%v)", oid, typ, err)
		}
		report.Manager(&report.ManagerInfo{
			UID:      adid,
			Uname:    adName,
			Business: 41,
			Type:     int(typ),
			Oid:      oid,
			Ctime:    now,
			Action:   action,
			Index:    []interface{}{sub.State, state},
			Content:  map[string]interface{}{"remark": remark},
		})
	}
	return
}

// FreezeSub freeze or unfreeze subject
func (s *Service) FreezeSub(c context.Context, adid int64, adName string, oids []int64, typ int32, freeze int32, remark string) (fails map[int64]string, err error) {
	var (
		state  int32
		action string
		sub    *model.Subject
		now    = time.Now()
	)
	fails = make(map[int64]string)
	for _, oid := range oids {
		if sub, err = s.subject(c, oid, typ); err != nil {
			log.Error("rpSvr.subject(oid,%d,type,%d)error(%v)", oid, typ, err)
			fails[oid] = ecode.Cause(err).Message()
			err = nil
			continue
		}
		// already freeze or already unfreeze
		if (sub.AttrVal(model.SubAttrFrozen) == model.AttrYes && freeze == 2) || (sub.AttrVal(model.SubAttrFrozen) == model.AttrNo && freeze != 2) {
			continue
		}
		switch freeze {
		case 0:
			// unfreeze and allow
			sub.AttrSet(model.AttrNo, model.SubAttrFrozen)
			state = model.SubStateNormal
			action = model.SujectUnfrozenAllow
		case 1:
			// unfreeze and forbid
			sub.AttrSet(model.AttrNo, model.SubAttrFrozen)
			state = model.SubStateForbid
			action = model.SujectUnfrozenForbid
		case 2:
			// freeze and forbid
			sub.AttrSet(model.AttrYes, model.SubAttrFrozen)
			state = model.SubStateForbid
			action = model.SujectFrozen
		default:
			err = ecode.ReplyIllegalSubState
			return
		}
		if _, err = s.dao.UpStateAndAttr(c, oid, typ, state, sub.Attr, now); err != nil {
			log.Error("s.UpSubjectAttr(%d,%d,%d,%d) error(%v)", oid, typ, sub.Attr, err)
			return
		}
		if err = s.dao.DelSubjectCache(c, oid, typ); err != nil {
			log.Error("s.dao.DeleteSubjectCache(%d,%d) state:%d error(%v)", oid, typ, err)
		}
		report.Manager(&report.ManagerInfo{
			UID:      adid,
			Uname:    adName,
			Business: 41,
			Type:     int(typ),
			Oid:      oid,
			Ctime:    now,
			Action:   action,
			Index:    []interface{}{sub.State, state},
			Content:  map[string]interface{}{"remark": remark},
		})
	}
	return
}

// SubjectLog returns operation logs by query parameters
func (s *Service) SubjectLog(c context.Context, sp model.LogSearchParam) (res *model.SubjectLogRes, err error) {
	res = &model.SubjectLogRes{
		Logs: []*model.SubjectLog{},
	}
	sp.Action = "subject_allow,subject_forbid,subject_frozen,subject_unfrozen_allow,subject_unfrozen_forbid"
	if sp.Pn <= 0 || sp.Ps <= 0 {
		return nil, ecode.RequestErr
	}
	reportData, err := s.dao.ReportLog(c, sp)
	if err != nil || reportData == nil {
		return
	}
	res.Page = reportData.Page
	res.Sort = reportData.Sort
	res.Order = reportData.Order
	exists := make(map[int64]bool)
	var oids []int64
	for _, data := range reportData.Result {
		if !exists[data.Oid] {
			exists[data.Oid] = true
			oids = append(oids, data.Oid)
		}
		var extra map[string]string
		err = json.Unmarshal([]byte(data.Content), &extra)
		if err != nil {
			log.Error("Subject Operation Log unmarshal failed(%v)", err)
			return
		}
		res.Logs = append(res.Logs, &model.SubjectLog{
			AdminID:   data.AdminID,
			AdminName: data.AdminName,
			Oid:       strconv.FormatInt(data.Oid, 10),
			Type:      data.Type,
			Remark:    extra["remark"],
			CTime:     data.Ctime,
			Action:    data.Action,
		})
	}
	if sp.Type != 0 && len(oids) > 0 {
		var subjects map[int64]*model.Subject
		subjects, err = s.subjects(c, oids, sp.Type)
		if err == nil {
			for _, data := range res.Logs {
				var oid int64
				oid, err = strconv.ParseInt(data.Oid, 10, 64)
				if err != nil {
					log.Error("strconv.ParseInt failed(%v)", err)
					return
				}
				if sub, ok := subjects[oid]; ok && sub != nil {
					if sub.AttrVal(model.SubAttrFrozen) == model.AttrYes {
						//frozen
						data.State = 2
					} else if sub.State == model.SubStateForbid {
						data.State = model.SubStateForbid
					} else {
						data.State = model.SubStateNormal
					}
				}
			}
		}
	}
	return
}