javascript - 限制下载次数

Question

我真的是 mongoDB 和 node.js 的新手，我试图限制我服务器中每个文件的下载数量。我正在使用 gridfs 将文件存储在数据库中，在生成一个能够下载文件的链接后，我需要限制每个文件的下载数量，但不知道该怎么做。

Answer 1

假设您使用 express 作为您的 node.js http 服务器，您可以这样做:

const app = require('express')();
const bucket = new mongodb.GridFSBucket(db, {
  chunkSizeBytes: 1024,
  bucketName: 'songs'
});

const downloads = {};

const fileURI = '/somefile.mp3';
const maxDownload = 100;

app.get(fileURI, function(req, res) {
  if (downloads[fileURI] <= maxDownload) {
      // pipe the file to res
      return bucket.openDownloadStreamByName('somefile.mp3').
      .pipe(res)
      .on('error', function(error) {
          console.error(error);
      })
      .on('finish', function() {
          console.log('done!');
          downloads[fileURI] = downloads[fileURI] || 0;
          downloads[fileURI]++;
      });
    } 

    return res.status(400).send({ message: 'download limit reached' });
});    

app.listen(8080);