debian-packages/vibe-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix waitForDataAsync compilation error for callbacks that have scoped destruction.

Browse source
This commit is contained in:
Sönke Ludwig 2018-09-03 12:35:05 +02:00
parent d8546b843b
commit 0b638634c9
1 changed files with 49 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
source/vibe/core/net.d
View file

@ -619,7 +619,7 @@ mixin(tracer);
if (is(typeof(() @safe { read_ready_callback(true); } ())))
{
mixin(tracer);
import vibe.core.core : setTimer;
import vibe.core.core : Timer, setTimer;
if (!m_context)
return WaitForDataAsyncStatus.noMoreData;
@ -632,17 +632,39 @@ mixin(tracer);
return rs ? WaitForDataAsyncStatus.dataAvailable : WaitForDataAsyncStatus.noMoreData;
}
auto tm = setTimer(timeout, {
eventDriver.sockets.cancelRead(m_socket);
read_ready_callback(false);
});
eventDriver.sockets.read(m_socket, m_context.readBuffer.peekDst(), IOMode.once,
(sock, st, nb) {
tm.stop();
assert(m_context.readBuffer.length == 0);
m_context.readBuffer.putN(nb);
read_ready_callback(m_context.readBuffer.length > 0);
});
static final class WaitContext {
CALLABLE callback;
TCPConnection connection;
Timer timer;
this(CALLABLE callback, TCPConnection connection, Duration timeout)
{
this.callback = callback;
this.connection = connection;
if (timeout < Duration.max)
this.timer = setTimer(timeout, &onTimeout);
}
void onTimeout()
{
eventDriver.sockets.cancelRead(connection.m_socket);
callback(false);
}
void onData(StreamSocketFD, IOStatus st, size_t nb)
{
if (timer) timer.stop();
assert(connection.m_context.readBuffer.length == 0);
connection.m_context.readBuffer.putN(nb);
callback(connection.m_context.readBuffer.length > 0);
}
}
// FIXME: make this work without a heap allocation!
auto context = new WaitContext(read_ready_callback, this, timeout);
eventDriver.sockets.read(m_socket, m_context.readBuffer.peekDst(),
IOMode.once, &context.onData);
return WaitForDataAsyncStatus.waiting;
}
@ -758,6 +780,20 @@ enum WaitForDataAsyncStatus {
waiting,
}
unittest { // test compilation of callback with scoped destruction
static struct CB {
~this() {}
this(this) {}
void opCall(bool) {}
}
void test() {
TCPConnection c;
CB cb;
c.waitForDataAsync(cb);
}
}
mixin validateConnectionStream!TCPConnection;
@ -1066,4 +1102,4 @@ class ReadTimeoutException: Exception
{
super(message, file, line, next);
}
}
}