|
23 | 23 | ) |
24 | 24 | from dqliteclient.node_store import MemoryNodeStore, NodeInfo, NodeStore |
25 | 25 | from dqliteclient.pool import ConnectionPool |
| 26 | +from dqlitewire import ( |
| 27 | + DEFAULT_MAX_CONTINUATION_FRAMES as _DEFAULT_MAX_CONTINUATION_FRAMES, |
| 28 | +) |
| 29 | +from dqlitewire import ( |
| 30 | + DEFAULT_MAX_TOTAL_ROWS as _DEFAULT_MAX_TOTAL_ROWS, |
| 31 | +) |
26 | 32 |
|
27 | 33 | __version__ = "0.1.3" |
28 | 34 |
|
@@ -54,8 +60,8 @@ async def connect( |
54 | 60 | *, |
55 | 61 | database: str = "default", |
56 | 62 | timeout: float = 10.0, |
57 | | - max_total_rows: int | None = 10_000_000, |
58 | | - max_continuation_frames: int | None = 100_000, |
| 63 | + max_total_rows: int | None = _DEFAULT_MAX_TOTAL_ROWS, |
| 64 | + max_continuation_frames: int | None = _DEFAULT_MAX_CONTINUATION_FRAMES, |
59 | 65 | trust_server_heartbeat: bool = False, |
60 | 66 | close_timeout: float = 0.5, |
61 | 67 | ) -> DqliteConnection: |
@@ -105,8 +111,8 @@ async def create_pool( |
105 | 111 | timeout: float = 10.0, |
106 | 112 | cluster: ClusterClient | None = None, |
107 | 113 | node_store: NodeStore | None = None, |
108 | | - max_total_rows: int | None = 10_000_000, |
109 | | - max_continuation_frames: int | None = 100_000, |
| 114 | + max_total_rows: int | None = _DEFAULT_MAX_TOTAL_ROWS, |
| 115 | + max_continuation_frames: int | None = _DEFAULT_MAX_CONTINUATION_FRAMES, |
110 | 116 | trust_server_heartbeat: bool = False, |
111 | 117 | close_timeout: float = 0.5, |
112 | 118 | ) -> ConnectionPool: |
|
0 commit comments