mirror of
https://github.com/nim-lang/Nim.git
synced 2026-01-08 22:13:29 +00:00
be991ed413
This change was done to avoid confusion with TCP/IP raw sockets. Native sockets module represents handling native system low level socket API in general and is not just limited anyhow to TCP/IP raw sockets. A stub lib/deprecated/pure/rawsockets.nim module has been added as compatibility layer for old code using rawsockets, so this change will not break existing code.
64 lines
2.1 KiB
Nim
64 lines
2.1 KiB
Nim
import nativesockets, asyncdispatch, macros
|
|
var p = newDispatcher()
|
|
var sock = newAsyncNativeSocket()
|
|
|
|
proc convertReturns(node, retFutureSym: NimNode): NimNode {.compileTime.} =
|
|
case node.kind
|
|
of nnkReturnStmt:
|
|
result = newCall(newIdentNode("complete"), retFutureSym, node[0])
|
|
else:
|
|
result = node
|
|
for i in 0 .. <node.len:
|
|
result[i] = convertReturns(node[i], retFutureSym)
|
|
|
|
macro async2(prc: stmt): stmt {.immediate.} =
|
|
expectKind(prc, nnkProcDef)
|
|
|
|
var outerProcBody = newNimNode(nnkStmtList)
|
|
|
|
# -> var retFuture = newFuture[T]()
|
|
var retFutureSym = newIdentNode("retFuture") #genSym(nskVar, "retFuture")
|
|
outerProcBody.add(
|
|
newVarStmt(retFutureSym,
|
|
newCall(
|
|
newNimNode(nnkBracketExpr).add(
|
|
newIdentNode("newFuture"),
|
|
prc[3][0][1])))) # Get type from return type of this proc.
|
|
|
|
# -> iterator nameIter(): FutureBase {.closure.} = <proc_body>
|
|
# Changing this line to: newIdentNode($prc[0].ident & "Iter") # will make it work.
|
|
var iteratorNameSym = genSym(nskIterator, $prc[0].ident & "Iter")
|
|
#var iteratorNameSym = newIdentNode($prc[0].ident & "Iter")
|
|
var procBody = prc[6].convertReturns(retFutureSym)
|
|
|
|
var closureIterator = newProc(iteratorNameSym, [newIdentNode("FutureBase")],
|
|
procBody, nnkIteratorDef)
|
|
closureIterator[4] = newNimNode(nnkPragma).add(newIdentNode("closure"))
|
|
outerProcBody.add(closureIterator)
|
|
|
|
# -> var nameIterVar = nameIter
|
|
# -> var first = nameIterVar()
|
|
var varNameIterSym = newIdentNode($prc[0].ident & "IterVar") #genSym(nskVar, $prc[0].ident & "IterVar")
|
|
var varNameIter = newVarStmt(varNameIterSym, iteratorNameSym)
|
|
outerProcBody.add varNameIter
|
|
var varFirstSym = genSym(nskVar, "first")
|
|
var varFirst = newVarStmt(varFirstSym, newCall(varNameIterSym))
|
|
outerProcBody.add varFirst
|
|
|
|
|
|
result = prc
|
|
|
|
# Remove the 'closure' pragma.
|
|
for i in 0 .. <result[4].len:
|
|
if result[4][i].ident == !"async":
|
|
result[4].del(i)
|
|
|
|
result[6] = outerProcBody
|
|
|
|
proc readStuff(): Future[string] {.async2.} =
|
|
var fut = connect(sock, "irc.freenode.org", Port(6667))
|
|
yield fut
|
|
var fut2 = recv(sock, 50)
|
|
yield fut2
|
|
return fut2.read
|