@push.rocks/smartproxy 19.6.2 → 19.6.7

package/ts/proxies/smart-proxy/models/metrics-types.ts

@@ -1,54 +1,106 @@
 /**
- * Interface for proxy statistics and metrics
+ * Interface for throughput sample data
  */
-export interface IProxyStats {
-  /**
-   * Get the current number of active connections
-   */
-  getActiveConnections(): number;
-  
-  /**
-   * Get connection counts grouped by route name
-   */
-  getConnectionsByRoute(): Map<string, number>;
+export interface IThroughputSample {
+  timestamp: number;
+  bytesIn: number;
+  bytesOut: number;
+}
+
+/**
+ * Interface for throughput data
+ */
+export interface IThroughputData {
+  in: number;
+  out: number;
+}
+
+/**
+ * Interface for time-series throughput data
+ */
+export interface IThroughputHistoryPoint {
+  timestamp: number;
+  in: number;
+  out: number;
+}
+
+/**
+ * Main metrics interface with clean, grouped API
+ */
+export interface IMetrics {
+  // Connection metrics
+  connections: {
+    active(): number;
+    total(): number;
+    byRoute(): Map<string, number>;
+    byIP(): Map<string, number>;
+    topIPs(limit?: number): Array<{ ip: string; count: number }>;
+  };
   
-  /**
-   * Get connection counts grouped by IP address
-   */
-  getConnectionsByIP(): Map<string, number>;
+  // Throughput metrics (bytes per second)
+  throughput: {
+    instant(): IThroughputData;      // Last 1 second
+    recent(): IThroughputData;       // Last 10 seconds  
+    average(): IThroughputData;      // Last 60 seconds
+    custom(seconds: number): IThroughputData;
+    history(seconds: number): Array<IThroughputHistoryPoint>;
+    byRoute(windowSeconds?: number): Map<string, IThroughputData>;
+    byIP(windowSeconds?: number): Map<string, IThroughputData>;
+  };
   
-  /**
-   * Get the total number of connections since proxy start
-   */
-  getTotalConnections(): number;
+  // Request metrics
+  requests: {
+    perSecond(): number;
+    perMinute(): number;
+    total(): number;
+  };
   
-  /**
-   * Get the current requests per second rate
-   */
-  getRequestsPerSecond(): number;
+  // Cumulative totals
+  totals: {
+    bytesIn(): number;
+    bytesOut(): number;
+    connections(): number;
+  };
   
-  /**
-   * Get total throughput (bytes transferred)
-   */
-  getThroughput(): { bytesIn: number; bytesOut: number };
+  // Performance metrics
+  percentiles: {
+    connectionDuration(): { p50: number; p95: number; p99: number };
+    bytesTransferred(): { 
+      in: { p50: number; p95: number; p99: number };
+      out: { p50: number; p95: number; p99: number };
+    };
+  };
 }
 
 /**
- * Extended interface for additional metrics helpers
+ * Configuration for metrics collection
  */
-export interface IProxyStatsExtended extends IProxyStats {
-  /**
-   * Get throughput rate (bytes per second) for last minute
-   */
-  getThroughputRate(): { bytesInPerSec: number; bytesOutPerSec: number };
+export interface IMetricsConfig {
+  enabled: boolean;
+  
+  // Sampling configuration
+  sampleIntervalMs: number;        // Default: 1000 (1 second)
+  retentionSeconds: number;        // Default: 3600 (1 hour)
   
-  /**
-   * Get top IPs by connection count
-   */
-  getTopIPs(limit?: number): Array<{ ip: string; connections: number }>;
+  // Performance tuning
+  enableDetailedTracking: boolean; // Per-connection byte history
+  enablePercentiles: boolean;      // Calculate percentiles
+  cacheResultsMs: number;          // Cache expensive calculations
   
-  /**
-   * Check if an IP has reached the connection limit
-   */
-  isIPBlocked(ip: string, maxConnectionsPerIP: number): boolean;
+  // Export configuration
+  prometheusEnabled: boolean;
+  prometheusPath: string;          // Default: /metrics
+  prometheusPrefix: string;        // Default: smartproxy_
+}
+
+/**
+ * Internal interface for connection byte tracking
+ */
+export interface IByteTracker {
+  connectionId: string;
+  routeName: string;
+  remoteIP: string;
+  bytesIn: number;
+  bytesOut: number;
+  lastUpdate: number;
 }

package/ts/proxies/smart-proxy/nftables-manager.ts

@@ -10,7 +10,7 @@ import type {
   TPortRange,
   INfTablesOptions
 } from './models/route-types.js';
-import type { ISmartProxyOptions } from './models/interfaces.js';
+import type { SmartProxy } from './smart-proxy.js';
 
 /**
  * Manages NFTables rules based on SmartProxy route configurations
@@ -25,9 +25,9 @@ export class NFTablesManager {
   /**
    * Creates a new NFTablesManager
    * 
-   * @param options The SmartProxy options
+   * @param smartProxy The SmartProxy instance
    */
-  constructor(private options: ISmartProxyOptions) {}
+  constructor(private smartProxy: SmartProxy) {}
   
   /**
    * Provision NFTables rules for a route
@@ -166,10 +166,10 @@ export class NFTablesManager {
       protocol: action.nftables?.protocol || 'tcp',
       preserveSourceIP: action.nftables?.preserveSourceIP !== undefined ? 
                         action.nftables.preserveSourceIP : 
-                        this.options.preserveSourceIP,
+                        this.smartProxy.settings.preserveSourceIP,
       useIPSets: action.nftables?.useIPSets !== false,
       useAdvancedNAT: action.nftables?.useAdvancedNAT,
-      enableLogging: this.options.enableDetailedLogging,
+      enableLogging: this.smartProxy.settings.enableDetailedLogging,
       deleteOnExit: true,
       tableName: action.nftables?.tableName || 'smartproxy'
     };

package/ts/proxies/smart-proxy/port-manager.ts

@@ -1,8 +1,7 @@
 import * as plugins from '../../plugins.js';
-import type { ISmartProxyOptions } from './models/interfaces.js';
-import { RouteConnectionHandler } from './route-connection-handler.js';
 import { logger } from '../../core/utils/logger.js';
 import { cleanupSocket } from '../../core/utils/socket-utils.js';
+import type { SmartProxy } from './smart-proxy.js';
 
 /**
  * PortManager handles the dynamic creation and removal of port listeners
@@ -16,8 +15,6 @@ import { cleanupSocket } from '../../core/utils/socket-utils.js';
  */
 export class PortManager {
   private servers: Map<number, plugins.net.Server> = new Map();
-  private settings: ISmartProxyOptions;
-  private routeConnectionHandler: RouteConnectionHandler;
   private isShuttingDown: boolean = false;
   // Track how many routes are using each port
   private portRefCounts: Map<number, number> = new Map();
@@ -25,16 +22,11 @@ export class PortManager {
   /**
    * Create a new PortManager
    * 
-   * @param settings The SmartProxy settings
-   * @param routeConnectionHandler The handler for new connections
+   * @param smartProxy The SmartProxy instance
    */
   constructor(
-    settings: ISmartProxyOptions,
-    routeConnectionHandler: RouteConnectionHandler
-  ) {
-    this.settings = settings;
-    this.routeConnectionHandler = routeConnectionHandler;
-  }
+    private smartProxy: SmartProxy
+  ) {}
 
   /**
    * Start listening on a specific port
@@ -70,7 +62,7 @@ export class PortManager {
       }
       
       // Delegate to route connection handler
-      this.routeConnectionHandler.handleConnection(socket);
+      this.smartProxy.routeConnectionHandler.handleConnection(socket);
     }).on('error', (err: Error) => {
       try {
         logger.log('error', `Server Error on port ${port}: ${err.message}`, {
@@ -86,7 +78,7 @@ export class PortManager {
     // Start listening on the port
     return new Promise<void>((resolve, reject) => {
       server.listen(port, () => {
-        const isHttpProxyPort = this.settings.useHttpProxy?.includes(port);
+        const isHttpProxyPort = this.smartProxy.settings.useHttpProxy?.includes(port);
         try {
           logger.log('info', `SmartProxy -> OK: Now listening on port ${port}${
             isHttpProxyPort ? ' (HttpProxy forwarding enabled)' : ''