@clonegod/ttd-core 3.1.43 → 3.1.44

package/dist/analyze/index.js

@@ -71,10 +71,29 @@ function getWsClient() {
         return null;
     if (!_wsClient) {
         const url = `ws://${host}:${service_ports_1.SERVICE_PORT.TRADE_ANALYZE_HTTP}/trade/analyze/ws/ingest`;
-        _wsClient = new ws_client_1.AnalyzeWsClient(url);
+        _wsClient = new ws_client_1.AnalyzeWsClient(url, () => buildHello());
     }
     return _wsClient;
 }
+function buildHello() {
+    let env = {};
+    try {
+        env = (0, core_env_1.getCoreEnv)() || {};
+    }
+    catch (_a) { }
+    const hasQuotePool = !!(env.quote_pool_address || env.quote_pool_name);
+    const hasGroup = !!env.group_id;
+    const kind = hasQuotePool ? 'QUOTE' : hasGroup ? 'TRADE' : 'OTHER';
+    return {
+        kind,
+        app_id: process.env.name || env.app_name || '',
+        pair: env.pair || '',
+        dex_id: env.dex_id || '',
+        group_id: env.group_id || '',
+        pool_name: env.quote_pool_name || '',
+        fee_rate: env.quote_pool_fee_rate ? Number(env.quote_pool_fee_rate) : undefined,
+    };
+}
 const WS_TYPES = new Set([
     'PreTradeEvent',
     'PriceMessageType',

package/dist/analyze/ws_client.d.ts

@@ -1,5 +1,15 @@
+export interface AnalyzeWsHello {
+    kind: 'QUOTE' | 'TRADE' | 'OTHER';
+    app_id?: string;
+    pair?: string;
+    dex_id?: string;
+    group_id?: string;
+    pool_name?: string;
+    fee_rate?: number;
+}
 export declare class AnalyzeWsClient {
     private readonly url;
+    private readonly helloFn?;
     private ws;
     private reconnectTimer;
     private connected;
@@ -7,7 +17,7 @@ export declare class AnalyzeWsClient {
     private readonly MAX_QUEUE;
     private droppedCount;
     private lastDroppedLogAt;
-    constructor(url: string);
+    constructor(url: string, helloFn?: () => AnalyzeWsHello);
     private connect;
     private scheduleReconnect;
     private flushQueue;

package/dist/analyze/ws_client.js

@@ -7,8 +7,9 @@ exports.AnalyzeWsClient = void 0;
 const ws_1 = __importDefault(require("ws"));
 const index_1 = require("../index");
 class AnalyzeWsClient {
-    constructor(url) {
+    constructor(url, helloFn) {
         this.url = url;
+        this.helloFn = helloFn;
         this.ws = null;
         this.reconnectTimer = null;
         this.connected = false;
@@ -24,6 +25,15 @@ class AnalyzeWsClient {
             this.ws.on('open', () => {
                 this.connected = true;
                 (0, index_1.log_info)(`[AnalyzeWs] connected: ${this.url}`);
+                if (this.helloFn) {
+                    try {
+                        const hello = this.helloFn();
+                        this.ws.send(JSON.stringify({ type: 'HELLO', data: hello }));
+                    }
+                    catch (e) {
+                        (0, index_1.log_debug)(`[AnalyzeWs] hello send failed: ${e.message}`);
+                    }
+                }
                 this.flushQueue();
             });
             this.ws.on('close', () => {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@clonegod/ttd-core",
-    "version": "3.1.43",
+    "version": "3.1.44",
     "description": "Common types and utilities for trading systems - use `npm run push` to publish",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",