package com.xmzs.midjourney.controller;
|
|
import cn.hutool.json.JSONUtil;
|
import com.xmzs.common.chat.constant.OpenAIConst;
|
import com.xmzs.common.core.domain.model.LoginUser;
|
import com.xmzs.common.core.exception.base.BaseException;
|
import com.xmzs.common.satoken.utils.LoginHelper;
|
import com.xmzs.midjourney.domain.InsightFace;
|
import com.xmzs.system.domain.bo.ChatMessageBo;
|
import com.xmzs.system.service.IChatMessageService;
|
import com.xmzs.system.service.IChatService;
|
import com.xmzs.system.service.ISseService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import okhttp3.MediaType;
|
import okhttp3.OkHttpClient;
|
import okhttp3.Request;
|
import okhttp3.Response;
|
import okio.Buffer;
|
import okio.BufferedSink;
|
import okio.GzipSink;
|
import okio.Okio;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.io.IOException;
|
import java.util.concurrent.TimeUnit;
|
|
@Api(tags = "任务查询")
|
@RestController
|
@RequestMapping("/mj")
|
@RequiredArgsConstructor
|
@Slf4j
|
public class FaceController {
|
|
@Value("${chat.apiKey}")
|
private String apiKey;
|
@Value("${chat.apiHost}")
|
private String apiHost;
|
|
@Autowired
|
private IChatService chatService;
|
|
@Autowired
|
private ISseService sseService;
|
|
@ApiOperation(value = "换脸")
|
@PostMapping("/insight-face/swap")
|
public String insightFace(@RequestBody InsightFace insightFace) {
|
// 查询是否是付费用户
|
sseService.checkUserGrade();
|
// 扣除接口费用
|
chatService.mjTaskDeduct("换脸", OpenAIConst.MJ_COST_TYPE2);
|
OkHttpClient client = new OkHttpClient.Builder()
|
.connectTimeout(30, TimeUnit.SECONDS) // 连接超时时间
|
.writeTimeout(30, TimeUnit.SECONDS) // 写入超时时间
|
.readTimeout(30, TimeUnit.SECONDS) // 读取超时时间
|
.build();
|
// 创建一个Request对象来配置你的请求
|
// 创建请求体（这里使用JSON作为媒体类型）
|
String jsonStr = JSONUtil.toJsonStr(insightFace);
|
|
MediaType JSON = MediaType.get("application/json; charset=utf-8");
|
okhttp3.RequestBody body = okhttp3.RequestBody.create(jsonStr, JSON);
|
Buffer buffer = new Buffer();
|
GzipSink gzipSink = new GzipSink(buffer);
|
BufferedSink gzipBufferedSink = Okio.buffer(gzipSink);
|
try {
|
body.writeTo(gzipBufferedSink);
|
gzipBufferedSink.close();
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
|
// 创建POST请求
|
Request request = new Request.Builder()
|
.header("mj-api-secret", apiKey)
|
.header("Content-Encoding", "gzip")
|
.url(apiHost + "mj/insight-face/swap") // 替换为你的URL
|
.post(body)
|
.build();
|
|
try (Response response = client.newCall(request).execute()) {
|
if (!response.isSuccessful()) throw new IOException("Unexpected code " + response);
|
if (response.body() != null) {
|
return response.body().string();
|
}
|
} catch (IOException e) {
|
log.error("换脸失败! {}", e.getMessage());
|
}
|
return null;
|
}
|
|
}
|
