@tdengine/websocket 3.2.2 → 3.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (275) hide show
  1. package/lib/package.json +64 -0
  2. package/lib/src/client/retryConfig.d.ts +10 -0
  3. package/lib/src/client/retryConfig.d.ts.map +1 -0
  4. package/lib/src/client/retryConfig.js +46 -0
  5. package/lib/src/client/wsClient.d.ts +24 -7
  6. package/lib/src/client/wsClient.d.ts.map +1 -1
  7. package/lib/src/client/wsClient.js +164 -137
  8. package/lib/src/client/wsConnector.d.ts +55 -9
  9. package/lib/src/client/wsConnector.d.ts.map +1 -1
  10. package/lib/src/client/wsConnector.js +522 -102
  11. package/lib/src/client/wsConnectorPool.d.ts +5 -1
  12. package/lib/src/client/wsConnectorPool.d.ts.map +1 -1
  13. package/lib/src/client/wsConnectorPool.js +67 -52
  14. package/lib/src/client/wsEventCallback.d.ts +3 -0
  15. package/lib/src/client/wsEventCallback.d.ts.map +1 -1
  16. package/lib/src/client/wsEventCallback.js +67 -8
  17. package/lib/src/client/wsResponse.d.ts +0 -3
  18. package/lib/src/client/wsResponse.d.ts.map +1 -1
  19. package/lib/src/client/wsResponse.js +0 -3
  20. package/lib/src/common/addressConnectionTracker.d.ts +11 -0
  21. package/lib/src/common/addressConnectionTracker.d.ts.map +1 -0
  22. package/lib/src/common/addressConnectionTracker.js +53 -0
  23. package/lib/src/common/config.d.ts +3 -0
  24. package/lib/src/common/config.d.ts.map +1 -1
  25. package/lib/src/common/config.js +6 -0
  26. package/lib/src/common/constant.d.ts +1 -0
  27. package/lib/src/common/constant.d.ts.map +1 -1
  28. package/lib/src/common/constant.js +6 -1
  29. package/lib/src/common/dsn.d.ts +30 -0
  30. package/lib/src/common/dsn.d.ts.map +1 -0
  31. package/lib/src/common/dsn.js +221 -0
  32. package/lib/src/common/taosResult.d.ts.map +1 -1
  33. package/lib/src/common/taosResult.js +0 -5
  34. package/lib/src/common/urlParser.d.ts +32 -0
  35. package/lib/src/common/urlParser.d.ts.map +1 -0
  36. package/lib/src/common/urlParser.js +201 -0
  37. package/lib/src/common/utils.d.ts +6 -1
  38. package/lib/src/common/utils.d.ts.map +1 -1
  39. package/lib/src/common/utils.js +74 -22
  40. package/lib/src/index.d.ts.map +1 -1
  41. package/lib/src/sql/wsSql.d.ts.map +1 -1
  42. package/lib/src/sql/wsSql.js +2 -2
  43. package/lib/src/stmt/FieldBindParams.d.ts.map +1 -1
  44. package/lib/src/stmt/wsColumnInfo.d.ts.map +1 -1
  45. package/lib/src/stmt/wsParams1.d.ts.map +1 -1
  46. package/lib/src/stmt/wsParams1.js +26 -26
  47. package/lib/src/stmt/wsParams2.d.ts.map +1 -1
  48. package/lib/src/stmt/wsParams2.js +0 -3
  49. package/lib/src/stmt/wsParamsBase.d.ts.map +1 -1
  50. package/lib/src/stmt/wsProto.d.ts.map +1 -1
  51. package/lib/src/stmt/wsProto.js +16 -16
  52. package/lib/src/stmt/wsStmt1.d.ts.map +1 -1
  53. package/lib/src/stmt/wsStmt2.d.ts +12 -4
  54. package/lib/src/stmt/wsStmt2.d.ts.map +1 -1
  55. package/lib/src/stmt/wsStmt2.js +182 -64
  56. package/lib/src/stmt/wsTableInfo.d.ts.map +1 -1
  57. package/lib/src/tmq/config.d.ts +4 -2
  58. package/lib/src/tmq/config.d.ts.map +1 -1
  59. package/lib/src/tmq/config.js +30 -13
  60. package/lib/src/tmq/constant.d.ts +4 -0
  61. package/lib/src/tmq/constant.d.ts.map +1 -1
  62. package/lib/src/tmq/constant.js +4 -0
  63. package/lib/src/tmq/tmqResponse.d.ts.map +1 -1
  64. package/lib/src/tmq/tmqResponse.js +1 -4
  65. package/lib/src/tmq/wsTmq.d.ts +4 -1
  66. package/lib/src/tmq/wsTmq.d.ts.map +1 -1
  67. package/lib/src/tmq/wsTmq.js +55 -27
  68. package/lib/test/bulkPulling/a.test.d.ts +2 -0
  69. package/lib/test/bulkPulling/a.test.d.ts.map +1 -0
  70. package/lib/test/bulkPulling/a.test.js +166 -0
  71. package/lib/test/bulkPulling/cloud.tmq.test.js +3 -2
  72. package/lib/test/bulkPulling/decimal.test.js +8 -8
  73. package/lib/test/bulkPulling/dsn.test.d.ts +2 -0
  74. package/lib/test/bulkPulling/dsn.test.d.ts.map +1 -0
  75. package/lib/test/bulkPulling/dsn.test.js +397 -0
  76. package/lib/test/bulkPulling/queryTables.test.js +1 -1
  77. package/lib/test/bulkPulling/retryConfig.test.d.ts +2 -0
  78. package/lib/test/bulkPulling/retryConfig.test.d.ts.map +1 -0
  79. package/lib/test/bulkPulling/retryConfig.test.js +34 -0
  80. package/lib/test/bulkPulling/schemaless.test.js +15 -14
  81. package/lib/test/bulkPulling/sql.failover.test.d.ts +2 -0
  82. package/lib/test/bulkPulling/sql.failover.test.d.ts.map +1 -0
  83. package/lib/test/bulkPulling/sql.failover.test.js +338 -0
  84. package/lib/test/bulkPulling/sql.test.js +116 -44
  85. package/lib/test/bulkPulling/stmt1.func.test.js +31 -30
  86. package/lib/test/bulkPulling/stmt1.type.test.js +1 -1
  87. package/lib/test/bulkPulling/stmt2.failover.test.d.ts +2 -0
  88. package/lib/test/bulkPulling/stmt2.failover.test.d.ts.map +1 -0
  89. package/lib/test/bulkPulling/stmt2.failover.test.js +313 -0
  90. package/lib/test/bulkPulling/stmt2.func.test.js +37 -36
  91. package/lib/test/bulkPulling/stmt2.init.failover.test.d.ts +2 -0
  92. package/lib/test/bulkPulling/stmt2.init.failover.test.d.ts.map +1 -0
  93. package/lib/test/bulkPulling/stmt2.init.failover.test.js +50 -0
  94. package/lib/test/bulkPulling/stmt2.type.test.js +1 -1
  95. package/lib/test/bulkPulling/tmq.config.test.d.ts +2 -0
  96. package/lib/test/bulkPulling/tmq.config.test.d.ts.map +1 -0
  97. package/lib/test/bulkPulling/tmq.config.test.js +77 -0
  98. package/lib/test/bulkPulling/tmq.failover.test.d.ts +2 -0
  99. package/lib/test/bulkPulling/tmq.failover.test.d.ts.map +1 -0
  100. package/lib/test/bulkPulling/tmq.failover.test.js +404 -0
  101. package/lib/test/bulkPulling/tmq.test.js +135 -14
  102. package/lib/test/bulkPulling/urlParser.test.d.ts +2 -0
  103. package/lib/test/bulkPulling/urlParser.test.d.ts.map +1 -0
  104. package/lib/test/bulkPulling/urlParser.test.js +452 -0
  105. package/lib/test/bulkPulling/wsClient.reconnect.integration.test.d.ts +2 -0
  106. package/lib/test/bulkPulling/wsClient.reconnect.integration.test.d.ts.map +1 -0
  107. package/lib/test/bulkPulling/wsClient.reconnect.integration.test.js +184 -0
  108. package/lib/test/bulkPulling/wsClient.recovery.test.d.ts +2 -0
  109. package/lib/test/bulkPulling/wsClient.recovery.test.d.ts.map +1 -0
  110. package/lib/test/bulkPulling/wsClient.recovery.test.js +104 -0
  111. package/lib/test/bulkPulling/wsConfig.dsn.test.d.ts +2 -0
  112. package/lib/test/bulkPulling/wsConfig.dsn.test.d.ts.map +1 -0
  113. package/lib/test/bulkPulling/wsConfig.dsn.test.js +39 -0
  114. package/lib/test/bulkPulling/wsConnectPool.test.js +7 -7
  115. package/lib/test/bulkPulling/wsConnector.failover.test.d.ts +2 -0
  116. package/lib/test/bulkPulling/wsConnector.failover.test.d.ts.map +1 -0
  117. package/lib/test/bulkPulling/wsConnector.failover.test.js +497 -0
  118. package/lib/test/bulkPulling/wsConnectorPool.key.test.d.ts +2 -0
  119. package/lib/test/bulkPulling/wsConnectorPool.key.test.d.ts.map +1 -0
  120. package/lib/test/bulkPulling/wsConnectorPool.key.test.js +52 -0
  121. package/lib/test/bulkPulling/wsConnectorPool.keyAuth.test.d.ts +2 -0
  122. package/lib/test/bulkPulling/wsConnectorPool.keyAuth.test.d.ts.map +1 -0
  123. package/lib/test/bulkPulling/wsConnectorPool.keyAuth.test.js +28 -0
  124. package/lib/test/bulkPulling/wsEventCallback.test.d.ts +2 -0
  125. package/lib/test/bulkPulling/wsEventCallback.test.d.ts.map +1 -0
  126. package/lib/test/bulkPulling/wsEventCallback.test.js +54 -0
  127. package/lib/test/bulkPulling/wsProxy.failover.integration.test.d.ts +2 -0
  128. package/lib/test/bulkPulling/wsProxy.failover.integration.test.d.ts.map +1 -0
  129. package/lib/test/bulkPulling/wsProxy.failover.integration.test.js +120 -0
  130. package/lib/test/bulkPulling/wsProxy.failover.test.d.ts +2 -0
  131. package/lib/test/bulkPulling/wsProxy.failover.test.d.ts.map +1 -0
  132. package/lib/test/bulkPulling/wsProxy.failover.test.js +465 -0
  133. package/lib/test/client/wsClient.recovery.test.d.ts +2 -0
  134. package/lib/test/client/wsClient.recovery.test.d.ts.map +1 -0
  135. package/lib/test/client/wsClient.recovery.test.js +122 -0
  136. package/lib/test/client/wsConnectPool.test.d.ts +2 -0
  137. package/lib/test/client/wsConnectPool.test.d.ts.map +1 -0
  138. package/lib/test/client/wsConnectPool.test.js +147 -0
  139. package/lib/test/client/wsConnector.failover.test.d.ts +2 -0
  140. package/lib/test/client/wsConnector.failover.test.d.ts.map +1 -0
  141. package/lib/test/client/wsConnector.failover.test.js +681 -0
  142. package/lib/test/client/wsConnector.leastConnections.test.d.ts +2 -0
  143. package/lib/test/client/wsConnector.leastConnections.test.d.ts.map +1 -0
  144. package/lib/test/client/wsConnector.leastConnections.test.js +71 -0
  145. package/lib/test/client/wsConnectorPool.key.test.d.ts +2 -0
  146. package/lib/test/client/wsConnectorPool.key.test.d.ts.map +1 -0
  147. package/lib/test/client/wsConnectorPool.key.test.js +127 -0
  148. package/lib/test/client/wsEventCallback.test.d.ts +2 -0
  149. package/lib/test/client/wsEventCallback.test.d.ts.map +1 -0
  150. package/lib/test/client/wsEventCallback.test.js +98 -0
  151. package/lib/test/common/addressConnectionTracker.test.d.ts +2 -0
  152. package/lib/test/common/addressConnectionTracker.test.d.ts.map +1 -0
  153. package/lib/test/common/addressConnectionTracker.test.js +74 -0
  154. package/lib/test/common/dsn.test.d.ts +2 -0
  155. package/lib/test/common/dsn.test.d.ts.map +1 -0
  156. package/lib/test/common/dsn.test.js +406 -0
  157. package/lib/test/common/log.test.d.ts +2 -0
  158. package/lib/test/common/log.test.d.ts.map +1 -0
  159. package/lib/test/common/log.test.js +54 -0
  160. package/lib/test/common/utils.test.d.ts +2 -0
  161. package/lib/test/common/utils.test.d.ts.map +1 -0
  162. package/lib/test/common/utils.test.js +13 -0
  163. package/lib/test/common/wsConfig.dsn.test.d.ts +2 -0
  164. package/lib/test/common/wsConfig.dsn.test.d.ts.map +1 -0
  165. package/lib/test/common/wsConfig.dsn.test.js +39 -0
  166. package/lib/test/helpers/utils.d.ts +27 -0
  167. package/lib/test/helpers/utils.d.ts.map +1 -0
  168. package/lib/test/helpers/utils.js +341 -0
  169. package/lib/test/helpers/wsFailoverProxy.d.ts +109 -0
  170. package/lib/test/helpers/wsFailoverProxy.d.ts.map +1 -0
  171. package/lib/test/helpers/wsFailoverProxy.js +420 -0
  172. package/lib/test/helpers/wsProxy.d.ts +110 -0
  173. package/lib/test/helpers/wsProxy.d.ts.map +1 -0
  174. package/lib/test/helpers/wsProxy.js +429 -0
  175. package/lib/test/sql/core/decimal.test.d.ts +2 -0
  176. package/lib/test/sql/core/decimal.test.d.ts.map +1 -0
  177. package/lib/test/sql/core/decimal.test.js +153 -0
  178. package/lib/test/sql/core/queryTables.test.d.ts +2 -0
  179. package/lib/test/sql/core/queryTables.test.d.ts.map +1 -0
  180. package/lib/test/sql/core/queryTables.test.js +506 -0
  181. package/lib/test/sql/core/schemaless.test.d.ts +2 -0
  182. package/lib/test/sql/core/schemaless.test.d.ts.map +1 -0
  183. package/lib/test/sql/core/schemaless.test.js +102 -0
  184. package/lib/test/sql/core/sql.test.d.ts +2 -0
  185. package/lib/test/sql/core/sql.test.d.ts.map +1 -0
  186. package/lib/test/sql/core/sql.test.js +324 -0
  187. package/lib/test/sql/decimal.test.d.ts +2 -0
  188. package/lib/test/sql/decimal.test.d.ts.map +1 -0
  189. package/lib/test/sql/decimal.test.js +153 -0
  190. package/lib/test/sql/failover/sql.failover.test.d.ts +2 -0
  191. package/lib/test/sql/failover/sql.failover.test.d.ts.map +1 -0
  192. package/lib/test/sql/failover/sql.failover.test.js +341 -0
  193. package/lib/test/sql/queryTables.test.d.ts +2 -0
  194. package/lib/test/sql/queryTables.test.d.ts.map +1 -0
  195. package/lib/test/sql/queryTables.test.js +506 -0
  196. package/lib/test/sql/schemaless.test.d.ts +2 -0
  197. package/lib/test/sql/schemaless.test.d.ts.map +1 -0
  198. package/lib/test/sql/schemaless.test.js +102 -0
  199. package/lib/test/sql/sql.failover.test.d.ts +2 -0
  200. package/lib/test/sql/sql.failover.test.d.ts.map +1 -0
  201. package/lib/test/sql/sql.failover.test.js +341 -0
  202. package/lib/test/sql/sql.test.d.ts +2 -0
  203. package/lib/test/sql/sql.test.d.ts.map +1 -0
  204. package/lib/test/sql/sql.test.js +324 -0
  205. package/lib/test/stmt/failover/stmt2.failover.mock.test.d.ts +2 -0
  206. package/lib/test/stmt/failover/stmt2.failover.mock.test.d.ts.map +1 -0
  207. package/lib/test/stmt/failover/stmt2.failover.mock.test.js +341 -0
  208. package/lib/test/stmt/failover/stmt2.failover.test.d.ts +2 -0
  209. package/lib/test/stmt/failover/stmt2.failover.test.d.ts.map +1 -0
  210. package/lib/test/stmt/failover/stmt2.failover.test.js +384 -0
  211. package/lib/test/stmt/stmt1.func.test.d.ts +2 -0
  212. package/lib/test/stmt/stmt1.func.test.d.ts.map +1 -0
  213. package/lib/test/stmt/stmt1.func.test.js +418 -0
  214. package/lib/test/stmt/stmt1.type.test.d.ts +2 -0
  215. package/lib/test/stmt/stmt1.type.test.d.ts.map +1 -0
  216. package/lib/test/stmt/stmt1.type.test.js +399 -0
  217. package/lib/test/stmt/stmt2.failover.mock.test.d.ts +2 -0
  218. package/lib/test/stmt/stmt2.failover.mock.test.d.ts.map +1 -0
  219. package/lib/test/stmt/stmt2.failover.mock.test.js +341 -0
  220. package/lib/test/stmt/stmt2.failover.test.d.ts +2 -0
  221. package/lib/test/stmt/stmt2.failover.test.d.ts.map +1 -0
  222. package/lib/test/stmt/stmt2.failover.test.js +384 -0
  223. package/lib/test/stmt/stmt2.func.test.d.ts +2 -0
  224. package/lib/test/stmt/stmt2.func.test.d.ts.map +1 -0
  225. package/lib/test/stmt/stmt2.func.test.js +537 -0
  226. package/lib/test/stmt/stmt2.type.test.d.ts +2 -0
  227. package/lib/test/stmt/stmt2.type.test.d.ts.map +1 -0
  228. package/lib/test/stmt/stmt2.type.test.js +401 -0
  229. package/lib/test/stmt/v1/stmt1.func.test.d.ts +2 -0
  230. package/lib/test/stmt/v1/stmt1.func.test.d.ts.map +1 -0
  231. package/lib/test/stmt/v1/stmt1.func.test.js +418 -0
  232. package/lib/test/stmt/v1/stmt1.type.test.d.ts +2 -0
  233. package/lib/test/stmt/v1/stmt1.type.test.d.ts.map +1 -0
  234. package/lib/test/stmt/v1/stmt1.type.test.js +399 -0
  235. package/lib/test/stmt/v2/stmt2.func.test.d.ts +2 -0
  236. package/lib/test/stmt/v2/stmt2.func.test.d.ts.map +1 -0
  237. package/lib/test/stmt/v2/stmt2.func.test.js +537 -0
  238. package/lib/test/stmt/v2/stmt2.type.test.d.ts +2 -0
  239. package/lib/test/stmt/v2/stmt2.type.test.d.ts.map +1 -0
  240. package/lib/test/stmt/v2/stmt2.type.test.js +401 -0
  241. package/lib/test/tmq/cloud/cloud.tmq.test.d.ts +2 -0
  242. package/lib/test/tmq/cloud/cloud.tmq.test.d.ts.map +1 -0
  243. package/lib/test/tmq/cloud/cloud.tmq.test.js +84 -0
  244. package/lib/test/tmq/cloud/tmq.cloud.test.d.ts +2 -0
  245. package/lib/test/tmq/cloud/tmq.cloud.test.d.ts.map +1 -0
  246. package/lib/test/tmq/cloud/tmq.cloud.test.js +82 -0
  247. package/lib/test/tmq/core/tmq.config.test.d.ts +2 -0
  248. package/lib/test/tmq/core/tmq.config.test.d.ts.map +1 -0
  249. package/lib/test/tmq/core/tmq.config.test.js +83 -0
  250. package/lib/test/tmq/core/tmq.test.d.ts +2 -0
  251. package/lib/test/tmq/core/tmq.test.d.ts.map +1 -0
  252. package/lib/test/tmq/core/tmq.test.js +513 -0
  253. package/lib/test/tmq/failover/tmq.failover.test.d.ts +2 -0
  254. package/lib/test/tmq/failover/tmq.failover.test.d.ts.map +1 -0
  255. package/lib/test/tmq/failover/tmq.failover.test.js +404 -0
  256. package/lib/test/tmq/tmq.cloud.test.d.ts +2 -0
  257. package/lib/test/tmq/tmq.cloud.test.d.ts.map +1 -0
  258. package/lib/test/tmq/tmq.cloud.test.js +82 -0
  259. package/lib/test/tmq/tmq.config.test.d.ts +2 -0
  260. package/lib/test/tmq/tmq.config.test.d.ts.map +1 -0
  261. package/lib/test/tmq/tmq.config.test.js +94 -0
  262. package/lib/test/tmq/tmq.failover.test.d.ts +2 -0
  263. package/lib/test/tmq/tmq.failover.test.d.ts.map +1 -0
  264. package/lib/test/tmq/tmq.failover.test.js +404 -0
  265. package/lib/test/tmq/tmq.test.d.ts +2 -0
  266. package/lib/test/tmq/tmq.test.d.ts.map +1 -0
  267. package/lib/test/tmq/tmq.test.js +513 -0
  268. package/lib/test/unit/connectionManager.test.d.ts +2 -0
  269. package/lib/test/unit/connectionManager.test.d.ts.map +1 -0
  270. package/lib/test/unit/connectionManager.test.js +91 -0
  271. package/lib/test/utils.d.ts +4 -0
  272. package/lib/test/utils.d.ts.map +1 -1
  273. package/lib/test/utils.js +11 -17
  274. package/package.json +1 -1
  275. package/readme.md +2 -2
@@ -0,0 +1,420 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || (function () {
19
+ var ownKeys = function(o) {
20
+ ownKeys = Object.getOwnPropertyNames || function (o) {
21
+ var ar = [];
22
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
23
+ return ar;
24
+ };
25
+ return ownKeys(o);
26
+ };
27
+ return function (mod) {
28
+ if (mod && mod.__esModule) return mod;
29
+ var result = {};
30
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
31
+ __setModuleDefault(result, mod);
32
+ return result;
33
+ };
34
+ })();
35
+ Object.defineProperty(exports, "__esModule", { value: true });
36
+ exports.WsFailoverProxy = void 0;
37
+ const http = __importStar(require("http"));
38
+ const websocket_1 = require("websocket");
39
+ const UPSTREAM_WS_URL = "ws://localhost:6041/ws";
40
+ const SUPPORTED_PATH = "/ws";
41
+ class WsFailoverProxy {
42
+ constructor(options) {
43
+ this._lockedPort = null;
44
+ this._httpServer = null;
45
+ this._wsServer = null;
46
+ this._running = false;
47
+ this._nextConnectionId = 1;
48
+ this._lifecycleChain = Promise.resolve();
49
+ this._tunnels = new Map();
50
+ this._eventLog = [];
51
+ if (!options.host || options.host.trim().length === 0) {
52
+ throw new Error("listen host must not be empty");
53
+ }
54
+ if (!Number.isInteger(options.port) ||
55
+ options.port < 0 ||
56
+ options.port > 65535) {
57
+ throw new Error(`invalid listen port: ${options.port}, expected 0-65535`);
58
+ }
59
+ this._listenHost = options.host;
60
+ this._requestedPort = options.port;
61
+ this._onEvent = options.onEvent;
62
+ this._control = {
63
+ restart: async (opts) => {
64
+ await this.restart(opts);
65
+ },
66
+ stop: async (reason) => {
67
+ await this.stop(reason);
68
+ },
69
+ isRunning: () => this.isRunning(),
70
+ };
71
+ }
72
+ static async create(options) {
73
+ const proxy = new WsFailoverProxy(options);
74
+ await proxy.start("initial start");
75
+ return proxy;
76
+ }
77
+ async start(reason) {
78
+ await this.enqueueLifecycle(async () => {
79
+ await this.startInternal(reason);
80
+ });
81
+ }
82
+ async stop(reason) {
83
+ await this.enqueueLifecycle(async () => {
84
+ await this.stopInternal(reason);
85
+ });
86
+ }
87
+ async restart(opts) {
88
+ await this.enqueueLifecycle(async () => {
89
+ const reason = opts?.reason || "restart";
90
+ const downtimeMs = Math.max(0, opts?.downtimeMs || 0);
91
+ await this.stopInternal(reason);
92
+ if (downtimeMs > 0) {
93
+ await sleep(downtimeMs);
94
+ }
95
+ await this.startInternal(reason);
96
+ });
97
+ }
98
+ isRunning() {
99
+ return this._running;
100
+ }
101
+ getPort() {
102
+ if (this._lockedPort === null) {
103
+ throw new Error("proxy has not started yet");
104
+ }
105
+ return this._lockedPort;
106
+ }
107
+ getHost() {
108
+ return this._listenHost;
109
+ }
110
+ getUrl() {
111
+ return `ws://${this._listenHost}:${this.getPort()}${SUPPORTED_PATH}`;
112
+ }
113
+ getEventLog() {
114
+ return [...this._eventLog];
115
+ }
116
+ async enqueueLifecycle(task) {
117
+ this._lifecycleChain = this._lifecycleChain.then(task, task);
118
+ await this._lifecycleChain;
119
+ }
120
+ async startInternal(reason) {
121
+ if (this._running) {
122
+ return;
123
+ }
124
+ const listenPort = this._lockedPort === null
125
+ ? this._requestedPort
126
+ : this._lockedPort;
127
+ const httpServer = http.createServer((_req, res) => {
128
+ res.statusCode = 404;
129
+ res.end();
130
+ });
131
+ await new Promise((resolve, reject) => {
132
+ const onError = (err) => {
133
+ httpServer.removeListener("listening", onListening);
134
+ reject(err);
135
+ };
136
+ const onListening = () => {
137
+ httpServer.removeListener("error", onError);
138
+ resolve();
139
+ };
140
+ httpServer.once("error", onError);
141
+ httpServer.once("listening", onListening);
142
+ httpServer.listen(listenPort, this._listenHost);
143
+ });
144
+ const address = httpServer.address();
145
+ if (!address || typeof address === "string") {
146
+ httpServer.close();
147
+ throw new Error("failed to get proxy listen address");
148
+ }
149
+ if (this._lockedPort === null) {
150
+ this._lockedPort = address.port;
151
+ }
152
+ const wsServer = new websocket_1.server({
153
+ httpServer,
154
+ autoAcceptConnections: false,
155
+ });
156
+ wsServer.on("request", (request) => {
157
+ this.handleClientRequest(request);
158
+ });
159
+ this._httpServer = httpServer;
160
+ this._wsServer = wsServer;
161
+ this._running = true;
162
+ this.emitEvent({
163
+ type: "proxy_started",
164
+ timestamp: Date.now(),
165
+ host: this._listenHost,
166
+ port: this.getPort(),
167
+ url: this.getUrl(),
168
+ reason,
169
+ });
170
+ }
171
+ async stopInternal(reason) {
172
+ if (!this._running && !this._httpServer && !this._wsServer) {
173
+ return;
174
+ }
175
+ this._running = false;
176
+ for (const tunnel of Array.from(this._tunnels.values())) {
177
+ this.closeTunnel(tunnel.id, `proxy stopping${reason ? `: ${reason}` : ""}`, "proxy");
178
+ }
179
+ if (this._wsServer) {
180
+ try {
181
+ this._wsServer.removeAllListeners("request");
182
+ this._wsServer.closeAllConnections();
183
+ this._wsServer.unmount();
184
+ }
185
+ catch (_err) {
186
+ // Ignore shutdown errors and continue releasing resources.
187
+ }
188
+ }
189
+ if (this._httpServer) {
190
+ await new Promise((resolve) => {
191
+ if (!this._httpServer || !this._httpServer.listening) {
192
+ resolve();
193
+ return;
194
+ }
195
+ this._httpServer.close(() => resolve());
196
+ });
197
+ }
198
+ this._wsServer = null;
199
+ this._httpServer = null;
200
+ this.emitEvent({
201
+ type: "proxy_stopped",
202
+ timestamp: Date.now(),
203
+ reason,
204
+ });
205
+ }
206
+ handleClientRequest(request) {
207
+ if (!this._running) {
208
+ request.reject(503, "proxy is not running");
209
+ return;
210
+ }
211
+ const path = request.resourceURL?.pathname || request.resource;
212
+ if (path !== SUPPORTED_PATH) {
213
+ request.reject(404, `unsupported path: ${path}`);
214
+ return;
215
+ }
216
+ const clientConn = request.accept(undefined, request.origin);
217
+ const connectionId = this._nextConnectionId;
218
+ this._nextConnectionId += 1;
219
+ const tunnel = {
220
+ id: connectionId,
221
+ closed: false,
222
+ messageSeq: 0,
223
+ clientConn,
224
+ pendingFrames: [],
225
+ };
226
+ this._tunnels.set(connectionId, tunnel);
227
+ this.emitEvent({
228
+ type: "client_connected",
229
+ timestamp: Date.now(),
230
+ connectionId,
231
+ path,
232
+ remoteAddress: request.remoteAddress,
233
+ });
234
+ clientConn.on("message", (message) => {
235
+ this.handleClientMessage(tunnel, message);
236
+ });
237
+ clientConn.on("close", (code, desc) => {
238
+ this.closeTunnel(connectionId, `client closed: ${code} ${desc}`, "client");
239
+ });
240
+ clientConn.on("error", (error) => {
241
+ this.emitError("client_error", error, connectionId);
242
+ this.closeTunnel(connectionId, `client error: ${error.message}`, "client");
243
+ });
244
+ this.connectUpstream(tunnel);
245
+ }
246
+ connectUpstream(tunnel) {
247
+ const upstreamClient = new websocket_1.client();
248
+ tunnel.upstreamClient = upstreamClient;
249
+ upstreamClient.on("connect", (upstreamConn) => {
250
+ if (tunnel.closed) {
251
+ upstreamConn.close();
252
+ return;
253
+ }
254
+ tunnel.upstreamConn = upstreamConn;
255
+ this.emitEvent({
256
+ type: "upstream_connected",
257
+ timestamp: Date.now(),
258
+ connectionId: tunnel.id,
259
+ upstreamUrl: UPSTREAM_WS_URL,
260
+ });
261
+ upstreamConn.on("message", (message) => {
262
+ this.handleUpstreamMessage(tunnel, message);
263
+ });
264
+ upstreamConn.on("close", (code, desc) => {
265
+ this.closeTunnel(tunnel.id, `upstream closed: ${code} ${desc}`, "upstream");
266
+ });
267
+ upstreamConn.on("error", (error) => {
268
+ this.emitError("upstream_error", error, tunnel.id);
269
+ this.closeTunnel(tunnel.id, `upstream error: ${error.message}`, "upstream");
270
+ });
271
+ this.flushPendingFrames(tunnel);
272
+ });
273
+ upstreamClient.on("connectFailed", (error) => {
274
+ this.emitError("upstream_connect_failed", error, tunnel.id);
275
+ this.closeTunnel(tunnel.id, `upstream connect failed: ${error.message}`, "upstream");
276
+ });
277
+ upstreamClient.connect(UPSTREAM_WS_URL);
278
+ }
279
+ handleClientMessage(tunnel, message) {
280
+ if (tunnel.closed) {
281
+ return;
282
+ }
283
+ if (message.type === "utf8") {
284
+ const rawData = message.utf8Data;
285
+ this.emitMessageEvent(tunnel, "client_to_upstream", false, rawData);
286
+ if (this.isConnected(tunnel.upstreamConn)) {
287
+ tunnel.upstreamConn.sendUTF(rawData);
288
+ return;
289
+ }
290
+ tunnel.pendingFrames.push({
291
+ isBinary: false,
292
+ data: rawData,
293
+ });
294
+ return;
295
+ }
296
+ const rawData = Buffer.from(message.binaryData);
297
+ this.emitMessageEvent(tunnel, "client_to_upstream", true, rawData);
298
+ if (this.isConnected(tunnel.upstreamConn)) {
299
+ tunnel.upstreamConn.sendBytes(rawData);
300
+ return;
301
+ }
302
+ tunnel.pendingFrames.push({
303
+ isBinary: true,
304
+ data: rawData,
305
+ });
306
+ }
307
+ handleUpstreamMessage(tunnel, message) {
308
+ if (tunnel.closed || !this.isConnected(tunnel.clientConn)) {
309
+ this.closeTunnel(tunnel.id, "client connection unavailable while forwarding upstream message", "client");
310
+ return;
311
+ }
312
+ if (message.type === "utf8") {
313
+ const rawData = message.utf8Data;
314
+ this.emitMessageEvent(tunnel, "upstream_to_client", false, rawData);
315
+ tunnel.clientConn.sendUTF(rawData);
316
+ return;
317
+ }
318
+ const rawData = Buffer.from(message.binaryData);
319
+ this.emitMessageEvent(tunnel, "upstream_to_client", true, rawData);
320
+ tunnel.clientConn.sendBytes(rawData);
321
+ }
322
+ flushPendingFrames(tunnel) {
323
+ if (!this.isConnected(tunnel.upstreamConn) || tunnel.closed) {
324
+ return;
325
+ }
326
+ while (tunnel.pendingFrames.length > 0) {
327
+ const frame = tunnel.pendingFrames.shift();
328
+ if (!frame) {
329
+ break;
330
+ }
331
+ if (frame.isBinary) {
332
+ tunnel.upstreamConn.sendBytes(frame.data);
333
+ }
334
+ else {
335
+ tunnel.upstreamConn.sendUTF(frame.data);
336
+ }
337
+ }
338
+ }
339
+ closeTunnel(connectionId, reason, source) {
340
+ const tunnel = this._tunnels.get(connectionId);
341
+ if (!tunnel || tunnel.closed) {
342
+ return;
343
+ }
344
+ tunnel.closed = true;
345
+ this._tunnels.delete(connectionId);
346
+ const isProxyInjectedFailure = source === "proxy";
347
+ this.safeCloseConnection(tunnel.clientConn, isProxyInjectedFailure);
348
+ this.safeCloseConnection(tunnel.upstreamConn, isProxyInjectedFailure);
349
+ this.emitEvent({
350
+ type: "tunnel_closed",
351
+ timestamp: Date.now(),
352
+ connectionId,
353
+ reason,
354
+ source,
355
+ });
356
+ }
357
+ isConnected(conn) {
358
+ return !!conn && conn.connected;
359
+ }
360
+ emitMessageEvent(tunnel, direction, isBinary, rawData) {
361
+ tunnel.messageSeq += 1;
362
+ this.emitEvent({
363
+ type: "message",
364
+ timestamp: Date.now(),
365
+ connectionId: tunnel.id,
366
+ direction,
367
+ isBinary,
368
+ byteLength: typeof rawData === "string"
369
+ ? Buffer.byteLength(rawData)
370
+ : rawData.byteLength,
371
+ rawData,
372
+ messageSeq: tunnel.messageSeq,
373
+ });
374
+ }
375
+ emitError(stage, error, connectionId, sourceEventType, details, notifyCallback = true) {
376
+ this.emitEvent({
377
+ type: "error",
378
+ timestamp: Date.now(),
379
+ connectionId,
380
+ error: normalizeError(error),
381
+ stage,
382
+ sourceEventType,
383
+ details,
384
+ }, notifyCallback);
385
+ }
386
+ emitEvent(event, notifyCallback = true) {
387
+ this._eventLog.push(event);
388
+ if (!notifyCallback || !this._onEvent) {
389
+ return;
390
+ }
391
+ void Promise.resolve(this._onEvent(event, this._control)).catch((err) => {
392
+ this.emitError("event_handler", err, event.connectionId, event.type, "event callback threw an error", false);
393
+ });
394
+ }
395
+ safeCloseConnection(conn, abnormal) {
396
+ try {
397
+ if (!this.isConnected(conn)) {
398
+ return;
399
+ }
400
+ if (abnormal) {
401
+ conn.close(1012, "proxy restart");
402
+ return;
403
+ }
404
+ conn.close();
405
+ }
406
+ catch (_err) {
407
+ // Ignore close failures.
408
+ }
409
+ }
410
+ }
411
+ exports.WsFailoverProxy = WsFailoverProxy;
412
+ function normalizeError(error) {
413
+ if (error instanceof Error) {
414
+ return error;
415
+ }
416
+ return new Error(String(error));
417
+ }
418
+ function sleep(ms) {
419
+ return new Promise((resolve) => setTimeout(resolve, ms));
420
+ }
@@ -0,0 +1,110 @@
1
+ type WsProxyDirection = "client_to_upstream" | "upstream_to_client";
2
+ type WsProxyEventType = "proxy_started" | "proxy_stopped" | "client_connected" | "upstream_connected" | "message" | "tunnel_closed" | "error";
3
+ interface WsProxyEventBase {
4
+ type: WsProxyEventType;
5
+ timestamp: number;
6
+ connectionId?: number;
7
+ }
8
+ export interface WsProxyStartedEvent extends WsProxyEventBase {
9
+ type: "proxy_started";
10
+ host: string;
11
+ port: number;
12
+ url: string;
13
+ reason?: string;
14
+ }
15
+ export interface WsProxyStoppedEvent extends WsProxyEventBase {
16
+ type: "proxy_stopped";
17
+ reason?: string;
18
+ }
19
+ export interface WsProxyClientConnectedEvent extends WsProxyEventBase {
20
+ type: "client_connected";
21
+ connectionId: number;
22
+ path: string;
23
+ remoteAddress: string;
24
+ }
25
+ export interface WsProxyUpstreamConnectedEvent extends WsProxyEventBase {
26
+ type: "upstream_connected";
27
+ connectionId: number;
28
+ upstreamUrl: string;
29
+ }
30
+ export interface WsProxyMessageEvent extends WsProxyEventBase {
31
+ type: "message";
32
+ connectionId: number;
33
+ direction: WsProxyDirection;
34
+ isBinary: boolean;
35
+ byteLength: number;
36
+ rawData: Buffer | string;
37
+ messageSeq: number;
38
+ }
39
+ export interface WsProxyTunnelClosedEvent extends WsProxyEventBase {
40
+ type: "tunnel_closed";
41
+ connectionId: number;
42
+ reason: string;
43
+ source: "client" | "upstream" | "proxy";
44
+ }
45
+ export interface WsProxyErrorEvent extends WsProxyEventBase {
46
+ type: "error";
47
+ error: Error;
48
+ stage: string;
49
+ sourceEventType?: WsProxyEventType;
50
+ details?: string;
51
+ }
52
+ export type WsProxyEvent = WsProxyStartedEvent | WsProxyStoppedEvent | WsProxyClientConnectedEvent | WsProxyUpstreamConnectedEvent | WsProxyMessageEvent | WsProxyTunnelClosedEvent | WsProxyErrorEvent;
53
+ export interface WsProxyControl {
54
+ restart(opts?: {
55
+ downtimeMs?: number;
56
+ reason?: string;
57
+ }): Promise<void>;
58
+ stop(reason?: string): Promise<void>;
59
+ isRunning(): boolean;
60
+ }
61
+ export type WsProxyEventHandler = (event: WsProxyEvent, control: WsProxyControl) => void | Promise<void>;
62
+ export interface WsProxyOptions {
63
+ host: string;
64
+ port: number;
65
+ onEvent?: WsProxyEventHandler;
66
+ }
67
+ export declare class WsProxy {
68
+ private readonly _listenHost;
69
+ private readonly _requestedPort;
70
+ private _lockedPort;
71
+ private readonly _onEvent?;
72
+ private readonly _control;
73
+ private _httpServer;
74
+ private _wsServer;
75
+ private _running;
76
+ private _nextConnectionId;
77
+ private _lifecycleChain;
78
+ private readonly _tunnels;
79
+ private readonly _eventLog;
80
+ constructor(options: WsProxyOptions);
81
+ static create(options: WsProxyOptions): Promise<WsProxy>;
82
+ start(reason?: string): Promise<void>;
83
+ stop(reason?: string): Promise<void>;
84
+ restart(opts?: {
85
+ downtimeMs?: number;
86
+ reason?: string;
87
+ }): Promise<void>;
88
+ isRunning(): boolean;
89
+ getPort(): number;
90
+ getHost(): string;
91
+ getUrl(): string;
92
+ getEventLog(): WsProxyEvent[];
93
+ private enqueueLifecycle;
94
+ private startInternal;
95
+ private stopInternal;
96
+ private handleClientRequest;
97
+ private connectUpstream;
98
+ private buildUpstreamUrl;
99
+ private handleClientMessage;
100
+ private handleUpstreamMessage;
101
+ private flushPendingFrames;
102
+ private closeTunnel;
103
+ private isConnected;
104
+ private emitMessageEvent;
105
+ private emitError;
106
+ private emitEvent;
107
+ private safeCloseConnection;
108
+ }
109
+ export {};
110
+ //# sourceMappingURL=wsProxy.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"wsProxy.d.ts","sourceRoot":"","sources":["../../../test/helpers/wsProxy.ts"],"names":[],"mappings":"AAWA,KAAK,gBAAgB,GAAG,oBAAoB,GAAG,oBAAoB,CAAC;AACpE,KAAK,gBAAgB,GACf,eAAe,GACf,eAAe,GACf,kBAAkB,GAClB,oBAAoB,GACpB,SAAS,GACT,eAAe,GACf,OAAO,CAAC;AAEd,UAAU,gBAAgB;IACtB,IAAI,EAAE,gBAAgB,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,mBAAoB,SAAQ,gBAAgB;IACzD,IAAI,EAAE,eAAe,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,mBAAoB,SAAQ,gBAAgB;IACzD,IAAI,EAAE,eAAe,CAAC;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,2BAA4B,SAAQ,gBAAgB;IACjE,IAAI,EAAE,kBAAkB,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,6BAA8B,SAAQ,gBAAgB;IACnE,IAAI,EAAE,oBAAoB,CAAC;IAC3B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,mBAAoB,SAAQ,gBAAgB;IACzD,IAAI,EAAE,SAAS,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,gBAAgB,CAAC;IAC5B,QAAQ,EAAE,OAAO,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,GAAG,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,wBAAyB,SAAQ,gBAAgB;IAC9D,IAAI,EAAE,eAAe,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,QAAQ,GAAG,UAAU,GAAG,OAAO,CAAC;CAC3C;AAED,MAAM,WAAW,iBAAkB,SAAQ,gBAAgB;IACvD,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,gBAAgB,CAAC;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,YAAY,GAClB,mBAAmB,GACnB,mBAAmB,GACnB,2BAA2B,GAC3B,6BAA6B,GAC7B,mBAAmB,GACnB,wBAAwB,GACxB,iBAAiB,CAAC;AAExB,MAAM,WAAW,cAAc;IAC3B,OAAO,CAAC,IAAI,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxE,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,SAAS,IAAI,OAAO,CAAC;CACxB;AAED,MAAM,MAAM,mBAAmB,GAAG,CAC9B,KAAK,EAAE,YAAY,EACnB,OAAO,EAAE,cAAc,KACtB,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,MAAM,WAAW,cAAc;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,mBAAmB,CAAC;CACjC;AAkBD,qBAAa,OAAO;IAChB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;IACxC,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAsB;IAChD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;IAE1C,OAAO,CAAC,WAAW,CAA4B;IAC/C,OAAO,CAAC,SAAS,CAAgC;IACjD,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,eAAe,CAAoC;IAC3D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAuC;IAChE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAsB;gBAEpC,OAAO,EAAE,cAAc;WA2BtB,MAAM,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC;IAMxD,KAAK,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMrC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMpC,OAAO,CAAC,IAAI,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAY7E,SAAS,IAAI,OAAO;IAIpB,OAAO,IAAI,MAAM;IAOjB,OAAO,IAAI,MAAM;IAIjB,MAAM,IAAI,MAAM;IAIhB,WAAW,IAAI,YAAY,EAAE;YAIf,gBAAgB;YAKhB,aAAa;YA2Db,YAAY;IA4C1B,OAAO,CAAC,mBAAmB;IAoD3B,OAAO,CAAC,eAAe;IAqDvB,OAAO,CAAC,gBAAgB;IAMxB,OAAO,CAAC,mBAAmB;IA+B3B,OAAO,CAAC,qBAAqB;IAsB7B,OAAO,CAAC,kBAAkB;IAiB1B,OAAO,CAAC,WAAW;IAyBnB,OAAO,CAAC,WAAW;IAInB,OAAO,CAAC,gBAAgB;IAqBxB,OAAO,CAAC,SAAS;IAmBjB,OAAO,CAAC,SAAS;IAiBjB,OAAO,CAAC,mBAAmB;CAiB9B"}