# HG changeset patch
# User Matthew Wild <mwild1@gmail.com>
# Date 1243370912 -3600
# Node ID 23725bfdeed5ddee41d3a9915f36bf01340e9a84
# Parent  e69fafc14491ebde0328b45fb193eef19ff228e3
net.adns: Add support for cancelling a non-blocking lookup, optionally calling the handler

diff -r e69fafc14491 -r 23725bfdeed5 net/adns.lua
--- a/net/adns.lua	Tue May 26 21:47:25 2009 +0100
+++ b/net/adns.lua	Tue May 26 21:48:32 2009 +0100
@@ -16,7 +16,7 @@
 				end
 				log("debug", "Records for %s not in cache, sending query (%s)...", qname, tostring(coroutine.running()));
 				dns.query(qname, qtype, qclass);
-				coroutine.yield(nil); -- Wait for reply
+				coroutine.yield({ qclass or "IN", qtype or "A", qname, coroutine.running()}); -- Wait for reply
 				log("debug", "Reply for %s (%s)", qname, tostring(coroutine.running()));
 				local ok, err = pcall(handler, dns.peek(qname, qtype, qclass));
 				if not ok then
@@ -25,6 +25,13 @@
 			end)(dns.peek(qname, qtype, qclass));
 end
 
+function cancel(handle, call_handler)
+	dns.cancel(handle);
+	if call_handler then
+		handle[4]()
+	end
+end
+
 function new_async_socket(sock)
 	local newconn = {};
 	local listener = {};