nim-codex/tests/codex/tp_example.nim

43 lines
711 B
Nim
Raw Normal View History

2024-08-07 11:25:50 +00:00
import pkg/taskpools
block: # Async without result
proc displayInt(x: int) =
stdout.write(x)
stdout.write(" - SUCCESS\n")
proc main() =
echo "\nSanity check 1: Printing 123456 654321 in parallel"
var tp = Taskpool.new(numThreads = 4)
tp.spawn displayInt(123456)
tp.spawn displayInt(654321)
tp.shutdown()
main()
block: # Async/Await
var tp: Taskpool
proc asyncFib(n: int): int {.gcsafe.} =
if n < 2:
return n
let x = tp.spawn asyncFib(n-1)
let y = asyncFib(n-2)
result = sync(x) + y
proc main2() =
echo "\nSanity check 2: fib(20)"
tp = Taskpool.new()
let f = asyncFib(20)
tp.shutdown()
doAssert f == 6765
main2()