com.alibaba.schedulerx.worker.master.TaskMasterPool Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of schedulerx2-worker Show documentation

There is a newer version: 1.12.2

package com.alibaba.schedulerx.worker.master;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

/**
 *
 * @author xiaomeng.hxm
 */
public enum TaskMasterPool {
    INSTANCE;

    private Map masterPool = Maps.newConcurrentMap();

    public TaskMaster get(long jobInstanceId) {
        return masterPool.get(jobInstanceId);
    }

    public void put(long jobInstanceId, TaskMaster taskMaster) {
        masterPool.put(jobInstanceId, taskMaster);
    }

    public void remove(long jobInstanceId) {
        TaskMaster taskMaster = masterPool.remove(jobInstanceId);
        if (taskMaster != null) {
            taskMaster.latchRelease();
        }
    }

    public boolean contains(long jobInstanceId) {
        return masterPool.containsKey(jobInstanceId);
    }

    public List getInstanceIds(long appGroupId) {
        List list = Lists.newArrayList();
        for(TaskMaster taskMaster : getAllTaskMaster()){
            if(taskMaster.getJobInstanceInfo().getAppGroupId() == appGroupId){
                list.add(taskMaster.getJobInstanceInfo().getJobInstanceId());
            }
        }
        return list;
    }

    public void clear() {
        masterPool.clear();
    }

    public Collection getAllTaskMaster(){
        return masterPool.values();
    }
}