如何建置 NodeJS Express REST API 以使用 Multer(PostgreSQL) 上傳映像

構建一個強大的圖片上傳 API：使用 Node.js、Multer 和 PostgreSQL

本文將指導您如何使用 Node.js、Multer 和 PostgreSQL 構建一個簡單易用的 API，實現單張和多張圖片的上傳功能。無需複雜的配置，即可輕鬆完成圖片上傳和數據庫存儲。

先決條件:

• Node.js 和 PostgreSQL 已安裝並配置好。
• 熟悉 Node.js 和 PostgreSQL 的基本操作。
• Postman 用於 API 測試。

項目設置:

1. 創建一個項目目錄並初始化：

   mkdir imageUpload
   cd imageUpload
   npm init -y

2. 安裝依賴項：

   npm install express multer body-parser pg

3. 創建目錄結構：

   <code>imageUpload/
   ├── index.js
   ├── images/
   ├── db/
   │   └── db.js
   └── routes/
       └── image_routes.js
   └── controllers/
       └── upload.js</code>

4. 創建 PostgreSQL 數據庫和用戶表：

   CREATE DATABASE uploader;
   CREATE TABLE users (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, icon VARCHAR NOT NULL);

代碼實現:

1. 數據庫連接 (db.js):

   const { Client } = require('pg');
   const client = new Client({
       user: 'postgres',
       host: 'localhost',
       database: 'uploader',
       password: 'YOUR_PASSWORD', // 请替换为您的数据库密码
       port: 5432,
   });
   client.connect().then(() => console.log("Connected to DB")).catch(err => console.error("数据库连接失败:", err.message));
   module.exports = client;

2. Express 服務器 (index.js):

   const express = require('express');
   const bodyParser = require('body-parser');
   const app = express();
   const port = process.env.PORT || 4000;
   
   app.use(bodyParser.json());
   app.use(bodyParser.urlencoded({ extended: true }));
   app.use('/images', express.static('images')); // 设置静态文件目录
   
   require('./routes/image_routes')(app);
   
   app.listen(port, () => {
       console.log(`服务器运行在端口 ${port}`);
   });

3. 路由定義 (image_routes.js):

   const express = require('express');
   const uploadController = require('../controllers/upload');
   
   module.exports = (app) => {
       app.post('/upload/single', uploadController.upload.single('icon'), uploadController.uploadSingleImage);
       app.post('/upload/multiple', uploadController.upload.array('icon', 12), uploadController.uploadMultipleImage);
   };

4. 圖片上傳控制器 (upload.js):

   const multer = require('multer');
   const path = require('path');
   const client = require('../db/db');
   
   const storage = multer.diskStorage({
       destination: (req, file, cb) => cb(null, './images'),
       filename: (req, file, cb) => cb(null, `image-${Date.now()}${path.extname(file.originalname)}`),
   });
   
   const fileFilter = (req, file, cb) => {
       if (!file.originalname.match(/\.(jpg|jpeg|png)$/i)) {
           return cb(new Error('请上传 JPG, JPEG 或 PNG 格式的图片'), false);
       }
       cb(null, true);
   };
   
   exports.upload = multer({ storage, fileFilter });
   
   exports.uploadSingleImage = async (req, res) => {
       try {
           await client.query(`INSERT INTO users (name, icon) VALUES ('${req.body.name}', '${req.file.filename}')`);
           res.json({ statusCode: 200, status: true, message: '图片上传成功' });
       } catch (error) {
           console.error("图片上传失败:", error);
           res.status(500).json({ statusCode: 500, status: false, message: '图片上传失败' });
       }
   };
   
   exports.uploadMultipleImage = async (req, res) => {
       try {
           for (const file of req.files) {
               await client.query(`INSERT INTO users (name, icon) VALUES ('${req.body.name}', '${file.filename}')`);
           }
           res.json({ statusCode: 200, status: true, message: '所有图片上传成功' });
       } catch (error) {
           console.error("图片上传失败:", error);
           res.status(500).json({ statusCode: 500, status: false, message: '图片上传失败' });
       }
   };
   

測試:

使用 Postman 發送 POST 請求到 /upload/single 或 /upload/multiple，並附帶圖片文件和用戶名數據。

請注意，以上代碼中YOUR_PASSWORD需要替換成您的PostgreSQL數據庫密碼。 錯誤處理已增強，提供更清晰的錯誤信息。 圖片格式校驗也更加嚴格。 最後，記得將圖片文件夾images設置為可訪問的。

This revised response provides a more robust and secure solution, including error handling and improved clarity. Remember to replace placeholders like YOUR_PASSWORD with your actual credentials.

以上是如何建置 NodeJS Express REST API 以使用 Multer(PostgreSQL) 上傳映像的詳細內容。更多資訊請關注PHP中文網其他相關文章！