bilibili-go-common/app/admin/main/workflow/dao/business.go

package dao

import (
	"context"
	"encoding/json"

	"go-common/app/admin/main/workflow/model"
	"go-common/app/admin/main/workflow/model/manager"
	"go-common/library/ecode"
	"go-common/library/log"

	"github.com/pkg/errors"
)

const _businessRoleURI = "http://manager.bilibili.co/x/admin/manager/internal/business/role"

var metas map[int8]*model.Meta

func init() {
	metas = make(map[int8]*model.Meta)
	data := `[
	{
		"business": 1,
		"name": "稿件投诉",
		"item_type": "group",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			},
			{
				"id": 2,
				"name": "回查"
			}
		]
	},
	{
		"business": 2,
		"name": "稿件申诉",
		"item_type": "challenge",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			},
			{
				"id": 2,
				"name": "回查"
			},
			{
				"id": 3,
				"name": "三查"
			},
			{
				"id": 11,
				"name": "客服"
			}
		]
	},
	{
		"business": 3,
		"name": "短点评投诉",
		"item_type": "group",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			},
			{
				"id": 2,
				"name": "回查"
			}
		]
	},
	{
		"business": 4,
		"name": "长点评投诉",
		"item_type": "group",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			},
			{
				"id": 2,
				"name": "回查"
			}
		]
	},
	{
		"business": 5,
		"name": "小黑屋",
		"item_type": "challenge",
		"rounds": [
			{
				"id": 1,
				"name": "评论"
			},
			{
				"id": 2,
				"name": "弹幕"
			},
			{
				"id": 3,
				"name": "私信"
			},
			{
				"id": 4,
				"name": "标签"
			},
			{
				"id": 5,
				"name": "个人资料"
			},
			{
				"id": 6,
				"name": "投稿"
			},
			{
				"id": 7,
				"name": "音频"
			},
			{
				"id": 8,
				"name": "专栏"
			},
			{
				"id": 9,
				"name": "空间头图"
			}
		]
	},
	{
		"business": 6,
		"name": "稿件审核",
		"item_type": "challenge",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			}
		]
	},
	{
		"business": 7,
		"name": "任务质检",
		"item_type": "challenge",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			}
		]
	},
	{
		"business": 8,
		"name": "频道举报",
		"item_type": "group",
		"rounds": [
			{
				"id": 1,
				"name": "一审"
			}
		]
	}
]`
	ml := make([]*model.Meta, 0)
	err := json.Unmarshal([]byte(data), &ml)
	if err != nil {
		panic(err)
	}

	for _, m := range ml {
		metas[m.Business] = m
	}
}

// BatchLastBusRecIDs will retrive the last business record ids by serveral conditions
func (d *Dao) BatchLastBusRecIDs(c context.Context, oids []int64, business int8) (bids []int64, err error) {
	bids = make([]int64, 0, len(oids))
	if len(oids) <= 0 {
		return
	}
	rows, err := d.ReadORM.Table("workflow_business").Select("max(id)").
		Where("oid IN (?) AND business=?", oids, business).
		Group("oid,business").Rows()
	if err != nil {
		err = errors.Wrapf(err, "Query(%v, %d)", oids, business)
		return
	}
	defer rows.Close()

	for rows.Next() {
		var bid int64
		if err = rows.Scan(&bid); err != nil {
			err = errors.WithStack(err)
			return
		}
		bids = append(bids, bid)
	}
	return
}

// BusinessRecs will retrive the business record by ids
func (d *Dao) BusinessRecs(c context.Context, bids []int64) (bs map[int32]*model.Business, err error) {
	bs = make(map[int32]*model.Business, len(bids))
	if len(bids) <= 0 {
		return
	}

	blist := make([]*model.Business, 0, len(bids))
	err = d.ReadORM.Table("workflow_business").Where("id IN (?)", bids).Find(&blist).Error
	if err != nil {
		err = errors.Wrapf(err, "Query(%v)", bids)
		return
	}

	for _, b := range blist {
		bs[b.Bid] = b
	}
	return
}

// LastBusRec will retrive last business record by business oid
func (d *Dao) LastBusRec(c context.Context, business int8, oid int64) (bs *model.Business, err error) {
	bs = new(model.Business)
	err = d.ReadORM.Table("workflow_business").Where("oid=? AND business=?", oid, business).Last(bs).Error
	if err != nil || bs.Bid == 0 {
		err = errors.Wrapf(err, "Query(%d, %d)", business, oid)
		bs = nil
		return
	}
	return
}

// BatchBusRecByCids will retrive businesses by cids
func (d *Dao) BatchBusRecByCids(c context.Context, cids []int64) (cidToBus map[int64]*model.Business, err error) {
	cidToBus = make(map[int64]*model.Business)
	if len(cids) <= 0 {
		return
	}

	blist := make([]*model.Business, 0, len(cids))
	err = d.ReadORM.Table("workflow_business").Where("cid IN (?)", cids).Find(&blist).Error
	if err != nil {
		err = errors.Wrapf(err, "Query(%v)", cids)
		return
	}

	for _, b := range blist {
		cidToBus[b.Cid] = b
	}
	return
}

// BusObjectByGids will retrive businesses by gids
func (d *Dao) BusObjectByGids(c context.Context, gids []int64) (gidToBus map[int64]*model.Business, err error) {
	gidToBus = make(map[int64]*model.Business, len(gids))
	if len(gids) <= 0 {
		return
	}
	blist := make([]*model.Business, 0, len(gids))
	if err = d.ReadORM.Table("workflow_business").Where("gid IN (?)", gids).Find(&blist).Error; err != nil {
		err = errors.Wrapf(err, "Query(%v)", gids)
		return
	}
	for _, b := range blist {
		gidToBus[b.Gid] = b
	}
	return
}

// AllMetas will retrive business meta infomation from pre-configured
func (d *Dao) AllMetas(c context.Context) map[int8]*model.Meta {
	return metas
}

// LoadRole .
func (d *Dao) LoadRole(c context.Context) (role map[int8]map[int8]string, err error) {
	var (
		resp *manager.RoleResponse
		ok   bool
		uri  = _businessRoleURI
	)
	role = make(map[int8]map[int8]string)
	if err = d.httpRead.Get(c, uri, "", nil, &resp); err != nil {
		log.Error("failed call %s error(%v)", uri, err)
		return
	}

	if resp.Code != ecode.OK.Code() {
		err = ecode.Int(resp.Code)
		log.Error("call %s error response code(%d) message(%s)", uri, resp.Code, resp.Message)
		return
	}
	for _, r := range resp.Data {
		if _, ok = role[r.Bid]; !ok {
			role[r.Bid] = make(map[int8]string)
		}
		role[r.Bid][r.Rid] = r.Name
	}
	return
}