wangsiyuan 2024-05-11 17:14:02 +08:00
commit 8c4c8b3294
13 changed files with 227 additions and 50 deletions

8
README.md Normal file
View File

@ -0,0 +1,8 @@
## wepush
## 打包
```
mvn clean package
```

View File

@ -45,9 +45,10 @@ CREATE TABLE device_info (
sn VARCHAR(255),
device_model VARCHAR(255),
last_online_time BIGINT,
status INT
status INT,
remark VARCHAR(255)
);
INSERT INTO device_info (is_monitored, time_out_period, device_brand, android_version, sn, device_model, last_online_time, status) VALUES
(1, 600, 'google', 'Android 10', UNIX_TIMESTAMP(), 1),
INSERT INTO device_info (is_monitored, time_out_period, device_brand, android_version, sn, device_model, last_online_time, status,remark) VALUES
(1, 1800, 'google', 'Android 10',,'Pixel 3', UNIX_TIMESTAMP(), 1,'转发设备'),

17
docker/Dockerfile Normal file
View File

@ -0,0 +1,17 @@
# 基础镜像使用OpenJDK官方镜像标签为17-jdk-slim这是Java 17的精简版环境
FROM openjdk:17-jdk-slim
# 设置容器内的工作目录。如果该目录不存在Docker将自动为你创建它。
WORKDIR /app
# 将构建生成的可执行jar包复制到容器内。假设jar包位于target/目录下并命名为wePush-1.0.0.jar。
COPY target/wePush-1.0.0.jar ./app.jar
# 配置容器启动时运行的命令这里用来启动Spring Boot应用。
# 使用非root用户运行Java应用是一个好的安全实践。
# 这里可以根据需要添加其他JVM参数比如内存限制等。
CMD ["java", "-jar", "/app/app.jar"]
# 暴露8080端口这是Spring Boot应用默认的端口。
# 注意,实际部署时可能需要根据应用的具体配置来调整端口号。
EXPOSE 8080

View File

@ -22,18 +22,28 @@ public class WeChatMsgForwarderController {
this.handleWeChatMsgService = handleWeChatMsgService;
}
/**
*
*
* @param accessToken 访
* @param weChatMsg
* @return
*/
@PostMapping("/wechat")
public ServerResponseEntity receiveMsg(@RequestHeader("accessToken") String accessToken,
@RequestBody @Valid WeChatMsg weChatMsg){
logger.info("CallInfoO: {}",weChatMsg.toString());
logger.info("weChatMsg: {}",weChatMsg.toString());
// 检查accessToken是否为空为空则返回错误信息
if(accessToken == null){
logger.warn("Access token is missing or empty");
return ServerResponseEntity.fail("accessToken cannot be empty.");
}
// 检查weChatMsg消息体是否有效无效则返回错误信息
if (weChatMsg.isInvalid()) {
logger.debug("CallInfo has invalid fields.");
return ServerResponseEntity.fail("json body value error.");
}
// 调用服务处理微信消息
return handleWeChatMsgService.handleMsg(accessToken,weChatMsg);
}
}

View File

@ -21,4 +21,5 @@ public class DeviceInfoDAO {
private String deviceModel;
private long lastOnlineTime;
private int status;
private String remark;
}

View File

@ -21,6 +21,5 @@ public class TextMessage {
public static class Text {
private String content;
}
}

View File

@ -12,35 +12,51 @@ import java.io.IOException;
public class WeChatAPI {
private final Logger logger = LoggerFactory.getLogger(WeChatAPI.class);
/**
* 访
* @param corpid ID
* @param corpsecret
* @return 访null
*/
public String getAccessToken(String corpid,String corpsecret){
// 创建OkHttpClient实例
OkHttpClient client = new OkHttpClient();
// 设置请求URL并添加查询参数
String url = "https://qyapi.weixin.qq.com/cgi-bin/gettoken";
HttpUrl.Builder urlBuilder = HttpUrl.parse(url).newBuilder();
urlBuilder.addQueryParameter("corpid", corpid);
urlBuilder.addQueryParameter("corpsecret", corpsecret);
// 构建请求
Request request = new Request.Builder()
.url(urlBuilder.build())
.build();
try (Response response = client.newCall(request).execute()) {
// 从响应中读取body内容
String responseBody = response.body().string();
logger.info("get qyWeChat AccessToken responseBody: {}", responseBody);
// 对响应内容进行JSON解析
ObjectMapper objectMapper1 = new ObjectMapper();
QyWeChatGetAccessTokenApiResponse qyWeChatGetAccessTokenApiResponse = objectMapper1.readValue(responseBody, QyWeChatGetAccessTokenApiResponse.class);
// 检查是否获取令牌成功
if (qyWeChatGetAccessTokenApiResponse.getErrcode() == 0){
logger.debug("get qyWeChat AccessToken Success");
// 返回访问令牌
return qyWeChatGetAccessTokenApiResponse.getAccess_token();
} else {
// 记录获取令牌失败的日志
logger.error("getAccessToken error: {}",qyWeChatGetAccessTokenApiResponse.getErrmsg());
return null;
}
} catch (IOException e) {
// 记录异常信息
logger.error("getAccessToken error:",e);
return null;
}
}
}

View File

@ -27,17 +27,42 @@ public class ApiSettingService {
}
return apiSetting;
}
/**
* API
* API
*
* 1.
* 2.
* 3. API
* 4.
* 5. APInull
* 6. API
*/
private void queryApiSetting() {
// 从用户配置中获取电话号码
String phoneNumberToSearch = userConfig.getPhoneNumber();
// 构建查询条件
QueryWrapper<ApiSetting> wrapper = new QueryWrapper<>();
wrapper.eq("phone_number", phoneNumberToSearch);
// 根据查询条件查询API设置信息
ApiSetting result = apiSettingMapper.selectOne(wrapper);
// 记录查询结果
logger.info("result: {}", result);
// 处理查询结果为空的情况
if (result == null) {
apiSetting = null;
// 记录错误日志
logger.error("updateAccessToken error");
}
// 更新API设置成员变量
apiSetting = result;
// 记录更新日志
logger.info("get apiSetting from mysql,accessToken: {}", apiSetting);
}
}

View File

@ -30,37 +30,67 @@ public class CallService {
this.tokenService = tokenService;
}
/**
* accessToken
* accessToken
*
*
* @param accessToken accessToken
* @param callInfo
* @return ID
*/
public ServerResponseEntity getPhoneNumber(String accessToken, CallInfo callInfo) {
// 通过token服务获取正确的accessToken
String correctAccessToken = tokenService.getApiAccessToken();
logger.info("accessToken: {} correctAccessToken: {}", accessToken, correctAccessToken);
// 验证提供的accessToken是否有效
if (!correctAccessToken.equals(accessToken)) {
return ServerResponseEntity.fail("Invalid accessToken");
}
// 从呼叫信息中获取电话号码和呼叫时间
String phoneNumber = callInfo.getPhoneNumber();
String callTime = callInfo.getCallTime();
// 获取企业微信的访问令牌
String qyAccessToken = tokenService.getAccessToken();
// 使用企业微信的accessToken、电话号码和呼叫时间发送消息并获取响应
QyWeChatSendMessageApiResponse qyWeChatSendMessageApiResponse = requestWithOkhttp(qyAccessToken, phoneNumber, callTime);
// 检查发送消息的响应代码
int code = qyWeChatSendMessageApiResponse.getErrcode();
if (code == 0) {
// 如果消息发送成功,构建并返回成功的服务器响应
SendMessageResonse sendMessageResonse = new SendMessageResonse();
sendMessageResonse.setMsgid(qyWeChatSendMessageApiResponse.getMsgid());
return ServerResponseEntity.success(sendMessageResonse);
} else {
// 如果消息发送失败,返回失败的服务器响应
return ServerResponseEntity.fail("request qyWeChat error,please check server error log.");
}
}
/**
*
*
* @param phoneNumber
* @param callTime
* @return TextCardMessage
*/
public TextCardMessage setTextCardMessage(String phoneNumber, String callTime) {
TextCardMessage textCardMessage = new TextCardMessage();
// 从api设置中获取用户信息并设置到textCardMessage中
textCardMessage.setTouser(apiSettingService.getApiSetting().getTouser());
textCardMessage.setMsgtype(apiSettingService.getApiSetting().getMsgtype());
textCardMessage.setAgentid(apiSettingService.getApiSetting().getAgentid());
textCardMessage.setEnable_duplicate_check(apiSettingService.getApiSetting().getEnableDuplicateCheck());
textCardMessage.setDuplicate_check_interval(apiSettingService.getApiSetting().getDuplicateCheckInterval());
// 创建TextCard对象并设置其内容
TextCardMessage.TextCard textCard = new TextCardMessage.TextCard();
textCard.setTitle("新来电通知");
// 设置描述信息,包含来电号码和来电时间
textCard.setDescription("<div class='gray'>您有一通新来电</div><div class='highlight'>来电号码: " + phoneNumber + "</div><div class='highlight'>来电时间: " + callTime + "</div>如果您认识这个号码,请及时回电。如果不认识,请注意可能是垃圾电话或诈骗电话。");
textCard.setUrl("https://kimgo.cn");
// 将TextCard对象设置到TextCardMessage中
textCardMessage.setTextcard(textCard);
@ -69,12 +99,23 @@ public class CallService {
return textCardMessage;
}
/**
* 使OkHttp
*
* @param accessToken 访访API
* @param phoneNumber
* @param callTime
* @return APInull
*/
public QyWeChatSendMessageApiResponse requestWithOkhttp(String accessToken, String phoneNumber, String callTime) {
// 构建发送文本卡片消息的URL
String url = qyWeChatURLService.getSendTextCardMessageUrl() + "?access_token=" + accessToken;
OkHttpClient client = new OkHttpClient();
// 设置文本卡片消息内容
TextCardMessage textCardMessage = setTextCardMessage(phoneNumber, callTime);
// 使用Jackson进行序列化
// 将消息对象序列化为JSON字符串
ObjectMapper objectMapper = new ObjectMapper();
String jsonBody;
try {
@ -84,22 +125,27 @@ public class CallService {
logger.error("JSON processing error", e);
return null;
}
// 构建请求体
// 构建HTTP请求体
RequestBody body = RequestBody.create(MediaType.parse("application/json; charset=utf-8"), jsonBody);
Request request = new Request.Builder().url(url).post(body).build();
try (Response response = client.newCall(request).execute()) {
// 从响应中读取内容
String responseBody = response.body().string();
logger.info("request sendMessage api ResponseBody: {}", responseBody);
// 反序列化API响应
ObjectMapper objectMapper1 = new ObjectMapper();
QyWeChatSendMessageApiResponse apiResponse = objectMapper1.readValue(responseBody, QyWeChatSendMessageApiResponse.class);
// 处理不同的API响应结果
if (apiResponse.getErrcode() == 0) {
logger.debug("Request qyWeChat Success.");
return apiResponse;
} else if (apiResponse.getErrcode() == 42001 || apiResponse.getErrcode() == 40014) {
// 处理访问令牌过期的情况,刷新令牌后重新发送消息
logger.info("Access token expired. Refreshing token...");
tokenService.setAccessToken(); // 一个方法来刷新accessToken
tokenService.setAccessToken(); // 刷新accessToken的方法
String newAccessToken = tokenService.getAccessToken();
return requestWithOkhttp(newAccessToken, phoneNumber, callTime);
} else {

View File

@ -102,7 +102,7 @@ public class HeartBeatService {
try {
DeviceInfoDAO deviceInfoDAO = deviceInfoDAOMapper.selectOne(queryWrapper);
if (deviceInfoDAO != null) {
logger.info("Data is queried in the database based on the SN");
//logger.info("Data is queried in the database based on the SN");
return deviceInfoDAO;
}
return null; // 数据不存在
@ -120,9 +120,8 @@ public class HeartBeatService {
* @return DeviceInfoDAO
*/
private DeviceInfoDAO convertToDeviceInfoDAO(DeviceInfo deviceInfo) {
logger.info("start convert to device info dao");
// 这里需要实现将 DeviceInfo 转换为 DeviceInfoDAO 的逻辑
// 示例代码(根据实际字段调整):
// logger.info("start convert to device info dao");
DeviceInfoDAO deviceInfoDAO = new DeviceInfoDAO();
deviceInfoDAO.setIsMonitored(1);
deviceInfoDAO.setDeviceBrand(deviceInfo.getDeviceBrand());

View File

@ -33,50 +33,65 @@ public class MonitorService {
this.deviceInfoDAOMapper = deviceInfoDAOMapper;
}
/**
* 线
* 480000线
* 线线
*
* @Scheduled fixedDelay
*/
@Scheduled(fixedDelay = 480000)
public void monitorOnlineDevices() {
List<DeviceInfoDAO> devices;
try {
// 尝试从数据库获取所有设备信息
devices = getAllDevices();
} catch (DatabaseQueryException e) {
// 查询设备信息出错时的处理逻辑
logger.error("monitor online devices error.");
return;
}
if (devices.isEmpty()) {
// 没有找到数据的处理逻辑
// 如果查询到的设备列表为空,记录日志并返回
logger.info("no data was queried.");
return;
}
for (DeviceInfoDAO device : devices) {
try {
// 如果设备不被监控,则跳过该设备的处理
if (device.getIsMonitored() != 1) {
logger.info("device is not monitored ,device SN: {}", device.getSN());
continue;
}
// 计算设备最后上线时间与当前时间的差值
long currentTimeMillis = System.currentTimeMillis();
long timeDifferenceMillis = currentTimeMillis - device.getLastOnlineTime();
long timeDifferenceSeconds = timeDifferenceMillis / 1000;
// 如果设备超出设定的超时周期未上线,则认为设备离线,并进行相应处理
if (timeDifferenceSeconds > device.getTimeOutPeriod()) {
logger.info("Device offline. SN: {}, LastOnlineTime: {}, TimeDifference: {}, TimeoutPeriod: {}",
device.getSN(), device.getLastOnlineTime(), timeDifferenceSeconds, device.getTimeOutPeriod());
updateStatus(device, 0); // 设备离线
//logger.info("Device offline. SN: {}, LastOnlineTime: {}, TimeDifference: {}, TimeoutPeriod: {}",
//device.getSN(), device.getLastOnlineTime(), timeDifferenceSeconds, device.getTimeOutPeriod());
updateStatus(device, 0); // 更新设备状态为离线
logger.info("Update the device status to offline.");
sendNotification(device);
sendNotification(device); // 发送设备离线通知
} else {
updateStatus(device, 1); // 设备在线
logger.info("update the device status to online.");
// 如果设备仍在设定的超时周期内上线,则认为设备在线,并更新设备状态
updateStatus(device, 1); // 更新设备状态为在线
//logger.info("update the device status to online.");
}
} catch (DatabaseUpdateException dbe) {
// 更新数据库出错时的处理逻辑
logger.error("update to mysql error.");
return;
}
catch (Exception e) {
} catch (Exception e) {
// 处理设备信息处理过程中出现的其他异常
logger.error("Error processing device with SN " + device.getSN(), e);
}
}
}
/**
* DeviceInfoDAOid
*
@ -95,60 +110,99 @@ public class MonitorService {
}
/**
*
*
* @param deviceInfoDAO 访
*
*/
private void sendNotification(DeviceInfoDAO deviceInfoDAO) {
// 创建发送企业微信消息的请求对象,并初始化它
SendQyWeChatMsgRequest sendQyWeChatMsgRequest = new SendQyWeChatMsgRequest(tokenService);
// 获取访问令牌
String accessToken = tokenService.getAccessToken();
// 获取发送文本卡片消息的URL
String url = qyWeChatURLService.getSendTextCardMessageUrl();
// 发送请求,并获取响应
QyWeChatSendMessageApiResponse qyWeChatSendMessageApiResponse = sendQyWeChatMsgRequest.okhttpRequest(url,
accessToken, setTextCardMessage(deviceInfoDAO));
// 如果发送响应为空,则记录发送失败
if (qyWeChatSendMessageApiResponse == null) {
logger.info("Send Notification Fail");
}
// 如果发送响应不为空且错误码为0记录发送成功
if (qyWeChatSendMessageApiResponse != null && qyWeChatSendMessageApiResponse.getErrcode() == 0) {
logger.info("Send Notification Success");
}
}
/**
*
*
* @param deviceInfoDAO 访
* @return TextCardMessage
*/
public TextCardMessage setTextCardMessage(DeviceInfoDAO deviceInfoDAO) {
TextCardMessage textCardMessage = new TextCardMessage();
// 设置消息的基本信息如接收用户、消息类型和企业ID
textCardMessage.setTouser(apiSettingService.getApiSetting().getTouser());
textCardMessage.setMsgtype(apiSettingService.getApiSetting().getMsgtype());
textCardMessage.setAgentid(apiSettingService.getApiSetting().getAgentid());
textCardMessage.setEnable_duplicate_check(0);
// 创建文本卡片对象,并设置其内容
TextCardMessage.TextCard textCard = new TextCardMessage.TextCard();
textCard.setTitle("设备掉线通知");
// 使用当前时间格式化字符串
String formattedCurrentTime = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
// 设置卡片的描述信息,包含设备的详细掉线通知
textCard.setDescription("<div class='gray'>设备信息</div>" +
"<div class='highlight'>品牌: " + deviceInfoDAO.getDeviceBrand() + "</div>" +
"<div class='highlight'>设备型号: " + deviceInfoDAO.getDeviceModel() + "</div>" +
"<div class='highlight'>安卓版本: " + deviceInfoDAO.getAndroidVersion() + "</div>" +
"<div class='highlight'>SN: " + deviceInfoDAO.getSN() + "</div>" +
"<div class='highlight'>监测到掉线时间: " + formattedCurrentTime + "</div>" +
"<div class='highlight'>备注: " + deviceInfoDAO.getRemark() + "</div>" +
"检测到设备已掉线,请及时查看设备状态.");
textCard.setUrl("https://kimgo.cn");
// 将TextCard对象设置到TextCardMessage中
// 将构建的TextCard对象设置到TextCardMessage中
textCardMessage.setTextcard(textCard);
// 记录文本卡片消息的内容
logger.info("TextCardMessage: {}", textCardMessage);
return textCardMessage;
}
/**
*
*
* @param device SN
* @param status
* @throws DatabaseUpdateException
*/
private void updateStatus(DeviceInfoDAO device, int status) throws DatabaseUpdateException {
try {
// 构建更新条件,设置更新的状态值
UpdateWrapper<DeviceInfoDAO> updateWrapper = new UpdateWrapper<>();
updateWrapper.eq("SN", device.getSN());
updateWrapper.set("status", status);
// 执行更新操作
int result = deviceInfoDAOMapper.update(null, updateWrapper);
// 根据更新结果进行日志记录
if (result > 0) {
logger.debug("Update successful for SN: " + device.getSN());
} else {
logger.warn("Update failed: No rows affected for SN: " + device.getSN());
}
} catch (Exception e) {
// 捕获异常,记录错误日志,并抛出数据库更新异常
logger.error("Error updating status for SN: " + device.getSN(), e);
throw new DatabaseUpdateException("update to mysql error.", e);
}
}
}

View File

@ -32,11 +32,13 @@ public class QyWeChatURLService {
QueryWrapper<QyWeChatURL> wrapper = new QueryWrapper<>();
wrapper.eq("phone_number", phoneNumberToSearch);
QyWeChatURL result = qyWeChatURLMapper.selectOne(wrapper);
if (result == null){
if (result != null){
logger.error("queryURL error");
}
logger.info("queryURL result {}",result.toString());
logger.info("queryURL result {}",result);
sendTextCardMessageUrl = result.getSendTextCardMessageUrl();
qyWechatGetTokenUrl = result.getQyWechatGetTokenUrl();
};
} else {
logger.error("queryURL error.");
}
}
}

View File

@ -11,7 +11,6 @@
<pattern>%date %level [%thread] %logger{10} [%file:%line] %msg%n</pattern>
</encoder>
</appender>
<root level="info">
<appender-ref ref="STDOUT"/>
<appender-ref ref="FILE"/>