353 lines
13 KiB
Java
353 lines
13 KiB
Java
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;
|
||
import java.util.concurrent.ExecutorService;
|
||
import java.util.concurrent.Executors;
|
||
|
||
/**
|
||
* 定时任务调度测试
|
||
*
|
||
* @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编码值
|
||
// 指定使用的字符编码（如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<Apply> applies = applyMapper.selectApplyListAllow(apply);
|
||
|
||
if (applies == null || applies.isEmpty()) {
|
||
throw new Exception("没有参与人员");
|
||
}
|
||
|
||
if (perform.getNum() == null || perform.getNum().isEmpty()){
|
||
throw new Exception("Num为空");
|
||
}
|
||
|
||
// String eValue = null;
|
||
// if (perform.getNum() == null || perform.getNum().isEmpty()) {
|
||
// 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;
|
||
// 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<String> response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());
|
||
//
|
||
// // 检查响应状态码
|
||
// if (response.statusCode() == 200) {
|
||
// // 打印响应体
|
||
// String responseBody = response.body();
|
||
// WeverseData weverseData = JSONObject.parseObject(responseBody, WeverseData.class);
|
||
// if (weverseData == null ||
|
||
// weverseData.getButton() == null ||
|
||
// weverseData.getButton().getWebLink() == null ||
|
||
// weverseData.getButton().getWebLink().getUrl() == null ||
|
||
// weverseData.getButton().getWebLink().getUrl().isEmpty()) {
|
||
// throw new Exception("button获取失败");
|
||
// }
|
||
// URI uri = new URI(weverseData.getButton().getWebLink().getUrl());
|
||
//
|
||
// String query = uri.getQuery();
|
||
// String[] params = query.split("&");
|
||
// Map<String, String> 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");
|
||
// } else {
|
||
// throw new Exception("button请求失败");
|
||
// }
|
||
//
|
||
// if (eValue == null || eValue.isEmpty()) {
|
||
// throw new Exception("button为空");
|
||
// }
|
||
// perform1.setNum(eValue);
|
||
// } else {
|
||
// eValue = perform.getNum();
|
||
// }
|
||
|
||
String eValue = perform.getNum();
|
||
Perform perform1 = new Perform();
|
||
// 创建HttpClient实例
|
||
HttpClient httpClient = HttpClient.newBuilder()
|
||
.connectTimeout(Duration.ofSeconds(10))
|
||
.build();
|
||
|
||
HttpRequest request = HttpRequest.newBuilder()
|
||
.uri(new URI("https://faneventapi.weverse.io/api/v1/user/events/" + eValue))
|
||
.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<String> response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());
|
||
if (response.statusCode() == 200) {
|
||
WeverseAgree weverseAgree = JSONObject.parseObject(response.body(), WeverseAgree.class);
|
||
System.out.println(weverseAgree);
|
||
if (weverseAgree == null || weverseAgree.getMetadata() == null ||
|
||
weverseAgree.getMetadata().getForm() == null ||
|
||
weverseAgree.getMetadata().getForm().getTerms() == null ||
|
||
weverseAgree.getMetadata().getForm().getTerms().isEmpty()) {
|
||
throw new Exception("协议未找到");
|
||
}
|
||
|
||
StringBuffer sb = new StringBuffer();
|
||
weverseAgree.getMetadata().getForm().getTerms().forEach(str -> {
|
||
sb.append(str.getId()).append(",");
|
||
});
|
||
sb.deleteCharAt(sb.length() - 1);
|
||
|
||
String agree = sb.toString();
|
||
if (!agree.isEmpty()) {
|
||
perform1.setAgree(agree);
|
||
}
|
||
log.info(agree);
|
||
}
|
||
|
||
perform1.setId(pid);
|
||
performMapper.updatePerform(perform1);
|
||
}
|
||
|
||
/**
|
||
* 抢票
|
||
*
|
||
* @param pid
|
||
*/
|
||
public void rob(Long pid, Long jobId) throws Exception {
|
||
log.info("pid:" + pid + " 任务开始了");
|
||
//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<Apply> 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<MyReq> req = new ArrayList<>();
|
||
ExecutorService executorService = Executors.newFixedThreadPool(100);
|
||
HttpClient client = HttpClient.newHttpClient();
|
||
// 创建一个请求列表
|
||
List<CompletableFuture<String>> futures = new ArrayList<>();
|
||
|
||
for (Apply a : applies) {
|
||
// 准备JSON负载
|
||
Map<String, Object> eventData = new HashMap<>();
|
||
eventData.put("eventId", Integer.parseInt(perform.getNum()));
|
||
eventData.put("language", "zh-cn");
|
||
Map<String, Object> 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);
|
||
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(request, a.getWeverseEmail()));
|
||
}
|
||
|
||
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 - perform.getDelayTime());
|
||
log.info("抢票开始了");
|
||
for (MyReq r : req) {
|
||
log.info("【用户】：" + r.getUser());
|
||
// 发起异步请求，但不等待结果
|
||
CompletableFuture<String> future2 = client.sendAsync(r.getRequest(), HttpResponse.BodyHandlers.ofString())
|
||
.thenApplyAsync(HttpResponse::body, executorService);
|
||
futures.add(future2);
|
||
}
|
||
|
||
// 等待所有请求完成
|
||
CompletableFuture.allOf(futures.toArray(new CompletableFuture[0])).join();
|
||
|
||
// 输出所有请求的响应（可选）
|
||
for (CompletableFuture<String> future : futures) {
|
||
try {
|
||
log.info(future.get());
|
||
} catch (Exception e) {
|
||
log.error(e.getMessage());
|
||
}
|
||
}
|
||
// 关闭线程池
|
||
executorService.shutdown();
|
||
|
||
//status
|
||
Perform pStatus2 = new Perform();
|
||
pStatus2.setStatus("3");
|
||
pStatus2.setId(pid);
|
||
performMapper.updatePerform(pStatus2);
|
||
|
||
for (Apply upApply : applies) {
|
||
Apply apply1 = new Apply();
|
||
apply1.setId(upApply.getId());
|
||
apply1.setStatus("6");
|
||
applyMapper.updateApply(apply1);
|
||
}
|
||
}
|
||
|
||
|
||
}
|