lone-services/services/analysis/internal/logic/analysis/modeImageLogic.go

package analysislogic

import (
	"context"
	"strconv"

	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	"lone-services/pkg/validate"
	analysis "lone-services/rpc/analysis/pb"
	"lone-services/services/analysis/internal/dao"
	"lone-services/services/analysis/internal/logic/common"
	"lone-services/services/analysis/internal/model"
	"lone-services/services/analysis/internal/svc"
	"lone-services/services/analysis/validator"

	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *ModeImageLogic) ModeImage(in *analysis.TaskIdReq) (*analysis.Response, error) {
	var req validator.TaskIdValidator
	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
		return outParams(msg), nil
	}
	if _, fail := authorizeTask(l.ctx, req.TaskId); fail != nil {
		return fail, nil
	}

	var row dao.Image
	err := model.ImageModel{}.Init().GetOne(modelbase.Params{
		Eq: map[string]string{
			"task_id": req.TaskId,
			"status":  strconv.Itoa(utils.StatusOk),
		},
		Order: "id desc",
	}, &row)
	if err != nil {
		l.Errorf("mode image %s: %v", req.TaskId, err)
		return common.FailResponse(utils.Fail), nil
	}
	images := map[string]string{}
	if row.Id > utils.NumberZero && row.Images != utils.StringEmpty {
		images = taskModeImages(row.Images)
	}

	data := make([]map[string]string, 0, len(skinModeList))
	for _, item := range skinModeList {
		data = append(data, map[string]string{
			"name":    item.name,
			"english": item.english,
			"chinese": item.chinese,
			"image":   images[item.name],
		})
	}
	return common.OkResponse(data), nil
}