Node.js中開發(fā)樹形結(jié)構(gòu)接口的實現(xiàn)

更新時間：2024年12月05日 10:14:47 作者：三線碼工

本文介紹了Node.js中開發(fā)樹形結(jié)構(gòu)接口的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

在現(xiàn)代 Web 開發(fā)中，樹形結(jié)構(gòu)的數(shù)據(jù)展示非常常見，例如文件系統(tǒng)、組織架構(gòu)、分類目錄等。本文將介紹如何在 Node.js 中開發(fā)一個返回樹形結(jié)構(gòu)數(shù)據(jù)的接口。我們將使用 Express 框架來處理 HTTP 請求，并使用 MySQL 數(shù)據(jù)庫來存儲分類數(shù)據(jù)。

項目初始化

首先，確保你已經(jīng)安裝了 Node.js 和 npm。然后，創(chuàng)建一個新的項目目錄并初始化 npm：

mkdir node-tree-api
cd node-tree-api
npm init -y

接下來，安裝所需的依賴包：

npm install express mysql2

設(shè)置數(shù)據(jù)庫

為了簡化示例，我們將使用 MySQL 數(shù)據(jù)庫。如果你還沒有安裝 MySQL，可以從 MySQL 官方網(wǎng)站下載并安裝。

創(chuàng)建一個新的數(shù)據(jù)庫，例如 tree_api_db：

CREATE DATABASE tree_api_db;
USE tree_api_db;

創(chuàng)建分類表

創(chuàng)建一個名為 categories 的表，包含以下字段：

id: 分類的唯一標(biāo)識
name: 分類的名稱
description: 分類的描述
parent_id: 父分類的 ID，頂級分類的 parent_id 為 NULL 或 0

CREATE TABLE categories (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    description TEXT,
    parent_id INT DEFAULT NULL
);

插入示例數(shù)據(jù)

插入一些示例數(shù)據(jù)以便測試：

INSERT INTO categories (name, description, parent_id) VALUES
('Electronics', 'Electronic products', NULL),
('Computers', 'Computer products', 1),
('Laptops', 'Laptop computers', 2),
('Desktops', 'Desktop computers', 2),
('Mobile Phones', 'Mobile phones', 1),
('Smartphones', 'Smart mobile phones', 5),
('Feature Phones', 'Feature mobile phones', 5),
('Tablets', 'Tablet devices', 1);

編寫樹形結(jié)構(gòu)查詢邏輯

在項目目錄中創(chuàng)建一個 db.js 文件來管理數(shù)據(jù)庫連接：

// db.js
const mysql = require('mysql2');

const connection = mysql.createConnection({
    host: 'localhost',
    user: 'your_mysql_user',
    password: 'your_mysql_password',
    database: 'tree_api_db'
});

connection.connect((err) => {
    if (err) {
        console.error('Error connecting to the database:', err.stack);
        return;
    }
    console.log('Connected to the database.');
});

module.exports = connection;

創(chuàng)建一個 utils.js 文件來處理樹形結(jié)構(gòu)的構(gòu)建：

// utils.js
function buildTree(categories) {
    const map = {};
    const roots = [];

    // 將每個分類放入 map 中
    categories.forEach(category => {
        map[category.id] = { ...category, children: [] };
    });

    // 構(gòu)建樹形結(jié)構(gòu)
    categories.forEach(category => {
        if (category.parent_id === null || category.parent_id === 0) {
            roots.push(map[category.id]);
        } else {
            if (map[category.parent_id]) {
                map[category.parent_id].children.push(map[category.id]);
            }
        }
    });

    return roots;
}

module.exports = { buildTree };

創(chuàng)建 Express 路由

創(chuàng)建一個 app.js 文件來設(shè)置 Express 應(yīng)用并定義路由：

// app.js
const express = require('express');
const db = require('./db');
const { buildTree } = require('./utils');

const app = express();
const port = 3000;

// 中間件，解析 JSON 請求體
app.use(express.json());

// 查詢分類表并以樹形結(jié)構(gòu)返回
app.get('/api/categories', (req, res) => {
    const sql = "SELECT id, name, description, parent_id FROM categories";

    db.query(sql, (err, results) => {
        if (err) {
            return res.status(500).send({ code: 0, msg: 'Database error', data: null });
        }

        // 構(gòu)建樹形結(jié)構(gòu)
        const tree = buildTree(results);

        // 返回樹形結(jié)構(gòu)的數(shù)據(jù)
        res.send({ code: 1, msg: '獲取分類成功', data: tree });
    });
});

app.listen(port, () => {
    console.log(`Server is running on http://localhost:${port}`);
});

測試接口

啟動你的 Node.js 應(yīng)用：

node app.js

然后，你可以使用工具如 Postman 或瀏覽器訪問 http://localhost:3000/api/categories 來測試新創(chuàng)建的接口。你應(yīng)該會看到類似以下的 JSON 響應(yīng)：

{
    "code": 1,
    "msg": "獲取分類成功",
    "data": [
        {
            "id": 1,
            "name": "Electronics",
            "description": "Electronic products",
            "parent_id": null,
            "children": [
                {
                    "id": 2,
                    "name": "Computers",
                    "description": "Computer products",
                    "parent_id": 1,
                    "children": [
                        {
                            "id": 3,
                            "name": "Laptops",
                            "description": "Laptop computers",
                            "parent_id": 2,
                            "children": []
                        },
                        {
                            "id": 4,
                            "name": "Desktops",
                            "description": "Desktop computers",
                            "parent_id": 2,
                            "children": []
                        }
                    ]
                },
                {
                    "id": 5,
                    "name": "Mobile Phones",
                    "description": "Mobile phones",
                    "parent_id": 1,
                    "children": [
                        {
                            "id": 6,
                            "name": "Smartphones",
                            "description": "Smart mobile phones",
                            "parent_id": 5,
                            "children": []
                        },
                        {
                            "id": 7,
                            "name": "Feature Phones",
                            "description": "Feature mobile phones",
                            "parent_id": 5,
                            "children": []
                        }
                    ]
                },
                {
                    "id": 8,
                    "name": "Tablets",
                    "description": "Tablet devices",
                    "parent_id": 1,
                    "children": []
                }
            ]
        }
    ]
}