Merge remote-tracking branch 'origin/main' into test

This commit is contained in:
Leufolium 2024-05-25 01:17:29 +08:00
commit c44f8d3cab
4 changed files with 42 additions and 4 deletions

View File

@ -51,11 +51,9 @@ func (es *ElasticSearch) getIndexStreamerAcct() string {
func (es *ElasticSearch) CreateStreamerAcct(ctx *gin.Context, streameraccts []*dbstruct.EsStreamerAcct) error { func (es *ElasticSearch) CreateStreamerAcct(ctx *gin.Context, streameraccts []*dbstruct.EsStreamerAcct) error {
bulk := es.clientMix.Bulk().Index(es.getIndexStreamerAcct()).Refresh("true") bulk := es.clientMix.Bulk().Index(es.getIndexStreamerAcct()).Refresh("true")
for _, streameracct := range streameraccts { for _, streameracct := range streameraccts {
bulk.Add(elastic.NewBulkCreateRequest().Doc(streameracct)) bulk.Add(elastic.NewBulkCreateRequest().Id(fmt.Sprint(streameracct.Mid)).Doc(streameracct))
} }
logger.Info("bulk: %v", bulk) _, err := bulk.Do(ctx)
rsp, err := bulk.Do(ctx)
logger.Info("rsp: %v", rsp)
return err return err
} }

View File

@ -703,6 +703,24 @@ func (l *AuthBusinessValidator) EnsureMediaIsSuccessfullyUploaded(media *dbstruc
return l return l
} }
func (l *AuthBusinessValidator) EnsureZoneThirdPartnerIsNotZoneCreater(fun func(*gin.Context, int64) (*dbstruct.Zone, error), zid int64, ztpMid int64) *AuthBusinessValidator {
l.oplist = append(l.oplist, func() {
// 查询空间
zone, err := fun(l.ctx, zid)
if err != nil {
logger.Error("Zone GetById fail, err: %v", err)
l.ec = errcode.ErrCodeZoneSrvFail
return
}
if zone.GetMid() == ztpMid {
l.ec = errcode.ErrCodeZoneThirdPartnerWrongThirdPartner
return
}
})
return l
}
// 执行校验 // 执行校验
func (a *AuthBusinessValidator) Validate() *AuthBusinessValidator { func (a *AuthBusinessValidator) Validate() *AuthBusinessValidator {
a.BusinessValidateStream.Validate() a.BusinessValidateStream.Validate()

View File

@ -1501,3 +1501,21 @@ func (s *Service) OpUpdateZoneMomentBusinessValidate(ctx *gin.Context, req *zone
return return
} }
// 代运营
func (s *Service) OpCreateZoneThirdPartnerBusinessValidate(ctx *gin.Context, req *zone_third_partner_proto.OpCreateReq) (ec errcode.ErrCode) {
ec = errcode.ErrCodeZoneThirdPartnerSrvOk
// 1.鉴权校验
result := businessvalidator.NewAuthBusinessValidator(ctx, req).
EnsureZoneThirdPartnerIsNotZoneCreater(_DefaultZone.GetById, req.ZoneThirdPartner.GetZid(), req.ZoneThirdPartner.GetThirdPartnerMid()).
Validate().
Collect()
ec, _ = result[0].(errcode.ErrCode)
if ec != errcode.ErrCodeOk {
logger.Error("OpCreateZoneThirdPartner business validation failed!")
return
}
return
}

View File

@ -3754,6 +3754,10 @@ func (s *Service) OpSetPrivateZoneMoment(ctx *gin.Context, req *zonemomentproto.
func (s *Service) OpCreateZoneThirdPartner(ctx *gin.Context, req *zone_third_partner_proto.OpCreateReq) (ec errcode.ErrCode) { func (s *Service) OpCreateZoneThirdPartner(ctx *gin.Context, req *zone_third_partner_proto.OpCreateReq) (ec errcode.ErrCode) {
ec = errcode.ErrCodeZoneThirdPartnerSrvOk ec = errcode.ErrCodeZoneThirdPartnerSrvOk
if ec = s.OpCreateZoneThirdPartnerBusinessValidate(ctx, req); ec != errcode.ErrCodeZoneThirdPartnerSrvOk {
return
}
err := _DefaultZoneThirdPartner.OpCreate(ctx, req) err := _DefaultZoneThirdPartner.OpCreate(ctx, req)
if err != nil { if err != nil {
logger.Error("OpCreate fail, req: %v, err: %v", util.ToJson(req), err) logger.Error("OpCreate fail, req: %v, err: %v", util.ToJson(req), err)