package com.atlassian.core.task;
   
   import java.util.HashMap;
   import java.util.Map;
   
   /**
    * A TaskManager implementation that manages multiple queues
    */
   public class DefaultMultiQueueTaskManager implements MultiQueueTaskManager {
      private final Map<String, TaskQueue> queues = new HashMap<String, TaskQueue>();
  
      public DefaultMultiQueueTaskManager(String queueName, TaskQueue queue) {
          addTaskQueue(queueName, queue);
      }
  
      public DefaultMultiQueueTaskManager(Map<String, TaskQueue> queues) {
          setTaskQueues(queues);
      }
  
      public void addTaskQueue(String name, TaskQueue queue) {
          if (queues.keySet().contains(name)) {
              throw new IllegalArgumentException("The queue specified already exists in the task manager");
          }
          queues.put(name, queue);
      }
  
      public TaskQueue removeTaskQueue(String queueName, TaskQueue taskQueue, boolean flush) {
          TaskQueue queue = getTaskQueue(queueName);
          if (queue != null && flush) {
              queue.flush();
          }
          return queue;
      }
  
      public TaskQueue getTaskQueue(String name) {
          return queues.get(name);
      }
  
      public void setTaskQueues(Map<String, TaskQueue> queues) {
          for (Map.Entry<String, TaskQueue> entry : queues.entrySet()) {
              addTaskQueue(entry.getKey(), entry.getValue());
          }
      }
  
      public void addTask(String queueName, Task task) {
          getTaskQueue(queueName).addTask(task);
      }
  
      public void flush(String queueName) {
          getTaskQueue(queueName).flush();
      }
  
      public void flush() {
          for (TaskQueue taskQueue : queues.values()) {
              taskQueue.flush();
          }
      }
  }