Commit fdbf7c8a authored by lgcareer's avatar lgcareer Committed by qiaozhanwei
Browse files

Fixed tasks_queue and tasks_kill did not exist in zookeeper #1696 (#1734)



Co-authored-by: default avatarelonlo <gamelifedong@gmail.com>
parent fddf4ef9
Loading
Loading
Loading
Loading
+20 −1
Original line number Diff line number Diff line
@@ -37,8 +37,27 @@ public class TaskQueueZkImpl implements ITaskQueue {

    private static final Logger logger = LoggerFactory.getLogger(TaskQueueZkImpl.class);

    private final ZookeeperOperator zookeeperOperator;

    @Autowired
    private ZookeeperOperator zookeeperOperator;
    public TaskQueueZkImpl(ZookeeperOperator zookeeperOperator) {
        this.zookeeperOperator = zookeeperOperator;

        try {
            String tasksQueuePath = getTasksPath(Constants.DOLPHINSCHEDULER_TASKS_QUEUE);
            String tasksCancelPath = getTasksPath(Constants.DOLPHINSCHEDULER_TASKS_KILL);

            for(String key : new String[]{tasksQueuePath,tasksCancelPath}){
                if(!zookeeperOperator.isExisted(key)){
                    zookeeperOperator.persist(key, "");
                    logger.info("create tasks queue parent node success : {}", key);
                }
            }
        } catch (Exception e) {
            logger.error("create tasks queue parent node failure", e);
        }
    }


    /**
     * get all tasks from tasks queue