testing ratelimit

libs/core/handlers/src/decorators/rate-limit.ts

@@ -0,0 +1,56 @@
+/**
+ * Rate limit configuration for handlers and operations
+ */
+export interface RateLimitConfig {
+  points: number;         // Number of allowed requests
+  duration: number;       // Time window in seconds
+  blockDuration?: number; // How long to block after limit is hit (seconds)
+}
+
+/**
+ * RateLimit decorator - configures rate limiting for handlers or operations
+ * 
+ * Can be applied to:
+ * - Classes: Sets default rate limit for all operations in the handler
+ * - Methods: Sets specific rate limit for individual operations (overrides handler-level)
+ * 
+ * @param config Rate limit configuration
+ * 
+ * @example
+ * // Handler-level rate limit
+ * @Handler('myHandler')
+ * @RateLimit({ points: 100, duration: 60 })
+ * class MyHandler extends BaseHandler {
+ *   // All operations inherit the 100/minute limit
+ * }
+ * 
+ * @example
+ * // Operation-specific rate limit
+ * @Operation('fetch-data')
+ * @RateLimit({ points: 10, duration: 60, blockDuration: 30 })
+ * async fetchData() {
+ *   // This operation is limited to 10/minute
+ * }
+ */
+export function RateLimit(config: RateLimitConfig) {
+  return function (target: any, propertyKey?: string, descriptor?: PropertyDescriptor): any {
+    if (propertyKey) {
+      // Method decorator - operation-specific rate limit
+      const constructor = target.constructor;
+      
+      // Initialize rate limits array if not exists
+      if (!constructor.__rateLimits) {
+        constructor.__rateLimits = {};
+      }
+      
+      // Store rate limit config for this operation
+      constructor.__rateLimits[propertyKey] = config;
+      
+      return descriptor;
+    } else {
+      // Class decorator - handler-level rate limit
+      (target as any).__handlerRateLimit = config;
+      return target;
+    }
+  };
+}

libs/core/handlers/src/index.ts

@@ -7,6 +7,8 @@ export {
   ScheduledOperation,
   Disabled,
 } from './decorators/decorators';
+export { RateLimit } from './decorators/rate-limit';
+export type { RateLimitConfig } from './decorators/rate-limit';
 export { createJobHandler } from './utils/create-job-handler';
 
 // Re-export commonly used types from @stock-bot/types for convenience