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

added Timer and enum return to TCPConnection.waitForDataAsync

Browse source
This commit is contained in:
Francesco Mecca 2018-02-23 00:27:04 +00:00
parent 0d6ba62f51
commit 20e32cf327
1 changed files with 20 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -586,29 +586,39 @@ mixin(tracer);
return m_context.readBuffer.length > 0;
}
void waitForDataAsync(void delegate(bool) @safe read_callback, Duration timeout = Duration.max)
enum WaitForDataAsyncStatus {
noMoreData,
dataAvailable,
waiting
}
WaitForDataAsyncStatus waitForDataAsync(void delegate(bool) @safe read_callback, Duration timeout = Duration.max)
{
mixin(tracer);
import vibe.core.core : runTask;
import vibe.core.core : runTask, setTimer, createTimer;
if (!m_context) {
runTask(read_callback, false);
return;
return WaitForDataAsyncStatus.noMoreData;
}
if (m_context.readBuffer.length > 0) {
runTask(read_callback, true);
return;
return WaitForDataAsyncStatus.dataAvailable;
}
if (timeout <= 0.seconds) {
auto rs = waitForData(0.seconds);
auto mode = rs ? WaitForDataAsyncStatus.dataAvailable : WaitForDataAsyncStatus.noMoreData;
return mode;
}
auto mode = timeout <= 0.seconds ? IOMode.immediate : IOMode.once;
bool cancelled;
IOStatus status;
size_t nbytes;
eventDriver.sockets.waitForData(m_socket,
(sock, st, nb) { if(st != IOStatus.ok) runTask(read_callback, false);
auto tm = setTimer(timeout, { eventDriver.sockets.cancelRead(m_socket);
runTask(read_callback, false); });
eventDriver.sockets.read(m_socket, m_context.readBuffer.peekDst(), IOMode.once,
(sock, st, nb) { tm.stop(); if(st != IOStatus.ok) runTask(read_callback, false);
else runTask(read_callback, true);
});
return WaitForDataAsyncStatus.waiting;
}
const(ubyte)[] peek() { return m_context ? m_context.readBuffer.peek() : null; }