From c394acafc8fe231cc64b6bd21e5e10733a2bfb0e Mon Sep 17 00:00:00 2001 From: Matthew Wild Date: Tue, 26 May 2009 21:47:25 +0100 Subject: net.dns: Add support for cancelling a coroutine-based request --- net/dns.lua | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'net/dns.lua') diff --git a/net/dns.lua b/net/dns.lua index 0b37c0c6..69049317 100644 --- a/net/dns.lua +++ b/net/dns.lua @@ -21,8 +21,8 @@ local require = require local coroutine, io, math, socket, string, table = coroutine, io, math, socket, string, table -local ipairs, next, pairs, print, setmetatable, tostring, assert, error = - ipairs, next, pairs, print, setmetatable, tostring, assert, error +local ipairs, next, pairs, print, setmetatable, tostring, assert, error, unpack = + ipairs, next, pairs, print, setmetatable, tostring, assert, error, unpack local get, set = ztact.get, ztact.set @@ -711,6 +711,12 @@ function resolver:feed(sock, packet) return response end +function resolver:cancel(data) + local cos = get (self.wanted, unpack(data, 1, 3)) + if cos then + cos[data[4]] = nil; + end +end function resolver:pulse () -- - - - - - - - - - - - - - - - - - - - - pulse @@ -847,6 +853,8 @@ function dns.query (...) -- - - - - - - - - - - - - - - - - - - - - - query function dns.feed (...) -- - - - - - - - - - - - - - - - - - - - - - feed return resolve (resolver.feed, ...) end +function dns.cancel(...) -- - - - - - - - - - - - - - - - - - - - - - cancel + return resolve(resolver.cancel, ...) end function dns:socket_wrapper_set (...) -- - - - - - - - - socket_wrapper_set return resolve (resolver.socket_wrapper_set, ...) end -- cgit v1.2.3