48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"video-factory/shortdrama/middleware"
|
|
"video-factory/shortdrama/model/dto"
|
|
"video-factory/shortdrama/service"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
type userModelConfig struct{}
|
|
|
|
var UserModelConfig = new(userModelConfig)
|
|
|
|
func (c *userModelConfig) GetUserConfig(ctx context.Context, req *dto.GetUserModelConfigReq) (res *dto.GetUserModelConfigRes, err error) {
|
|
r := g.RequestFromCtx(ctx)
|
|
userId := middleware.GetUserId(r)
|
|
if userId <= 0 {
|
|
return nil, fmt.Errorf("未登录")
|
|
}
|
|
cfg := service.UserModelConfigService.GetUserConfig(ctx, userId, req.ModelType)
|
|
return &dto.GetUserModelConfigRes{UserModelConfig: cfg}, nil
|
|
}
|
|
|
|
func (c *userModelConfig) SaveUserConfig(ctx context.Context, req *dto.SaveUserModelConfigReq) (res *struct{}, err error) {
|
|
r := g.RequestFromCtx(ctx)
|
|
userId := middleware.GetUserId(r)
|
|
if userId <= 0 {
|
|
return nil, fmt.Errorf("未登录")
|
|
}
|
|
return nil, service.UserModelConfigService.SaveUserConfigs(ctx, userId, req.Configs)
|
|
}
|
|
|
|
func (c *userModelConfig) GetUserModelList(ctx context.Context, req *dto.GetUserModelListReq) (res *dto.GetUserModelListRes, err error) {
|
|
r := g.RequestFromCtx(ctx)
|
|
userId := middleware.GetUserId(r)
|
|
if req.Page <= 0 {
|
|
req.Page = 1
|
|
}
|
|
if req.PageSize <= 0 {
|
|
req.PageSize = 20
|
|
}
|
|
return service.UserModelConfigService.GetUserModelListPage(ctx, userId, req.Page, req.PageSize, req.ModelType, req.Keyword), nil
|
|
}
|