halt-rate 0.3.0 → 0.4.0

package/dist/index.mjs

@@ -484,7 +484,8 @@ var RateLimiter = class {
       return decision2;
     }
     const store = this.store;
-    let algorithm = this.algorithmCache.get(policy.name);
+    const cacheKey = `${policy.name}|${policy.algorithm}|${policy.limit}|${policy.window}|${policy.burst}`;
+    let algorithm = this.algorithmCache.get(cacheKey);
     if (!algorithm) {
       if (policy.algorithm === "token_bucket" /* TOKEN_BUCKET */) {
         algorithm = new TokenBucket(policy.burst, policy.limit, policy.window);
@@ -498,7 +499,7 @@ var RateLimiter = class {
       } else {
         throw new Error(`Algorithm ${policy.algorithm} not implemented`);
       }
-      this.algorithmCache.set(policy.name, algorithm);
+      this.algorithmCache.set(cacheKey, algorithm);
     }
     const span = this.otelTracer?.startSpan?.("halt.check", { attributes: { policy: policy.name, key } });
     const state = store.get(storageKey);
@@ -1093,6 +1094,70 @@ function getPlanPolicy(planName) {
   return PLAN_TIERS[normalized];
 }
 
+// src/core/registry.ts
+var PolicyRegistry = class {
+  constructor(initial = []) {
+    this.policies = /* @__PURE__ */ new Map();
+    for (const p of initial) this.register(p);
+  }
+  /** Add or replace a policy (keyed by `policy.name`). */
+  register(policy) {
+    this.policies.set(policy.name, policy);
+    return this;
+  }
+  get(name) {
+    return this.policies.get(name);
+  }
+  has(name) {
+    return this.policies.has(name);
+  }
+  /** Mutate fields of an existing policy at runtime (e.g. `{ limit: 500 }`). */
+  update(name, patch) {
+    const existing = this.policies.get(name);
+    if (!existing) throw new Error(`Unknown policy: ${name}`);
+    const updated = { ...existing, ...patch, name: existing.name };
+    if (patch.limit !== void 0 && patch.burst === void 0) {
+      delete updated.burst;
+    }
+    this.policies.set(name, updated);
+    return updated;
+  }
+  remove(name) {
+    return this.policies.delete(name);
+  }
+  list() {
+    return [...this.policies.values()];
+  }
+  /**
+   * Build a resolver for the limiter's `policy` option.
+   * @param selector maps a request to a registered policy name.
+   */
+  resolver(selector) {
+    return (request) => {
+      const name = selector(request);
+      const policy = this.policies.get(name);
+      if (!policy) throw new Error(`Unknown policy: ${name}`);
+      return policy;
+    };
+  }
+};
+function cachedPolicyResolver(loader, options = {}) {
+  const ttlMs = options.ttlMs ?? 5e3;
+  const keyFn = options.key ?? (() => "__default__");
+  const cache = /* @__PURE__ */ new Map();
+  return async (request) => {
+    const key = keyFn(request);
+    const now = Date.now();
+    const hit = cache.get(key);
+    if (hit && hit.expires > now) {
+      return hit.policy;
+    }
+    const policy = await loader(request);
+    cache.set(key, { policy, expires: now + ttlMs });
+    return policy;
+  };
+}
+
 // src/core/quota.ts
 var QuotaPeriod = /* @__PURE__ */ ((QuotaPeriod2) => {
   QuotaPeriod2["HOURLY"] = "hourly";
@@ -1611,6 +1676,6 @@ var OpenTelemetryMetrics = class {
   }
 };
 
-export { Algorithm, CompositeTelemetry, InMemoryStore, KeyStrategy, LoggingTelemetry, MetricsTelemetry, OpenTelemetryMetrics, PENALTY_LENIENT, PENALTY_MODERATE, PENALTY_STRICT, PenaltyManager, QUOTA_ENTERPRISE_MONTHLY, QUOTA_FREE_DAILY, QUOTA_FREE_MONTHLY, QUOTA_PRO_DAILY, QUOTA_PRO_MONTHLY, QuotaManager, QuotaPeriod, RateLimiter, RedisStore, StatsCollector, extractors_exports as extractors, isAtomicStore, normalizePolicy, presets_exports as presets, toHeaders };
+export { Algorithm, CompositeTelemetry, InMemoryStore, KeyStrategy, LoggingTelemetry, MetricsTelemetry, OpenTelemetryMetrics, PENALTY_LENIENT, PENALTY_MODERATE, PENALTY_STRICT, PenaltyManager, PolicyRegistry, QUOTA_ENTERPRISE_MONTHLY, QUOTA_FREE_DAILY, QUOTA_FREE_MONTHLY, QUOTA_PRO_DAILY, QUOTA_PRO_MONTHLY, QuotaManager, QuotaPeriod, RateLimiter, RedisStore, StatsCollector, cachedPolicyResolver, extractors_exports as extractors, isAtomicStore, normalizePolicy, presets_exports as presets, toHeaders };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map