256 lines
7.5 KiB
Go
256 lines
7.5 KiB
Go
|
package mcndao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
adminmodel "go-common/app/admin/main/mcn/model"
|
|||
|
|
"go-common/app/interface/main/mcn/model"
|
|||
|
|
"go-common/app/interface/main/mcn/model/mcnmodel"
|
|||
|
|
"go-common/app/interface/main/mcn/tool/validate"
|
|||
|
|
"go-common/library/ecode"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
|
|||
|
|
"github.com/jinzhu/gorm"
|
|||
|
|
"github.com/pkg/errors"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
signNotInStates = []model.MCNSignState{model.MCNSignStateOnDelete, model.MCNSignStateOnPreOpen, model.MCNSignStateOnExpire, model.MCNSignStateOnClear}
|
|||
|
|
// 从高到低，先找到的返回
|
|||
|
|
// 状态优先级：
|
|||
|
|
// 0： 封禁
|
|||
|
|
// 1： 签约
|
|||
|
|
// 2：
|
|||
|
|
statePriority = []model.MCNSignState{
|
|||
|
|
model.MCNSignStateOnBlock,
|
|||
|
|
model.MCNSignStateOnSign,
|
|||
|
|
model.MCNSignStateOnReject,
|
|||
|
|
model.MCNSignStateOnReview,
|
|||
|
|
model.MCNSignStateNoApply,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// mcnSign = mcnmodel.McnSign{}
|
|||
|
|
// mcnUp = mcnmodel.McnUp{}
|
|||
|
|
|
|||
|
|
// UpPermissionApplyCannotApplyStates 在这此状态下，不能再申请改变Up主
|
|||
|
|
UpPermissionApplyCannotApplyStates = []adminmodel.MCNUPPermissionState{
|
|||
|
|
adminmodel.MCNUPPermissionStateNoAuthorize, // 待Up主同意
|
|||
|
|
adminmodel.MCNUPPermissionStateReview, // 待审核
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpSignedStates up signed state
|
|||
|
|
UpSignedStates = []model.MCNUPState{
|
|||
|
|
model.MCNUPStateOnSign,
|
|||
|
|
model.MCNUPStateOnPreOpen,
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// GetMcnSignState .
|
|||
|
|
// mcnList, it's all mcn sign found with the state
|
|||
|
|
// state, the mcn's state of qualified, if multiple state found, will be return in priority
|
|||
|
|
func (d *Dao) GetMcnSignState(fields string, mcnMid int64) (mcn *mcnmodel.McnSign, state model.MCNSignState, err error) {
|
|||
|
|
var mcnList []*mcnmodel.McnSign
|
|||
|
|
if err = d.mcndb.Select(fields).Where("mcn_mid=? and state not in(?)", mcnMid, signNotInStates).Find(&mcnList).Error; err != nil {
|
|||
|
|
err = errors.WithStack(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(mcnList) == 0 {
|
|||
|
|
log.Warn("mcn not exist, mcn id=%d", mcnMid)
|
|||
|
|
err = ecode.NothingFound
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var stateMap = make(map[model.MCNSignState]*mcnmodel.McnSign)
|
|||
|
|
for _, v := range mcnList {
|
|||
|
|
stateMap[model.MCNSignState(v.State)] = v
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, v := range statePriority {
|
|||
|
|
if mcnValue, ok := stateMap[v]; ok {
|
|||
|
|
state = v
|
|||
|
|
mcn = mcnValue
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetUpBind .
|
|||
|
|
func (d *Dao) GetUpBind(query interface{}, args ...interface{}) (upList []*mcnmodel.McnUp, err error) {
|
|||
|
|
if err = d.mcndb.Select("*").Where(query, args...).Find(&upList).Error; err != nil {
|
|||
|
|
if err == gorm.ErrRecordNotFound {
|
|||
|
|
err = nil
|
|||
|
|
} else {
|
|||
|
|
log.Error("fail to get bind up from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BindUp .
|
|||
|
|
func (d *Dao) BindUp(up *mcnmodel.McnUp, sign *mcnmodel.McnSign, arg *mcnmodel.McnBindUpApplyReq) (result *mcnmodel.McnUp, affectedRow int64, err error) {
|
|||
|
|
if arg == nil || sign == nil {
|
|||
|
|
return nil, 0, ecode.ServerErr
|
|||
|
|
}
|
|||
|
|
var db *gorm.DB
|
|||
|
|
if up == nil {
|
|||
|
|
up = &mcnmodel.McnUp{
|
|||
|
|
SignID: sign.ID,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
arg.CopyTo(up)
|
|||
|
|
|
|||
|
|
// 如果绑定自己，那么直接接受
|
|||
|
|
if sign.McnMid == arg.UpMid {
|
|||
|
|
up.State = model.MCNUPStateOnSign
|
|||
|
|
// 签约周期为MCN的签约周期
|
|||
|
|
if up.BeginDate == 0 {
|
|||
|
|
up.BeginDate = sign.BeginDate
|
|||
|
|
}
|
|||
|
|
if up.EndDate == 0 {
|
|||
|
|
up.EndDate = sign.EndDate
|
|||
|
|
}
|
|||
|
|
var (
|
|||
|
|
now = time.Now()
|
|||
|
|
date = time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, time.Local)
|
|||
|
|
)
|
|||
|
|
if up.BeginDate.Time().After(date) {
|
|||
|
|
up.State = model.MCNUPStateOnPreOpen
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 清除状态为未授权
|
|||
|
|
up.State = model.MCNUPStateNoAuthorize
|
|||
|
|
if !validate.RegHTTPCheck.MatchString(arg.ContractLink) || !validate.RegHTTPCheck.MatchString(arg.UpAuthLink) {
|
|||
|
|
log.Error("contract link or up auth link is not http, arg=%v", arg)
|
|||
|
|
err = ecode.RequestErr
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 判断开始时间与结束时间
|
|||
|
|
if up.BeginDate == 0 || up.EndDate < up.BeginDate {
|
|||
|
|
log.Error("begin date is after end date, arg=%v", arg)
|
|||
|
|
err = ecode.MCNUpBindUpDateError
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
db = d.mcndb.Save(up)
|
|||
|
|
affectedRow, err = db.RowsAffected, db.Error
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("save bind up info fail, err=%s, sign=%v", err, sign)
|
|||
|
|
err = ecode.ServerErr
|
|||
|
|
}
|
|||
|
|
result = up
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateBindUp .
|
|||
|
|
func (d *Dao) UpdateBindUp(values map[string]interface{}, query interface{}, args ...interface{}) (affectedRow int64, err error) {
|
|||
|
|
var db = d.mcndb.Table(mcnmodel.TableNameMcnUp).Where(query, args...).Updates(values)
|
|||
|
|
affectedRow, err = db.RowsAffected, db.Error
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("fail to update bind up, err=%s", err)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//UpConfirm up confrim
|
|||
|
|
func (d *Dao) UpConfirm(arg *mcnmodel.McnUpConfirmReq, state model.MCNUPState) (err error) {
|
|||
|
|
var tx = d.mcndb.Begin()
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r != nil || err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
var changeMap = map[string]interface{}{
|
|||
|
|
"state": state,
|
|||
|
|
"state_change_time": time.Now(),
|
|||
|
|
}
|
|||
|
|
if arg.Choice {
|
|||
|
|
changeMap["confirm_time"] = time.Now()
|
|||
|
|
}
|
|||
|
|
err = tx.Table(mcnmodel.TableNameMcnUp).
|
|||
|
|
Where("id=? and up_mid=? and state=?", arg.BindID, arg.UpMid, model.MCNUPStateNoAuthorize).
|
|||
|
|
Updates(changeMap).Error
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("fail to update db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 表示同意
|
|||
|
|
if arg.Choice {
|
|||
|
|
// 驳回其他的绑定请求
|
|||
|
|
err = tx.Table(mcnmodel.TableNameMcnUp).
|
|||
|
|
Where("id !=? and up_mid=? and state=?", arg.BindID, arg.UpMid, model.MCNUPStateNoAuthorize).
|
|||
|
|
Updates(map[string]interface{}{
|
|||
|
|
"state": model.MCNUPStateOnRefuse,
|
|||
|
|
"state_change_time": time.Now(),
|
|||
|
|
}).Error
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("fail to update db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return tx.Commit().Error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetBindInfo .
|
|||
|
|
func (d *Dao) GetBindInfo(arg *mcnmodel.McnUpGetBindReq) (res *mcnmodel.McnGetBindReply, err error) {
|
|||
|
|
var result mcnmodel.McnGetBindReply
|
|||
|
|
err = d.mcndb.Raw(`select s.company_name, s.mcn_mid, u.up_auth_link, u.id as bind_id, u.permission as new_permission
|
|||
|
|
from mcn_up as u inner join mcn_sign as s
|
|||
|
|
on s.id = u.sign_id
|
|||
|
|
where u.id = ? and u.up_mid=? and u.state = 0;`, arg.BindID, arg.UpMid).Find(&result).Error
|
|||
|
|
res = &result
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//GetMcnOldInfo 获取冷却中的信息
|
|||
|
|
func (d *Dao) GetMcnOldInfo(mcnMid int64) (res *mcnmodel.McnSign, err error) {
|
|||
|
|
res = new(mcnmodel.McnSign)
|
|||
|
|
err = d.mcndb.Where("mcn_mid=? and state=?", mcnMid, model.MCNSignStateOnCooling).Order("id desc").Limit(1).Find(res).Error
|
|||
|
|
if err != nil && err != gorm.ErrRecordNotFound {
|
|||
|
|
log.Error("fail to get db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//GetUpPermissionApply 从permission apply表中读取数据
|
|||
|
|
func (d *Dao) GetUpPermissionApply(fields string, query interface{}, args ...interface{}) (res []*mcnmodel.McnUpPermissionApply, err error) {
|
|||
|
|
err = d.mcndb.Select(fields).Where(query, args...).Find(&res).Error
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("fail to get db, err=%v", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetUpPermissionBindInfo .
|
|||
|
|
func (d *Dao) GetUpPermissionBindInfo(arg *mcnmodel.McnUpGetBindReq) (res *mcnmodel.McnGetBindReply, err error) {
|
|||
|
|
var result mcnmodel.McnGetBindReply
|
|||
|
|
err = d.mcndb.Raw(`select s.company_name, u.mcn_mid, u.up_auth_link, u.id as bind_id, u.old_permission, u.new_permission
|
|||
|
|
from mcn_up_permission_apply as u inner join mcn_sign as s
|
|||
|
|
on s.id = u.sign_id
|
|||
|
|
where u.id = ? and u.up_mid=? and u.state = 0;`, arg.BindID, arg.UpMid).Find(&result).Error
|
|||
|
|
res = &result
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//UpPermissionConfirm up confrim
|
|||
|
|
func (d *Dao) UpPermissionConfirm(arg *mcnmodel.McnUpConfirmReq, state adminmodel.MCNUPPermissionState) (err error) {
|
|||
|
|
var db = d.mcndb
|
|||
|
|
var changeMap = map[string]interface{}{
|
|||
|
|
"state": state,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = db.Table(mcnmodel.TableMcnUpPermissionApply).
|
|||
|
|
Where("id=? and up_mid=? and state=?", arg.BindID, arg.UpMid, adminmodel.MCNUPPermissionStateNoAuthorize).
|
|||
|
|
Updates(changeMap).Error
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("fail to update db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|