|
12 | 12 | def _make_handler(): |
13 | 13 | async def handler(): |
14 | 14 | pass |
| 15 | + |
15 | 16 | return handler |
16 | 17 |
|
17 | 18 |
|
18 | 19 | def bench_routing(): |
19 | 20 | tree = RadixTree() |
20 | 21 |
|
21 | 22 | # Register realistic routes (REST API pattern) |
22 | | - prefixes = ["users", "posts", "comments", "tags", "categories", "products", "orders", "payments"] |
| 23 | + prefixes = [ |
| 24 | + "users", |
| 25 | + "posts", |
| 26 | + "comments", |
| 27 | + "tags", |
| 28 | + "categories", |
| 29 | + "products", |
| 30 | + "orders", |
| 31 | + "payments", |
| 32 | + ] |
23 | 33 | routes_registered = 0 |
24 | 34 |
|
25 | 35 | for prefix in prefixes: |
26 | | - tree.insert(Route( |
27 | | - path=f"/{prefix}", |
28 | | - handler=_make_handler(), |
29 | | - methods=frozenset({"GET", "HEAD"}), |
30 | | - name=f"list_{prefix}", |
31 | | - )) |
32 | | - tree.insert(Route( |
33 | | - path=f"/{prefix}", |
34 | | - handler=_make_handler(), |
35 | | - methods=frozenset({"POST"}), |
36 | | - name=f"create_{prefix}", |
37 | | - )) |
38 | | - tree.insert(Route( |
39 | | - path=f"/{prefix}/{{id:int}}", |
40 | | - handler=_make_handler(), |
41 | | - methods=frozenset({"GET", "HEAD"}), |
42 | | - name=f"get_{prefix}", |
43 | | - )) |
44 | | - tree.insert(Route( |
45 | | - path=f"/{prefix}/{{id:int}}", |
46 | | - handler=_make_handler(), |
47 | | - methods=frozenset({"PUT"}), |
48 | | - name=f"update_{prefix}", |
49 | | - )) |
50 | | - tree.insert(Route( |
51 | | - path=f"/{prefix}/{{id:int}}", |
52 | | - handler=_make_handler(), |
53 | | - methods=frozenset({"DELETE"}), |
54 | | - name=f"delete_{prefix}", |
55 | | - )) |
| 36 | + tree.insert( |
| 37 | + Route( |
| 38 | + path=f"/{prefix}", |
| 39 | + handler=_make_handler(), |
| 40 | + methods=frozenset({"GET", "HEAD"}), |
| 41 | + name=f"list_{prefix}", |
| 42 | + ) |
| 43 | + ) |
| 44 | + tree.insert( |
| 45 | + Route( |
| 46 | + path=f"/{prefix}", |
| 47 | + handler=_make_handler(), |
| 48 | + methods=frozenset({"POST"}), |
| 49 | + name=f"create_{prefix}", |
| 50 | + ) |
| 51 | + ) |
| 52 | + tree.insert( |
| 53 | + Route( |
| 54 | + path=f"/{prefix}/{{id:int}}", |
| 55 | + handler=_make_handler(), |
| 56 | + methods=frozenset({"GET", "HEAD"}), |
| 57 | + name=f"get_{prefix}", |
| 58 | + ) |
| 59 | + ) |
| 60 | + tree.insert( |
| 61 | + Route( |
| 62 | + path=f"/{prefix}/{{id:int}}", |
| 63 | + handler=_make_handler(), |
| 64 | + methods=frozenset({"PUT"}), |
| 65 | + name=f"update_{prefix}", |
| 66 | + ) |
| 67 | + ) |
| 68 | + tree.insert( |
| 69 | + Route( |
| 70 | + path=f"/{prefix}/{{id:int}}", |
| 71 | + handler=_make_handler(), |
| 72 | + methods=frozenset({"DELETE"}), |
| 73 | + name=f"delete_{prefix}", |
| 74 | + ) |
| 75 | + ) |
56 | 76 | routes_registered += 5 |
57 | 77 |
|
58 | 78 | # Also add some nested routes |
59 | 79 | for prefix in prefixes[:4]: |
60 | | - tree.insert(Route( |
61 | | - path=f"/{prefix}/{{id:int}}/comments", |
62 | | - handler=_make_handler(), |
63 | | - methods=frozenset({"GET", "HEAD"}), |
64 | | - name=f"{prefix}_comments", |
65 | | - )) |
66 | | - tree.insert(Route( |
67 | | - path=f"/{prefix}/{{id:int}}/comments/{{comment_id:int}}", |
68 | | - handler=_make_handler(), |
69 | | - methods=frozenset({"GET", "HEAD"}), |
70 | | - name=f"{prefix}_comment_detail", |
71 | | - )) |
| 80 | + tree.insert( |
| 81 | + Route( |
| 82 | + path=f"/{prefix}/{{id:int}}/comments", |
| 83 | + handler=_make_handler(), |
| 84 | + methods=frozenset({"GET", "HEAD"}), |
| 85 | + name=f"{prefix}_comments", |
| 86 | + ) |
| 87 | + ) |
| 88 | + tree.insert( |
| 89 | + Route( |
| 90 | + path=f"/{prefix}/{{id:int}}/comments/{{comment_id:int}}", |
| 91 | + handler=_make_handler(), |
| 92 | + methods=frozenset({"GET", "HEAD"}), |
| 93 | + name=f"{prefix}_comment_detail", |
| 94 | + ) |
| 95 | + ) |
72 | 96 | routes_registered += 2 |
73 | 97 |
|
74 | 98 | print(f"Registered {routes_registered} routes across {len(prefixes)} resources\n") |
|
0 commit comments