node簡單實現一個更改頭像功能

前言

一直想寫這篇文章，無奈因爲要考試的緣由，一直在複習，拖延到如今才寫🤣，以前用 node 的 express 框架寫了個小項目，裏面有個上傳圖片的功能，這裏記錄一下如何實現（我使用的是 ejs）📝

思路

1. 首先，當用戶點擊上傳頭像，更新頭像的時候，將頭像上傳到項目的一個文件夾裏面（我是存放在項目的public/images/img裏面），而且將圖像名重命名（能夠以時間戳來命名）。

   

2. 同時圖片在項目的路徑插入到用戶表的當前用戶的 userpicturepath 裏面
3. 而後更新用戶的 session，將圖片裏面的路徑賦值給 session 的裏面的picture屬性裏面
4. <img> 的 src 獲取到當前用戶的session裏面的 picture 的值，最後動態刷新頁面頭像就換成了用戶上傳的頭像了

實現效果

代碼

ejs部分

<img class="nav-user-photo" src="<%= user.picture.replace(/public(\/.*)/, "$1") %>" alt="Photo" style="height: 40px;"/>

<form enctype="multipart/form-data" method="post" name="fileInfo">
    <input type="file" accept="image/png,image/jpg" id="picUpload" name="file">
</form>

<button type="button" class="btn btn-primary" id="modifyPicV">肯定</button>
複製代碼

js部分

document.querySelector('#modifyPicV').addEventListener('click', function () {
    let formData = new FormData();
    formData.append("file",$("input[name='file']")[0].files[0]);//把文件對象插到formData對象上
    console.log(formData.get('file'));
    $.ajax({
        url:'/modifyPic',
        type:'post',
        data: formData,
        processData: false,  // 不處理數據
        contentType: false,   // 不設置內容類型
        success:function () {
            alert('success');
            location.reload();
        },
    })
});
複製代碼

路由部分，使用formidable，這是一個Node.js模塊，用於解析表單數據，尤爲是文件上傳

let express = require('express');
let router = express.Router();
let fs = require('fs');
let {User} = require('../data/db');
let formidable = require('formidable');
let cacheFolder = 'public/images/';//放置路徑
router.post('/modifyPic', function (req, res, next) {
    let userDirPath = cacheFolder + "Img";
    if (!fs.existsSync(userDirPath)) {
        fs.mkdirSync(userDirPath);//建立目錄
    }
    let form = new formidable.IncomingForm(); //建立上傳表單
    form.encoding = 'utf-8'; //設置編碼
    form.uploadDir = userDirPath; //設置上傳目錄
    form.keepExtensions = true; //保留後綴
    form.maxFieldsSize = 2 * 1024 * 1024; //文件大小
    form.type = true;
    form.parse(req, function (err, fields, files) {
        if (err) {
            return res.json(err);
        }
        let extName = ''; //後綴名
        switch (files.file.type) {
            case 'image/pjpeg':
                extName = 'jpg';
                break;
            case 'image/jpeg':
                extName = 'jpg';
                break;
            case 'image/png':
                extName = 'png';
                break;
            case 'image/x-png':
                extName = 'png';
                break;
        }
        if (extName.length === 0) {
            return res.json({
                msg: '只支持png和jpg格式圖片'
            });
        } else {
            let avatarName = '/' + Date.now() + '.' + extName;
            let newPath = form.uploadDir + avatarName;
            fs.renameSync(files.file.path, newPath); //重命名
            console.log(newPath)
            //更新表
            User.update({
                picture: newPath
            }, {
                where: {
                    username: req.session.user.username
                }
            }).then(function (data) {
                if (data[0] !== undefined) {
                    User.findAll({
                        where: {
                            username: req.session.user.username
                        }
                    }).then(function (data) {
                        if (data[0] !== undefined) {
                            req.session.user.picture = data[0].dataValues.picture;
                            res.send(true);
                        } else {
                            res.send(false);
                        }
                    })
                }
            }).catch(function (err) {
                console.log(err);
            });
        }
    });
});
複製代碼