xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-10-09 11:36:38 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

屏蔽mongodb 操作

Browse Source
This commit is contained in:
xuzhenjun 2023-04-15 01:01:51 +08:00
parent c617071746
commit f3d64ab88c
1 changed files with 50 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
app/api/chat-stream/route.ts
View File

@ -51,59 +51,59 @@ async function createStream(req: NextRequest) {
export async function POST(req: NextRequest) { export async function POST(req: NextRequest) {
const accessCode = req.headers.get("access-code"); const accessCode = req.headers.get("access-code");
try { try {
// 使用 clientPromise 连接到 MongoDB 数据库 // // 使用 clientPromise 连接到 MongoDB 数据库
const client = await clientPromise; // const client = await clientPromise;
// 选择数据库和集合 // // 选择数据库和集合
const db = client.db("chat_db"); // const db = client.db("chat_db");
const usersCollection = db.collection("users"); // const usersCollection = db.collection("users");
// 查询用户数据 // // 查询用户数据
const user = await usersCollection.findOne({ key: accessCode }); // const user = await usersCollection.findOne({ key: accessCode });
console.log(user, accessCode); // console.log(user, accessCode);
const tips = // const tips =
"您的链接授权已过期，为了避免恶意盗刷，\n 请关注微信公众号【coder思维】\n回复关键词：`ai` 获取授权链接 \n ![](/wx.png)"; // "您的链接授权已过期，为了避免恶意盗刷，\n 请关注微信公众号【coder思维】\n回复关键词：`ai` 获取授权链接 \n ![](/wx.png)";
if (!user) { // if (!user) {
return new Response(tips); // return new Response(tips);
} // }
console.log("compare: "); // console.log("compare: ");
console.log( // console.log(
user["expire"] < new Date().getTime(), // user["expire"] < new Date().getTime(),
user["expire"], // user["expire"],
new Date().getTime(), // new Date().getTime(),
); // );
if (user["expire"] < new Date().getTime()) { // if (user["expire"] < new Date().getTime()) {
// 判断用户是否过期 // // 判断用户是否过期
return new Response(tips); // return new Response(tips);
} // }
// 创建查询条件 // // 创建查询条件
// 计算24小时前的时间戳 // // 计算24小时前的时间戳
const currentTime = new Date(); // const currentTime = new Date();
const startTime = new Date(currentTime.getTime() - 24 * 60 * 60 * 1000); // const startTime = new Date(currentTime.getTime() - 24 * 60 * 60 * 1000);
// 创建查询条件 // // 创建查询条件
const query = { // const query = {
username: user["username"], // username: user["username"],
create_time: { $gte: startTime }, // create_time: { $gte: startTime },
}; // };
const collection = db.collection("chat_logs"); // const collection = db.collection("chat_logs");
// 根据查询条件查询记录数 // // 根据查询条件查询记录数
const recordCount = await collection.countDocuments(query); // const recordCount = await collection.countDocuments(query);
if (recordCount > 1000) { // if (recordCount > 1000) {
// 判断用户是否超过1000条记录 // // 判断用户是否超过1000条记录
return new Response( // return new Response(
"您的聊天提问已超过1000条，为了避免恶意盗刷，请过稍后再试，24小时内只能提问1000条", // "您的聊天提问已超过1000条，为了避免恶意盗刷，请过稍后再试，24小时内只能提问1000条",
); // );
} // }
// 创建聊天记录 // // 创建聊天记录
const chatLog = { // const chatLog = {
username: user["username"], // username: user["username"],
create_time: new Date().getTime(), // create_time: new Date().getTime(),
content: req.body, // content: req.body,
}; // };
collection.insertOne(chatLog).then(() => { // collection.insertOne(chatLog).then(() => {
//client.close(); // //client.close();
}); // });
const stream = await createStream(req); const stream = await createStream(req);
return new Response(stream); return new Response(stream);