用途介绍
Midjourney是一个目前优秀的AI画图工具，不挂梯无法直接访问
本代码主要用于搭建镜像站使用
适合人群
本代码不适合新手，建议使用过okhttp、且具有二开能力的同学使用~
实现原理
通过调用发送信息接口发送请求，通过轮询房间消息接口判断是否作图完成
发送的时候带上我们存储好的cookie信息即可
轮询房间消息接口是为了避免模拟网页实际的websocket连接，那玩意解密没解出来...
准备
1. 开通Midjourney会员的Discord账号
2. 新建很多房间（因为房间消息接口默认就50条数据，也就是说每个房间最多并行50个图片生成任务）
功能预览
上代码
代码功能：
1. 调用Midjourney生成并将生成好的内容发送微信消息给用户（这个是在微信公众号平台做的，大家做网页版的话自己改下输出到网页就可以）
2. 进行合规性检测（生成点不得了的东西后果你懂得，这里用的是百度的文本审核服务，5万次以下免费）
3. 如果用户输入是中文，就翻译为英文再发送
midjourneyLog
midjourney_log表字段如下代码所示
需要使用通用mapper哦~
package com.example.midjourney.bean.pojo;import lombok.Data;
import tk.mybatis.mapper.annotation.NameStyle;
import tk.mybatis.mapper.code.Style;import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;
import javax.persistence.Table;
import java.util.Date;@Data
@NameStyle(Style.camelhump)
@Table(name = "midjourney_log")
public class MidjourneyLog {@Id@GeneratedValue(strategy = GenerationType.IDENTITY)private Long id;private Integer memberId;private String channel;private String prompt;private Integer type;private Integer status;private String imgFile;private Integer roomId;private String uuid;private Date createTime;}
MidJourneyBiz
请求唯一id没啥规律，一段时间内不一致就行，试了没啥校验
另外请求头除了cookie需要保证和 authorization、x-super-properties对应上，三者对应上实测2个月都不会掉线！
package com.example.midjourney.biz;import cn.hutool.core.img.ImgUtil;
import cn.hutool.core.io.FileUtil;
import com.alibaba.fastjson.JSON;
import com.example.midjourney.bean.BaiduTextCensor;
import com.example.midjourney.bean.BaiduTextCensorData;
import com.example.midjourney.bean.MidMsg;
import com.example.midjourney.bean.pojo.Discord;
import com.example.midjourney.bean.pojo.MidjourneyLog;
import com.example.midjourney.bean.pojo.RoomInfo;
import com.example.midjourney.contant.Constant;
import com.example.midjourney.enums.MedjourneyLogType;
import com.example.midjourney.service.DiscordService;
import com.example.midjourney.service.MidjourneyLogService;
import com.example.midjourney.service.MemberService;
import com.example.midjourney.service.RoomInfoService;
import com.example.midjourney.util.*;
import com.google.common.base.Joiner;
import com.google.common.base.Splitter;
import com.google.common.collect.Lists;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import okhttp3.*;
import org.apache.logging.log4j.util.Strings;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;import javax.annotation.Resource;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.math.BigDecimal;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.UUID;
import java.util.stream.Collectors;import static com.example.midjourney.contant.Constant.*;/*** 启动就运行，不停刷新消息列表*/
@Slf4j
@Service
public class MidJourneyBiz {private static BigDecimal bigDecimal = new BigDecimal("1099765990370980513").add(new BigDecimal(System.currentTimeMillis()));@Resourceprivate WeChatBiz weChatBiz;@Resourceprivate TranslateBiz translateBiz;@Resourceprivate MidjourneyLogService midjourneyLogService;@Resourceprivate MemberService memberService;@Resourceprivate RoomInfoService roomInfoService;@Resourceprivate DiscordService discordService;public void buildImg(Long id) {MidjourneyLog midjourneyLog = midjourneyLogService.findById(id);try {if (midjourneyLog.getStatus() != 0) {return;}if (isFailMsg(midjourneyLog.getChannel(), midjourneyLog)) {log.info("[失败提示] 消息校验不通过 log:{}", midjourneyLog);sendBuildFail(midjourneyLog);return;}String prompt = midjourneyLog.getPrompt();if (midjourneyLog.getType() == MedjourneyLogType.BIG_IMG.getCode()) {downImg(midjourneyLog);return;}BaiduTextCensor censor = null;if (TextUtil.isHaveChinese(prompt)) {censor = BaiduUtil.textCensor(prompt);}prompt = cleanMsg(prompt);if (isBlackWord(prompt)) {sendSensitive(midjourneyLog);return;}if (Objects.isNull(censor)) {censor = BaiduUtil.textCensor(prompt);}midjourneyLogService.updatePrompt(midjourneyLog, prompt);if (censor.getConclusionType() == 2 || censor.getConclusionType() == 3) {sendSensitive(midjourneyLog, censor);return;}if (midjourneyLog.getType() == MedjourneyLogType.ITERATIVE.getCode()) {//暂不支持sendBuildFail(midjourneyLog);return;}RoomInfo roomInfo = roomInfoService.findIdleRoom();if (Objects.isNull(roomInfo)) {log.error("[并发超出警报] 当前并发次数已经无法满足！！！");sendBuildFail(midjourneyLog);return;}if (sendMsg(prompt, roomInfo)) {midjourneyLogService.updateRoom(midjourneyLog, roomInfo);} else {log.error("[发送失败] 发送信息失败，请检查");sendBuildFail(midjourneyLog);}} catch (Throwable t) {sendBuildFail(midjourneyLog);}}@SneakyThrowsprivate void downImg(MidjourneyLog midjourneyLog) {MidjourneyLog lastLog = midjourneyLogService.findLastNormalLog(midjourneyLog.getMemberId());log.info("[下载图片] {}", lastLog);if (Objects.isNull(lastLog)) {log.info("[失败提示] 找不到上一次的图片 log:{}", midjourneyLog);sendBuildFail(midjourneyLog);return;}String imgFile = lastLog.getImgFile();File file = new File(imgFile);if (!file.exists()) {log.info("[失败提示] 上一次图片不存在 log:{}", lastLog);sendBuildFail(midjourneyLog);return;}cutAndSendImg(midjourneyLog, file);}private void cutAndSendImg(MidjourneyLog midjourneyLog, File oldFile) throws IOException {BufferedImage bufferedImage = ImageIO.read(new FileInputStream(oldFile));int width = bufferedImage.getWidth();int height = bufferedImage.getHeight();String msg = midjourneyLog.getPrompt().trim();String newFileName;if (msg.equalsIgnoreCase("u1")) {newFileName = oldFile.getPath() + "_u1.png";ImgUtil.cut(oldFile, FileUtil.file(newFileName),new Rectangle(0, 0, width / 2, height / 2));} else if (msg.equalsIgnoreCase("u2")) {newFileName = oldFile.getPath() + "_u2.png";ImgUtil.cut(oldFile, FileUtil.file(newFileName),new Rectangle(width / 2, 0, width / 2, height / 2));} else if (msg.equalsIgnoreCase("u3")) {newFileName = oldFile.getPath() + "_u3.png";ImgUtil.cut(oldFile, FileUtil.file(newFileName),new Rectangle(0, height / 2, width / 2, height / 2));} else if (msg.equalsIgnoreCase("u4")) {newFileName = oldFile.getPath() + "_u4.png";ImgUtil.cut(oldFile, FileUtil.file(newFileName),new Rectangle(width / 2, height / 2, width / 2, height / 2));} else {sendBuildFail(midjourneyLog);return;}String mediaId = weChatBiz.sendImg(newFileName, midjourneyLog.getChannel());log.info("[mediaId] {}", mediaId);if (Strings.isNotEmpty(mediaId)) {String wxId = memberService.selectWxidById(midjourneyLog.getMemberId());if (Strings.isEmpty(wxId)) {sendBuildFail(midjourneyLog);return;}midjourneyLogService.updateFinish(midjourneyLog);weChatBiz.sendImgMsg(wxId, mediaId, midjourneyLog.getChannel());}}private boolean isFailMsg(String channel, MidjourneyLog midjourneyLog) {return Objects.isNull(midjourneyLog) || Strings.isEmpty(midjourneyLog.getPrompt()) || Strings.isEmpty(channel)|| Objects.isNull(midjourneyLog.getMemberId()) || checkNotRunMsg(midjourneyLog);}public boolean checkNotRunMsg(MidjourneyLog midjourneyLog) {String prompt = midjourneyLog.getPrompt();if (midjourneyLog.getType() == MedjourneyLogType.BIG_IMG.getCode()) {return !(prompt.equalsIgnoreCase("u1") || prompt.equalsIgnoreCase("u2")|| prompt.equalsIgnoreCase("u3") || prompt.equalsIgnoreCase("u4"));} else if (midjourneyLog.getType() == MedjourneyLogType.ITERATIVE.getCode()) {return !(prompt.equalsIgnoreCase("v1") || prompt.equalsIgnoreCase("v2")|| prompt.equalsIgnoreCase("v3") || prompt.equalsIgnoreCase("v4"));} else {return false;}}private String cleanMsg(String msg) {msg = msg.replace("—", "--").replace("-- ", "--").replace("-- ", "--").replace("-- ", "--").replace("，", ",").replace("/", "").replace("--v", " --v ").replace("--niji", " --niji ").replace("--ar", " --ar ").replace("--aspect", " --ar ").replace("--chaos", " --chaos ").replace("--c", " --c ").replace("--no", " --no ").replace("--quality", " --quality ").replace("--q", " --q ").replace("--repeat", " --repeat ").replace("--s", " --s ").replace("--upbeta", " --upbeta ").trim();if (TextUtil.isHaveChinese(msg)) {msg = translateBiz.translate(msg);}msg = msg.replace("，", ",").replace("/", "").replace("--v5", " --v 5 ").replace("--niji5", " --niji 5 ").trim();if (!msg.contains("--niji") && !msg.contains("--v") && msg.length() > 3) {msg = msg + " --v 5";}return msg;}public void sendBuildFail(MidjourneyLog midjourneyLog) {String wxId = memberService.selectWxidById(midjourneyLog.getMemberId());CallBackUtil.failCallBack(midjourneyLog.getChannel(), MIDJOURNEY, midjourneyLog.getId());midjourneyLogService.updateFail(midjourneyLog);weChatBiz.sendTextMsg(wxId, "抱歉，您的消息【" + midjourneyLog.getPrompt() + "】处理失败，已为您退换对应电量",midjourneyLog.getChannel());}@Scheduled(cron = "34 * * * * ? ")public void checkImg() {//房间号清理cleanRoomNumber();//查询所有进行中的任务List<MidjourneyLog> logs = midjourneyLogService.selectAllDoing().stream().filter(log -> Objects.nonNull(log.getRoomId())).collect(Collectors.toList());//找到超时任务进行关闭List<MidjourneyLog> failLogs = logs.stream().filter(log -> System.currentTimeMillis() - log.getCreateTime().getTime() > MAX_WAIT_TIME).collect(Collectors.toList());failLogs.forEach(this::sendBuildFail);//剩余任务整理出来房间号logs.removeAll(failLogs);Set<Integer> roomSet = logs.stream().map(MidjourneyLog::getRoomId).collect(Collectors.toSet());//轮询当前进度for (Integer roomId : roomSet) {RoomInfo roomInfo = roomInfoService.findById(roomId);if (Objects.isNull(roomInfo)) {log.error("[room没找到] roomId:{} 没找到对应房间，看一下是不是挂了", roomId);continue;}Discord discord = discordService.findById(roomInfo.getDiscordId());List<MidMsg> midMsgs = readNowList(roomInfo, discord);log.info("[消息列表] {}", midMsgs);checkAndSendMsg(midMsgs, logs, roomInfo);}}private void cleanRoomNumber() {List<RoomInfo> roomInfos = roomInfoService.findAll();for (RoomInfo roomInfo : roomInfos) {int count = midjourneyLogService.selectRoomOnUse(roomInfo.getId());roomInfo.setNowNumber(count);roomInfoService.update(roomInfo);}}private void checkAndSendMsg(List<MidMsg> midMsgs, List<MidjourneyLog> logs, RoomInfo roomInfo) {List<MidjourneyLog> roomLogs = logs.stream().filter(log -> log.getRoomId().equals(roomInfo.getId())).collect(Collectors.toList());midMsgs.stream().filter(this::isPrintOk).forEach(m -> roomLogs.forEach(midjourey -> {if (m.getContent().startsWith(MID_JOURNEY_HEAD + filterHead(midjourey.getPrompt()))) {log.info("[Midjourney 配对] msg:{} key:{}", m, midjourey.getPrompt());String wxid = memberService.selectWxidById(midjourey.getMemberId());String url = m.getAttachments().get(0).getUrl();String localPath = Constant.FILE_PATH + UUID.randomUUID() + "." + FileUtil.getSuffix(url);ImgDownUtil.getImage(url.replace("https://", "http://"), localPath);localPath = localPath.replace("\\", "/");if ("webp".equals(FileUtil.getSuffix(localPath))) {com.example.midjourney.util.ImgUtil.webpToPng(localPath, localPath + ".png");localPath = localPath + ".png";}String mediaId = weChatBiz.sendImg(localPath, midjourey.getChannel());log.info("[mediaId] {}", mediaId);if (Strings.isNotEmpty(mediaId)) {weChatBiz.sendImgMsg(wxid, mediaId, midjourey.getChannel());weChatBiz.sendTextMsg(wxid, "下载高清大图口令：\n\n☆左上图回复：U1\n\n☆右上图回复：U2" +"\n\n☆左下图回复：U3\n\n☆右下图回复：U4\n\n下载大图也会扣电量哦~", midjourey.getChannel());midjourey.setImgFile(localPath);midjourneyLogService.updateFinish(midjourey);}}}));}private String filterHead(String prompt) {return Lists.newArrayList(Splitter.on("--").split(prompt)).stream().findFirst().orElse("").trim();}private boolean isPrintOk(MidMsg midMsg) {if (Objects.isNull(midMsg)|| Strings.isEmpty(midMsg.getContent())|| !midMsg.getContent().contains(MID_FIND_LEFT)) {return false;}String str = midMsg.getContent().substring(midMsg.getContent().indexOf(MID_FIND_LEFT));return !str.contains("%") && !str.contains("Waiting");}private boolean isBlackWord(String msg) {msg = msg.toLowerCase();for (String s : Constant.MidjourneyBlackWord) {if (msg.contains(s.toLowerCase())) {return true;}}return false;}private void sendSensitive(MidjourneyLog midjourneyLog) {String wxid = memberService.selectWxidById(midjourneyLog.getMemberId());midjourneyLogService.updateSensitive(midjourneyLog);weChatBiz.sendTextMsg(wxid, "【违规提示】输入内容包含违禁词，依法进行屏蔽。", midjourneyLog.getChannel());}/*** 审核不通过通知用户*/private void sendSensitive(MidjourneyLog midjourneyLog, BaiduTextCensor censor) {String reason = Joiner.on("、").join(Safes.of(censor.getData()).stream().map(BaiduTextCensorData::getMsg).collect(Collectors.toList()));reason = Strings.isEmpty(reason) ? "包含敏感信息" : reason;String fullText = "【违规提示】输入内容因 " + reason + " ，依法进行屏蔽。（百度提供审核能力）";String wxid = memberService.selectWxidById(midjourneyLog.getMemberId());midjourneyLogService.updateSensitive(midjourneyLog);weChatBiz.sendTextMsg(wxid, fullText, midjourneyLog.getChannel());}public List<MidMsg> readNowList(RoomInfo roomInfo, Discord discord) {OkHttpClient client = new OkHttpClient().newBuilder().build();Request request = new Request.Builder().url("https://discord.com/api/v9/channels/" + roomInfo.getDiscordChannelId() + "/messages?limit=50").get().addHeader("authority", "discord.com").addHeader("accept", "*/*").addHeader("accept-language", "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6").addHeader("authorization", discord.getAuthorization()).addHeader("cache-control", "no-cache").addHeader("cookie", discord.getCookie()).addHeader("pragma", "no-cache").addHeader("referer", "https://discord.com/channels/" + roomInfo.getDiscordGuildId() +"/" + roomInfo.getDiscordChannelId()).addHeader("sec-ch-ua", "\"Not_A Brand\";v=\"99\", \"Microsoft Edge\";v=\"109\", \"Chromium\";v=\"109\"").addHeader("sec-ch-ua-mobile", "?0").addHeader("sec-ch-ua-platform", "\"Windows\"").addHeader("sec-fetch-dest", "empty").addHeader("sec-fetch-mode", "cors").addHeader("sec-fetch-site", "same-origin").addHeader("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36 Edg/109.0.1518.70").addHeader("x-debug-options", "bugReporterEnabled").addHeader("x-discord-locale", "zh-CN").addHeader("x-super-properties", discord.getSuperProperties()).build();try {Response response = client.newCall(request).execute();String string = Objects.requireNonNull(response.body()).string();response.close();return JSON.parseArray(string, MidMsg.class);} catch (Throwable t) {log.error("[发生意外 读取消息失败] room: {}", roomInfo);}return Lists.newArrayList();}@SneakyThrowspublic boolean sendMsg(String msg, RoomInfo roomInfo) {Discord discord = discordService.findById(roomInfo.getDiscordId());bigDecimal = bigDecimal.add(new BigDecimal(1000));OkHttpClient client = new OkHttpClient().newBuilder().build();RequestBody body = new MultipartBody.Builder().setType(MultipartBody.FORM).addFormDataPart("payload_json", "{\"type\":2,\"application_id\":\"936929561302675456\"," +"\"guild_id\":\"" + roomInfo.getDiscordGuildId() + "\"," +"\"channel_id\":\"" + roomInfo.getDiscordChannelId() + "\"," +"\"session_id\":\"" + discord.getSessionId() + "\"," +"\"data\":{\"version\":\"1077969938624553050\"," +"\"id\":\"938956540159881230\"," +"\"name\":\"imagine\"," +"\"type\":1,\"options\":[" +"{\"type\":3,\"name\":\"prompt\"," +"\"value\":\"" + TextUtil.cleanString(msg) + "\"}]," +"\"application_command\":" +"{\"id\":\"938956540159881230\",\"application_id\":\"936929561302675456\",\"version\":" +"\"1077969938624553050\",\"default_member_permissions\":null,\"type\":1,\"nsfw\":false," +"\"name\":\"imagine\",\"description\":\"Create images with Midjourney\",\"dm_permission\":true," +"\"contexts\":null,\"options\":[{\"type\":3,\"name\":\"prompt\"," +"\"description\":\"The prompt to imagine\",\"required\":true}]}," +"\"attachments\":[]},\"nonce\":\"" + bigDecimal.toString() + "\"} ").build();Request request = new Request.Builder().url("https://discord.com/api/v9/interactions").method("POST", body).addHeader("authority", "discord.com").addHeader("accept", "*/*").addHeader("accept-language", "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6").addHeader("authorization", discord.getAuthorization()).addHeader("cache-control", "no-cache").addHeader("cookie", discord.getCookie()).addHeader("origin", "https://discord.com").addHeader("pragma", "no-cache").addHeader("referer", "https://discord.com/channels/" + roomInfo.getDiscordGuildId() +"/" + roomInfo.getDiscordChannelId()).addHeader("sec-ch-ua", "\"Not_A Brand\";v=\"99\", \"Microsoft Edge\";v=\"109\", \"Chromium\";v=\"109\"").addHeader("sec-ch-ua-mobile", "?0").addHeader("sec-ch-ua-platform", "\"Windows\"").addHeader("sec-fetch-dest", "empty").addHeader("sec-fetch-mode", "cors").addHeader("sec-fetch-site", "same-origin").addHeader("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36 Edg/109.0.1518.70").addHeader("x-debug-options", "bugReporterEnabled").addHeader("x-discord-locale", "zh-CN").addHeader("x-super-properties", discord.getSuperProperties()).build();Response response = client.newCall(request).execute();if (response.code() == 204) {roomInfoService.addOnce(roomInfo);response.close();return true;}response.close();log.info("[midjourney发消息失败] {}", response.code());return false;}
}
MidjourneyLogService
package com.example.midjourney.service;import com.example.midjourney.bean.pojo.MidjourneyLog;
import com.example.midjourney.bean.pojo.RoomInfo;import java.util.List;public interface MidjourneyLogService {MidjourneyLog findById(Long id);MidjourneyLog findLastNormalLog(Integer memberId);MidjourneyLog findLastNormalOrIterative(Integer memberId);void updateRoom(MidjourneyLog midjourneyLog, RoomInfo roomInfo);List<MidjourneyLog> selectAllDoing();void updateFail(MidjourneyLog midjourneyLog);void updateSensitive(MidjourneyLog midjourneyLog);void updatePrompt(MidjourneyLog midjourneyLog, String prompt);void updateFinish(MidjourneyLog midjourey);int selectRoomOnUse(Integer id);}MidjourneyLogServiceImpl
package com.example.midjourney.service.impl;import com.example.midjourney.bean.pojo.MidjourneyLog;
import com.example.midjourney.bean.pojo.RoomInfo;
import com.example.midjourney.mapper.MidjourneyLogMapper;
import com.example.midjourney.service.MidjourneyLogService;
import com.example.midjourney.util.Safes;
import org.springframework.stereotype.Service;import javax.annotation.Resource;
import java.util.List;@Service
public class MidjourneyLogServiceImpl implements MidjourneyLogService {@Resourceprivate MidjourneyLogMapper midjourneyLogMapper;@Overridepublic MidjourneyLog findById(Long id) {return midjourneyLogMapper.selectByPrimaryKey(id);}@Overridepublic MidjourneyLog findLastNormalLog(Integer memberId) {return midjourneyLogMapper.findLastNormalLog(memberId);}@Overridepublic MidjourneyLog findLastNormalOrIterative(Integer memberId) {return midjourneyLogMapper.findLastNormalOrIterative(memberId);}@Overridepublic void updateRoom(MidjourneyLog midjourneyLog, RoomInfo roomInfo) {midjourneyLog.setRoomId(roomInfo.getId());midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);}@Overridepublic List<MidjourneyLog> selectAllDoing() {MidjourneyLog midjourneyLog = new MidjourneyLog();midjourneyLog.setStatus(0);return Safes.of(midjourneyLogMapper.select(midjourneyLog));}@Overridepublic void updateFail(MidjourneyLog midjourneyLog) {midjourneyLog.setStatus(500);midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);}@Overridepublic void updateSensitive(MidjourneyLog midjourneyLog) {midjourneyLog.setStatus(-1);midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);}@Overridepublic void updatePrompt(MidjourneyLog midjourneyLog, String prompt) {midjourneyLog.setPrompt(prompt);midjourneyLogMapper.updateByPrimaryKeySelective(midjourneyLog);}@Overridepublic void updateFinish(MidjourneyLog midjourey) {midjourey.setStatus(1);midjourneyLogMapper.updateByPrimaryKeySelective(midjourey);}@Overridepublic int selectRoomOnUse(Integer id) {return midjourneyLogMapper.selectRoomOnUse(id);}
}
MidjourneyLogMapper
这里用了通用mapper插件
package com.example.midjourney.mapper;import com.example.midjourney.bean.pojo.MidjourneyLog;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import tk.mybatis.mapper.common.Mapper;public interface MidjourneyLogMapper extends Mapper<MidjourneyLog> {String COLUMN = " id, member_id AS memberId, prompt, channel, type, " +"status, img_file AS imgFile, room_id AS roomId ";@Select("SELECT " + COLUMN + " FROM midjourney_log WHERE member_id = #{memberId} AND type = 0 ORDER BY id DESC LIMIT 1")MidjourneyLog findLastNormalLog(@Param("memberId") Integer memberId);@Select("SELECT " + COLUMN + " FROM midjourney_log WHERE member_id = #{memberId} AND (type = 0 OR type = 2)" +" ORDER BY id DESC LIMIT 1")MidjourneyLog findLastNormalOrIterative(Integer memberId);@Select("SELECT COUNT(*) AS count FROM midjourney_log WHERE room_id = #{id} AND status = 0")Integer selectRoomOnUse(@Param("id") Integer id);}Constant
常量类可以根据需要自定调整敏感词
敏感词写上审核不过，大家自己想想脑补下吧，各种不好的词都往那个敏感词List里面自己加吧！
package com.example.midjourney.contant;import com.example.midjourney.bean.UserInfo;
import com.example.midjourney.biz.WeChatBiz;
import com.example.midjourney.util.BaiduUtil;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;public class Constant {public static ExecutorService threadPool = Executors.newFixedThreadPool(20);public static Map<String, UserInfo> imgWordUserMap = Maps.newConcurrentMap();public static String MID_JOURNEY_HEAD = "**";public static String MID_FIND_LEFT = "** - <";public static String FILE_PATH = "D:\\000img\\";public static String CHATGPT = "chatgpt";public static String MIDJOURNEY = "midjourney";public static final String INSUFFICIENT = "insufficient_quota";public static Map<String, String> WX_TOKEN_CHANNEL = Maps.newHashMap();public static Map<String, String> WX_TOKEN_MAP = Maps.newHashMap();public static String BAIDU_TOKEN = BaiduUtil.queryBaiduToken();public static Map<String, String> CHANNEL_CALL_BACK = Maps.newHashMap();/*** 最大等待时间15分钟*/public static long MAX_WAIT_TIME = 1000 * 60 * 15;public static List<String> MidjourneyBlackWord = Lists.newArrayList("写了审核不过，大家自己发挥想象吧...");}
