Use DO fetch

glotcode · Jul 21, 2024 · 9787d6e · 9787d6e
1 parent f60e940
commit 9787d6e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/glot_cloudflare/functions/api/run.ts b/glot_cloudflare/functions/api/run.ts
@@ -19,7 +19,8 @@ export const onRequestPost: PagesFunction<Env & StringRecord> = async (context)
 
  const id = context.env.RATE_LIMITER.idFromName(ip);
  const stub = context.env.RATE_LIMITER.get(id);
- const stats = stub.increment({ maxRequests: 10, periodDuration: 60 * 1000 });
+ const response = await stub.fetch(context.request);
+ const stats = await response.text();
 
  console.log(stats)
 

diff --git a/glot_cloudflare_rate_limiter/src/rate_limiter.ts b/glot_cloudflare_rate_limiter/src/rate_limiter.ts
@@ -16,6 +16,11 @@ export class RateLimiter extends DurableObject {
  private startTimestamp: number = 0;
  private requestCount: number = 0;
 
+ async fetch(request: Request): Promise<Response> {
+ const stats = await this.increment({ maxRequests: 10, periodDuration: 60 * 1000 });
+ return new Response(JSON.stringify(stats), { status: 200 });
+ }
+
  async increment(config: RateLimitConfig): Promise<RateLimitStats> {
  const now = Date.now();
  const elapsedTime = now - this.startTimestamp;