pg-ratelimit 0.1.0 → 0.2.0

package/dist/index.cjs

@@ -276,6 +276,10 @@ var Ratelimit = class {
   durable;
   synchronousCommit;
   cleanupProbability;
+  inMemoryBlock;
+  maxBlockedKeys;
+  limitValue;
+  blockedKeys;
   // Pre-parsed durations (only the relevant one is used per algorithm)
   windowMs;
   intervalMs;
@@ -303,6 +307,10 @@ var Ratelimit = class {
       this.windowMs = 0;
       this.intervalMs = toMs(this.algorithm.interval);
     }
+    this.inMemoryBlock = "inMemoryBlock" in config && config.inMemoryBlock === true;
+    this.maxBlockedKeys = this.inMemoryBlock && "maxBlockedKeys" in config ? config.maxBlockedKeys ?? 1e4 : 1e4;
+    this.blockedKeys = /* @__PURE__ */ new Map();
+    this.limitValue = this.algorithm.type === "tokenBucket" ? this.algorithm.maxTokens : this.algorithm.tokens;
   }
   static fixedWindow(tokens, window) {
     return { type: "fixedWindow", tokens, window };
@@ -316,6 +324,16 @@ var Ratelimit = class {
   async limit(key, opts) {
     const rate = opts?.rate ?? 1;
     const now = this.clock();
+    const nowMs = now.getTime();
+    if (this.inMemoryBlock && rate > 0) {
+      const cachedReset = this.blockedKeys.get(key);
+      if (cachedReset !== void 0) {
+        if (cachedReset > nowMs) {
+          return { success: false, limit: this.limitValue, remaining: 0, reset: cachedReset };
+        }
+        this.blockedKeys.delete(key);
+      }
+    }
     await ensureTables(this.pool);
     const ctx = {
       pool: this.pool,
@@ -344,6 +362,16 @@ var Ratelimit = class {
         );
         break;
     }
+    if (this.inMemoryBlock) {
+      if (rate < 0) {
+        this.blockedKeys.delete(key);
+      } else if (!result.success) {
+        this.blockedKeys.set(key, result.reset);
+        if (this.blockedKeys.size > this.maxBlockedKeys) {
+          this.sweepExpired(nowMs);
+        }
+      }
+    }
     if (Math.random() < this.cleanupProbability) {
       void this.pool.query(`DELETE FROM ${this.table} WHERE prefix = $1 AND expires_at < $2`, [
         this.prefix,
@@ -473,6 +501,14 @@ var Ratelimit = class {
       console.debug("pg-ratelimit resetUsedTokens:", sql, [this.prefix, key]);
     }
     await this.pool.query(sql, [this.prefix, key]);
+    this.blockedKeys.delete(key);
+  }
+  sweepExpired(nowMs) {
+    for (const [k, reset] of this.blockedKeys) {
+      if (reset <= nowMs) {
+        this.blockedKeys.delete(k);
+      }
+    }
   }
 };
 // Annotate the CommonJS export names for ESM import in node:

package/dist/index.d.cts

@@ -23,6 +23,18 @@ interface LimitResult {
     reset: number;
 }
 type Clock = () => Date;
+type InMemoryBlockConfig = {
+    inMemoryBlock?: false;
+} | {
+    inMemoryBlock: true;
+    maxBlockedKeys?: number;
+};
+type DurableConfig = {
+    durable?: false;
+} | {
+    durable: true;
+    synchronousCommit?: boolean;
+};
 type RatelimitConfig = {
     pool: Pool;
     limiter: Algorithm;
@@ -30,12 +42,7 @@ type RatelimitConfig = {
     debug?: boolean;
     clock?: Clock;
     cleanupProbability?: number;
-} & ({
-    durable?: false;
-} | {
-    durable: true;
-    synchronousCommit?: boolean;
-});
+} & InMemoryBlockConfig & DurableConfig;
 
 declare class Ratelimit {
     private readonly pool;
@@ -47,6 +54,10 @@ declare class Ratelimit {
     private readonly durable;
     private readonly synchronousCommit;
     private readonly cleanupProbability;
+    private readonly inMemoryBlock;
+    private readonly maxBlockedKeys;
+    private readonly limitValue;
+    private readonly blockedKeys;
     private readonly windowMs;
     private readonly intervalMs;
     constructor(config: RatelimitConfig);
@@ -64,6 +75,7 @@ declare class Ratelimit {
         reset: number;
     }>;
     resetUsedTokens(key: string): Promise<void>;
+    private sweepExpired;
 }
 
 declare const TABLE_SQL: string;

package/dist/index.d.ts

@@ -23,6 +23,18 @@ interface LimitResult {
     reset: number;
 }
 type Clock = () => Date;
+type InMemoryBlockConfig = {
+    inMemoryBlock?: false;
+} | {
+    inMemoryBlock: true;
+    maxBlockedKeys?: number;
+};
+type DurableConfig = {
+    durable?: false;
+} | {
+    durable: true;
+    synchronousCommit?: boolean;
+};
 type RatelimitConfig = {
     pool: Pool;
     limiter: Algorithm;
@@ -30,12 +42,7 @@ type RatelimitConfig = {
     debug?: boolean;
     clock?: Clock;
     cleanupProbability?: number;
-} & ({
-    durable?: false;
-} | {
-    durable: true;
-    synchronousCommit?: boolean;
-});
+} & InMemoryBlockConfig & DurableConfig;
 
 declare class Ratelimit {
     private readonly pool;
@@ -47,6 +54,10 @@ declare class Ratelimit {
     private readonly durable;
     private readonly synchronousCommit;
     private readonly cleanupProbability;
+    private readonly inMemoryBlock;
+    private readonly maxBlockedKeys;
+    private readonly limitValue;
+    private readonly blockedKeys;
     private readonly windowMs;
     private readonly intervalMs;
     constructor(config: RatelimitConfig);
@@ -64,6 +75,7 @@ declare class Ratelimit {
         reset: number;
     }>;
     resetUsedTokens(key: string): Promise<void>;
+    private sweepExpired;
 }
 
 declare const TABLE_SQL: string;

package/dist/index.js

@@ -249,6 +249,10 @@ var Ratelimit = class {
   durable;
   synchronousCommit;
   cleanupProbability;
+  inMemoryBlock;
+  maxBlockedKeys;
+  limitValue;
+  blockedKeys;
   // Pre-parsed durations (only the relevant one is used per algorithm)
   windowMs;
   intervalMs;
@@ -276,6 +280,10 @@ var Ratelimit = class {
       this.windowMs = 0;
       this.intervalMs = toMs(this.algorithm.interval);
     }
+    this.inMemoryBlock = "inMemoryBlock" in config && config.inMemoryBlock === true;
+    this.maxBlockedKeys = this.inMemoryBlock && "maxBlockedKeys" in config ? config.maxBlockedKeys ?? 1e4 : 1e4;
+    this.blockedKeys = /* @__PURE__ */ new Map();
+    this.limitValue = this.algorithm.type === "tokenBucket" ? this.algorithm.maxTokens : this.algorithm.tokens;
   }
   static fixedWindow(tokens, window) {
     return { type: "fixedWindow", tokens, window };
@@ -289,6 +297,16 @@ var Ratelimit = class {
   async limit(key, opts) {
     const rate = opts?.rate ?? 1;
     const now = this.clock();
+    const nowMs = now.getTime();
+    if (this.inMemoryBlock && rate > 0) {
+      const cachedReset = this.blockedKeys.get(key);
+      if (cachedReset !== void 0) {
+        if (cachedReset > nowMs) {
+          return { success: false, limit: this.limitValue, remaining: 0, reset: cachedReset };
+        }
+        this.blockedKeys.delete(key);
+      }
+    }
     await ensureTables(this.pool);
     const ctx = {
       pool: this.pool,
@@ -317,6 +335,16 @@ var Ratelimit = class {
         );
         break;
     }
+    if (this.inMemoryBlock) {
+      if (rate < 0) {
+        this.blockedKeys.delete(key);
+      } else if (!result.success) {
+        this.blockedKeys.set(key, result.reset);
+        if (this.blockedKeys.size > this.maxBlockedKeys) {
+          this.sweepExpired(nowMs);
+        }
+      }
+    }
     if (Math.random() < this.cleanupProbability) {
       void this.pool.query(`DELETE FROM ${this.table} WHERE prefix = $1 AND expires_at < $2`, [
         this.prefix,
@@ -446,6 +474,14 @@ var Ratelimit = class {
       console.debug("pg-ratelimit resetUsedTokens:", sql, [this.prefix, key]);
     }
     await this.pool.query(sql, [this.prefix, key]);
+    this.blockedKeys.delete(key);
+  }
+  sweepExpired(nowMs) {
+    for (const [k, reset] of this.blockedKeys) {
+      if (reset <= nowMs) {
+        this.blockedKeys.delete(k);
+      }
+    }
   }
 };
 export {

package/package.json

@@ -1,28 +1,25 @@
 {
   "name": "pg-ratelimit",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "PostgreSQL-backed rate limiting for Node.js",
-  "license": "MIT",
-  "author": "Max Malm",
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/benjick/pg-ratelimit.git",
-    "directory": "packages/pg-ratelimit"
-  },
-  "homepage": "https://benjick.js.org/pg-ratelimit",
-  "bugs": {
-    "url": "https://github.com/benjick/pg-ratelimit/issues"
-  },
   "keywords": [
-    "rate-limit",
-    "ratelimit",
     "postgres",
     "postgresql",
+    "rate-limit",
     "rate-limiting",
+    "ratelimit",
     "throttle"
   ],
-  "engines": {
-    "node": ">=18.0.0"
+  "homepage": "https://benjick.js.org/pg-ratelimit",
+  "bugs": {
+    "url": "https://github.com/benjick/pg-ratelimit/issues"
+  },
+  "license": "MIT",
+  "author": "Max Malm",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/benjick/pg-ratelimit.git",
+    "directory": "packages/pg-ratelimit"
   },
   "files": [
     "dist"
@@ -43,6 +40,10 @@
       }
     }
   },
+  "scripts": {
+    "build": "tsup",
+    "test": "vitest run"
+  },
   "devDependencies": {
     "@testcontainers/postgresql": "^10.0.0",
     "@types/pg": "^8.0.0",
@@ -54,8 +55,7 @@
   "peerDependencies": {
     "pg": "^8.0.0"
   },
-  "scripts": {
-    "build": "tsup",
-    "test": "vitest run"
+  "engines": {
+    "node": ">=18.0.0"
   }
-}
+}

package/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2026 Max Malm
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.