TestController.java 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981
  1. package com.backendsys.modules;
  2. import cn.hutool.core.util.IdUtil;
  3. import cn.hutool.json.JSONArray;
  4. import cn.hutool.json.JSONObject;
  5. import cn.hutool.json.JSONUtil;
  6. import com.backendsys.aspect.RateLimiting;
  7. import com.backendsys.aspect.QueuingPoll;
  8. import com.backendsys.modules.common.config.security.utils.HttpRequestUtil;
  9. import com.backendsys.modules.common.config.security.utils.JwtUtil;
  10. import com.backendsys.modules.common.utils.Result;
  11. import com.backendsys.modules.sdk.baidu.yunapp.entity.ExecuteScriptParams;
  12. import com.backendsys.modules.sdk.baidu.yunapp.service.YunappService;
  13. import com.backendsys.modules.sdk.volcengine.entity.VisualFaceSwapV2;
  14. import com.backendsys.modules.sdk.volcengine.service.VolcengineService;
  15. import com.backendsys.modules.system.entity.SysUser;
  16. import com.backendsys.service.TestService;
  17. import com.backendsys.utils.ResourceUtil;
  18. import com.fasterxml.jackson.core.JsonProcessingException;
  19. import com.fasterxml.jackson.databind.ObjectMapper;
  20. import cn.afterturn.easypoi.word.WordExportUtil;
  21. import com.backendsys.utils.MD5Util;
  22. import com.rabbitmq.client.Channel;
  23. import com.rabbitmq.client.GetResponse;
  24. import com.tencentcloudapi.common.Credential;
  25. //import io.github.pigmesh.ai.deepseek.core.DeepSeekClientImpl;
  26. //import io.github.pigmesh.ai.deepseek.core.OpenAiHttpException;
  27. //import io.github.pigmesh.ai.deepseek.core.chat.ChatCompletionResponse;
  28. import com.volcengine.service.visual.IVisualService;
  29. import com.volcengine.service.visual.impl.VisualServiceImpl;
  30. import io.jsonwebtoken.Claims;
  31. import jakarta.annotation.PostConstruct;
  32. import jakarta.servlet.ServletContext;
  33. import org.apache.poi.xwpf.usermodel.XWPFDocument;
  34. import org.redisson.api.*;
  35. import org.springframework.amqp.core.*;
  36. import org.springframework.amqp.core.Queue;
  37. import org.springframework.amqp.rabbit.annotation.RabbitListener;
  38. import org.springframework.amqp.rabbit.connection.Connection;
  39. import org.springframework.amqp.rabbit.core.RabbitAdmin;
  40. import org.springframework.amqp.rabbit.core.RabbitTemplate;
  41. import org.springframework.beans.factory.annotation.Autowired;
  42. import org.springframework.beans.factory.annotation.Value;
  43. import org.springframework.context.annotation.Lazy;
  44. import org.springframework.data.redis.core.RedisTemplate;
  45. import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
  46. import org.springframework.web.bind.annotation.*;
  47. import java.awt.*;
  48. import java.io.*;
  49. import java.nio.charset.StandardCharsets;
  50. import java.nio.file.Path;
  51. import java.nio.file.Paths;
  52. import java.security.NoSuchAlgorithmException;
  53. import java.util.*;
  54. import java.util.List;
  55. import java.util.concurrent.CompletableFuture;
  56. import java.util.concurrent.ExecutionException;
  57. import java.util.concurrent.TimeUnit;
  58. import com.tencentcloudapi.common.CommonClient;
  59. import com.tencentcloudapi.common.exception.TencentCloudSDKException;
  60. @RestController
  61. @RequestMapping("/api/test")
  62. @SuppressWarnings({"rawtypes", "unchecked"})
  63. public class TestController {
  64. @Autowired
  65. private JwtUtil jwtUtil;
  66. @GetMapping("/testExtractToken")
  67. public Claims extractToken(String token) {
  68. Claims claims = jwtUtil.extractAllClaims(token);
  69. System.out.println("claims = " + claims);
  70. return claims;
  71. }
  72. @Value("${tencent.facefusion.secret-id}")
  73. private String SECRET_ID;
  74. @Value("${tencent.facefusion.secret-key}")
  75. private String SECRET_KEY;
  76. // @Autowired
  77. // private AmqpAdmin amqpAdmin;
  78. @Autowired
  79. private RabbitTemplate rabbitTemplate;
  80. // private void initRabbitMQ() {
  81. // System.out.println("-- initRabbitMQ. --");
  82. //
  83. // // 交换机
  84. // amqpAdmin.declareExchange(new DirectExchange("demo.exchange", true, false));
  85. //
  86. // // 队列 (TTL + 死信)
  87. // Map<String, Object> args = new HashMap<>();
  88. // args.put("x-message-ttl", 5000); // 消息 TTL:60 秒(单位毫秒)
  89. // amqpAdmin.declareQueue(new Queue("demo.queue", true, false, false, args));
  90. //
  91. //// // 队列
  92. //// amqpAdmin.declareQueue(new Queue("demo.queue", true));
  93. //
  94. // // 把队列 demo.queue 绑定到交换机 demo.exchange,路由键设置为 order.create
  95. // amqpAdmin.declareBinding(
  96. // BindingBuilder.bind(new Queue("demo.queue"))
  97. // .to(new DirectExchange("demo.exchange"))
  98. // .with("order.create")
  99. // );
  100. // // 单队列 (无死信)
  101. // amqpAdmin.declareQueue(new Queue("demo.queue", true));
  102. // }
  103. @GetMapping("/testRabbitMQ/send")
  104. public String send() {
  105. // initRabbitMQ();
  106. // rabbitTemplate.convertAndSend("demo.exchange", "order.create", "Hello RabbitMQ!");
  107. // rabbitTemplate.convertAndSend("", "demo.queue", "Hello RabbitMQ!");
  108. SysUser sysUser = new SysUser();
  109. sysUser.setUsername(UUID.randomUUID().toString());
  110. System.out.println("[Demo][RabbitMQ-发送]:" + sysUser);
  111. // rabbitTemplate.convertAndSend("", "demo.queue", sysUser);
  112. rabbitTemplate.convertAndSend("demo.exchange", "demo.create", sysUser);
  113. return "ok";
  114. }
  115. @GetMapping("/testRabbitMQ/poll")
  116. public String poll() throws Exception {
  117. // 1. 拿到连接和 channel(不要每次 new)
  118. Connection connection = rabbitTemplate.getConnectionFactory().createConnection();
  119. Channel channel = connection.createChannel(false);
  120. // 2. 手动拉消息,第二个参数传 false 表示“不要自动 ack”
  121. // GetResponse resp = channel.basicGet("demo.queue", false);
  122. GetResponse resp = channel.basicGet("demo.queue", false);
  123. if (resp == null) {
  124. channel.close();
  125. connection.close();
  126. return "no message";
  127. }
  128. String body = new String(resp.getBody(), StandardCharsets.UTF_8);
  129. System.out.println("[Demo][RabbitMQ-手动处理]:" + body);
  130. // 3. 用同一个 channel 去 ack
  131. channel.basicAck(resp.getEnvelope().getDeliveryTag(), false);
  132. channel.close();
  133. connection.close();
  134. return "acked: " + body;
  135. // try {
  136. // channel.basicAck(msg.getMessageProperties().getDeliveryTag(), false);
  137. // } finally {
  138. // channel.close();
  139. // }
  140. // return "acked: " + body;
  141. }
  142. // @Autowired
  143. // private DeepSeekClientImpl deepSeekClient;
  144. //
  145. // // https://javaai.pig4cloud.com/deepseek/quickstart
  146. // // sse 流式返回
  147. // @GetMapping(value = "/deepseek/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
  148. // public Flux<ChatCompletionResponse> chat(String prompt) {
  149. // try {
  150. // Flux<ChatCompletionResponse> flux = deepSeekClient.chatFluxCompletion(prompt);
  151. // return flux;
  152. // } catch (OpenAiHttpException e) {
  153. // throw new CustException(e.getMessage());
  154. // }
  155. // }
  156. // // 进阶配置(多轮对话)
  157. // @GetMapping(value = "/chat/advanced", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
  158. // public Flux<ChatCompletionResponse> advancedChat(String prompt) {
  159. // ChatCompletionRequest request = ChatCompletionRequest.builder()
  160. // .model(ChatCompletionModel.DEEPSEEK_REASONER) // 指定推理模型
  161. // .addUserMessage(prompt) // 用户当前问题
  162. // .addAssistantMessage("上轮对话结果") // 历史助手回复
  163. // .addSystemMessage("你是一名AI助理") // 系统角色设定
  164. // .maxTokens(1000) // 最大生成 token 数
  165. // .temperature(0.7) // 生成多样性控制
  166. // .tools(yourFunctionTools) // Function Calling 工具
  167. // .responseFormat(ChatResponseFormat.JSON) // 结构化响应
  168. // .build();
  169. //
  170. // return deepSeekClient.chatFluxCompletion(request);
  171. // }
  172. @Autowired
  173. private ServletContext servletContext;
  174. @GetMapping("/tomcat-version")
  175. public String getTomcatVersion() {
  176. String serverInfo = servletContext.getServerInfo();
  177. return "当前Tomcat版本: " + serverInfo;
  178. }
  179. /*
  180. // SubmitMediaModeration 提交审核任务
  181. // DescribeMediaModeration 查询审核结果
  182. // SubmitLongVideoFaceFusionJobPro 视频人脸融合异步处理任务,成功提交任务后返回任务的Job id 、预计完成时间以及目前队列长度。
  183. // QueryVideoFaceFusionJob 通过Job Id查询视频人脸融合任务的进度和状态
  184. */
  185. /**
  186. * [腾讯云-人脸融合] 提交审核任务 (视频)
  187. */
  188. @PostMapping("SubmitMediaModeration")
  189. public String SubmitMediaModeration(@RequestBody Map<String, Object> requestBody) throws TencentCloudSDKException {
  190. String videoUrl = (String) requestBody.get("videoUrl");
  191. // String videoUrl = "https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/video/3.mp4";
  192. Credential cred = new Credential(SECRET_ID, SECRET_KEY);
  193. CommonClient client = new CommonClient("facefusion", "2022-09-27", cred, "ap-guangzhou");
  194. String param = "{\"VideoUrl\":\"" + videoUrl + "\"}";
  195. String resp = client.call("SubmitMediaModeration", param);
  196. System.out.println(resp);
  197. return resp;
  198. }
  199. /**
  200. * [腾讯云-人脸融合] 查询审核结果
  201. */
  202. @GetMapping("DescribeMediaModeration")
  203. public String DescribeMediaModeration(String JobId) throws TencentCloudSDKException {
  204. Credential cred = new Credential(SECRET_ID, SECRET_KEY);
  205. CommonClient client = new CommonClient("facefusion", "2022-09-27", cred, "ap-guangzhou");
  206. String param = "{\"JobId\":\"" + JobId + "\"}";
  207. String resp = client.call("DescribeMediaModeration", param);
  208. System.out.println(resp);
  209. return resp;
  210. }
  211. /**
  212. * 视频人脸融合异步处理任务,成功提交任务后返回任务的Job id 、预计完成时间以及目前队列长度。
  213. */
  214. @PostMapping("/SubmitLongVideoFaceFusionJobPro")
  215. public String SubmitLongVideoFaceFusionJobPro(@RequestBody Map<String, Object> requestBody) {
  216. try {
  217. // 从Map中获取参数
  218. List<Map<String, Object>> TemplateInfos = (List<Map<String, Object>>) requestBody.get("TemplateInfos");
  219. List<Map<String, Object>> MergeInfos = (List<Map<String, Object>>) requestBody.get("MergeInfos");
  220. String VideoUrl = (String) requestBody.get("VideoUrl");
  221. Integer SwapModelType = (Integer) requestBody.get("SwapModelType");
  222. /*
  223. https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/face/fe-1.png
  224. https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/face/fe-2.jpg
  225. https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/face/m-1.png
  226. https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/face/m-2.png
  227. */
  228. // String imgUrl = "https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/face/fe-2.jpg";
  229. // String videoUrl = "https://guangzhou-1320301544.cos-website.ap-guangzhou.myqcloud.com/test/ai-face/video/3.mp4";
  230. Credential cred = new Credential(SECRET_ID, SECRET_KEY);
  231. // FacefusionClient client = new FacefusionClient(cred, "ap-guangzhou");
  232. CommonClient client = new CommonClient("facefusion", "2022-09-27", cred, "ap-guangzhou");
  233. JSONObject param = new JSONObject();
  234. JSONArray paramMergeInfos = JSONUtil.parseArray(MergeInfos);
  235. JSONArray paramTemplateInfos = JSONUtil.parseArray(TemplateInfos);
  236. System.out.println("paramMergeInfos:");
  237. System.out.println(paramMergeInfos);
  238. System.out.println("paramTemplateInfos:");
  239. System.out.println(paramTemplateInfos);
  240. param.set("MergeInfos", MergeInfos);
  241. param.set("TemplateInfos", TemplateInfos);
  242. param.set("VideoUrl", VideoUrl);
  243. param.set("SwapModelType", SwapModelType);
  244. System.out.println("param:");
  245. System.out.println(param);
  246. String resp = client.call("SubmitLongVideoFaceFusionJobPro", JSONUtil.toJsonStr(param));
  247. System.out.println(resp);
  248. return resp;
  249. } catch (TencentCloudSDKException e) {
  250. System.out.println(e);
  251. System.out.println(e.getMessage());
  252. return e.getMessage();
  253. }
  254. }
  255. /**
  256. * 通过Job Id查询视频人脸融合任务的进度和状态
  257. */
  258. @GetMapping("QueryVideoFaceFusionJob")
  259. public String QueryVideoFaceFusionJob(String JobId) throws TencentCloudSDKException {
  260. Credential cred = new Credential(SECRET_ID, SECRET_KEY);
  261. CommonClient client = new CommonClient("facefusion", "2022-09-27", cred, "ap-guangzhou");
  262. String param = "{\"JobId\":\"" + JobId + "\"}";
  263. String resp = client.call("QueryVideoFaceFusionJob", param);
  264. System.out.println(resp);
  265. return resp;
  266. }
  267. @Value("${file.upload.directory}") // 配置保存文件的目录
  268. private String uploadDirectory;
  269. private static final ObjectMapper MAPPER = new ObjectMapper();
  270. // @Autowired
  271. // private HttpExchangeService httpExchangeService;
  272. // @PostMapping("testWebHook")
  273. // public Result testWebHook(@RequestBody Map<String, Object> requestBody, HttpServletRequest request) {
  274. // System.out.println("------- testWebHook --------");
  275. //// try {
  276. //// String requestBody = StreamUtils.copyToString(request.getInputStream(), StandardCharsets.UTF_8);
  277. //// System.out.println(requestBody);
  278. //// } catch (IOException e) {
  279. //// e.printStackTrace();
  280. //// }
  281. // System.out.println("===========================================");
  282. // System.out.println("requestBody:");
  283. // System.out.println(requestBody);
  284. // System.out.println("----------------------------");
  285. // System.out.println("requestBody (JSON):");
  286. // System.out.println(JSONUtil.parseObj(requestBody));
  287. // System.out.println("----------------------------");
  288. // String token = request.getHeader("X-Codeup-Token");
  289. // System.out.println("X-Codeup-Token: " + token);
  290. // System.out.println("===========================================");
  291. // return null;
  292. // }
  293. //
  294. @Autowired
  295. private TestService testService;
  296. @GetMapping("testThreadPool")
  297. public String testThreadPool() throws ExecutionException, InterruptedException {
  298. testService.testThreadPool();
  299. return "ok";
  300. }
  301. @RateLimiting(key = "test", limit = 5, duration = 10)
  302. @GetMapping("testSemaphore")
  303. public String testSemaphore() {
  304. // Semaphore semaphore = new Semaphore(2);
  305. // try {
  306. // // 尝试获取一个许可
  307. // semaphore.acquire();
  308. // // 模拟服务操作
  309. // System.out.println("服务被访问,当前活跃线程数:" + semaphore.availablePermits());
  310. // // 假设服务操作需要一些时间
  311. // Thread.sleep(5000);
  312. // } catch (InterruptedException e) {
  313. // Thread.currentThread().interrupt();
  314. // System.out.println("线程被中断");
  315. // } finally {
  316. // // 释放许可
  317. // semaphore.release();
  318. // System.out.println("服务访问结束,释放许可,当前活跃线程数:" + semaphore.availablePermits());
  319. // }
  320. return "ok";
  321. }
  322. // @Autowired
  323. // private SysResourceService sysResourceService;
  324. //
  325. // @GetMapping("testConsumptionPoint")
  326. // public Result testConsumptionPoint(String resource_type, String resource_tag) throws IOException {
  327. // sysResourceService.consumptionPoint(resource_type, resource_tag, null);
  328. // return Result.success("扣除成功");
  329. // }
  330. // @Autowired
  331. // private SDKTencentCOSService sdkTencentCOSService;
  332. //
  333. // @PostMapping("testUploadOrigin")
  334. // public Result testUploadOrigin(@RequestBody UploadOriginDTO uploadOriginDTO) throws IOException {
  335. // return Result.success(sdkTencentCOSService.uploadOrigin(uploadOriginDTO));
  336. // }
  337. @GetMapping("testLocalPath")
  338. public Path testLocalPath() {
  339. return Paths.get(".");
  340. }
  341. @GetMapping("testDownloadLocal")
  342. public String testDownloadLocal() {
  343. // return ResourceUtil.downloadImage("https://dashscope-result-bj.oss-cn-beijing.aliyuncs.com/466b5214/20240417/172326_0_2a130824-cffe-4d18-8455-f98c18e4900e.png?Expires=1713432207&OSSAccessKeyId=LTAI5tQZd8AEcZX6KZV4G8qL&Signature=439TeHWxweXDJ%2FE8mAQ3UQJx7XI%3D");
  344. return ResourceUtil.downloadImage("http://cos.daoguyujiamcn.com/customer-pipline/1/65119e58-9940-458c-8911-750d23f9fbf1/idcard/b.pdf", uploadDirectory);
  345. }
  346. //
  347. //
  348. // // 【腾讯云 COS】
  349. // // 文档中心 > 对象存储 > API 文档 > Object 接口 > 基本操作 > PUT Object
  350. // // https://cloud.tencent.com/document/product/436/7749
  351. // // https://github.com/tencentyun/cos-java-sdk-v5/blob/master/src/main/java/com/qcloud/cos/demo/PutObjectDemo.java
  352. // static COSClient cosClient = createCli();
  353. //
  354. // static COSClient createCli() {
  355. // return createCli("ap-hongkong");
  356. // }
  357. //
  358. // static COSClient createCli(String region) {
  359. // // 初始化用户身份信息(secretId, secretKey)
  360. // COSCredentials cred = new BasicCOSCredentials("AKID3zlNxRjstjnohWFnDUfeVBj3CJH7mFaK","IXgzFKB71rXOCxlS4BTdtCYuJbP8h7Xr");
  361. // // 设置bucket的区域, COS地域的简称请参照 https://www.qcloud.com/document/product/436/6224
  362. // ClientConfig clientConfig = new ClientConfig(new Region(region));
  363. // // 生成cos客户端
  364. // return new COSClient(cred, clientConfig);
  365. // }
  366. //
  367. // /**
  368. // * 腾讯云 COS - 上传文件
  369. // */
  370. // @GetMapping("testTencentSDK/putObject")
  371. // public String testTencentSDKPutObject() {
  372. // String bucketName = "storage-1320301544";
  373. //
  374. // // 获取当前日期
  375. // Date currentDate = new Date();
  376. // SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss");
  377. // String dateFolder = dateFormat.format(currentDate);
  378. //
  379. // System.out.println("dateFolder: " + dateFolder);
  380. //
  381. // String fileName = "README.md";
  382. // String key = dateFolder + "/" + fileName;
  383. //
  384. // String localPath = fileName;
  385. //
  386. // // 计算过期时间为 1 分钟后
  387. // Date currentTime = new Date();
  388. // long expiresTime = currentTime.getTime() + 60*1000; // 当前时间毫秒数 + 1分钟的毫秒数
  389. // ObjectMetadata objectMetadata = new ObjectMetadata();
  390. // objectMetadata.setHeader("expires", new Date(expiresTime));
  391. //
  392. // PutObjectRequest putObjectRequest = new PutObjectRequest(bucketName, key, new File(localPath));
  393. // putObjectRequest.withMetadata(objectMetadata);
  394. //
  395. // PutObjectResult putObjectResult = cosClient.putObject(putObjectRequest);
  396. //
  397. // System.out.println(putObjectResult.getRequestId());
  398. //
  399. // GetObjectRequest getObjectRequest = new GetObjectRequest(bucketName, key);
  400. // COSObject cosObject = cosClient.getObject(getObjectRequest);
  401. // System.out.println(cosObject.getObjectMetadata().getRequestId());
  402. //
  403. // cosClient.shutdown();
  404. // return "ok";
  405. // }
  406. //
  407. // 文档中心 > SDK 中心 > Java
  408. // https://cloud.tencent.com/document/sdk/Java
  409. // 【腾讯云-混元大模型】
  410. // https://cloud.tencent.com/document/api/1729/101843
  411. // private final static Charset UTF8 = StandardCharsets.UTF_8;
  412. //// // 需要设置环境变量 TENCENTCLOUD_SECRET_ID,值为示例的 AKIDz8krbsJ5yKBZQpn74WFkmLPx3*******
  413. //// private final static String SECRET_ID = "AKID3zlNxRjstjnohWFnDUfeVBj3CJH7mFaK";
  414. //// // 需要设置环境变量 TENCENTCLOUD_SECRET_KEY,值为示例的 Gu5t9xGARNpq86cd98joQYCN3*******
  415. //// private final static String SECRET_KEY = "IXgzFKB71rXOCxlS4BTdtCYuJbP8h7Xr";
  416. //
  417. // @Value("tencent.hunyuan.secret-id")
  418. // private String SECRET_ID;
  419. // @Value("tencent.hunyuan.secret-key")
  420. // private String SECRET_KEY;
  421. //
  422. //
  423. // private final static String CT_JSON = "application/json; charset=utf-8";
  424. // public static byte[] hmac256(byte[] key, String msg) throws Exception {
  425. // Mac mac = Mac.getInstance("HmacSHA256");
  426. // SecretKeySpec secretKeySpec = new SecretKeySpec(key, mac.getAlgorithm());
  427. // mac.init(secretKeySpec);
  428. // return mac.doFinal(msg.getBytes(UTF8));
  429. // }
  430. // public static String sha256Hex(String s) throws Exception {
  431. // MessageDigest md = MessageDigest.getInstance("SHA-256");
  432. // byte[] d = md.digest(s.getBytes(UTF8));
  433. // return DatatypeConverter.printHexBinary(d).toLowerCase();
  434. // }
  435. // @GetMapping("testTencentSDK/hunyuan")
  436. // public Result testTencentSDKHunyuan() throws Exception {
  437. //
  438. // try {
  439. // String result = "";
  440. // Credential cred = new Credential(SECRET_ID, SECRET_KEY);
  441. // ClientProfile clientProfile = new ClientProfile();
  442. // HunyuanClient client = new HunyuanClient(cred, "ap-guangzhou", clientProfile);
  443. //
  444. // ChatStdRequest req = new ChatStdRequest();
  445. // Message msg = new Message();
  446. // msg.setRole("user");
  447. // msg.setContent("给我一句诗,不超过10个字");
  448. // req.setMessages(new Message[]{ msg });
  449. //
  450. // ChatStdResponse resp = client.ChatStd(req);
  451. //
  452. // Gson gson = new GsonBuilder().create();
  453. // for (SSEResponseModel.SSE e : resp) {
  454. //
  455. // System.out.println(e.Data);
  456. //
  457. // /**
  458. // * .Data:
  459. // * {"Note":"以上内容为AI生成,不代表开发者立场,请勿删除或修改本标记","Choices":[{"FinishReason":"","Delta":{"Role":"assistant","Content":"当然"}}],"Created":1709618061,"Id":"e73c0a71-5c98-4893-ba90-ad5056d5871a","Usage":{"PromptTokens":7,"CompletionTokens":1,"TotalTokens":8}}
  460. // * e.Data:
  461. // * {"Note":"以上内容为AI生成,不代表开发者立场,请勿删除或修改本标记","Choices":[{"FinishReason":"","Delta":{"Role":"assistant","Content":"可以"}}],"Created":1709618061,"Id":"e73c0a71-5c98-4893-ba90-ad5056d5871a","Usage":{"PromptTokens":7,"CompletionTokens":2,"TotalTokens":9}}
  462. // * e.Data:
  463. // * {"Note":"以上内容为AI生成,不代表开发者立场,请勿删除或修改本标记","Choices":[{"FinishReason":"","Delta":{"Role":"assistant","Content":","}}],"Created":1709618061,"Id":"e73c0a71-5c98-4893-ba90-ad5056d5871a","Usage":{"PromptTokens":7,"CompletionTokens":3,"TotalTokens":10}}
  464. // * e.Data:
  465. // */
  466. // JSONObject dataObject = JSONUtil.parseObj(e.Data);
  467. // JSONArray choicesArray = dataObject.getJSONArray("Choices");
  468. // if (choicesArray != null && choicesArray.size() > 0) {
  469. // // 获取第一个元素
  470. // JSONObject firstChoice = choicesArray.getJSONObject(0);
  471. // System.out.println("firstChoice:");
  472. // System.out.println(firstChoice);
  473. // // 获取 Delta 对象
  474. // JSONObject deltaObject = firstChoice.getJSONObject("Delta");
  475. // System.out.println("deltaObject:");
  476. // System.out.println(deltaObject);
  477. // // 获取 content 的值
  478. // String content = deltaObject.getStr("Content");
  479. // System.out.println("Content: " + content);
  480. //
  481. // result += content;
  482. // System.out.println("result: " + result);
  483. // System.out.println("---------------------------------------------");
  484. // }
  485. // }
  486. // return Result.success(result);
  487. //
  488. // } catch (TencentCloudSDKException e) {
  489. // e.printStackTrace();
  490. // }
  491. // return null;
  492. // }
  493. // @GetMapping("testQueue")
  494. // @QueuingPoll
  495. // public String testQueue(@RequestParam Integer tenantId, @RequestParam Integer time) throws InterruptedException {
  496. // Thread currentThread = Thread.currentThread();
  497. // String threadName = currentThread.getName();
  498. // System.out.println(threadName + " 开始休眠: " + time);
  499. // Thread.sleep(time);
  500. // System.out.println(threadName + " 休眠结束: " + time);
  501. // return threadName;
  502. // }
  503. @GetMapping("testMeitu/login")
  504. public JSONObject testMeituLogin(String code, String access_key) {
  505. System.out.println("code: " + code + ", access_key: " + access_key);
  506. System.out.println("---------------------------------------------------");
  507. String response = "{\"code\": 0,\"msg\": \"success\", \"data\": {\"id\": \"enjdj328ydhhw3u43yjhdj\", \"name\": \"张三\",\"avatar_url\": \"http://www.**.com/avatar/icon.jpg\"}}";
  508. JSONObject jsonObject = JSONUtil.parseObj(response);
  509. return jsonObject;
  510. }
  511. @Autowired
  512. private RedisTemplate redisTemplate;
  513. @GetMapping("testQueue")
  514. @QueuingPoll
  515. public String testQueue(String taskId) {
  516. redisTemplate.opsForList().rightPush("demoQueue", taskId);
  517. Long listSize = redisTemplate.opsForList().size("demoQueue");
  518. System.out.println("队伍总人数: " + listSize);
  519. List listData = redisTemplate.opsForList().range("demoQueue", 0, -1);
  520. System.out.println("队伍信息: " + listData.toString() + ", taskId: " + taskId + ", listData.get(0): " + listData.get(0));
  521. Integer index = 0;
  522. if (listData.size() == 0 || (listData.size() > 0 && listData.get(0).toString().equals(taskId))) {
  523. // 创建一个 CompletableFuture 来执行异步任务
  524. CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
  525. // 执行业务逻辑
  526. System.out.println("开始休眠..");
  527. try {
  528. Thread.sleep(5000);
  529. } catch (InterruptedException e) {
  530. throw new RuntimeException(e);
  531. }
  532. System.out.println("休眠结束..");
  533. //
  534. Long removeNum = redisTemplate.opsForList().remove("demoQueue", 0, taskId);
  535. System.out.println("离开队伍: " + removeNum);
  536. System.out.println("-------------------------------------");
  537. });
  538. return "操作成功";
  539. } else {
  540. index = listData.indexOf(taskId);
  541. return "排队中:" + index + " / " + listSize;
  542. }
  543. }
  544. /**
  545. * 这里多个key用":" 拼接,后续定时任务踢出队列拆分使用。
  546. * @param keys
  547. * @return
  548. */
  549. public static String buildResourcesSetValue(String... keys){
  550. return String.join(":",keys);
  551. }
  552. // @GetMapping("testQueue")
  553. // public String testQueue() throws InterruptedException {
  554. // // 定义Redis信号量,允许的最大并发数为1
  555. // RSemaphore semaphore = redissonClient.getSemaphore("demoQueue");
  556. // int maxConcurrentRequests = 2; // 允许的最大并发请求数
  557. //
  558. // try {
  559. // // 尝试获取信号量,设定超时时间为3秒
  560. // if (!semaphore.tryAcquire(maxConcurrentRequests, 3, TimeUnit.SECONDS)) {
  561. // // 如果无法获取信号量,说明并发数已满,抛出异常
  562. // long waitingPosition = getWaitingPosition();
  563. // throw new CustException("正在排队中,第" + waitingPosition + "位 / 共" + maxConcurrentRequests + "位");
  564. // }
  565. // // 执行业务逻辑
  566. // System.out.println("开始休眠..");
  567. // Thread.sleep(5000);
  568. // System.out.println("休眠结束..");
  569. //
  570. // } catch (InterruptedException e) {
  571. // // 处理中断异常
  572. // throw new CustException("处理中断异常");
  573. //
  574. // } finally {
  575. // // 释放信号量
  576. //// if (semaphore.isHeldByCurrentThread()) {
  577. // System.out.println("semaphore:");
  578. // System.out.println(semaphore);
  579. // System.out.println(semaphore.isExists());
  580. // semaphore.release();
  581. //// }
  582. // }
  583. //
  584. //
  585. // // 返回结果
  586. // return "操作成功";
  587. //
  588. // }
  589. //
  590. // // 获取当前排队位置的方法
  591. // private long getWaitingPosition() {
  592. // RScoredSortedSet<Object> scoreSet = redissonClient.getScoredSortedSet("demoQueue");
  593. //// long queueSize = scoreSet.size();
  594. //// System.out.println("(getWaitingPosition) queueSize: " + queueSize);
  595. // long waitingPosition = 0;
  596. // // 遍历有序集合,计算当前请求之前的请求数量
  597. // Iterator<Object> iterator = scoreSet.iterator(0);
  598. // while (iterator.hasNext()) {
  599. // Object obj = iterator.next();
  600. // Double score = scoreSet.getScore(obj);
  601. // if (score > System.currentTimeMillis()) {
  602. // waitingPosition++;
  603. // }
  604. // }
  605. // return waitingPosition;
  606. // }
  607. /* 字符串 转 JSONObject */
  608. @GetMapping("testJSONObject")
  609. public JSONObject testGson() {
  610. String jsonString = " {\"Note\":\"以上内容为AI生成,不代表开发者立场,请勿删除或修改本标记\",\"Choices\":[{\"FinishReason\":\"stop\",\"Delta\":{\"Role\":\"assistant\",\"Content\":\"\"}}],\"Created\":1709617106,\"Id\":\"7511d79a-6318-4da1-883a-35c8be97f67f\",\"Usage\":{\"PromptTokens\":7,\"CompletionTokens\":23,\"TotalTokens\":30}}";
  611. JSONObject jsonObject = JSONUtil.parseObj(jsonString);
  612. return jsonObject;
  613. }
  614. // 【火山引擎】
  615. // - 公共错误代码
  616. // https://www.volcengine.com/docs/6369/68677
  617. // - 获取个性化内容
  618. // https://www.volcengine.com/docs/6392/75762
  619. // - 模型广场
  620. // https://www.volcengine.com/docs/82379/1099503
  621. // @GetMapping("testHuoshanSDK")
  622. // public String testHuoShanSDK() throws Exception {
  623. // // 注意示例代码安全,代码泄漏会导致AK/SK泄漏,有极大的安全风险。
  624. // String ak = "AKLTMjM2OTNmNTViNGQ3NDM0NmI1YjU4MmU2ZjY5YzEwM2I";
  625. // String sk = "WlRZMk5qRTJOR1ZrTURnMk5EVmhZVGc1TmpVeE4yUTJNelkzWXpVeU1XVQ==";
  626. //
  627. // BusinessSecurityService businessSecurityService = BusinessSecurityServiceImpl.getInstance();
  628. // // call below method if you dont set ak and sk in ~/.volc/config
  629. //
  630. // businessSecurityService.setAccessKey(ak);
  631. // businessSecurityService.setSecretKey(sk);
  632. // // risk detection
  633. // try {
  634. // RiskDetectionRequest riskDetectionRequest = new RiskDetectionRequest();
  635. //// riskDetectionRequest.setLimit(3);
  636. //
  637. // RiskDetectionResponse riskDetectionResponse = businessSecurityService.RiskDetection(riskDetectionRequest);
  638. // System.out.println(JSON.toJSONString(riskDetectionResponse));
  639. // } catch (Exception e) {
  640. // e.printStackTrace();
  641. // }
  642. //
  643. // return "1";
  644. // }
  645. // @GetMapping("testMeituSDK")
  646. // public SdkResponseDTO.Result testMeituSDK() throws URISyntaxException {
  647. // //使用前需提前申请 AccessKey 和 secret,具体获取,请对接商务
  648. // String accessKey = "306129caf4574016be9c09a5b0ecdcad";
  649. // String secret = "a202a3f4f0a844109cea84af79ad9b40";
  650. //
  651. // // 创建AiClient实例
  652. // AiClient aiClient = new AiClient(accessKey,secret);
  653. //
  654. // // 【美图AI - 文生图】
  655. // // https://ai.meitu.com/doc/?id=228&type=api&lang=zh
  656. //
  657. // // 文生图的请求对象
  658. // InferenceParamsDTO inferenceParamsDTO = new InferenceParamsDTO();
  659. // // 推理业务配置中获取中约定的模型平台
  660. // inferenceParamsDTO.setModelPlatform("diffusers");
  661. // // 推理业务配置中获取的模型
  662. // inferenceParamsDTO.setModel("braBeautifulRealistic_V5_diffusers_v0.1.zip");
  663. // // 提示词
  664. // inferenceParamsDTO.setPrompt("a girl");
  665. // // 方向提示词
  666. // inferenceParamsDTO.setNegativePrompt("easy negative");
  667. // // 推理业务配置中获取中约定的高
  668. // inferenceParamsDTO.setHeight(512);
  669. // // 推理业务配置中获取中约定的宽
  670. // inferenceParamsDTO.setHeight(512);
  671. // // 推理提交推理任务,并等待返回推理图片地址
  672. // SdkResponseDTO.Result result = aiClient.inference(inferenceParamsDTO);
  673. // return result;
  674. // }
  675. @GetMapping("testWordExport")
  676. public Map<String, Object> SimpleWordExport() {
  677. Map<String, Object> map = new HashMap<String, Object>();
  678. map.put("title", "Easypoi 标题");
  679. map.put("content", "JueYue 内容,在家也能享受绿色健康美味!用纯天然无添加剂的食材,带给你和家人最纯净的美味。从选购新鲜食材到烘焙出炉,每一步都充满温馨与乐趣。让我们一起回归自然,享受绿色健康的蛋糕时光!");
  680. try {
  681. XWPFDocument doc = WordExportUtil.exportWord07("D:/excel/template.docx", map);
  682. FileOutputStream fos = new FileOutputStream("D:/excel/simple.docx");
  683. doc.write(fos);
  684. fos.close();
  685. } catch (Exception e) {
  686. e.printStackTrace();
  687. }
  688. return map;
  689. }
  690. @GetMapping("testGetUUID")
  691. public String testGetUUID() {
  692. return UUID.randomUUID().toString();
  693. }
  694. @GetMapping("testGetShortUUID")
  695. public String testGetShortUUID(Integer length) {
  696. // return UUID.randomUUID().toString()
  697. // .replace("-", "")
  698. // .substring(0, length);
  699. return IdUtil.simpleUUID().substring(0, length);
  700. }
  701. /**
  702. * BCrypt算法进行加密
  703. */
  704. @GetMapping("testPassword")
  705. public String testPassword(String value) {
  706. String password = Optional.ofNullable(value).orElse("123456");
  707. BCryptPasswordEncoder encoder = new BCryptPasswordEncoder();
  708. String encodedPassword = encoder.encode(password);
  709. boolean matched = encoder.matches(password, encodedPassword);
  710. return "Value: " + value + "\nencodedPassword: " + encodedPassword + "\nmatched: " + matched;
  711. }
  712. /**
  713. * 使用 MD5Util 工具类
  714. */
  715. @GetMapping("testMD5")
  716. public String testMD5(String value) throws NoSuchAlgorithmException {
  717. String password = Optional.ofNullable(value).orElse("123456");
  718. String version = "0.0.48";
  719. return "Value: " + value + "\n" +
  720. "MD5: " + MD5Util.encrypt(password) + "\n" +
  721. "Version: " + version;
  722. }
  723. @GetMapping("/testJSON")
  724. public String testJSON() throws JsonProcessingException {
  725. Integer[] a = MAPPER.readValue("1,2", Integer[].class);
  726. System.out.println("a: ");
  727. System.out.println(a);
  728. return "1";
  729. }
  730. @GetMapping("testFont")
  731. public String[] testFont() {
  732. GraphicsEnvironment ge = GraphicsEnvironment.getLocalGraphicsEnvironment();
  733. String[] fontNames = ge.getAvailableFontFamilyNames();
  734. for (String fontName : fontNames) {
  735. System.out.println(fontName);
  736. }
  737. return fontNames;
  738. }
  739. /**
  740. * Redisson 加锁
  741. * @return
  742. */
  743. @Lazy
  744. @Autowired
  745. RedissonClient redissonClient;
  746. @GetMapping("testRedisson")
  747. public String testGetRedisson() throws InterruptedException {
  748. RLock lock = redissonClient.getLock("my-lock");
  749. lock.tryLock(3, TimeUnit.SECONDS); // 阻塞式等待。默认加的锁是30s时间
  750. try {
  751. // 1、锁的自动续期,运行期间自动给锁续上新的30s,无需担心业务时间长,锁过期会自动被释放
  752. // 2、加锁的业务只要运行完成,就不会给当前锁续期,即使不手动释放锁,锁默认在30s后自动释放,避免死锁
  753. System.out.println("加锁成功,执行业务代码..."+Thread.currentThread().getId());
  754. Thread.sleep(3000);
  755. } catch (Exception e) {
  756. e.printStackTrace();
  757. } finally {
  758. System.out.println("释放锁..."+Thread.currentThread().getId());
  759. lock.unlock();
  760. }
  761. return "get lock!";
  762. }
  763. /**
  764. * Redisson 一般用于商品秒杀活动:
  765. */
  766. /*
  767. @Component
  768. public class DistributedLockExample {
  769. @Autowired
  770. private Redisson redisson;
  771. private RLock distributedLock;
  772. @PostConstruct
  773. private void init() {
  774. distributedLock = redisson.getLock("myDistributedLock");
  775. }
  776. public void performOperation() {
  777. try {
  778. // 尝试获取分布式锁,最多等待10秒
  779. boolean lockAcquired = distributedLock.tryLock(10, TimeUnit.SECONDS);
  780. if (lockAcquired) {
  781. // 获取锁成功,执行需要加锁的业务逻辑
  782. // ...
  783. } else {
  784. // 获取锁失败,处理异常情况
  785. // ...
  786. }
  787. } catch (InterruptedException e) {
  788. // 处理中断异常
  789. // ...
  790. } finally {
  791. // 释放分布式锁
  792. distributedLock.unlock();
  793. }
  794. }
  795. }
  796. */
  797. public static Object resource1 = new Object();
  798. public static Object resource2 = new Object();
  799. @GetMapping("testDeadLock")
  800. public String testDeadLock() {
  801. RLock lock = redissonClient.getLock("thread-lock");
  802. // 线程1
  803. new Thread(() -> {
  804. //System.out.println("------- lock thread-1 -------");
  805. lock.lock();
  806. synchronized (resource2) {
  807. System.out.println(Thread.currentThread().getName() + "获取了 resource2 锁");
  808. try {
  809. Thread.sleep(3000);
  810. } catch (InterruptedException e) {
  811. throw new RuntimeException(e);
  812. }
  813. synchronized (resource1) {
  814. System.out.println(Thread.currentThread().getName() + "获取了 resource1 锁");
  815. lock.unlock();
  816. }
  817. }
  818. }).start();
  819. // 线程2
  820. new Thread(() -> {
  821. //System.out.println("------- lock thread-2 -------");
  822. lock.lock();
  823. synchronized (resource1) {
  824. System.out.println(Thread.currentThread().getName() + "获取了 resource1 锁");
  825. try {
  826. Thread.sleep(3000);
  827. } catch (InterruptedException e) {
  828. throw new RuntimeException(e);
  829. }
  830. synchronized (resource2) {
  831. System.out.println(Thread.currentThread().getName() + "获取了 resource2 锁");
  832. lock.unlock();
  833. }
  834. }
  835. }).start();
  836. return "ok";
  837. }
  838. }