अपस्टैश और नेक्स्ट.जेएस के साथ अपने SaaS के लिए एक विश्वसनीय कोटा प्रणाली बनाएं

डेटाबेस स्कीमा को परिभाषित करना

model User {
 id Int @id @default(autoincrement())
 planId String @default("FREE")
 email String @unique
 quota Int @default(0)
 spreadsheets Spreadsheet[]
}
model Spreadsheet {
 id Int @id @default(autoincrement())
 userId Int
 user User @relation(fields: [userId], references: [id])
 content String
}

• द planId फ़ील्ड वह योजना है जिसकी उपयोगकर्ता ने सदस्यता ली है।
• द quota फ़ील्ड चालू माह में उपयोगकर्ता द्वारा किए गए एपीआई अनुरोधों की संख्या है।
• द Spreadsheet मॉडल एक स्प्रेडशीट का प्रतिनिधित्व करता है, जो User से जुड़ा होता है .
• ए User एकाधिक Spreadsheet हो सकते हैं इससे जुड़ा हुआ.

उपयोगकर्ता का कोटा बढ़ाया जा रहा है

// pages/api/spreadsheets/[id]/index.ts
import type { NextApiRequest, NextApiResponse } from "next";
 
const MAX_FREE_TIER_QUOTA = 1000;
 
export default async function handler(
 req: NextApiRequest,
 res: NextApiResponse,
) {
 // Get the ID of the spreadsheet from the URL.
 const { id } = req.params;
 
 // Get the spreadsheet and its associated user from the database.
 const spreadsheet = await prisma.spreadsheet.findUnique({
 where: { id: Number(id) },
 include: { user: true },
 });
 
 // Verify that the spreadsheet exists.
 if (!spreadsheet) {
 return res.status(404).json({ error: "Spreadsheet not found" });
 }
 
 // Verify that the user is not over the quota.
 if (spreadsheet.user.quota >= MAX_FREE_TIER_QUOTA) {
 return res.status(429).json({ error: "Quota exceeded" });
 }
 
 // Increase the user's quota by 1.
 await prisma.user.update({
 where: { id: spreadsheet.user.id },
 data: { quota: { increment: 1 } },
 });
 
 // Return the spreadsheet's content.
 return res.json({ content: spreadsheet.content });
}

• आपका डेटाबेस धीमा हो सकता है, इससे आपकी प्रतिक्रिया का समय धीमा हो जाएगा।
• आपका डेटाबेस आपके सर्वर से दूर हो सकता है, इससे आपका प्रतिक्रिया समय फिर से धीमा हो जाएगा।
• इस कोड को एज संगत बनाना कठिन है , आपको दुनिया भर में कई डेटाबेस प्रतिकृतियां बनाने की आवश्यकता है (ऐसा करना आसान काम नहीं है!) .

उपयोगकर्ता का कोटा बढ़ाने के लिए अपस्टैश का उपयोग करना

// pages/api/spreadsheets/[id]/index.ts
import type { NextApiRequest, NextApiResponse } from "next";
 
import { Redis } from "@upstash/redis";
 
const MAX_FREE_TIER_QUOTA = 1000;
 
// Generate an Upstash Redis instance using environment variables.
// Make sure those are defined in your .env file.
const redis = new Redis({
 url: process.env.UPSTASH_REDIS_REST_URL!,
 token: process.env.UPSTASH_REDIS_REST_TOKEN!,
});
 
export default async function handler(
 req: NextApiRequest,
 res: NextApiResponse,
) {
 // Get the ID of the spreadsheet from the URL.
 const { id } = req.params;
 
 // Get the spreadsheet and its associated user from the database.
 const spreadsheet = await prisma.spreadsheet.findUnique({
 where: { id: Number(id) },
 include: { user: true },
 });
 
 // Verify that the spreadsheet exists.
 if (!spreadsheet) {
 return res.status(404).json({ error: "Spreadsheet not found" });
 }
 
 // Redis quota key, unique for each user by its ID.
 const quotaKey = `user:${spreadsheet.user.id}:quota`;
 
 // Retrieve the user's quota from Redis.
 const quota = await redis.incr(quotaKey);
 
 // If the key did not exist before, the returned value will be 1
 // and we set the expiration to 1 day.
 if (quota === 1) {
 await redis.expire(quotaKey, 60 * 60 * 24);
 }
 
 // Verify that the user is not over the quota.
 if (quota > MAX_FREE_TIER_QUOTA) {
 return res.status(429).json({ error: "Quota exceeded" });
 }
 
 // Return the spreadsheet's content.
 return res.json({ content: spreadsheet.content });
}

निष्कर्ष

इसे लाइव देखें