jyapayne/nim-faststreams
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Slightly more flexible pipelines (to be documented)

Browse source
This commit is contained in:
Zahary Karadjov 2020-05-07 01:18:58 +03:00
commit 8b863f7798
No known key found for this signature in database
GPG key ID: C8936F8A3073D609
2 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
faststreams/pipelines.nim
View file

@ -305,20 +305,29 @@ macro executePipeline*(start: AsyncInputStream, steps: varargs[untyped]): untype
stepInput = stepOutput
var RetTypeExpr = copy steps[^1]
RetTypeExpr.insert(1, newCall("default", ident"AsyncOutputStream"))
RetTypeExpr.insert(1, newCall("default", ident"AsyncInputStream"))
var closingCall = steps[^1]
closingCall.insert(1, newDotExpr(stepInput, ident"output"))
closingCall.insert(1, newCall(bindSym"initReader", stepInput))
pipelineBody.add quote do:
await allFutures(`pipelineSteps`)
return `closingCall`
`closingCall`
result = quote do:
type RetType = type(`RetTypeExpr`)
type UserOpRetType = type(`RetTypeExpr`)
when UserOpRetType is Future:
type RetType = type(default(UserOpRetType).read)
else:
type RetType = UserOpRetType
proc pipelineProc(`stream`: AsyncInputStream): Future[RetType] {.async.} =
`pipelineBody`
when UserOpRetType is Future:
var f = `pipelineBody`
return await(f)
else:
return `pipelineBody`
pipelineProc(`start`)