52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
// Code scaffolded by goctl. Safe to edit.
|
|
// goctl 1.10.1
|
|
|
|
package logic
|
|
|
|
import (
|
|
"context"
|
|
|
|
"img_infinite_canvas/internal/domain/design"
|
|
sharingmodule "img_infinite_canvas/internal/modules/sharing"
|
|
"img_infinite_canvas/internal/svc"
|
|
"img_infinite_canvas/internal/types"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type SubmitGeneratorTaskLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewSubmitGeneratorTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SubmitGeneratorTaskLogic {
|
|
return &SubmitGeneratorTaskLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *SubmitGeneratorTaskLogic) SubmitGeneratorTask(req *types.GeneratorTaskSubmitRequest) (resp *types.GeneratorTaskSubmitResponse, err error) {
|
|
if access, ok := sharingmodule.AccessFromContext(l.ctx); ok && req.ProjectId != access.ProjectID {
|
|
return nil, sharingmodule.ErrNotFound
|
|
}
|
|
task, err := l.svcCtx.DesignService.SubmitGeneratorTask(l.ctx, design.GeneratorTaskSubmitRequest{
|
|
CID: req.Cid,
|
|
ProjectID: req.ProjectId,
|
|
GeneratorName: req.GeneratorName,
|
|
InputArgs: toDomainGeneratorTaskInputArgs(req.InputArgs),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &types.GeneratorTaskSubmitResponse{
|
|
Code: 0,
|
|
Message: "success",
|
|
Data: types.GeneratorTaskSubmitData{
|
|
GeneratorTaskId: task.GeneratorTaskID,
|
|
},
|
|
}, nil
|
|
}
|