forked from rongchao/epmet-cloud-rizhao
wangxianzhang
3 years ago
7 changed files with 312 additions and 260 deletions
@ -0,0 +1,213 @@ |
|||
package com.epmet.processor; |
|||
|
|||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|||
import com.epmet.commons.tools.distributedlock.DistributedLock; |
|||
import com.epmet.commons.tools.exception.ExceptionUtils; |
|||
import com.epmet.constant.EpidemicConstant; |
|||
import com.epmet.dao.IcSyncJobDao; |
|||
import com.epmet.entity.IcSyncJobEntity; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.redisson.api.RLock; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.util.CollectionUtils; |
|||
|
|||
import javax.annotation.PostConstruct; |
|||
import java.net.Inet4Address; |
|||
import java.net.InetAddress; |
|||
import java.net.UnknownHostException; |
|||
import java.util.List; |
|||
import java.util.concurrent.ExecutorService; |
|||
import java.util.concurrent.TimeUnit; |
|||
|
|||
/** |
|||
* @ClassName BaseSyncProcessor |
|||
* @Description 抽象数据同步处理器基类。所有基于该模式的同步器都应当继承该类。自定义Scheduler之后，调用基类的scanAndExeJobs()方法。 |
|||
* 实现需要实现的方法(定义分布式锁的key，定义同步数据的类型)。有特殊逻辑的可以重写相关方法 |
|||
* @Author wangxianzhang |
|||
* @Date 2022/11/19 下午11:16 |
|||
*/ |
|||
@Slf4j |
|||
public abstract class AbstractDataSyncJobProcessor { |
|||
|
|||
@Autowired |
|||
private ExecutorService executorService; |
|||
|
|||
@Autowired |
|||
private IcSyncJobDao icSyncJobDao; |
|||
|
|||
@Autowired |
|||
private DistributedLock distributedLock; |
|||
|
|||
/** |
|||
* 最多能运行几条线程。如果有特殊情况，可以通过抽象方法每个子类单独定义 |
|||
*/ |
|||
private int MAX_EXECUTING_NUMBER = 3; |
|||
|
|||
/** |
|||
* 本机ip |
|||
*/ |
|||
private String localIp; |
|||
|
|||
{ |
|||
try { |
|||
// 获取本机ip
|
|||
InetAddress localHost = Inet4Address.getLocalHost(); |
|||
localIp = localHost.getHostAddress(); |
|||
log.info("【抽象定时数据同步器】获取本机ip为:{}", localIp); |
|||
} catch (UnknownHostException e) { |
|||
log.error("【抽象定时数据同步器】获取本机ip失败"); |
|||
} |
|||
|
|||
} |
|||
|
|||
/** |
|||
* 扫描并执行 |
|||
* @author wxz |
|||
* @date 2022/11/20 上午12:10 |
|||
* |
|||
* |
|||
*/ |
|||
void scanAndExecWaitingJobs() { |
|||
|
|||
int executingTaskNumber = getExecutingTaskNumber(); |
|||
if (executingTaskNumber >= MAX_EXECUTING_NUMBER) { |
|||
// 单个实例下，最多只允许同时3条线程运行
|
|||
return; |
|||
} |
|||
|
|||
// 还可以运行几条线程
|
|||
int leftCount = MAX_EXECUTING_NUMBER - executingTaskNumber; |
|||
RLock lock = null; |
|||
try { |
|||
lock = distributedLock.getLock(getDistributeLockKey(), getDistributeLockLeaseTime(), getDistributeLockWaitTime(), TimeUnit.SECONDS); |
|||
// 查询可执行的任务列表，并且异步执行
|
|||
List<IcSyncJobEntity> icSyncJobToExec = icSyncJobDao.selectJobListByStatus("waiting", getJobType(), null, leftCount); |
|||
if (!CollectionUtils.isEmpty(icSyncJobToExec)) { |
|||
// 异步提交任务
|
|||
submitAsyncJob(icSyncJobToExec); |
|||
} |
|||
} catch (Exception e) { |
|||
log.error("【异步数据更新】出错:{}", ExceptionUtils.getErrorStackTrace(e)); |
|||
} finally { |
|||
if (lock != null) { |
|||
lock.unlock(); |
|||
} |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 扫描并且执行被中断的任务 |
|||
* @author wxz |
|||
* @date 2022/11/20 下午10:29 |
|||
* |
|||
* |
|||
*/ |
|||
@PostConstruct |
|||
void scanAndContinueInteruptedJobs() { |
|||
// 此处不需要加锁，因为已经按照ip查询了，查到的一定是本机之前执行过的，其他机器查不到该条
|
|||
List<IcSyncJobEntity> interuptedJobs = icSyncJobDao.selectJobListByStatus("processing", getJobType(), localIp,null); |
|||
if (!CollectionUtils.isEmpty(interuptedJobs)) { |
|||
submitAsyncJob(interuptedJobs); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 提交异步任务 |
|||
* @author wxz |
|||
* @date 2022/11/20 下午10:37 |
|||
* * @param icSyncJobToExec |
|||
* |
|||
*/ |
|||
void submitAsyncJob(List<IcSyncJobEntity> icSyncJobToExec) { |
|||
for (IcSyncJobEntity jobEntity : icSyncJobToExec) { |
|||
// 将此任务状态修改为执行中
|
|||
updateJobStatus(jobEntity.getId(), EpidemicConstant.OPERATION_STATUS_PROCESSING, localIp); |
|||
executorService.submit(() -> { |
|||
try { |
|||
execJobTask(jobEntity); |
|||
} finally { |
|||
// 更新任务状态为结束
|
|||
updateJobStatus(jobEntity.getId(), EpidemicConstant.OPERATION_STATUS_FINISH, null); |
|||
} |
|||
}); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 先检查一下，有几个任务需要执行 |
|||
* @author wxz |
|||
* @date 2022/11/19 下午10:57 |
|||
* @return int |
|||
*/ |
|||
private int getExecutingTaskNumber() { |
|||
// 查询本机正在处理中的任务列表
|
|||
LambdaQueryWrapper<IcSyncJobEntity> executingListQuery = new LambdaQueryWrapper<>(); |
|||
executingListQuery.eq(IcSyncJobEntity::getOperationStatus, EpidemicConstant.OPERATION_STATUS_PROCESSING); |
|||
executingListQuery.eq(IcSyncJobEntity::getProcessorIp, localIp); |
|||
return icSyncJobDao.selectCount(executingListQuery); |
|||
} |
|||
|
|||
/** |
|||
* 更新任务状态 |
|||
* @author wxz |
|||
* @date 2022/11/8 下午8:25 |
|||
* @param id |
|||
* @param status |
|||
|
|||
*/ |
|||
private void updateJobStatus(String id, String status, String processorIp) { |
|||
LambdaQueryWrapper<IcSyncJobEntity> query = new LambdaQueryWrapper<>(); |
|||
query.eq(IcSyncJobEntity::getId, id); |
|||
|
|||
IcSyncJobEntity updateEntity = new IcSyncJobEntity(); |
|||
updateEntity.setOperationStatus(status); |
|||
updateEntity.setProcessorIp(processorIp); |
|||
icSyncJobDao.update(updateEntity, query); |
|||
} |
|||
|
|||
/** |
|||
* 执行任务 |
|||
* @author wxz |
|||
* @date 2022/11/20 上午12:06 |
|||
* |
|||
* |
|||
*/ |
|||
protected abstract void execJobTask(IcSyncJobEntity jobEntity); |
|||
|
|||
/** |
|||
* 获取分布式锁的key |
|||
* @author wxz |
|||
* @date 2022/11/20 上午12:21 |
|||
* |
|||
* |
|||
*/ |
|||
protected abstract String getDistributeLockKey(); |
|||
|
|||
/** |
|||
* 任务类型 |
|||
* @author wxz |
|||
* @date 2022/11/20 上午12:25 |
|||
* |
|||
* * @return String |
|||
*/ |
|||
protected abstract String getJobType(); |
|||
|
|||
/** |
|||
* 分布式锁的续期时间 |
|||
* @author wxz |
|||
* @date 2022/11/20 下午10:52 |
|||
* |
|||
* * @return long |
|||
*/ |
|||
protected abstract long getDistributeLockLeaseTime(); |
|||
|
|||
/** |
|||
* 分布式锁的等待时间 |
|||
* @author wxz |
|||
* @date 2022/11/20 下午10:52 |
|||
* |
|||
* * @return long |
|||
*/ |
|||
protected abstract long getDistributeLockWaitTime(); |
|||
|
|||
} |
|||
Loading…
Reference in new issue