From c5f71789698618dcf5851217b7c40aee6bfbc3ea Mon Sep 17 00:00:00 2001
From: platfowner <seo@comcom.ai>
Date: Thu, 8 Aug 2024 18:06:21 +0900
Subject: [PATCH] Reuse existing ip addresses in event handler

---
 event-handler/event-channel-manager.js | 5 ++---
 json_rpc/event-handler.js              | 6 +++---
 test/unit/event-handler.test.js        | 5 ++---
 3 files changed, 7 insertions(+), 9 deletions(-)

diff --git a/event-handler/event-channel-manager.js b/event-handler/event-channel-manager.js
index a57d1245e..c9b700089 100644
--- a/event-handler/event-channel-manager.js
+++ b/event-handler/event-channel-manager.js
@@ -23,12 +23,11 @@ class EventChannelManager {
     this.heartbeatInterval = null;
   }
 
-  async getNetworkInfo() {
-    const ipAddr = await getIpAddress(NodeConfigs.HOSTING_ENV === HostingEnvs.COMCOM || NodeConfigs.HOSTING_ENV === HostingEnvs.LOCAL);
+  getNetworkInfo() {
+    const ipAddr = (NodeConfigs.HOSTING_ENV === HostingEnvs.COMCOM || NodeConfigs.HOSTING_ENV === HostingEnvs.LOCAL) ? this.node.ipAddrInternal : this.node.ipAddrExternal;
     const eventHandlerUrl = new URL(`ws://${ipAddr}:${NodeConfigs.EVENT_HANDLER_PORT}`);
     return {
       url: eventHandlerUrl.toString(),
-      port: NodeConfigs.EVENT_HANDLER_PORT,
       maxNumEventChannels: NodeConfigs.MAX_NUM_EVENT_CHANNELS,
       numEventChannels: this.getNumEventChannels(),
       maxNumEventFilters: NodeConfigs.MAX_NUM_EVENT_FILTERS,
diff --git a/json_rpc/event-handler.js b/json_rpc/event-handler.js
index 7141ab6a8..9efb47b65 100644
--- a/json_rpc/event-handler.js
+++ b/json_rpc/event-handler.js
@@ -8,12 +8,12 @@ const { JSON_RPC_METHODS } = require('./constants');
 module.exports = function getEventHandlerApis(eventHandler) {
   return {
     // NOTE(cshcomcom): Async function doesn't need a done parameter. (Ref: https://www.npmjs.com/package/jayson#promises)
-    [JSON_RPC_METHODS.NET_GET_EVENT_HANDLER_NETWORK_INFO]: async function(args) {
+    [JSON_RPC_METHODS.NET_GET_EVENT_HANDLER_NETWORK_INFO]: function(args, done) {
       const beginTime = Date.now();
-      const result = await eventHandler.eventChannelManager.getNetworkInfo();
+      const result = eventHandler.eventChannelManager.getNetworkInfo();
       const latency = Date.now() - beginTime;
       trafficStatsManager.addEvent(TrafficEventTypes.JSON_RPC_GET, latency);
-      return JsonRpcUtil.addProtocolVersion({ result });
+      done(null, JsonRpcUtil.addProtocolVersion({ result }));
     },
 
     [JSON_RPC_METHODS.AIN_GET_EVENT_HANDLER_FILTER_INFO]: function(args, done) {
diff --git a/test/unit/event-handler.test.js b/test/unit/event-handler.test.js
index 93400af85..4b8fdadfc 100644
--- a/test/unit/event-handler.test.js
+++ b/test/unit/event-handler.test.js
@@ -372,14 +372,13 @@ describe('EventHandler Test', () => {
     it('getNetworkInfo', async () => {
       const intIp = await getIpAddress(true);
       const intUrl = new URL(`ws://${intIp}:${NodeConfigs.EVENT_HANDLER_PORT}`);
-      const networkInfo = await eventChannelManager.getNetworkInfo();
+      const networkInfo = eventChannelManager.getNetworkInfo();
       assert.deepEqual(networkInfo, {
+        url: intUrl.toString(),
         maxNumEventChannels: NodeConfigs.MAX_NUM_EVENT_CHANNELS,
         numEventChannels: 0,
         maxNumEventFilters: NodeConfigs.MAX_NUM_EVENT_FILTERS,
         numEventFilters: 0,
-        url: intUrl.toString(),
-        port: NodeConfigs.EVENT_HANDLER_PORT,
       });
     });
   });