在nodejs mongodb中添加url缩短计数器

const express = require('express');
const app = express();
const MongoClient = require('mongodb').MongoClient;
const url = 'mongodb://localhost:27017'; // MongoDB连接URL
const dbName = 'url_shortener'; // 数据库名称

// 其他代码

app.get('/shorten', (req, res) => {
  const originalUrl = req.query.url; // 获取原始URL参数

  // 生成短URL代码
  // ...

  // 将原始URL和短URL保存到数据库
  MongoClient.connect(url, (err, client) => {
    if (err) {
      console.error(err);
      res.status(500).send('数据库连接错误');
    } else {
      const db = client.db(dbName);
      const collection = db.collection('urls');

      collection.insertOne({ originalUrl, shortUrl, count: 0 }, (err, result) => {
        if (err) {
          console.error(err);
          res.status(500).send('数据库插入错误');
        } else {
          res.send({ shortUrl });
        }
        client.close();
      });
    }
  });
});

app.get('/:shortUrl', (req, res) => {
  const shortUrl = req.params.shortUrl;

  MongoClient.connect(url, (err, client) => {
    if (err) {
      console.error(err);
      res.status(500).send('数据库连接错误');
    } else {
      const db = client.db(dbName);
      const collection = db.collection('urls');

      collection.findOneAndUpdate(
        { shortUrl },
        { $inc: { count: 1 } }, // 每次访问增加计数器
        { returnOriginal: false },
        (err, result) => {
          if (err) {
            console.error(err);
            res.status(500).send('数据库更新错误');
          } else if (result.value) {
            res.redirect(result.value.originalUrl);
          } else {
            res.status(404).send('未找到对应的URL');
          }
          client.close();
        }
      );
    }
  });
});

const port = 3000;

app.listen(port, () => {
  console.log(`服务器已启动，正在监听端口 ${port}`);
});