Update server.js

server.js

@@ -3,41 +3,32 @@ const express = require('express');
 const bodyParser = require('body-parser');
 const cors = require('cors');
 const requestIp = require('request-ip');
-const crypto = require('crypto'); // 用于生成设备指纹的哈希值
-const { URL } = require('url'); // 用于验证 URL 格式
+const crypto = require('crypto');
+const { URL } = require('url');
 
 const app = express();
 
 // === 配置常量 ===
-const PORT = process.env.PORT; // 动态端口由系统提供
-const HOST = '0.0.0.0';        // 允许来自外部的连接
+const PORT = process.env.PORT || 3000;
+const HOST = '0.0.0.0';
+console.log(`Starting server! PORT: ${PORT}`);
 
 // === 中间件配置 ===
-
-// CORS 配置
 const corsOptions = {
-    origin: '*', // 允许所有来源的请求
+    origin: '*',
     methods: ['GET', 'POST', 'OPTIONS'],
     allowedHeaders: ['Content-Type'],
 };
 app.use(cors(corsOptions));
-
-// 使用 Body-Parser 解析 JSON 请求体
 app.use(bodyParser.json());
-
-// 使用 request-ip 中间件获取客户端 IP
 app.use(requestIp.mw());
 
-// === 内存存储及清理 ===
-
-// URL 存储字典（以用户唯一标识符为键）
+// === 内存存储 ===
 const urlMap = new Map();
-
-// 超时时间配置（以毫秒为单位）
 const EXPIRATION_TIME = 90 * 1000; // 90 秒
-const CLEANUP_INTERVAL = 60 * 1000; // 每 60 秒清理一次
+const CLEANUP_INTERVAL = 60 * 1000;
 
-// 定时清理过期数据
+// 定期清理任务
 setInterval(() => {
     const now = Date.now();
     const expiredTime = now - EXPIRATION_TIME;
@@ -45,91 +36,73 @@ setInterval(() => {
     for (const [userId, { timestamp }] of urlMap) {
         if (timestamp < expiredTime) {
             console.log(`Deleting expired data for user: ${userId}`);
-            urlMap.delete(userId); // 删除过期数据
+            urlMap.delete(userId);
         }
     }
 }, CLEANUP_INTERVAL);
 
-// === 辅助函数 ===
-
-// 生成设备指纹
-const generateDeviceFingerprint = (req) => {
-    const ip = req.clientIp || '';
-    const userAgent = req.headers['user-agent'] || '';
-    const acceptLanguage = req.headers['accept-language'] || '';
-    const connection = req.headers['connection'] || '';
-    const encoding = req.headers['accept-encoding'] || '';
-    const forwardedFor = req.headers['x-forwarded-for'] || '';
-
-    // 将关键信息合并生成唯一指纹
-    const rawFingerprint = `${ip}-${userAgent}-${acceptLanguage}-${connection}-${encoding}-${forwardedFor}`;
-
-    // 使用 SHA-256 哈希算法生成指纹
-    const fingerprint = crypto.createHash('sha256').update(rawFingerprint).digest('hex');
-    return fingerprint;
-};
-
 // === 路由 ===
 
-// 存储 URL（POST 请求）
+// 健康检查
+app.get("/", (req, res) => {
+    res.status(200).send({ message: "Server is healthy and running." });
+});
+
+// 存储 URL
 app.post('/storeURL', (req, res) => {
-    const url = req.body.url; // 从请求体中解析 URL
-    const ip = req.clientIp; // 获取客户端 IP
+    const url = req.body.url;
+    const ip = req.clientIp;
 
-    // 验证 URL 是否存在并合法
     if (!url) {
-        return res.status(400).json({ error: 'URL is required.' });
+        return res.status(400).json({ success: false, message: 'URL is required.' });
     }
     try {
-        new URL(url); // 验证 URL 格式
+        new URL(url);
     } catch (err) {
-        return res.status(400).json({ error: 'Invalid URL format.' });
+        return res.status(400).json({ success: false, message: 'Invalid URL format.' });
     }
 
-    // 生成用户唯一标识符（包括 IP 和设备指纹）
-    const deviceFingerprint = generateDeviceFingerprint(req);
-    const userId = `${ip}-${deviceFingerprint}`; // 结合 IP 和设备指纹生成唯一标识符
-
-    // 存储到字典中
+    const fingerprint = crypto.createHash('sha256').update(`${ip}-${req.headers['user-agent']}`).digest('hex');
+    const userId = `${ip}-${fingerprint}`;
     urlMap.set(userId, { url, timestamp: Date.now() });
-    console.log(`Stored URL for user: ${userId}`);
 
-    // 返回成功响应
-    res.json({ message: 'URL stored successfully.', userId });
+    console.log(`Stored URL for user: ${userId}`);
+    res.json({ success: true, message: 'URL stored successfully.', userId });
 });
 
-// 获取 URL（GET 请求）
+// 获取 URL
 app.get('/getURL', (req, res) => {
-    const ip = req.clientIp; // 获取客户端 IP
-
-    // 生成用户唯一标识符（包括 IP 和设备指纹）
-    const deviceFingerprint = generateDeviceFingerprint(req);
-    const userId = `${ip}-${deviceFingerprint}`;
+    const ip = req.clientIp;
+    const fingerprint = crypto.createHash('sha256').update(`${ip}-${req.headers['user-agent']}`).digest('hex');
+    const userId = `${ip}-${fingerprint}`;
 
-    // 查询字典获取存储的 URL
     if (urlMap.has(userId)) {
         const storedData = urlMap.get(userId);
-        storedData.timestamp = Date.now(); // 更新数据时间戳
-        urlMap.set(userId, storedData); // 保存更新后的数据
+        storedData.timestamp = Date.now();
+        urlMap.set(userId, storedData);
 
         console.log(`Retrieved URL for user: ${userId}`);
-        return res.json({ url: storedData.url });
+        res.json({ success: true, url: storedData.url });
     } else {
         console.error(`No URL found for user: ${userId}`);
-        return res.status(404).json({ error: 'URL not found for this user.' });
+        res.status(404).json({ success: false, message: 'URL not found for this user.' });
     }
 });
 
-// 测试路由
-app.get("/", (req, res) => {
-    res.send("Welcome to the URL Server! The server is running.");
-});
-
 // === 启动服务器 ===
 app.listen(PORT, HOST, (err) => {
     if (err) {
         console.error(`Error starting server: ${err}`);
-        return;
+        process.exit(1);
     }
     console.log(`Server running on http://${HOST}:${PORT}`);
 });
+
+// === 全局错误处理 ===
+process.on('uncaughtException', (err) => {
+    console.error('Uncaught Exception:', err);
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+    console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+});