轻松上传新头像，彰显个性风采！

2024-01-21 16:16:33

好的，以下是以node.js实现头像上传功能的一篇文章：

背景

在许多网络应用中，用户头像上传功能是不可或缺的。它允许用户选择并上传自己的头像，彰显其个人特色。作为开发者，我们自然希望能提供这种基本的功能。本文将介绍如何在Node.js中实现头像上传功能。

实现步骤

安装必要的依赖项。

npm install multer

创建头像上传路由。

app.post('/upload-avatar', upload.single('avatar'), (req, res) => {
  // 处理头像上传逻辑
});

设置头像上传选项。

const upload = multer({
  dest: 'public/images/img',
  limits: {
    fileSize: 1000000 // 1MB
  },
  fileFilter: (req, file, cb) => {
    // 检查文件类型
    if (file.mimetype.startsWith('image/')) {
      cb(null, true);
    } else {
      cb(new Error('Invalid file type'), false);
    }
  }
});

处理头像上传逻辑。

const avatarName = Date.now() + '-' + req.file.originalname;
const avatarPath = path.join(__dirname, '..', 'public', 'images', 'img', avatarName);

fs.rename(req.file.path, avatarPath, (err) => {
  if (err) {
    res.status(500).json({ message: '头像上传失败！' });
  } else {
    // 更新用户头像
    User.findByIdAndUpdate(req.user._id, { avatar: avatarName }, (err, user) => {
      if (err) {
        res.status(500).json({ message: '头像更新失败！' });
      } else {
        res.status(200).json({ message: '头像更新成功！' });
      }
    });
  }
});