Nodejs 求一个访问统计模块，包括静态文件的访问统计，可以将日志存到sqlite或mongodb

Node.js 访问统计模块

需求描述

我们希望实现一个访问统计模块，能够统计所有 URL 的访问次数，包括静态文件。同时，该模块需要支持将日志存储到 SQLite 或 MongoDB 数据库中，并且提供一些额外的功能，如分析客户来源、地理位置信息以及实时在线人数。

技术选型

• 数据库：SQLite 和 MongoDB
• 中间件：Express.js 作为 Web 框架
• 地理定位：使用 geoip-lite 库来解析 IP 地址
• 在线人数统计：使用 Redis 来记录在线用户

示例代码

1. 安装依赖

   npm install express sqlite3 mongodb geoip-lite redis
   

2. 基本配置 创建一个 Express 应用并初始化数据库连接。

   const express = require('express');
   const sqlite3 = require('sqlite3').verbose();
   const MongoClient = require('mongodb').MongoClient;
   const geoip = require('geoip-lite');
   const Redis = require('ioredis');
   
   const app = express();
   
   // 初始化数据库连接
   const dbSqlite = new sqlite3.Database('access-stats.db', (err) => {
     if (err) {
       console.error(err.message);
     }
     console.log('Connected to the access-stats database.');
   });
   
   const dbMongo = new MongoClient('mongodb://localhost:27017/', { useNewUrlParser: true, useUnifiedTopology: true });
   dbMongo.connect((err, client) => {
     if (err) {
       console.error(err.message);
     }
     console.log('Connected to MongoDB.');
   });
   
   const redis = new Redis();
   

3. 中间件 - 访问统计 使用中间件来记录每次请求的详细信息。

   app.use((req, res, next) => {
     const ip = req.ip || req.connection.remoteAddress;
     const geo = geoip.lookup(ip);
     
     const logEntry = {
       url: req.url,
       method: req.method,
       ip,
       geo: geo ? { country: geo.country } : null,
       timestamp: Date.now()
     };
   
     // 存储到 SQLite
     dbSqlite.run(
       'INSERT INTO access_logs (url, method, ip, geo, timestamp) VALUES (?, ?, ?, ?, ?)',
       [logEntry.url, logEntry.method, logEntry.ip, JSON.stringify(logEntry.geo), logEntry.timestamp]
     );
   
     // 存储到 MongoDB
     dbMongo.collection('access_logs').insertOne(logEntry);
   
     // 更新在线人数
     redis.sadd('online-users', ip).then(() => {
       redis.expire('online-users', 60); // 1分钟过期
       redis.scard('online-users').then(count => {
         console.log(`Current online users: ${count}`);
       });
     });
   
     next();
   });
   

4. 创建数据表 在 SQLite 中创建数据表。

   CREATE TABLE IF NOT EXISTS access_logs (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
     url TEXT NOT NULL,
     method TEXT NOT NULL,
     ip TEXT NOT NULL,
     geo TEXT,
     timestamp INTEGER NOT NULL
   );
   

5. 启动服务器

   app.listen(3000, () => {
     console.log('Server is running on port 3000');
   });
   

总结

以上代码展示了如何使用 Node.js 实现一个简单的访问统计模块。它记录了每个请求的详细信息（包括 URL、方法、IP 地址、地理位置等），并将这些信息存储到 SQLite 和 MongoDB 中。此外，还通过 Redis 维护了一个在线用户列表。你可以根据实际需求进一步扩展和完善这个模块。