diff --git a/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/AliYunAiApi.java b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/AliYunAiApi.java
new file mode 100644
index 0000000..04b3084
--- /dev/null
+++ b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/AliYunAiApi.java
@@ -0,0 +1,87 @@
+package edu.whut.smilepicturebackend.api.aliyunai;
+
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
+import cn.hutool.json.JSONUtil;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskRequest;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskResponse;
+import edu.whut.smilepicturebackend.api.aliyunai.model.GetOutPaintingTaskResponse;
+import edu.whut.smilepicturebackend.exception.BusinessException;
+import edu.whut.smilepicturebackend.exception.ErrorCode;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+@Slf4j
+@Component
+public class AliYunAiApi {
+
+ // 读取配置文件
+ @Value("${smile-picture.aliyun.apiKey}")
+ private String apiKey;
+
+ // 创建任务地址
+ public static final String CREATE_OUT_PAINTING_TASK_URL = "https://dashscope.aliyuncs.com/api/v1/services/aigc/image2image/out-painting";
+
+ // 查询任务状态
+ public static final String GET_OUT_PAINTING_TASK_URL = "https://dashscope.aliyuncs.com/api/v1/tasks/%s";
+
+
+ /**
+ * 创建任务
+ *
+ * @param createOutPaintingTaskRequest
+ * @return
+ */
+ public CreateOutPaintingTaskResponse createOutPaintingTask(CreateOutPaintingTaskRequest createOutPaintingTaskRequest) {
+ if (createOutPaintingTaskRequest == null) {
+ throw new BusinessException(ErrorCode.OPERATION_ERROR, "扩图参数为空");
+ }
+ // 发送请求
+ HttpRequest httpRequest = HttpRequest.post(CREATE_OUT_PAINTING_TASK_URL)
+ .header("Authorization", "Bearer " + apiKey)
+ // 必须开启异步处理 enable
+ .header("X-DashScope-Async", "enable")
+ .header("Content-Type", "application/json")
+ .body(JSONUtil.toJsonStr(createOutPaintingTaskRequest));
+ // 处理响应
+ //这段代码会在 try 块结束时（正常返回或抛出异常）自动调用 httpResponse.close()，从而释放所有底层资源
+ try (HttpResponse httpResponse = httpRequest.execute()) {
+ if (!httpResponse.isOk()) {
+ log.error("请求异常：{}", httpResponse.body());
+ throw new BusinessException(ErrorCode.OPERATION_ERROR, "AI 扩图失败");
+ }
+ CreateOutPaintingTaskResponse createOutPaintingTaskResponse = JSONUtil.toBean(httpResponse.body(), CreateOutPaintingTaskResponse.class);
+ if (createOutPaintingTaskResponse.getCode() != null) {
+ String errorMessage = createOutPaintingTaskResponse.getMessage();
+ log.error("请求异常：{}", errorMessage);
+ throw new BusinessException(ErrorCode.OPERATION_ERROR, "AI 扩图失败，" + errorMessage);
+ }
+ return createOutPaintingTaskResponse;
+ }
+ }
+
+ /**
+ * 查询创建的任务结果
+ *
+ * @param taskId
+ * @return
+ */
+ public GetOutPaintingTaskResponse getOutPaintingTask(String taskId) {
+ if (StrUtil.isBlank(taskId)) {
+ throw new BusinessException(ErrorCode.OPERATION_ERROR, "任务 ID 不能为空");
+ }
+ // 处理响应
+ String url = String.format(GET_OUT_PAINTING_TASK_URL, taskId);
+ try (HttpResponse httpResponse = HttpRequest.get(url)
+ .header("Authorization", "Bearer " + apiKey)
+ .execute()) {
+ if (!httpResponse.isOk()) {
+ log.error("请求异常：{}", httpResponse.body());
+ throw new BusinessException(ErrorCode.OPERATION_ERROR, "获取任务结果失败");
+ }
+ return JSONUtil.toBean(httpResponse.body(), GetOutPaintingTaskResponse.class);
+ }
+ }
+}
diff --git a/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/CreateOutPaintingTaskRequest.java b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/CreateOutPaintingTaskRequest.java
new file mode 100644
index 0000000..b39fa33
--- /dev/null
+++ b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/CreateOutPaintingTaskRequest.java
@@ -0,0 +1,112 @@
+package edu.whut.smilepicturebackend.api.aliyunai.model;
+
+import cn.hutool.core.annotation.Alias;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.io.Serializable;
+
+/**
+ * 创建扩图任务请求
+ */
+@Data
+public class CreateOutPaintingTaskRequest implements Serializable {
+
+ /**
+ * 模型，例如 "image-out-painting"
+ */
+ private String model = "image-out-painting";
+
+ /**
+ * 输入图像信息
+ */
+ private Input input;
+
+ /**
+ * 图像处理参数
+ */
+ private Parameters parameters;
+
+ @Data
+ public static class Input {
+ /**
+ * 必选，图像 URL
+ */
+ @Alias("image_url") //java类中一般写驼峰式的，但是大模型接收的是image_url 所以用@Alias
+ private String imageUrl;
+ }
+
+ @Data
+ public static class Parameters implements Serializable {
+ /**
+ * 可选，逆时针旋转角度，默认值 0，取值范围 [0, 359]
+ */
+ private Integer angle;
+
+ /**
+ * 可选，输出图像的宽高比，默认空字符串，不设置宽高比
+ * 可选值：["", "1:1", "3:4", "4:3", "9:16", "16:9"]
+ */
+ @Alias("output_ratio")
+ private String outputRatio;
+
+ /**
+ * 可选，图像居中，在水平方向上按比例扩展，默认值 1.0，范围 [1.0, 3.0]
+ */
+ @Alias("x_scale")
+ @JsonProperty("xScale")
+ private Float xScale;
+
+ /**
+ * 可选，图像居中，在垂直方向上按比例扩展，默认值 1.0，范围 [1.0, 3.0]
+ */
+ @Alias("y_scale")
+ @JsonProperty("yScale")
+ private Float yScale;
+
+ /**
+ * 可选，在图像上方添加像素，默认值 0
+ */
+ @Alias("top_offset")
+ private Integer topOffset;
+
+ /**
+ * 可选，在图像下方添加像素，默认值 0
+ */
+ @Alias("bottom_offset")
+ private Integer bottomOffset;
+
+ /**
+ * 可选，在图像左侧添加像素，默认值 0
+ */
+ @Alias("left_offset")
+ private Integer leftOffset;
+
+ /**
+ * 可选，在图像右侧添加像素，默认值 0
+ */
+ @Alias("right_offset")
+ private Integer rightOffset;
+
+ /**
+ * 可选，开启图像最佳质量模式，默认值 false
+ * 若为 true，耗时会成倍增加
+ */
+ @Alias("best_quality")
+ private Boolean bestQuality;
+
+ /**
+ * 可选，限制模型生成的图像文件大小，默认值 true
+ * - 单边长度 <= 10000：输出图像文件大小限制为 5MB 以下
+ * - 单边长度 > 10000：输出图像文件大小限制为 10MB 以下
+ */
+ @Alias("limit_image_size")
+ private Boolean limitImageSize;
+
+ /**
+ * 可选，添加 "Generated by AI" 水印，默认值 true
+ */
+ @Alias("add_watermark")
+ private Boolean addWatermark = false;
+ }
+}
\ No newline at end of file
diff --git a/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/CreateOutPaintingTaskResponse.java b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/CreateOutPaintingTaskResponse.java
new file mode 100644
index 0000000..40a9607
--- /dev/null
+++ b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/CreateOutPaintingTaskResponse.java
@@ -0,0 +1,60 @@
+package edu.whut.smilepicturebackend.api.aliyunai.model;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/**
+ * 创建扩图任务响应类
+ */
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+public class CreateOutPaintingTaskResponse {
+
+ private Output output;
+
+ /**
+ * 表示任务的输出信息
+ */
+ @Data
+ public static class Output {
+
+ /**
+ * 任务 ID
+ */
+ private String taskId;
+
+ /**
+ * 任务状态
+ *
+ * - PENDING：排队中
+ * - RUNNING：处理中
+ * - SUSPENDED：挂起
+ * - SUCCEEDED：执行成功
+ * - FAILED：执行失败
+ * - UNKNOWN：任务不存在或状态未知
+ *
+ */
+ private String taskStatus;
+ }
+
+ /**
+ * 接口错误码。
+ * 接口成功请求不会返回该参数。
+ */
+ private String code;
+
+ /**
+ * 接口错误信息。
+ * 接口成功请求不会返回该参数。
+ */
+ private String message;
+
+ /**
+ * 请求唯一标识。
+ * 可用于请求明细溯源和问题排查。
+ */
+ private String requestId;
+
+}
\ No newline at end of file
diff --git a/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/GetOutPaintingTaskResponse.java b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/GetOutPaintingTaskResponse.java
new file mode 100644
index 0000000..58ee6aa
--- /dev/null
+++ b/src/main/java/edu/whut/smilepicturebackend/api/aliyunai/model/GetOutPaintingTaskResponse.java
@@ -0,0 +1,111 @@
+package edu.whut.smilepicturebackend.api.aliyunai.model;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/**
+ * 查询扩图任务响应类
+ */
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+public class GetOutPaintingTaskResponse {
+
+ /**
+ * 请求唯一标识
+ */
+ private String requestId;
+
+ /**
+ * 输出信息
+ */
+ private Output output;
+
+ /**
+ * 表示任务的输出信息
+ */
+ @Data
+ public static class Output {
+
+ /**
+ * 任务 ID
+ */
+ private String taskId;
+
+ /**
+ * 任务状态
+ *
+ * - PENDING：排队中
+ * - RUNNING：处理中
+ * - SUSPENDED：挂起
+ * - SUCCEEDED：执行成功
+ * - FAILED：执行失败
+ * - UNKNOWN：任务不存在或状态未知
+ *
+ */
+ private String taskStatus;
+
+ /**
+ * 提交时间
+ * 格式：YYYY-MM-DD HH:mm:ss.SSS
+ */
+ private String submitTime;
+
+ /**
+ * 调度时间
+ * 格式：YYYY-MM-DD HH:mm:ss.SSS
+ */
+ private String scheduledTime;
+
+ /**
+ * 结束时间
+ * 格式：YYYY-MM-DD HH:mm:ss.SSS
+ */
+ private String endTime;
+
+ /**
+ * 输出图像的 URL
+ */
+ private String outputImageUrl;
+
+ /**
+ * 接口错误码
+ * 接口成功请求不会返回该参数
+ */
+ private String code;
+
+ /**
+ * 接口错误信息
+ * 接口成功请求不会返回该参数
+ */
+ private String message;
+
+ /**
+ * 任务指标信息
+ */
+ private TaskMetrics taskMetrics;
+ }
+
+ /**
+ * 表示任务的统计信息
+ */
+ @Data
+ public static class TaskMetrics {
+
+ /**
+ * 总任务数
+ */
+ private Integer total;
+
+ /**
+ * 成功任务数
+ */
+ private Integer succeeded;
+
+ /**
+ * 失败任务数
+ */
+ private Integer failed;
+ }
+}
\ No newline at end of file
diff --git a/src/main/java/edu/whut/smilepicturebackend/controller/PictureController.java b/src/main/java/edu/whut/smilepicturebackend/controller/PictureController.java
index 39c60e1..c6e0ec9 100644
--- a/src/main/java/edu/whut/smilepicturebackend/controller/PictureController.java
+++ b/src/main/java/edu/whut/smilepicturebackend/controller/PictureController.java
@@ -1,7 +1,11 @@
package edu.whut.smilepicturebackend.controller;
+import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import edu.whut.smilepicturebackend.annotation.AuthCheck;
+import edu.whut.smilepicturebackend.api.aliyunai.AliYunAiApi;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskResponse;
+import edu.whut.smilepicturebackend.api.aliyunai.model.GetOutPaintingTaskResponse;
import edu.whut.smilepicturebackend.api.imagesearch.ImageSearchApiFacade;
import edu.whut.smilepicturebackend.api.imagesearch.model.ImageSearchResult;
import edu.whut.smilepicturebackend.common.BaseResponse;
@@ -24,7 +28,6 @@ import edu.whut.smilepicturebackend.service.UserService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
-import org.springframework.util.DigestUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
@@ -42,6 +45,7 @@ public class PictureController {
private final UserService userService;
private final PictureService pictureService;
private final SpaceService spaceService;
+ private final AliYunAiApi aliYunAiApi;
@@ -305,4 +309,32 @@ public class PictureController {
pictureService.editPictureByBatch(pictureEditByBatchRequest, loginUser);
return ResultUtils.success(true);
}
+
+ /**
+ *创建 AI 扩图任务
+ * @param createPictureOutPaintingTaskRequest
+ * @param request
+ * @return
+ */
+ @PostMapping("/out_painting/create_task")
+// @SaSpaceCheckPermission(value = SpaceUserPermissionConstant.PICTURE_EDIT)
+ public BaseResponse createPictureOutPaintingTask(@RequestBody CreatePictureOutPaintingTaskRequest createPictureOutPaintingTaskRequest,
+ HttpServletRequest request) {
+ if (createPictureOutPaintingTaskRequest == null || createPictureOutPaintingTaskRequest.getPictureId() == null) {
+ throw new BusinessException(ErrorCode.PARAMS_ERROR);
+ }
+ User loginUser = userService.getLoginUser(request);
+ CreateOutPaintingTaskResponse response = pictureService.createPictureOutPaintingTask(createPictureOutPaintingTaskRequest, loginUser);
+ return ResultUtils.success(response);
+ }
+
+ /**
+ * 查询 AI 扩图任务 ，前端轮询结果，当结果中有SUCCESS，那么同时取出该响应中的结果。
+ */
+ @GetMapping("/out_painting/get_task")
+ public BaseResponse getPictureOutPaintingTask(String taskId) {
+ ThrowUtils.throwIf(StrUtil.isBlank(taskId), ErrorCode.PARAMS_ERROR);
+ GetOutPaintingTaskResponse task = aliYunAiApi.getOutPaintingTask(taskId);
+ return ResultUtils.success(task);
+ }
}
diff --git a/src/main/java/edu/whut/smilepicturebackend/manager/upload/PictureUploadTemplate.java b/src/main/java/edu/whut/smilepicturebackend/manager/upload/PictureUploadTemplate.java
index 5e65d35..1c1c6c9 100644
--- a/src/main/java/edu/whut/smilepicturebackend/manager/upload/PictureUploadTemplate.java
+++ b/src/main/java/edu/whut/smilepicturebackend/manager/upload/PictureUploadTemplate.java
@@ -58,6 +58,7 @@ public abstract class PictureUploadTemplate {
String uploadPath = String.format("/%s/%s/%s",projectName, uploadPathPrefix, uploadFilename);
File file = null;
try {
+ log.info("uploadPath"+uploadPath);
// 3. 创建临时文件，获取文件到服务器
file = File.createTempFile(uploadPath, null);
// 处理文件来源
diff --git a/src/main/java/edu/whut/smilepicturebackend/manager/upload/UrlPictureUpload.java b/src/main/java/edu/whut/smilepicturebackend/manager/upload/UrlPictureUpload.java
index 5157ddd..a580f6a 100644
--- a/src/main/java/edu/whut/smilepicturebackend/manager/upload/UrlPictureUpload.java
+++ b/src/main/java/edu/whut/smilepicturebackend/manager/upload/UrlPictureUpload.java
@@ -9,6 +9,7 @@ import cn.hutool.http.Method;
import edu.whut.smilepicturebackend.exception.BusinessException;
import edu.whut.smilepicturebackend.exception.ErrorCode;
import edu.whut.smilepicturebackend.exception.ThrowUtils;
+import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import java.io.File;
@@ -21,6 +22,7 @@ import java.util.List;
* URL 图片上传
*/
@Service
+@Slf4j
public class UrlPictureUpload extends PictureUploadTemplate {
@Override
@@ -46,14 +48,15 @@ public class UrlPictureUpload extends PictureUploadTemplate {
* @return 不含点的扩展名，比如 "jpg","png"，取不出时返回空串
*/
protected String fetchAndValidateExtension(String fileUrl) {
+ log.info("收到的fileurl:{}",fileUrl);
try (HttpResponse resp = HttpUtil
.createRequest(Method.HEAD, fileUrl)
.execute()) {
if (resp.getStatus() != HttpStatus.HTTP_OK) {
- throw new BusinessException(
- ErrorCode.OPERATION_ERROR, "文件不存在或不可访问");
+ // 网络或权限问题时直接降级
+ log.warn("HEAD 请求未返回 200，status = {}，将从 URL 中提取后缀", resp.getStatus());
+ return "";
}
-
// 1) Content-Type 验证 & 提取扩展名
String ct = resp.header("Content-Type");
String ext = "";
@@ -95,16 +98,21 @@ public class UrlPictureUpload extends PictureUploadTemplate {
@Override
protected String getOriginFilename(Object inputSource) {
String fileUrl = (String) inputSource;
- // 1) HEAD 验证并拿扩展名（只这一处会发 HEAD）
- String ext = fetchAndValidateExtension(fileUrl);
- // 2) fallback：若服务器没返回类型，再从 URL 中简单截取
+ // 先把 query string 去掉
+ int qIdx = fileUrl.indexOf('?');
+ String cleanUrl = qIdx > 0 ? fileUrl.substring(0, qIdx) : fileUrl;
+
+ // 1) 尝试 HEAD 拿扩展名，不可用时 ext == ""
+ String ext = fetchAndValidateExtension(cleanUrl);
+
+ // 2) fallback：若 ext 为空，就直接返回去掉参数后的 URL
if (StrUtil.isBlank(ext)) {
- ext = FileUtil.extName(fileUrl);
+ return cleanUrl;
+ } else {
+ // 3) 从 cleanUrl 中拿到 baseName，再拼回去
+ String base = FileUtil.mainName(FileUtil.getName(cleanUrl));
+ return base + "." + ext;
}
- ThrowUtils.throwIf(ext==null,ErrorCode.PARAMS_ERROR,"不正确的图片格式");
- // 3) 拿到 baseName，然后拼回去
- String base = FileUtil.mainName(FileUtil.getName(fileUrl));
- return StrUtil.isNotBlank(ext) ? base + "." + ext : base;
}
@Override
diff --git a/src/main/java/edu/whut/smilepicturebackend/model/dto/picture/CreatePictureOutPaintingTaskRequest.java b/src/main/java/edu/whut/smilepicturebackend/model/dto/picture/CreatePictureOutPaintingTaskRequest.java
new file mode 100644
index 0000000..6d2ae6f
--- /dev/null
+++ b/src/main/java/edu/whut/smilepicturebackend/model/dto/picture/CreatePictureOutPaintingTaskRequest.java
@@ -0,0 +1,24 @@
+package edu.whut.smilepicturebackend.model.dto.picture;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskRequest;
+import lombok.Data;
+
+import java.io.Serializable;
+
+/**
+ * 创建扩图任务请求
+ */
+@Data
+public class CreatePictureOutPaintingTaskRequest implements Serializable {
+
+ /**
+ * 图片 id
+ */
+ private Long pictureId;
+
+ /**
+ * 扩图参数
+ */
+ private CreateOutPaintingTaskRequest.Parameters parameters;
+
+ private static final long serialVersionUID = 1L;
+}
\ No newline at end of file
diff --git a/src/main/java/edu/whut/smilepicturebackend/service/PictureService.java b/src/main/java/edu/whut/smilepicturebackend/service/PictureService.java
index 18909bc..e1e440d 100644
--- a/src/main/java/edu/whut/smilepicturebackend/service/PictureService.java
+++ b/src/main/java/edu/whut/smilepicturebackend/service/PictureService.java
@@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.IService;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskResponse;
import edu.whut.smilepicturebackend.api.imagesearch.model.ImageSearchResult;
import edu.whut.smilepicturebackend.model.dto.picture.*;
import edu.whut.smilepicturebackend.model.entity.Picture;
@@ -144,4 +145,11 @@ public interface PictureService extends IService {
*/
void editPictureByBatch(PictureEditByBatchRequest pictureEditByBatchRequest, User loginUser);
+ /**
+ * 创建扩图任务
+ *
+ * @param createPictureOutPaintingTaskRequest
+ * @param loginUser
+ */
+ CreateOutPaintingTaskResponse createPictureOutPaintingTask(CreatePictureOutPaintingTaskRequest createPictureOutPaintingTaskRequest, User loginUser);
}
diff --git a/src/main/java/edu/whut/smilepicturebackend/service/impl/PictureServiceImpl.java b/src/main/java/edu/whut/smilepicturebackend/service/impl/PictureServiceImpl.java
index e414f4b..8ab359d 100644
--- a/src/main/java/edu/whut/smilepicturebackend/service/impl/PictureServiceImpl.java
+++ b/src/main/java/edu/whut/smilepicturebackend/service/impl/PictureServiceImpl.java
@@ -12,6 +12,9 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import edu.whut.smilepicturebackend.api.aliyunai.AliYunAiApi;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskRequest;
+import edu.whut.smilepicturebackend.api.aliyunai.model.CreateOutPaintingTaskResponse;
import edu.whut.smilepicturebackend.api.imagesearch.ImageSearchApiFacade;
import edu.whut.smilepicturebackend.api.imagesearch.model.ImageSearchResult;
import edu.whut.smilepicturebackend.exception.BusinessException;
@@ -76,6 +79,8 @@ public class PictureServiceImpl extends ServiceImpl
private final CosManager cosManager;
private final SpaceService spaceService;
private final TransactionTemplate transactionTemplate;
+ private final AliYunAiApi aliYunAiApi;
+
@Override
public void validPicture(Picture picture) {
ThrowUtils.throwIf(picture == null, ErrorCode.PARAMS_ERROR);
@@ -154,6 +159,7 @@ public class PictureServiceImpl extends ServiceImpl
// 根据 inputSource 的类型区分上传方式!!
PictureUploadTemplate pictureUploadTemplate = filePictureUpload;
if (inputSource instanceof String) {
+ log.info("收到 upload/url 请求，url = {}", inputSource);
pictureUploadTemplate = urlPictureUpload;
}
//上传到腾讯云COS上
@@ -680,6 +686,26 @@ public class PictureServiceImpl extends ServiceImpl
ThrowUtils.throwIf(!result, ErrorCode.OPERATION_ERROR, "批量编辑失败");
}
+ @Override
+ public CreateOutPaintingTaskResponse createPictureOutPaintingTask(CreatePictureOutPaintingTaskRequest createPictureOutPaintingTaskRequest, User loginUser) {
+ // 获取图片信息
+ Long pictureId = createPictureOutPaintingTaskRequest.getPictureId();
+ Picture picture = this.getById(pictureId);
+ if (picture == null) {
+ throw new BusinessException(ErrorCode.NOT_FOUND_ERROR, "图片不存在");
+ }
+ // 校验权限，已经改为使用注解鉴权
+// checkPictureAuth(loginUser, picture);
+ // 创建扩图任务
+ CreateOutPaintingTaskRequest createOutPaintingTaskRequest = new CreateOutPaintingTaskRequest();
+ CreateOutPaintingTaskRequest.Input input = new CreateOutPaintingTaskRequest.Input();
+ input.setImageUrl(picture.getUrl());
+ createOutPaintingTaskRequest.setInput(input);
+ createOutPaintingTaskRequest.setParameters(createPictureOutPaintingTaskRequest.getParameters());
+ // 创建任务
+ return aliYunAiApi.createOutPaintingTask(createOutPaintingTaskRequest);
+ }
+
/**
* nameRule 格式：图片-{序号} =>图片-1 图片-2 ...
*
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 2998472..63ba4f1 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -68,4 +68,8 @@ cos:
secretId: ${smile-picture.cos.client.secretId}
secretKey: ${smile-picture.cos.client.secretKey}
region: ${smile-picture.cos.client.region}
- bucket: ${smile-picture.cos.client.bucket}
\ No newline at end of file
+ bucket: ${smile-picture.cos.client.bucket}
+
+smile-picture:
+ aliyun:
+ apiKey: ${smile-picture.aliyun.apiKey}
\ No newline at end of file