Use standard Resolv library instead of net/dns

purcell · Aug 30, 2012 · db00e9d · db00e9d
1 parent 65fa0f2
commit db00e9d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/Gemfile b/Gemfile
@@ -2,5 +2,3 @@ source 'http://rubygems.org'
 source 'http://gems.github.com'
 
 gem 'whois'
-gem 'net-dns'
-
diff --git a/bin/postfix-policy-whois b/bin/postfix-policy-whois
@@ -2,12 +2,14 @@
 
 $0='postfix-policy-registrar'
 
-require 'net/dns'
+require 'resolv'
 require 'whois'
 
 def nameserver_for(domain)
-  if answer = Net::DNS::Resolver.start(domain, Net::DNS::NS).answer.first
-    answer.nsdname.sub(/\.$/, '')
+  begin
+    Resolv::DNS.new.getresource(domain, Resolv::DNS::Resource::IN::NS).name.to_s
+  rescue Resolv::ResolvError
+    nil
   end
 end