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

switch to asyncAwait

Browse source
This commit is contained in:
Boris-Barboris 2017-11-11 15:45:44 +00:00 committed by Sönke Ludwig
parent 55e7866497
commit 6634cbc645
2 changed files with 23 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -195,18 +195,18 @@ TCPConnection connectTCP(NetworkAddress addr, NetworkAddress bind_address = anyA
scope uaddr = new RefAddress(addr.sockAddr, addr.sockAddrLen); scope uaddr = new RefAddress(addr.sockAddr, addr.sockAddrLen);
scope baddr = new RefAddress(bind_address.sockAddr, bind_address.sockAddrLen); scope baddr = new RefAddress(bind_address.sockAddr, bind_address.sockAddrLen);
auto result = asyncAwaitUninterruptible!(ConnectCallback, auto result = asyncAwait!(ConnectCallback,
cb => eventDriver.sockets.connectStream(uaddr, baddr, cb), cb => eventDriver.sockets.connectStream(uaddr, baddr, cb),
(ConnectCallback cb, StreamSocketFD sock_fd) { (ConnectCallback cb, StreamSocketFD sock_fd) {
eventDriver.sockets.cancelConnectStream(sock_fd); eventDriver.sockets.cancelConnectStream(sock_fd);
} }
)(timeout); )(timeout);
if (result[1] == ConnectStatus.cancelled) enforce(result.completed, "Failed to connect to " ~ addr.toString() ~
result[1] = ConnectStatus.timeout; ": timeout");
enforce(result[1] == ConnectStatus.connected, "Failed to connect to "~addr.toString()~": "~result[1].to!string); enforce(result.results[1] == ConnectStatus.connected, "Failed to connect to "~addr.toString()~": "~result.results[1].to!string);
return TCPConnection(result[0], uaddr); return TCPConnection(result.results[0], uaddr);
} (); } ();
} }

16
source/vibe/internal/async.d
View file

@ -1,6 +1,6 @@
module vibe.internal.async; module vibe.internal.async;
import std.traits : ParameterTypeTuple; import std.traits : ParameterTypeTuple, ReturnType;
import std.typecons : tuple; import std.typecons : tuple;
import vibe.core.core : hibernate, switchToTask; import vibe.core.core : hibernate, switchToTask;
import vibe.core.task : InterruptException, Task; import vibe.core.task : InterruptException, Task;
@ -20,14 +20,23 @@ auto asyncAwait(Callback, alias action, alias cancel)(Duration timeout, string f
{ {
static struct R { static struct R {
bool completed = true; bool completed = true;
typeof(waitable.results) results; ParameterTypeTuple!Callback results;
} }
R ret; R ret;
static if (is(ReturnType!action == void)) {
alias waitable = Waitable!(Callback, alias waitable = Waitable!(Callback,
action, action,
(cb) { ret.completed = false; cancel(cb); }, (cb) { ret.completed = false; cancel(cb); },
(ParameterTypeTuple!Callback r) { ret.results = r; } (ParameterTypeTuple!Callback r) { ret.results = r; }
); );
}
else {
alias waitable = Waitable!(Callback,
action,
(cb, waitres) { ret.completed = false; cancel(cb, waitres); },
(ParameterTypeTuple!Callback r) { ret.results = r; }
);
}
asyncAwaitAny!(true, waitable)(timeout, func); asyncAwaitAny!(true, waitable)(timeout, func);
return ret; return ret;
} }
@ -60,7 +69,8 @@ template Waitable(CB, alias WAIT, alias CANCEL, alias DONE)
"CANCEL must be callable with a parameter of type "~CB.stringof); "CANCEL must be callable with a parameter of type "~CB.stringof);
else else
static assert(is(typeof(CANCEL(CB.init, typeof(WAIT(CB.init)).init))), static assert(is(typeof(CANCEL(CB.init, typeof(WAIT(CB.init)).init))),
"CANCEL must be callable with parameters ("~CB.stringof~", "~typeof(WAIT(CB.init)).stringof~")"); "CANCEL must be callable with parameters "~CB.stringof~
" and "~typeof(WAIT(CB.init)).stringof);
static assert(is(typeof(DONE(ParameterTypeTuple!CB.init))), static assert(is(typeof(DONE(ParameterTypeTuple!CB.init))),
"DONE must be callable with types "~ParameterTypeTuple!CB.stringof); "DONE must be callable with types "~ParameterTypeTuple!CB.stringof);