2024-02-21 19:51:43 -07:00
|
|
|
import std/os
|
|
|
|
|
import std/sequtils
|
|
|
|
|
|
|
|
|
|
import chronos
|
|
|
|
|
import chronos/threadsync
|
|
|
|
|
import chronos/unittest2/asynctests
|
|
|
|
|
import taskpools
|
|
|
|
|
|
|
|
|
|
import apatheia/queues
|
|
|
|
|
|
2024-02-21 20:07:24 -07:00
|
|
|
type
|
|
|
|
|
DataObj = ref object
|
|
|
|
|
data: seq[char]
|
|
|
|
|
|
2024-02-21 19:51:43 -07:00
|
|
|
proc worker(data: seq[char], queue: SignalQueue[int]) =
|
2024-02-21 20:07:24 -07:00
|
|
|
os.sleep(5000)
|
2024-02-21 19:51:43 -07:00
|
|
|
echo "worker: ", data
|
|
|
|
|
discard queue.send(data.len())
|
|
|
|
|
|
2024-02-21 20:07:24 -07:00
|
|
|
proc finalizer(obj: DataObj) =
|
|
|
|
|
echo "FINALIZE!!"
|
|
|
|
|
|
|
|
|
|
proc runTest(tp: TaskPool, queue: SignalQueue[int]) {.async.} =
|
|
|
|
|
## init
|
|
|
|
|
var data = "hello world!".toSeq
|
|
|
|
|
var obj: DataObj
|
|
|
|
|
new(obj, finalizer)
|
|
|
|
|
|
|
|
|
|
echo "spawn worker"
|
|
|
|
|
tp.spawn worker(data, queue)
|
|
|
|
|
|
|
|
|
|
let res = await wait(queue)
|
|
|
|
|
check res.get() == 12
|
|
|
|
|
|
2024-02-21 19:51:43 -07:00
|
|
|
suite "async tests":
|
|
|
|
|
|
|
|
|
|
var tp = Taskpool.new(num_threads = 2) # Default to the number of hardware threads.
|
|
|
|
|
var queue = newSignalQueue[int]()
|
|
|
|
|
|
|
|
|
|
asyncTest "test":
|
|
|
|
|
|
2024-02-21 20:07:24 -07:00
|
|
|
await runTest(tp, queue)
|
|
|
|
|
GC_fullCollect()
|
2024-02-21 19:51:43 -07:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|