biService/rpc/internal/logic/materialsavelogic.go

package logic

import (
	"bp.jydev.jianyu360.cn/BaseService/biService/entity"
	IC "bp.jydev.jianyu360.cn/BaseService/biService/rpc/internal/config"
	"bp.jydev.jianyu360.cn/BaseService/biService/rpc/internal/svc"
	"bp.jydev.jianyu360.cn/BaseService/biService/rpc/pb"
	"bp.jydev.jianyu360.cn/BaseService/biService/service"
	fpb "bp.jydev.jianyu360.cn/BaseService/fileCenter/rpc/pb"
	"context"
	"fmt"
	"github.com/gogf/gf/v2/util/gconv"
	"github.com/zeromicro/go-zero/core/logx"
	"log"
	"path"
	"strings"
)

type MaterialSaveLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewMaterialSaveLogic(ctx context.Context, svcCtx *svc.ServiceContext) *MaterialSaveLogic {
	return &MaterialSaveLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *MaterialSaveLogic) MaterialSave(in *pb.MaterialSaveReq) (*pb.MaterialSaveResp, error) {
	// todo: add your logic here and delete this line
	//先插入消息记录表
	msgId, mid, err := service.MaterialSave(in, IC.IC.MaterialMsg.Title, fmt.Sprintf(IC.IC.MaterialMsg.Content, in.CreateUser))
	if msgId == 0 || err != nil {
		return &pb.MaterialSaveResp{
			ErrorCode: -1,
			ErrorMsg:  err.Error(),
		}, nil
	}

	//获取分发人的userid
	userIdArr := service.GetSendUserId(in.ReceiveUserId, in.EntId)
	log.Println("物料分发给：", userIdArr)
	if in.QrcodeUrl == "" {
		return &pb.MaterialSaveResp{
			ErrorCode:  1,
			ErrorMsg:   "",
			MsgId:      msgId,
			MaterialId: mid,
			UserIdArr:  userIdArr,
		}, nil
	}
	positionArr := strings.Split(in.ReceiveUserId, ",")
	for k, v := range strings.Split(in.ReceiveUserName, ",") {
		if len(strings.Split(in.FileUrl, ",")) == 0 {
			break
		}
		personCode := service.GetPersonCode(gconv.Int64(positionArr[k]))
		for _, val := range strings.Split(in.FileUrl, ",") {
			go func(fileUrl, name string, positionId int64) {
				imgByte, err := service.PersonImgSaveComposite(fileUrl, in.QrcodeUrl, name, personCode, positionId)
				log.Println(imgByte)
				if err != nil {
					log.Println("合成图片出错：", err)
					return
				}
				log.Println("图片名称", path.Base(fileUrl))
				up, err := entity.FileCenterRpc.Upload(context.Background(), &fpb.UploadReq{
					File:           imgByte,
					OssBucketName:  entity.OssBucketName,
					OssUrl:         entity.OssUrl,
					Name:           path.Base(fileUrl),
					NeedEncryption: false,
				})
				//log.Println(err)
				if up == nil || up.Url == "" {
					log.Println("上传合成图片失败：", err)
					return
				}
				ok := service.PersonImageSave(up.Url, msgId, positionId, mid)
				if !ok {
					log.Println("合成图片存库失败")
					return
				}
			}(val, v, gconv.Int64(positionArr[k]))
		}
	}

	return &pb.MaterialSaveResp{
		ErrorCode:  1,
		ErrorMsg:   "",
		MsgId:      msgId,
		MaterialId: mid,
		UserIdArr:  userIdArr,
	}, nil

}