From e86ff63f91506c83370534d7d10746a92d1b2224 Mon Sep 17 00:00:00 2001
From: Eugen Wissner <belka@caraus.de>
Date: Sun, 12 Feb 2017 18:51:00 +0100
Subject: [PATCH] Add DuplexTransport.close for the selector transport

---
 source/tanya/async/event/selector.d | 29 +++++++++++++++++++++++++++++
 source/tanya/async/transport.d      | 13 +++++++++++++
 2 files changed, 42 insertions(+)

diff --git a/source/tanya/async/event/selector.d b/source/tanya/async/event/selector.d
index 9251b31..13254ce 100644
--- a/source/tanya/async/event/selector.d
+++ b/source/tanya/async/event/selector.d
@@ -37,6 +37,8 @@ package class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
 
 	private Protocol protocol_;
 
+	private bool closing;
+
 	/// Received notification that the underlying socket is write-ready.
 	package bool writeReady;
 
@@ -115,6 +117,25 @@ package class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
 		protocol_ = protocol;
 	}
 
+	/**
+	 * Returns $(D_PARAM true) if the transport is closing or closed.
+	 */
+	bool isClosing() const pure nothrow @safe @nogc
+	{
+		return closing;
+	}
+
+	/**
+	 * Close the transport.
+	 *
+	 * Buffered data will be flushed.  No more data will be received.
+	 */
+	void close() @nogc
+	{
+		closing = true;
+		loop.reify(this, EventMask(Event.read, Event.write), EventMask(Event.write));
+	}
+
 	/**
 	 * Invokes the watcher callback.
 	 */
@@ -124,6 +145,10 @@ package class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
 		{
 			protocol.received(output[0 .. $]);
 			output.clear();
+			if (isClosing() && input.length == 0)
+			{
+				loop.kill(this);
+			}
 		}
 		else
 		{
@@ -281,6 +306,10 @@ abstract class SelectorLoop : Loop
 			kill(transport, exception);
 			return false;
 		}
+		if (transport.input.length == 0 && transport.isClosing())
+		{
+			kill(transport);
+		}
 		return true;
 	}
 
diff --git a/source/tanya/async/transport.d b/source/tanya/async/transport.d
index f18fbee..4550522 100644
--- a/source/tanya/async/transport.d
+++ b/source/tanya/async/transport.d
@@ -73,6 +73,19 @@ interface DuplexTransport : ReadTransport, WriteTransport
 	{
 		assert(protocol !is null);
 	}
+
+
+	/**
+	 * Returns $(D_PARAM true) if the transport is closing or closed.
+	 */
+	bool isClosing() const pure nothrow @safe @nogc;
+
+	/**
+	 * Close the transport.
+	 *
+	 * Buffered data will be flushed.  No more data will be received.
+	 */
+	void close() @nogc;
 }
 
 /**