Trying to use the @upstash/ratelimit package with ioredis…

Author: matt-aitken

apps/webapp/app/services/apiRateLimit.server.ts

@@ -0,0 +1,37 @@
+import Redis, { RedisOptions } from "ioredis";
+import { Ratelimit } from "@upstash/ratelimit";
+import { LoaderFunction, LoaderFunctionArgs } from "@remix-run/server-runtime";
+import { env } from "~/env.server";
+
+export function createRedisRateLimitClient(): ConstructorParameters<typeof Ratelimit>[0]["redis"] {
+  const redis = new Redis({
+    port: env.REDIS_PORT,
+    host: env.REDIS_HOST,
+    username: env.REDIS_USERNAME,
+    password: env.REDIS_PASSWORD,
+    enableAutoPipelining: true,
+    ...(env.REDIS_TLS_DISABLED === "true" ? {} : { tls: {} }),
+  });
+
+  return {
+    sadd: async <TData>(key: string, ...members: TData[]): Promise<number> => {
+      return redis.sadd(key, members as (string | Buffer | number)[]);
+    },
+    eval: <TArgs extends unknown[], TData = unknown>(
+      ...args: [script: string, keys: string[], args: TArgs]
+    ): Promise<TData> => {
+      return redis.eval(...args) as TData;
+    },
+  };
+}
+
+const ratelimitter = new Ratelimit({
+  redis: createRedisRateLimitClient(),
+  limiter: Ratelimit.cachedFixedWindow(10, "10s"),
+  ephemeralCache: new Map(),
+  analytics: true,
+});
+
+export const rateLimit = {
+  loader: (args: LoaderFunctionArgs): ReturnType<LoaderFunction> => {},
+};

apps/webapp/package.json

@@ -96,6 +96,7 @@
     "@trigger.dev/yalt": "workspace:*",
     "@types/pg": "8.6.6",
     "@uiw/react-codemirror": "^4.19.5",
+    "@upstash/ratelimit": "^1.0.1",
     "@whatwg-node/fetch": "^0.9.14",
     "class-variance-authority": "^0.5.2",
     "clsx": "^1.2.1",
@@ -136,6 +137,7 @@
     "react-stately": "^3.29.1",
     "react-use": "^17.4.0",
     "recharts": "^2.8.0",
+    "redis": "^4.6.13",
     "remix-auth": "^3.6.0",
     "remix-auth-email-link": "2.0.2",
     "remix-auth-github": "^1.6.0",