| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.oodt.cas.resource.scheduler; |
| |
| //OODT imports |
| import org.apache.oodt.cas.resource.structs.exceptions.QueueManagerException; |
| |
| //JDK imports |
| import java.util.LinkedHashMap; |
| import java.util.LinkedHashSet; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Vector; |
| |
| /** |
| * |
| * @author bfoster |
| * @version $Revision$ |
| * |
| * <p> |
| * The Queue to Node Mapping Manager |
| * </p> |
| */ |
| public class QueueManager { |
| |
| protected Map<String, LinkedHashSet<String>> queueToNodesMapping; |
| |
| public QueueManager() { |
| this.queueToNodesMapping = new LinkedHashMap<String, LinkedHashSet<String>>(); |
| } |
| |
| public synchronized boolean containsQueue(String queueName) { |
| return this.queueToNodesMapping.containsKey(queueName); |
| } |
| |
| public synchronized void addNodeToQueue(String nodeId, String queueName) throws QueueManagerException { |
| if (queueName == null || !this.queueToNodesMapping.containsKey(queueName)) |
| throw new QueueManagerException("Queue '" + queueName + "' does not exist"); |
| |
| // add node to queue |
| LinkedHashSet<String> nodes = this.queueToNodesMapping.get(queueName); |
| if (nodes == null) |
| nodes = new LinkedHashSet<String>(); |
| nodes.add(nodeId); |
| |
| // put node list back into map |
| this.queueToNodesMapping.put(queueName, nodes); |
| } |
| |
| public synchronized void addQueue(String queueName) throws QueueManagerException { |
| if (queueName != null && !this.queueToNodesMapping.containsKey(queueName)) |
| this.queueToNodesMapping.put(queueName, new LinkedHashSet<String>()); |
| } |
| |
| public synchronized List<String> getNodes(String queueName) throws QueueManagerException { |
| if (queueName != null && this.queueToNodesMapping.containsKey(queueName)) |
| return new Vector<String>(this.queueToNodesMapping.get(queueName)); |
| else |
| throw new QueueManagerException("Queue '" + queueName + "' does not exist"); |
| } |
| |
| public synchronized List<String> getQueues() throws QueueManagerException { |
| return new Vector<String>(this.queueToNodesMapping.keySet()); |
| } |
| |
| public synchronized List<String> getQueues(String nodeId) throws QueueManagerException { |
| Vector<String> queueNames = new Vector<String>(); |
| for (String queueName : this.queueToNodesMapping.keySet()) |
| if (this.queueToNodesMapping.get(queueName).contains(nodeId)) |
| queueNames.add(queueName); |
| return queueNames; |
| } |
| |
| public synchronized void removeNodeFromQueue(String nodeId, String queueName) throws QueueManagerException { |
| if (queueName != null && this.queueToNodesMapping.containsKey(queueName)) |
| this.queueToNodesMapping.get(queueName).remove(nodeId); |
| else |
| throw new QueueManagerException("Queue '" + queueName + "' does not exist"); |
| } |
| |
| public synchronized void removeQueue(String queueName) throws QueueManagerException { |
| if (queueName != null) |
| this.queueToNodesMapping.remove(queueName); |
| } |
| |
| } |
| |