package alice.topology.manager.keeparive;

import alice.codesegment.CodeSegment;
import alice.datasegment.CommandType;
import alice.datasegment.Receiver;

public class RemoveTask extends CodeSegment {
    private Receiver info1 = ids.create(CommandType.TAKE); // list
    private Receiver info2 = ids.create(CommandType.TAKE); // remove task

    public RemoveTask(){
        info1.setKey("_SCHEDULER", this);
        info2.setKey("_REMOVETASK", this);
    }

    @Override
    public void run() {
        ListManager list = info1.asClass(ListManager.class); // explicit cast
        TaskInfo task = info2.asClass(TaskInfo.class);
        TaskExecuter exec = TaskExecuter.getInstance();
        if (!exec.compareAndSkip(task)){
            // case: remove task is still in the waiting queue.
            boolean removeFlag = list.deleteTask(task);
            if (!removeFlag){
                ods.put("_REMOVETASK", task);
            }
        }
        ods.update("_SCHEDULER", list);
    }
}