-
-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
ARC works for async on Windows (#13179)
- Loading branch information
Showing
5 changed files
with
110 additions
and
40 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,69 @@ | ||
| discard """ | ||
| output: "5000" | ||
| cmd: "nim c --gc:arc $file" | ||
| """ | ||
|
|
||
| import asyncdispatch, asyncnet, nativesockets, net, strutils, os | ||
|
|
||
| var msgCount = 0 | ||
|
|
||
| const | ||
| swarmSize = 50 | ||
| messagesToSend = 100 | ||
|
|
||
| var clientCount = 0 | ||
|
|
||
| proc sendMessages(client: AsyncFD) {.async.} = | ||
| for i in 0 ..< messagesToSend: | ||
| await send(client, "Message " & $i & "\c\L") | ||
|
|
||
| proc launchSwarm(port: Port) {.async.} = | ||
| for i in 0 ..< swarmSize: | ||
| var sock = createAsyncNativeSocket() | ||
|
|
||
| await connect(sock, "localhost", port) | ||
| await sendMessages(sock) | ||
| closeSocket(sock) | ||
|
|
||
| proc readMessages(client: AsyncFD) {.async.} = | ||
| # wrapping the AsyncFd into a AsyncSocket object | ||
| var sockObj = newAsyncSocket(client) | ||
| var (ipaddr, port) = sockObj.getPeerAddr() | ||
| doAssert ipaddr == "127.0.0.1" | ||
| (ipaddr, port) = sockObj.getLocalAddr() | ||
| doAssert ipaddr == "127.0.0.1" | ||
| while true: | ||
| var line = await recvLine(sockObj) | ||
| if line == "": | ||
| closeSocket(client) | ||
| clientCount.inc | ||
| break | ||
| else: | ||
| if line.startswith("Message "): | ||
| msgCount.inc | ||
| else: | ||
| doAssert false | ||
|
|
||
| proc createServer(port: Port) {.async.} = | ||
| var server = createAsyncNativeSocket() | ||
| block: | ||
| var name: Sockaddr_in | ||
| name.sin_family = typeof(name.sin_family)(toInt(AF_INET)) | ||
| name.sin_port = htons(uint16(port)) | ||
| name.sin_addr.s_addr = htonl(INADDR_ANY) | ||
| if bindAddr(server.SocketHandle, cast[ptr SockAddr](addr(name)), | ||
| sizeof(name).Socklen) < 0'i32: | ||
| raiseOSError(osLastError()) | ||
|
|
||
| discard server.SocketHandle.listen() | ||
| while true: | ||
| asyncCheck readMessages(await accept(server)) | ||
|
|
||
| asyncCheck createServer(Port(10335)) | ||
| asyncCheck launchSwarm(Port(10335)) | ||
| while true: | ||
| poll() | ||
| if clientCount == swarmSize: break | ||
|
|
||
| assert msgCount == swarmSize * messagesToSend | ||
| echo msgCount |