@geekmidas/rate-limit 0.1.0 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@geekmidas/rate-limit",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "private": false,
   "type": "module",
   "exports": {
@@ -12,18 +12,34 @@
   },
   "repository": {
     "type": "git",
-    "url": "https://github.com/geekmidas/toolbox.git"
+    "url": "https://github.com/geekmidas/toolbox"
   },
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
   },
-  "dependencies": {
-    "@geekmidas/logger": "0.0.1",
-    "@geekmidas/services": "0.0.1",
-    "@geekmidas/cache": "0.0.7"
+  "dependencies": {},
+  "peerDependencies": {
+    "@geekmidas/cache": "^0.1.0",
+    "@geekmidas/logger": "^0.3.0",
+    "@geekmidas/services": "^0.1.0"
+  },
+  "peerDependenciesMeta": {
+    "@geekmidas/cache": {
+      "optional": true
+    },
+    "@geekmidas/logger": {
+      "optional": true
+    },
+    "@geekmidas/services": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@geekmidas/cache": "^0.1.0",
+    "@geekmidas/logger": "^0.3.0",
+    "@geekmidas/services": "^0.1.0"
   },
-  "peerDependencies": {},
   "scripts": {
     "ts": "tsc --noEmit --skipLibCheck src/**/*.ts"
   }

package/src/__benchmarks__/rateLimit.bench.ts

@@ -0,0 +1,110 @@
+import { InMemoryCache } from '@geekmidas/cache/memory';
+import { bench, describe } from 'vitest';
+import {
+  type RateLimitConfig,
+  type RateLimitContext,
+  type RateLimitData,
+  checkRateLimit,
+} from '../index';
+
+// Helper to create a mock context
+function createContext(ip: string): RateLimitContext {
+  return {
+    header: (key: string) => (key === 'x-forwarded-for' ? ip : undefined),
+    services: {},
+    logger: {
+      info: () => {},
+      error: () => {},
+      debug: () => {},
+      warn: () => {},
+    } as any,
+    session: {},
+    path: '/test',
+    method: 'GET',
+  };
+}
+
+describe('Rate Limiting', () => {
+  const cache = new InMemoryCache<RateLimitData>();
+  const config: RateLimitConfig<RateLimitData> = {
+    limit: 100,
+    windowMs: 60000,
+    cache,
+  };
+
+  bench('checkRateLimit - single IP', async () => {
+    const ctx = createContext('127.0.0.1');
+    await checkRateLimit(config, ctx);
+  });
+
+  bench('checkRateLimit - varying IPs', async () => {
+    const ip = `192.168.1.${Math.floor(Math.random() * 255)}`;
+    const ctx = createContext(ip);
+    await checkRateLimit(config, ctx);
+  });
+});
+
+describe('Rate Limiting - High Volume', () => {
+  bench('100 requests same IP', async () => {
+    const cache = new InMemoryCache<RateLimitData>();
+    const config: RateLimitConfig<RateLimitData> = {
+      limit: 1000,
+      windowMs: 60000,
+      cache,
+    };
+    const ctx = createContext('10.0.0.1');
+
+    for (let i = 0; i < 100; i++) {
+      await checkRateLimit(config, ctx);
+    }
+  });
+
+  bench('100 requests different IPs', async () => {
+    const cache = new InMemoryCache<RateLimitData>();
+    const config: RateLimitConfig<RateLimitData> = {
+      limit: 100,
+      windowMs: 60000,
+      cache,
+    };
+
+    for (let i = 0; i < 100; i++) {
+      const ctx = createContext(`10.0.0.${i}`);
+      await checkRateLimit(config, ctx);
+    }
+  });
+});
+
+describe('Rate Limiting - Window Sizes', () => {
+  bench('1 second window', async () => {
+    const cache = new InMemoryCache<RateLimitData>();
+    const config: RateLimitConfig<RateLimitData> = {
+      limit: 10,
+      windowMs: 1000,
+      cache,
+    };
+    const ctx = createContext('127.0.0.1');
+    await checkRateLimit(config, ctx);
+  });
+
+  bench('1 minute window', async () => {
+    const cache = new InMemoryCache<RateLimitData>();
+    const config: RateLimitConfig<RateLimitData> = {
+      limit: 100,
+      windowMs: 60000,
+      cache,
+    };
+    const ctx = createContext('127.0.0.1');
+    await checkRateLimit(config, ctx);
+  });
+
+  bench('1 hour window', async () => {
+    const cache = new InMemoryCache<RateLimitData>();
+    const config: RateLimitConfig<RateLimitData> = {
+      limit: 1000,
+      windowMs: 3600000,
+      cache,
+    };
+    const ctx = createContext('127.0.0.1');
+    await checkRateLimit(config, ctx);
+  });
+});