kukuy 1.6.0 → 1.9.0

package/src/utils/HealthChecker.js

@@ -3,8 +3,9 @@ const https = require('https');
 const url = require('url');
 
 class HealthChecker {
-  constructor() {
+  constructor(balancer = null) {
     this.timeout = 5000; // 5 segundos de timeout
+    this.balancer = balancer; // Referencia al balanceador para acceder a los hooks
   }
 
   async checkServerHealth(server) {
@@ -22,6 +23,9 @@ class HealthChecker {
         }
       };
 
+      // Emitir hook antes de realizar el check
+      await this.executeHook('onHealthCheckStart', { server, checkTime: new Date() });
+
       return new Promise((resolve) => {
         const request = server.protocol === 'https:'
           ? https.request(options)
@@ -33,25 +37,76 @@ class HealthChecker {
 
           // Considerar saludable si obtenemos una respuesta exitosa
           const isHealthy = res.statusCode >= 200 && res.statusCode < 400;
-          resolve(isHealthy);
+
+          // Emitir hook después de completar el check
+          this.executeHook('onHealthCheckComplete', {
+            server,
+            isHealthy,
+            statusCode: res.statusCode,
+            checkTime: new Date()
+          }).then(() => {
+            // Resolver la promesa después de emitir el hook
+            resolve(isHealthy);
+          }).catch(() => {
+            // En caso de error al emitir el hook, resolver igualmente
+            resolve(isHealthy);
+          });
         });
 
         request.on('error', (err) => {
-          // No imprimir errores de health check en consola para evitar spam
-          resolve(false);
+          // Emitir hook cuando ocurre un error en el check
+          this.executeHook('onHealthCheckError', {
+            server,
+            error: err,
+            checkTime: new Date()
+          }).then(() => {
+            // No imprimir errores de health check en consola para evitar spam
+            resolve(false);
+          }).catch(() => {
+            resolve(false);
+          });
         });
 
         request.on('timeout', () => {
-          // No imprimir errores de timeout de health check en consola
-          resolve(false);
+          // Emitir hook cuando ocurre un timeout en el check
+          this.executeHook('onHealthCheckTimeout', {
+            server,
+            checkTime: new Date()
+          }).then(() => {
+            // No imprimir errores de timeout de health check en consola
+            resolve(false);
+          }).catch(() => {
+            resolve(false);
+          });
         });
 
         request.end();
       });
     } catch (error) {
+      // Emitir hook cuando ocurre una excepción
+      await this.executeHook('onHealthCheckException', {
+        server,
+        error,
+        checkTime: new Date()
+      });
       return false;
     }
   }
+
+  /**
+   * Execute a hook if the balancer reference is available
+   * @param {string} hookName - Name of the hook to execute
+   * @param {Object} data - Data to pass to the hook
+   */
+  async executeHook(hookName, data) {
+    if (this.balancer && this.balancer.hookManager) {
+      try {
+        await this.balancer.hookManager.executeHooks(hookName, data);
+      } catch (error) {
+        console.error(`Error executing ${hookName} hook:`, error);
+      }
+    }
+  }
 }
 
 module.exports = { HealthChecker };