|
@@ -5,7 +5,7 @@ import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
|
|
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
|
|
|
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
|
|
|
import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImagePageReqVO;
|
|
|
-import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImagePublicListReqVO;
|
|
|
+import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImagePublicPageReqVO;
|
|
|
import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO;
|
|
|
import org.apache.ibatis.annotations.Mapper;
|
|
|
|
|
@@ -39,15 +39,16 @@ public interface AiImageMapper extends BaseMapperX<AiImageDO> {
|
|
|
.orderByDesc(AiImageDO::getId));
|
|
|
}
|
|
|
|
|
|
+ default PageResult<AiImageDO> selectPage(AiImagePublicPageReqVO pageReqVO) {
|
|
|
+ return selectPage(pageReqVO, new LambdaQueryWrapperX<AiImageDO>()
|
|
|
+ .eqIfPresent(AiImageDO::getPublicStatus, Boolean.TRUE)
|
|
|
+ .likeIfPresent(AiImageDO::getPrompt, pageReqVO.getPrompt())
|
|
|
+ .orderByDesc(AiImageDO::getId));
|
|
|
+ }
|
|
|
+
|
|
|
default List<AiImageDO> selectListByStatusAndPlatform(Integer status, String platform) {
|
|
|
return selectList(AiImageDO::getStatus, status,
|
|
|
AiImageDO::getPlatform, platform);
|
|
|
}
|
|
|
|
|
|
- default PageResult<AiImageDO> selectPageOfPublicList(AiImagePublicListReqVO releaseListReqVO) {
|
|
|
- return selectPage(releaseListReqVO, new LambdaQueryWrapperX<AiImageDO>()
|
|
|
- .eqIfPresent(AiImageDO::getPublicStatus, Boolean.TRUE)
|
|
|
- .eqIfPresent(AiImageDO::getPrompt, releaseListReqVO.getPrompt())
|
|
|
- .orderByDesc(AiImageDO::getId));
|
|
|
- }
|
|
|
}
|