package logic import ( "context" "lone-services/pkg/modelbase" "lone-services/pkg/utils" "lone-services/pkg/validate" product "lone-services/rpc/product/pb" "lone-services/services/task/internal/dao" "lone-services/services/task/internal/model" "strconv" "lone-services/rpc/task/pb" "lone-services/services/task/internal/svc" "github.com/zeromicro/go-zero/core/logx" ) type InfoLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InfoLogic { return &InfoLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *InfoLogic) Info(in *task.IdRequest) (*task.Response, error) { var v validate.InfoValidator if fail := l.checkParams(in, &v); fail != nil { return fail, nil } var info dao.TaskInfo modelObj := model.TaskModel{}.Init() w := modelbase.Params{Eq: map[string]string{"id": strconv.Itoa(int(in.Id))}} err := modelObj.GetOne(w, &info) if err != nil { return l.fail(utils.ErrorNotFund) } if info.ProductId < utils.NumberOne { return l.fail(utils.ErrorNotFund) } cli, err := svc.GetRpcClient(l.svcCtx.ProductSvcName) if err != nil { logx.Errorf("get rpc client err: %v", err) return l.fail(utils.ErrorInternalServer) } productClient := product.NewProductClient(cli.Conn()) productInfo, err := productClient.InfoById(context.Background(), &product.InfoByIdReq{Id: info.ProductId}) if err != nil { logx.Errorf("get products err: %v", err) return l.fail(utils.ErrorInternalServer) } if info.Id < utils.NumberOne { return l.fail(utils.ErrorNotFund) } info.ProductName = productInfo.Name return l.ok(info) }