Nodejs 利用socket.io监听访问量

Node.js 利用 Socket.io 监听访问量

在这篇文章中，我将介绍如何使用 Node.js、Express 和 MongoDB 结合 Socket.io 来实现一个简单的网页访问量统计系统。我们将通过 Socket.io 实时监听客户端的连接和断开事件，并在每次连接时记录访问信息。

数据模型

首先，我们需要定义数据模型来存储点击事件。这里我们使用 Mongoose 操作 MongoDB：

// models/click.js
const mongoose = require('mongoose');

const clickSchema = new mongoose.Schema({
  intime: { type: Date, default: Date.now }, // 进入时间
  outtime: { type: Date, default: null },   // 退出时间
  referrer: String,                         // 上一步 URL
  address: String,                          // IP 地址
  user: String                              // 用户标识（暂用 IP 地址）
});

const Click = mongoose.model('Click', clickSchema);

module.exports = Click;

状态管理

接下来，我们需要设置 Mongoose 连接 MongoDB：

// status.js
const mongoose = require('mongoose');

mongoose.connect('mongodb://localhost/blog', { useNewUrlParser: true, useUnifiedTopology: true });

const db = mongoose.connection;

module.exports = { mongoose, db };

后端逻辑

在后端，我们将处理用户的连接和断开事件，并更新数据库中的记录：

// models/user.js
const Click = require('./click');
const { mongoose } = require('../status');

class User {
  constructor() {}

  async intoclick(referrer, dd) {
    const click = new Click({
      intime: new Date(),
      outtime: null,
      referrer,
      address: dd.ip,
      user: dd.ip
    });

    try {
      await click.save();
      dd._id = click._id;
    } catch (err) {
      console.error(err);
    }
  }

  async outclick(dd) {
    try {
      await Click.updateOne(
        { _id: dd._id },
        { $set: { outtime: new Date() } }
      );
    } catch (err) {
      console.error(err);
    }
  }
}

module.exports = User;

Socket.io 配置

现在，我们可以配置 Socket.io 来监听客户端的连接和断开事件：

// server.js
const express = require('express');
const http = require('http');
const socketIo = require('socket.io');
const User = require('./models/user');

const app = express();
const server = http.createServer(app);
const io = socketIo(server);

const user = new User();

io.on('connection', (socket) => {
  console.log('A user connected:', socket.id);

  socket.on('intoclick', (data) => {
    user.intoclick(data.referrer, { ip: socket.handshake.address.address, _id: 0 });
  });

  socket.on('disconnect', () => {
    user.outclick({ ip: socket.handshake.address.address, _id: 0 });
  });
});

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

前端代码

最后，在前端页面中，我们将初始化 Socket.io 并发送连接请求：

<script src="/socket.io/socket.io.js"></script>
<script>
  window.onload = function() {
    var socket = io.connect('http://localhost:3000');
    socket.emit('intoclick', { referrer: document.referrer || "" });
  };
</script>

以上就是使用 Node.js、Express、MongoDB 和 Socket.io 实现网页访问量统计的完整流程。希望对你有所帮助！