package com.ruoyi.web.task; import com.alibaba.fastjson2.JSONObject; import com.fasterxml.jackson.databind.ObjectMapper; import com.ruoyi.quartz.domain.SysJob; import com.ruoyi.quartz.service.ISysJobService; import com.ruoyi.system.domain.Apply; import com.ruoyi.system.domain.Perform; import com.ruoyi.system.mapper.ApplyMapper; import com.ruoyi.system.mapper.PerformMapper; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; import javax.crypto.Mac; import javax.crypto.spec.SecretKeySpec; import java.net.URI; import java.net.URLDecoder; import java.net.URLEncoder; import java.net.http.HttpClient; import java.net.http.HttpRequest; import java.net.http.HttpResponse; import java.time.Duration; import java.util.*; import java.util.concurrent.CompletableFuture; /** * 定时任务调度测试 * * @author ruoyi */ @Component("ryTask") public class RyTask { @Autowired private ISysJobService jobService; @Autowired private PerformMapper performMapper; @Autowired private ApplyMapper applyMapper; private static final Logger log = LoggerFactory.getLogger(RyTask.class); /** * 获取wmd * * @return */ private String getWMD(String data) { try { // 密钥(secret key) String secret = "1b9cb6378d959b45714bec49971ade22e6e24e42"; // 要哈希的消息数据 // 创建HMAC算法实例,指定算法为HmacSHA1 Mac hmac = Mac.getInstance("HmacSHA1"); // 创建密钥对象 SecretKeySpec secretKey = new SecretKeySpec(secret.getBytes(), "HmacSHA1"); // 初始化HMAC算法实例 hmac.init(secretKey); // 计算消息的HMAC值,结果是一个字节数组 byte[] hmacBytes = hmac.doFinal(data.getBytes()); // 将字节数组转换为Base64编码的字符串 String base64Hmac = Base64.getEncoder().encodeToString(hmacBytes); // 输出生成的HMAC Base64编码值 System.out.println("HMAC-SHA1 (Base64): " + base64Hmac); // 指定使用的字符编码(如UTF-8) String charset = "UTF-8"; // 进行URL编码 return URLEncoder.encode(base64Hmac, charset); } catch (Exception e) { log.error(e.getMessage()); } return null; } /** * 获取参数 * * @param pid * @param jobId * @throws Exception */ public void get(Long pid, Long jobId) throws Exception { //执行完成停止任务 SysJob job = new SysJob(); job.setJobId(jobId); job.setStatus("1"); jobService.changeStatus(job); //演出 Perform perform = performMapper.selectPerformById(pid); if (perform == null) { throw new Exception("未找到演出"); } //人员 Apply apply = new Apply(); apply.setPerformId(pid); List applies = applyMapper.selectApplyListAllow(apply); if (applies == null || applies.isEmpty()) { throw new Exception("没有参与人员"); } long time = System.currentTimeMillis(); String data = "/notice/v1.0/notice-" + perform.getNotice() + "?fieldSet=noticeV1&appId=be4d79eb8fc7bd008ee82c8ec4ff6fd4&language=zh-cn&os=WEB&platform=WEB&wpf=pc"; String base = "https://global.apis.naver.com/weverse/wevweb"; String wmd = getWMD(data + time); if (wmd == null || wmd.isEmpty()) { throw new Exception("wmd解析失败"); } // 创建HttpClient实例 HttpClient httpClient = HttpClient.newBuilder() .connectTimeout(Duration.ofSeconds(10)) .build(); // 创建HttpRequest String url = base + data + "&wmsgpad=" + time + "&wmd=" + wmd; System.out.println(url); HttpRequest request = HttpRequest.newBuilder() .uri(new URI(url)) .GET() .header("Authorization", "Bearer " + applies.get(0).getToken()) .header("Origin", "https://weverse.io") .header("Referer", "https://weverse.io/") .header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36") .build(); // 发送请求并获取响应 HttpResponse response = httpClient.send(request, HttpResponse.BodyHandlers.ofString()); String eValue = null; // 检查响应状态码 if (response.statusCode() == 403) { // 打印响应体 String responseBody = response.body(); WeverseResponse weverseResponse = JSONObject.parseObject(responseBody, WeverseResponse.class); if (weverseResponse == null || weverseResponse.getData() == null || weverseResponse.getData().getButton() == null || weverseResponse.getData().getButton().getWebLink() == null || weverseResponse.getData().getButton().getWebLink().getUrl() == null || weverseResponse.getData().getButton().getWebLink().getUrl().isEmpty()) { throw new Exception("button获取失败"); } URI uri = new URI(weverseResponse.getData().getButton().getWebLink().getUrl()); String query = uri.getQuery(); String[] params = query.split("&"); Map paramMap = new HashMap<>(); for (String param : params) { String[] keyValue = param.split("="); String key = URLDecoder.decode(keyValue[0], "UTF-8"); String value = URLDecoder.decode(keyValue[1], "UTF-8"); paramMap.put(key, value); } // 获取参数e的值 eValue = paramMap.get("e"); System.out.println(weverseResponse); } else { throw new Exception("button请求失败"); } if (eValue == null || eValue.isEmpty()) { throw new Exception("button为空"); } Perform perform1 = new Perform(); perform1.setId(pid); perform1.setNum(eValue); performMapper.updatePerform(perform1); } /** * 抢票 * * @param pid */ public void rob(Long pid, Long jobId) throws Exception { //status Perform pStatus1 = new Perform(); pStatus1.setStatus("2"); pStatus1.setId(pid); performMapper.updatePerform(pStatus1); //执行完成停止任务 SysJob job = new SysJob(); job.setJobId(jobId); job.setStatus("1"); jobService.changeStatus(job); //演出 Perform perform = performMapper.selectPerformById(pid); if (perform == null) { throw new Exception("未找到演出"); } //URL String url = String.format("https://faneventapi.weverse.io/api/v1/events/%s/register", perform.getNum()); //人员 Apply apply = new Apply(); apply.setPerformId(pid); List applies = applyMapper.selectApplyListAllow(apply); if (applies == null || applies.isEmpty()) { throw new Exception("没有参与人员"); } if (perform.getAgree() == null || "".equals(perform.getAgree())) { throw new Exception("协议为空"); } //协议 String[] agree = perform.getAgree().split(","); int[] agrees = Arrays.stream(agree).mapToInt(Integer::parseInt).toArray(); List req = new ArrayList<>(); for (Apply a : applies) { // 准备JSON负载 Map eventData = new HashMap<>(); eventData.put("eventId", Integer.parseInt(perform.getNum())); eventData.put("language", "zh-cn"); Map data = new HashMap<>(); data.put("birthDate", a.getBirthday()); data.put("phoneNumber", a.getCountryCode() + " " + a.getPhoneNum()); data.put("agreedTermIds", agrees); eventData.put("data", data); ObjectMapper mapper = new ObjectMapper(); String jsonPayload = mapper.writeValueAsString(eventData); System.out.println(jsonPayload); HttpClient client = HttpClient.newHttpClient(); HttpRequest request = HttpRequest.newBuilder() .uri(URI.create(url)) .header("Content-Type", "application/json") .header("Authorization", "Bearer " + a.getToken()) .header("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36") .POST(HttpRequest.BodyPublishers.ofString(jsonPayload)) .build(); req.add(new MyReq(client, request, a.getName())); } Calendar calendar = Calendar.getInstance(); calendar.setTime(perform.getStartTime()); calendar.add(Calendar.HOUR_OF_DAY, -1); long start = calendar.getTimeInMillis(); long now = System.currentTimeMillis(); Thread.sleep(start - now); log.info("抢票开始了"); for (MyReq r : req) { log.info("【用户】:" + r.getUser()); // 发起异步请求,但不等待结果 CompletableFuture> future2 = r.getClient().sendAsync(r.getRequest(), HttpResponse.BodyHandlers.ofString()); future2.thenAccept(response -> { // 处理响应(例如,记录日志) log.info("Response received: " + response.statusCode()); log.info(response.body()); }).exceptionally(ex -> { // 处理异常 ex.printStackTrace(); return null; }); } //status Perform pStatus2 = new Perform(); pStatus2.setStatus("3"); pStatus2.setId(pid); performMapper.updatePerform(pStatus2); } }