zookeeper初体验之关于解决quartz重复执行任务的一种思路

前阵子工作中遇到了一个很麻烦的问题。
本人所在的项目组做了一个机遇quartz集群的任务系统。通俗点讲就是用quartz框架(quartz是一款能跑定时任务的框架支持复杂的时间表达式)
来执行定时任务。但是这里定时任务的并发数很多,就出现了一个问题,同一个trigger被多个机器重复的触发了,这就造成了执行的任务数目
比预期的多很多。领导就让我处理这个问题。
开始我以为是这个框架本身的配置有问题,结果翻了很多资料还是没解决(这里不过多讲这个,有兴趣可以留言)。那么问题出在哪里呢?
quartz的任务工作的方式是这样的。当任务达到触发条件的时候(当这条任务满足qrtz_cron_triggers表中定义的相关的时间表达式的时候)
qrtz_triggers表对应的这条记录的状态发生改变,同时下次触发时间根据时间表达式做出改变,同时根据sched_name找到qrtz_job_details
表中的具体job去执行,下面就是具体的业务了。我这里的问题就出在同一时间内(前后相差几ms)多台机器触发了同一条trigger。然而这个
我是没有办法解决的或者说不想动quartz的源码(有朋友能从这一步就把问题搞定的可以交流一下),所以我就顺着quartz的工作流程继续往下
到了job这里,由于多个trigger被触发所以执行了多次job,那么我是否能通过让他只执行一次job来防止重复执行呢。如果有一种方法可以
让这个job执行一次就可以达到我的要求了(选择在job这一步处理其实还是因为不想动源码,到java这里我就好办了)。正好我对zookeeper
有一些了解,zookeeper恰好有一种注册机制可以解决这个问题。
回顾一下zookeeper关于节点注册的用法:
zookeeper只可以注册一个同名节点如果节点已经存在则返回nodeexits.
那么运用到我这个场景就是当任务进入job之后用job id(同时触发的这几个job的id是一样的)去向zookeeper完成注册,由于
id是一样的那么只能有一个注册成功,只要在注册成功的条件下我才允许task。这样就保证了不做重复的运算。
具体如下:

public class PlatformQuartzJobBean extends QuartzJobBean {
	private String path = "/zk_triggerID";
	private String lock = "/zk_lock";
	private static ZooKeeper zk = null;
	static{
		try {
			zk = new ZooKeeper(PropsUtil.get("zooKeeperUrl")+":"+PropsUtil.get("zooKeeperPort"), 50000,new ZKWatcher());
		} catch (IOException e) {
			logger.error(e.getMessage(),e);
		}
	}
	//任务执行的具体逻辑
	protected void executeInternal(JobExecutionContext jobexecutioncontext)
		throws JobExecutionException {
		Trigger trigger = jobexecutioncontext.getTrigger();
		String triggerName = trigger.getKey().getName();//triggername是唯一的
		boolean createSuccess = false;
		boolean doTask = false; //不对zookeeper注册执行任务
		int childrens = 0;
		List<String> children = null;
		boolean onDelete = false;//是否获取删除节点的权限
		try {
			zk.create(path+"/"+triggerName, triggerName.getBytes(), 
					Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);//如果注册出现问题说明节点存在是重复的任务
			createSuccess = true;
			children = zk.getChildren(path, false);
			if(children != null){
				childrens = children.size();
			}
			if(childrens>99){//节点个数达到100个就执行删除操作
				try {
					zk.create(lock+"/dodelete", "dodelete".getBytes(), 
							Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);
					onDelete = true;
				} catch (KeeperException e1) {
					if(e1 instanceof NodeExistsException){
						logger.info("already on delete!");
					}else{
						logger.error(e1.getMessage(),e1);
					}
				} catch (InterruptedException e1) {
					logger.error(e1.getMessage(),e1);
				}
				
			}
			//执行具体的任务
			execuTask(trigger,triggerName,jobexecutioncontext,af);
		} catch (KeeperException e) {
			if(e instanceof NodeExistsException){
				logger.info("already on do");
			}else if(e instanceof ConnectionLossException){
				logger.info("ConnectionLoss ,do task without registered!!");
				doTask = true;
			}else if(e instanceof SessionExpiredException){
				logger.info("session expired ,do task without registered!!");
				doTask = true;
				try {
					zk = new ZooKeeper(PropsUtil.get("zooKeeperUrl")+":"+PropsUtil.get("zooKeeperPort"), 50000,new ZKWatcher());
				} catch (IOException e1) {
					logger.error(e1.getMessage(),e1);
				}
			}else{
				logger.error(e.getMessage(),e);
			}
		} catch (InterruptedException e) {
			logger.error(e.getMessage(),e);
		}
		if(createSuccess && onDelete){//如果创建成功并且root下有执行删除的权利
			try {
				for(String str:children){
					zk.delete(path+"/"+str, -1);
				}
			} catch (KeeperException e1) {
				logger.error(e1.getMessage(),e1);
			} catch (InterruptedException e1) {
				logger.error(e1.getMessage(),e1);
			}finally{
				if(onDelete){
					try {
						zk.delete(lock+"/dodelete", -1);
					} catch (InterruptedException e) {
						logger.error(e.getMessage(),e);
					} catch (KeeperException e) {
						if(e instanceof ConnectionLossException){
							logger.info("ConnectionLoss ,reconnect zk!!");
							try {
								zk.close();//人为失效,删除dodelete节点
								zk = new ZooKeeper(PropsUtil.get("zooKeeperUrl")+":"+PropsUtil.get("zooKeeperPort"), 50000,new ZKWatcher());
							} catch (InterruptedException e1) {
								e1.printStackTrace();
							} catch (IOException e1) {
								e1.printStackTrace();
							}
							
						}else{
							logger.error(e.getMessage(),e);
						}
						
					}
				}
			}
			
		}
		//如果出现connectloss和sessionexpired 可能是网络有点问题找不到zookeeper就不管重复问题了完成任务为最重要的
		if(doTask){//如果出现connectloss和sessionexpired 就直接执行任务
			execuTask(trigger,triggerName,jobexecutioncontext,af);
		}
	
	}
	
}

整个过程就是:当job开始的时候去向zookeeper申请注册,只有当注册成功的时候才执行业务,失败则退出job。同时由于我这里是每天循环的
定时任务所以当zookeeper下的节点数目达到一定的个数的时候加一个删除锁(就是向zookeeper create一个ondetele节点),同时删除之前
的triggername节点,这样保证了明天这些任务可以继续完成。至此,任务重复执行的问题就解决了。下一篇博客将简单的介绍一下zookeeper和
zookeeper的布置,虽然网上这方面东西很多,不过自己写出来(自己实践过可以用的),以后可以直接拿来用。。

posted @ 2016-08-05 15:38  一腿狗毛  阅读(7702)  评论(2编辑  收藏  举报