27 lines
718 B
Go
27 lines
718 B
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"video-factory/shortdrama/model/dto"
|
|
"video-factory/shortdrama/service"
|
|
)
|
|
|
|
type prompt struct{}
|
|
|
|
var Prompt = new(prompt)
|
|
|
|
func (c *prompt) List(ctx context.Context, req *dto.ListPromptReq) (res *dto.ListPromptRes, err error) {
|
|
list := service.PromptService.GetList(ctx)
|
|
return &dto.ListPromptRes{List: list}, nil
|
|
}
|
|
|
|
func (c *prompt) Get(ctx context.Context, req *dto.GetPromptReq) (res *dto.GetPromptRes, err error) {
|
|
p := service.PromptService.GetByName(ctx, req.Name)
|
|
return &dto.GetPromptRes{Prompt: p}, nil
|
|
}
|
|
|
|
func (c *prompt) Save(ctx context.Context, req *dto.SavePromptReq) (res *struct{}, err error) {
|
|
return nil, service.PromptService.Save(ctx, &req.Prompt)
|
|
}
|