trafficlunar/tomodachi-share
1
0
Fork 0
mirror of https://github.com/trafficlunar/tomodachi-share.git synced 2026-06-17 04:54:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: replace ioredis with node-redis

Browse source
This commit is contained in:
trafficlunar 2026-01-31 16:57:04 +00:00
parent 4405aa50c3
commit bdab417708
3 changed files with 120 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/lib/rate-limit.ts
View file

@ -1,8 +1,12 @@
import { NextRequest, NextResponse } from "next/server";
import { Redis } from "ioredis";
import { createClient } from "redis";
import { auth } from "./auth";
const redis = new Redis(process.env.REDIS_URL!);
const client = await createClient({
url: process.env.REDIS_URL,
})
.on("error", (err) => console.error("Redis client error", err))
.connect();
const windowSize = 60;
interface RateLimitData {
@ -41,18 +45,13 @@ export class RateLimit {
const expireAt = currentWindow + windowSize;
try {
// Create a Redis transaction
const tx = redis.multi();
tx.incr(key);
tx.expireat(key, expireAt);
// Execute transaction and get the count
const results = await tx.exec();
if (!results) {
// Execute a Redis transaction and get the count
const [result] = await client.multi().incr(key).expireAt(key, expireAt).exec();
if (!result) {
throw new Error("Redis transaction failed");
}
const count = results[0][1] as number;
const count = result as unknown as number;
const success = count <= this.maxRequests;
const remaining = Math.max(0, this.maxRequests - count);