CachedDNSToSwitchMapping.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.net;
  19. import java.util.ArrayList;
  20. import java.util.List;
  21. import java.util.Map;
  22. import java.util.concurrent.ConcurrentHashMap;
  23. /**
  24.  * A cached implementation of DNSToSwitchMapping that takes an
  25.  * raw DNSToSwitchMapping and stores the resolved network location in 
  26.  * a cache. The following calls to a resolved network location
  27.  * will get its location from the cache. 
  28.  *
  29.  */
  30. public class CachedDNSToSwitchMapping implements DNSToSwitchMapping {
  31.   private Map<String, String> cache = new ConcurrentHashMap<String, String>();
  32.   protected DNSToSwitchMapping rawMapping;
  33.   
  34.   public CachedDNSToSwitchMapping(DNSToSwitchMapping rawMapping) {
  35.     this.rawMapping = rawMapping;
  36.   }
  37.   
  38.   public List<String> resolve(List<String> names) {
  39.     // normalize all input names to be in the form of IP addresses
  40.     names = NetUtils.normalizeHostNames(names);
  41.     
  42.     List <String> result = new ArrayList<String>(names.size());
  43.     if (names.isEmpty()) {
  44.       return result;
  45.     }
  46.     // find out all names without cached resolved location
  47.     List<String> unCachedHosts = new ArrayList<String>(names.size());
  48.     for (String name : names) {
  49.       if (cache.get(name) == null) {
  50.         unCachedHosts.add(name);
  51.       } 
  52.     }
  53.     
  54.     // Resolve those names
  55.     List<String> rNames = rawMapping.resolve(unCachedHosts);
  56.     
  57.     // Cache the result
  58.     if (rNames != null) {
  59.       for (int i=0; i<unCachedHosts.size(); i++) {
  60.         cache.put(unCachedHosts.get(i), rNames.get(i));
  61.       }
  62.     }
  63.     
  64.     // Construct the result
  65.     for (String name : names) {
  66.       //now everything is in the cache
  67.       String networkLocation = cache.get(name);
  68.       if (networkLocation != null) {
  69.         result.add(networkLocation);
  70.       } else { //resolve all or nothing
  71.         return null;
  72.       }
  73.     }
  74.     return result;
  75.   }
  76. }