46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
|
|
// Code scaffolded by goctl. Safe to edit.
|
||
|
|
// goctl 1.10.1
|
||
|
|
|
||
|
|
package logic
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"img_infinite_canvas/internal/domain/design"
|
||
|
|
"img_infinite_canvas/internal/svc"
|
||
|
|
"img_infinite_canvas/internal/types"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RunNodeActionAsyncLogic struct {
|
||
|
|
logx.Logger
|
||
|
|
ctx context.Context
|
||
|
|
svcCtx *svc.ServiceContext
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRunNodeActionAsyncLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RunNodeActionAsyncLogic {
|
||
|
|
return &RunNodeActionAsyncLogic{
|
||
|
|
Logger: logx.WithContext(ctx),
|
||
|
|
ctx: ctx,
|
||
|
|
svcCtx: svcCtx,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *RunNodeActionAsyncLogic) RunNodeActionAsync(req *types.NodeActionRequest) (resp *types.ProjectResponse, err error) {
|
||
|
|
project, err := l.svcCtx.DesignService.RunNodeActionAsync(l.ctx, req.Id, req.NodeId, design.NodeActionRequest{
|
||
|
|
Action: req.Action,
|
||
|
|
Prompt: req.Prompt,
|
||
|
|
ImageModel: req.ImageModel,
|
||
|
|
ImageSize: req.ImageSize,
|
||
|
|
Mask: req.Mask,
|
||
|
|
Selection: req.Selection,
|
||
|
|
Options: req.Options,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return toProjectResponse(project), nil
|
||
|
|
}
|