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

本文將指導(dǎo)您如何使用 Node.js、Multer 和 PostgreSQL 構(gòu)建一個簡單易用的 API，實現(xiàn)單張和多張圖片的上傳功能。無需復(fù)雜的配置，即可輕松完成圖片上傳和數(shù)據(jù)庫存儲。

先決條件:

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

項目設(shè)置:

1. 創(chuàng)建一個項目目錄并初始化：

   mkdir imageUpload
   cd imageUpload
   npm init -y

2. 安裝依賴項：

   npm install express multer body-parser pg

3. 創(chuàng)建目錄結(jié)構(gòu)：

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

4. 創(chuàng)建 PostgreSQL 數(shù)據(jù)庫和用戶表：

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

代碼實現(xiàn):

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

   const { Client } = require('pg');
   const client = new Client({
       user: 'postgres',
       host: 'localhost',
       database: 'uploader',
       password: 'YOUR_PASSWORD', // 請?zhí)鎿Q為您的數(shù)據(jù)庫密碼
       port: 5432,
   });
   client.connect().then(() => console.log("Connected to DB")).catch(err => console.error("數(shù)據(jù)庫連接失敗:", err.message));
   module.exports = client;

2. Express 服務(wù)器 (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')); // 設(shè)置靜態(tài)文件目錄
   
   require('./routes/image_routes')(app);
   
   app.listen(port, () => {
       console.log(`服務(wù)器運行在端口 ${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 發(fā)送 POST 請求到 /upload/single 或 /upload/multiple，并附帶圖片文件和用戶名數(shù)據(jù)。

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

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.

以上是如何構(gòu)建 NodeJS Express REST API 以使用 Multer(PostgreSQL) 上傳圖像的詳細內(nèi)容。更多信息請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！