hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-10-31 10:18:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加mj web socket 测试类

Browse Source
This commit is contained in:
cherishsince
2024-04-05 08:46:41 +08:00
parent 6adb8d657c
commit 28b2fad7b6
2 changed files with 51 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/demo/wss/Main.java
View File

@@ -1,39 +0,0 @@
package cn.iocoder.yudao.framework.ai.midjourney.demo.wss;
import cn.iocoder.yudao.framework.ai.midjourney.demo.jad.DiscordAccount;
import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.SpringUserWebSocketStarter;
import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.UserMessageListener;
import java.util.Scanner;
/**
* author: fansili
* time: 2024/4/3 16:40
*/
public class Main {
public static void main(String[] args) {
String token = "NTY5MDY4NDAxNzEyOTU1Mzky.G4-Fu0.MzD-7ll-ElbXTTgDPHF-WS_UyhMAfbKN3WyyBc";
DiscordHelper discordHelper = new DiscordHelper();
DiscordAccount discordAccount = new DiscordAccount();
discordAccount.setUserToken(token);
discordAccount.setGuildId("1221445697157468200");
discordAccount.setChannelId("1221445862962630706");
var messageListener = new UserMessageListener();
var webSocketStarter = new SpringUserWebSocketStarter(discordHelper.getWss(), null, discordAccount, messageListener);
try {
webSocketStarter.start();
} catch (Exception e) {
throw new RuntimeException(e);
}
Scanner scanner = new Scanner(System.in);
scanner.nextLine();
}
}

51
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MjWebSocketTests.java Normal file
View File

@@ -0,0 +1,51 @@
package cn.iocoder.yudao.framework.ai.image;
import cn.hutool.core.io.FileUtil;
import cn.iocoder.yudao.framework.ai.midjourney.MidjourneyConfig;
import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.MjMessageListener;
import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.MjWebSocketStarter;
import org.junit.Before;
import org.junit.Test;
import java.io.File;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
/**
* author: fansili
* time: 2024/4/3 16:40
*/
public class MjWebSocketTests {
private MidjourneyConfig midjourneyConfig;
@Before
public void setup() {
String token = "OTcwNDc3NzQxMjUyMTY5NzI4.GJcVxa.VrzMii8dsHOJAPZn4Mw8GuEo7_nIUJij9JIHD4";
Map<String, String> requestTemplates = new HashMap<>();
List<File> files = FileUtil.loopFiles("/Users/fansili/projects/github/ruoyi-vue-pro/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/resources/http-body");
for (File file : files) {
requestTemplates.put(file.getName().replace(".json", ""), FileUtil.readUtf8String(file));
}
this.midjourneyConfig = new MidjourneyConfig(token, "1225414986084388926", "1225414986587832385", requestTemplates);
}
@Test
public void startSocketTest() {
String wssUrl = "wss://gateway.discord.gg";
var messageListener = new MjMessageListener(midjourneyConfig);
var webSocketStarter = new MjWebSocketStarter(wssUrl, null, midjourneyConfig, messageListener);
try {
webSocketStarter.start();
} catch (Exception e) {
throw new RuntimeException(e);
}
Scanner scanner = new Scanner(System.in);
scanner.nextLine();
}
}