blob: 3de1436431cd3f3b575fbe3e3709ec8e27146c6d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
discard """
output: '''
x
e
done
'''
"""
#[
xxx move this to tests/stdlib/tasyncjs.nim
]#
import std/asyncjs
block:
# demonstrate forward definition for js
proc y(e: int): Future[string] {.async.}
proc e: int {.discardable.} =
echo "e"
return 2
proc x(e: int): Future[void] {.async.} =
var s = await y(e)
if e > 2:
return
echo s
e()
proc y(e: int): Future[string] {.async.} =
if e > 0:
return await y(0)
else:
return "x"
discard x(2)
import std/sugar
from std/strutils import contains
var witness: seq[string]
proc fn(n: int): Future[int] {.async.} =
if n >= 7:
raise newException(ValueError, "foobar: " & $n)
if n > 0:
var ret = 1 + await fn(n-1)
witness.add $(n, ret)
return ret
else:
return 10
proc asyncFact(n: int): Future[int] {.async.} =
if n > 0: result = n * await asyncFact(n-1)
else: result = 1
proc asyncIdentity(n: int): Future[int] {.async.} =
if n > 0: result = 1 + await asyncIdentity(n-1)
else: result = 0
proc main() {.async.} =
block: # then
let x = await fn(4)
.then((a: int) => a.float)
.then((a: float) => $a)
doAssert x == "14.0"
doAssert witness == @["(1, 11)", "(2, 12)", "(3, 13)", "(4, 14)"]
doAssert (await fn(2)) == 12
let x2 = await fn(4).then((a: int) => (discard)).then(() => 13)
doAssert x2 == 13
let x4 = await asyncFact(3).then(asyncIdentity).then(asyncIdentity).then((a:int) => a * 7).then(asyncIdentity)
doAssert x4 == 3 * 2 * 7
block: # bug #17177
proc asyncIdentityNested(n: int): Future[int] {.async.} = return n
let x5 = await asyncFact(3).then(asyncIdentityNested)
doAssert x5 == 3 * 2
when false: # xxx pending bug #17254
let x6 = await asyncFact(3).then((a:int) {.async.} => a * 11)
doAssert x6 == 3 * 2 * 11
block: # catch
var reason: Error
await fn(6).then((a: int) => (witness.add $a)).catch((r: Error) => (reason = r))
doAssert reason == nil
await fn(7).then((a: int) => (discard)).catch((r: Error) => (reason = r))
doAssert reason != nil
doAssert reason.name == "Error"
doAssert "foobar: 7" in $reason.message
echo "done" # justified here to make sure we're running this, since it's inside `async`
block asyncPragmaInType:
type Handler = proc () {.async.}
proc foo() {.async.} = discard
var x: Handler = foo
discard main()
|