BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 28b2fad7b612e3b1ff4684da022788ba93a3cd77

Authored by cherishsince
1 parent 6adb8d65

增加mj web socket 测试类

Inline Side-by-side
Showing 2 changed files with 51 additions and 39 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/demo/wss/Main.java deleted
View file @6adb8d6
1 -package cn.iocoder.yudao.framework.ai.midjourney.demo.wss;  
2 -  
3 -import cn.iocoder.yudao.framework.ai.midjourney.demo.jad.DiscordAccount;  
4 -import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.SpringUserWebSocketStarter;  
5 -import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.UserMessageListener;  
6 -  
7 -import java.util.Scanner;  
8 -  
9 -/**  
10 - * author: fansili  
11 - * time: 2024/4/3 16:40  
12 - */  
13 -public class Main {  
14 -  
15 - public static void main(String[] args) {  
16 - String token = "NTY5MDY4NDAxNzEyOTU1Mzky.G4-Fu0.MzD-7ll-ElbXTTgDPHF-WS_UyhMAfbKN3WyyBc";  
17 -  
18 -  
19 - DiscordHelper discordHelper = new DiscordHelper();  
20 - DiscordAccount discordAccount = new DiscordAccount();  
21 - discordAccount.setUserToken(token);  
22 - discordAccount.setGuildId("1221445697157468200");  
23 - discordAccount.setChannelId("1221445862962630706");  
24 -  
25 -  
26 - var messageListener = new UserMessageListener();  
27 - var webSocketStarter = new SpringUserWebSocketStarter(discordHelper.getWss(), null, discordAccount, messageListener);  
28 -  
29 - try {  
30 - webSocketStarter.start();  
31 - } catch (Exception e) {  
32 - throw new RuntimeException(e);  
33 - }  
34 -  
35 -  
36 - Scanner scanner = new Scanner(System.in);  
37 - scanner.nextLine();  
38 - }  
39 -}  
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MjWebSocketTests.java 0 → 100644
  View file @28b2fad
  1 +package cn.iocoder.yudao.framework.ai.image;
  2 +
  3 +import cn.hutool.core.io.FileUtil;
  4 +import cn.iocoder.yudao.framework.ai.midjourney.MidjourneyConfig;
  5 +import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.MjMessageListener;
  6 +import cn.iocoder.yudao.framework.ai.midjourney.demo.wss.user.MjWebSocketStarter;
  7 +import org.junit.Before;
  8 +import org.junit.Test;
  9 +
  10 +import java.io.File;
  11 +import java.util.HashMap;
  12 +import java.util.List;
  13 +import java.util.Map;
  14 +import java.util.Scanner;
  15 +
  16 +/**
  17 + * author: fansili
  18 + * time: 2024/4/3 16:40
  19 + */
  20 +public class MjWebSocketTests {
  21 +
  22 + private MidjourneyConfig midjourneyConfig;
  23 +
  24 + @Before
  25 + public void setup() {
  26 + String token = "OTcwNDc3NzQxMjUyMTY5NzI4.GJcVxa.VrzMii8dsHOJAPZn4Mw8GuEo7_nIUJij9JIHD4";
  27 + Map<String, String> requestTemplates = new HashMap<>();
  28 + 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");
  29 + for (File file : files) {
  30 + requestTemplates.put(file.getName().replace(".json", ""), FileUtil.readUtf8String(file));
  31 + }
  32 + this.midjourneyConfig = new MidjourneyConfig(token, "1225414986084388926", "1225414986587832385", requestTemplates);
  33 + }
  34 +
  35 + @Test
  36 + public void startSocketTest() {
  37 + String wssUrl = "wss://gateway.discord.gg";
  38 + var messageListener = new MjMessageListener(midjourneyConfig);
  39 + var webSocketStarter = new MjWebSocketStarter(wssUrl, null, midjourneyConfig, messageListener);
  40 +
  41 + try {
  42 + webSocketStarter.start();
  43 + } catch (Exception e) {
  44 + throw new RuntimeException(e);
  45 + }
  46 +
  47 +
  48 + Scanner scanner = new Scanner(System.in);
  49 + scanner.nextLine();
  50 + }
  51 +}