View Javadoc

1   package org.apache.helix.controller.stages;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *   http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.util.Collections;
23  import java.util.HashMap;
24  import java.util.List;
25  import java.util.Map;
26  
27  import org.apache.helix.model.Message;
28  import org.apache.helix.model.Partition;
29  
30  
31  public class MessageThrottleStageOutput
32  {
33    private final Map<String, Map<Partition, List<Message>>> _messagesMap;
34  
35    public MessageThrottleStageOutput()
36    {
37      _messagesMap = new HashMap<String, Map<Partition, List<Message>>>();
38    }
39  
40    public void addMessages(String resourceName,
41                            Partition partition,
42                            List<Message> selectedMessages)
43    {
44      if (!_messagesMap.containsKey(resourceName))
45      {
46        _messagesMap.put(resourceName, new HashMap<Partition, List<Message>>());
47      }
48      _messagesMap.get(resourceName).put(partition, selectedMessages);
49  
50    }
51  
52    public List<Message> getMessages(String resourceName, Partition partition)
53    {
54      Map<Partition, List<Message>> map = _messagesMap.get(resourceName);
55      if (map != null)
56      {
57        return map.get(partition);
58      }
59      return Collections.emptyList();
60    }
61  }