java JNDI 自己指定DNS服務器解析域名IP

目標：

1、自己指定域名服務器

2、指定多域名服務器

3、獲取所有IP

實現參考：

JNDI

http://docs.oracle.com/javase/1.4.2/docs/guide/jndi/jndi-dns.html

http://mindprod.com/jgloss/dns.html

詳細參考自己查看

實現：

       /**
       * 獲取DNS服務器信息
       *
       * @param domain  要獲取DNS信息的域名
       * @param provider      DNS服務器
       * @param types   信息類型 "A"(IP信息)，"MX"
       * @param timeout 請求超時
       * @param retryCount    重試次數
       *
       * @return 所有信息組成的數組
       *
       * @throws NamingException
       *        
       */

       @SuppressWarnings("rawtypes" )
       public static ArrayList<String> getDNSRecs(String domain, String provider,
                  String [] types, int timeout, int retryCount) throws NamingException {
            
            ArrayList<String> results = new ArrayList<String>(15);
            
            Hashtable<String, String> env = new Hashtable<String, String>();
            
            env.put( "java.naming.factory.initial" ,
                         "com.sun.jndi.dns.DnsContextFactory" );
            
             //設置域名服務器
            env.put(Context. PROVIDER_URL, "dns://" + provider);
            
             // 連接時間
            env.put( "com.sun.jndi.dns.timeout.initial" , String.valueOf(timeout));
            
             // 連接次數
            env.put( "com.sun.jndi.dns.timeout.retries" , String.valueOf(retryCount));
            
            DirContext ictx = new InitialDirContext(env);
            Attributes attrs = ictx.getAttributes(domain, types);
            
             for (Enumeration e = attrs.getAll(); e.hasMoreElements();) {
                  Attribute a = (Attribute) e.nextElement();
                   int size = a.size();
                   for (int i = 0; i < size; i++) {
                        results.add((String) a.get(i));
                  }
            }
             return results;
      }

       /**
       * 獲取域名所有IP
       * @param domain  域名
       * @param dnsServers    DNS服務器列表
       * @param timeout 請求超時
       * @param retryCount    重試次數
       * @return
       */
       public static Set<String> getAllIP(String domain, String[] dnsServers,
                   int timeout, int retryCount) {
            Set<String> ips = new HashSet<String>();
            
             for(String dnsServer: dnsServers) {
                  List<String> ipList;
                   try {
                        ipList = getDNSRecs(domain, dnsServer, new String[]{"A"},
                                    timeout, retryCount);
                  } catch (NamingException e) {
                         continue;
                  }
                  ips.addAll(ipList);
            }
            
             return ips;
      }