ElasticJob案列

配置文件

elasticjob:
  serverlists: 192.168.0.205:2181
#  serverlists: 192.168.1.188:2181
  namespace: tcs-job
  cron: 0 0/30 * * * ?

 

 

import com.dangdang.ddframe.job.reg.zookeeper.ZookeeperConfiguration;
import com.dangdang.ddframe.job.reg.zookeeper.ZookeeperRegistryCenter;
import com.paascloud.provider.job.ElasticJobListener;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.Resource;

@Configuration
@Data
@ConfigurationProperties(prefix = "elasticjob")
public class ElasticJobConfig {

    private String serverlists;

    private String namespace;

    @Bean
    public ZookeeperConfiguration zkConfig() {
        return new ZookeeperConfiguration(serverlists, namespace);
    }

    @Bean(initMethod = "init")
    public ZookeeperRegistryCenter regCenter(ZookeeperConfiguration config) {
        return new ZookeeperRegistryCenter(config);
    }

    @Bean
    public ElasticJobListener elasticJobListener() {
        //初始化要给定超时多少秒重连
        return new ElasticJobListener(100, 100);
    }
}

import com.dangdang.ddframe.job.api.ShardingContext;
import com.dangdang.ddframe.job.api.simple.SimpleJob;
import com.paascloud.DateUtil;
import com.paascloud.provider.mapper.AssignmentCrontabJobMapper;
import com.paascloud.provider.model.pojo.AssignmentCrontabJobPojo;
import com.paascloud.provider.service.IAddCronService;
import com.paascloud.provider.utils.DateUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.List;

@Slf4j
@Component
public class AddCrontabElasticJob implements SimpleJob {

    @Autowired
    private IAddCronService addCronService;

    @Override
    public void execute(ShardingContext shardingContext) {
        //打印出任务相关信息，JobParameter用于传递任务的ID
        log.info("任务名：{}, 片数：{}, id={}", shardingContext.getJobName(), shardingContext.getShardingTotalCount(),
                shardingContext.getJobParameter());
        addCronService.loadCronList();
    }
}

import com.dangdang.ddframe.job.api.simple.SimpleJob;
import com.dangdang.ddframe.job.config.JobCoreConfiguration;
import com.dangdang.ddframe.job.config.simple.SimpleJobConfiguration;
import com.dangdang.ddframe.job.lite.config.LiteJobConfiguration;
import com.dangdang.ddframe.job.lite.spring.api.SpringJobScheduler;
import com.dangdang.ddframe.job.reg.zookeeper.ZookeeperRegistryCenter;
import com.paascloud.provider.mapper.AssignmentCrontabJobMapper;
import com.paascloud.provider.model.pojo.AssignmentCrontabJobPojo;
import com.paascloud.provider.service.IAddCronService;
import com.paascloud.provider.utils.DateUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.util.Date;
import java.util.List;


@Component
@Slf4j
public class ElasticJobHandler {
    @Resource
    private ZookeeperRegistryCenter registryCenter;
    @Resource
    private ElasticJobListener elasticJobListener;

    @Autowired
    private PushMessageElasticJob pushMessageElasticJob;

    @Autowired
    private AddCrontabElasticJob addCrontabElasticJob;

    @Value("${elasticjob.cron}")
    private String cron;

    /**
     * @param jobName
     * @param jobClass
     * @param shardingTotalCount
     * @param cron
     * @param id                 数据ID
     * @return
     */
    private static LiteJobConfiguration.Builder simpleJobConfigBuilder(String jobName,
                                                                       Class<? extends SimpleJob> jobClass,
                                                                       int shardingTotalCount,
                                                                       String cron,
                                                                       String id) {
        return LiteJobConfiguration.newBuilder(new SimpleJobConfiguration(
                JobCoreConfiguration.newBuilder(jobName, cron, shardingTotalCount).jobParameter(id).build(), jobClass.getCanonicalName()));
    }

    /**
     * 添加一个定时任务
     *
     * @param jobName            任务名
     * @param cron               表达式
     * @param shardingTotalCount 分片数
     */
    public void addJob(String jobName, String cron, Integer shardingTotalCount, String id) {
        LiteJobConfiguration jobConfig = simpleJobConfigBuilder(jobName, PushMessageElasticJob.class, shardingTotalCount, cron, id)
                .overwrite(true).build();
        new SpringJobScheduler(pushMessageElasticJob, registryCenter, jobConfig, elasticJobListener).init();
    }

    @PostConstruct
    public void intit() {
        log.info("初始化任务。。。");
        LiteJobConfiguration jobConfig = simpleJobConfigBuilder("定时任务", PushMessageElasticJob.class, 1, cron, "-1")
                .overwrite(true).build();
        new SpringJobScheduler(addCrontabElasticJob, registryCenter, jobConfig, elasticJobListener).init();
    }


}

import com.dangdang.ddframe.job.executor.ShardingContexts;
import com.dangdang.ddframe.job.lite.api.listener.AbstractDistributeOnceElasticJobListener;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.Date;

public class ElasticJobListener  extends AbstractDistributeOnceElasticJobListener {

    @Autowired
    ElasticJobHandler elasticJobHandler;

    public ElasticJobListener(long startedTimeoutMilliseconds, long completedTimeoutMilliseconds) {
        super(startedTimeoutMilliseconds, completedTimeoutMilliseconds);
    }

    @Override
    public void doBeforeJobExecutedAtLastStarted(ShardingContexts shardingContexts) {
//        System.out.println("========初始化任务前========");
//        System.out.println(shardingContexts.getJobParameter());
//        System.out.println(new Date());
    }

    @Override
    public void doAfterJobExecutedAtLastCompleted(ShardingContexts shardingContexts) {
//        System.out.println("=======初始化任务后=============");
    }
}

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.dangdang.ddframe.job.api.ShardingContext;
import com.dangdang.ddframe.job.api.simple.SimpleJob;
import com.paascloud.PublicUtil;
import com.paascloud.model.dto.JpushDto;
import com.paascloud.provider.mapper.AssignmentConfMapper;
import com.paascloud.provider.model.pojo.AssignmentConfPojo;
import com.paascloud.provider.model.pojo.AssignmentUserSendPojo;
import com.paascloud.rabbitmq.sender.JpushSender;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;

@Slf4j
@Component
public class PushMessageElasticJob implements SimpleJob {

    @Autowired
    private AssignmentConfMapper confMapper;

    @Autowired
    private JpushSender jpushSender;
    @Override
    public void execute(ShardingContext shardingContext) {
        //打印出任务相关信息，JobParameter用于传递任务的ID
        log.info("任务名：{}, 片数：{}, id={}", shardingContext.getJobName(), shardingContext.getShardingTotalCount(),
                shardingContext.getJobParameter());
        AssignmentConfPojo pojo = confMapper.selectByPrimaryKey(shardingContext.getJobParameter());
        if (null != pojo) {
            String parramStr = JSONObject.toJSONString(pojo);
            JpushDto jpushDto = new JpushDto();
            jpushDto.setTitle("任务日历");
            jpushDto.setData(parramStr);
            jpushDto.setType("1");
            jpushDto.setMsg("我是极光推任务");
            List<AssignmentUserSendPojo> assignmentUserSendPojoList = confMapper.selectAllUserByConfId(pojo.getId());
            if(PublicUtil.isNotEmpty(assignmentUserSendPojoList)) {
                List<String> userList=new ArrayList<>();
                assignmentUserSendPojoList.stream().forEach(o->{
                    userList.add(o.getRemindUser());
                });
                jpushDto.setAlias(userList);
                try {
                    jpushSender.sendMessage(jpushDto);           // 发送mq消息
                } catch (Exception e) {
                    log.info("极光推送失败" + e);
                }
            }
            log.info("========我是极光推任务==========={}==================", JSON.toJSON(pojo));
        }
    }
}

import com.paascloud.provider.service.IAddCronService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Component
@Slf4j
public class StartLoadCronJob {

    @Autowired
    private IAddCronService addCronService;

    @PostConstruct
    public void intit(){
        log.info("系统启动中。。。增加近半小时任务");
        addCronService.loadCronList();
    }

}