Node.js案例 - 记账本

一、背景与问题

在开发个人财务管理类应用时，记账本系统是最基础的模块之一。传统做法是使用关系型数据库存储账目数据，但随着业务增长，需要考虑数据一致性、并发控制、性能优化等问题。

Node.js作为异步非阻塞的JavaScript运行环境，非常适合处理高并发的记账场景。但开发者常陷入以下困境：

如何在不使用ORM的情况下直接操作数据库？
如何保证多用户同时记账时的数据一致性？
如何在保证性能的同时处理复杂的账目计算？

本文将通过一个完整的记账本系统案例，深入解析Node.js在数据持久化、并发控制和性能优化方面的实践。

二、基本原理

1. Node.js的事件驱动架构

Node.js通过事件循环(Event Loop)处理异步操作，其核心机制是：

const fs = require('fs');
fs.readFile('file.txt', (err, data) => {
  if (err) throw err;
  console.log(data.toString());
});

在记账本系统中，文件读写、数据库操作等I/O操作均通过回调函数处理，避免阻塞主线程。

2. 数据库连接池原理

使用连接池可以显著提升数据库访问性能：

const mysql = require('mysql');
const pool = mysql.createPool({
  host: 'localhost',
  user: 'root',
  password: '123456',
  database: 'accounting'
});

pool.getConnection((err, connection) => {
  if (err) throw err;
  connection.query('SELECT * FROM accounts', (err, results) => {
    if (err) throw err;
    console.log(results);
    connection.release();
  });
});

连接池通过维护空闲连接队列，避免重复创建连接的开销。

三、环境准备

1. 开发环境配置

# 安装依赖
npm init -y
npm install express mysql2 dotenv

创建.env文件配置数据库参数：

DB_HOST=localhost
DB_USER=root
DB_PASSWORD=123456
DB_NAME=accounting

2. 项目结构设计

accounting-system/
├── src/
│   ├── db/
│   │   └── index.js        # 数据库连接
│   ├── models/
│   │   └── account.js      # 账目模型
│   ├── routes/
│   │   └── account.js      # 路由
│   └── server.js           # 启动文件
├── .env
├── package.json
└── README.md

四、核心实现

1. 数据库连接池实现

// src/db/index.js
const mysql = require('mysql2/promise');
require('dotenv').config();

const pool = mysql.createPool({
  host: process.env.DB_HOST,
  user: process.env.DB_USER,
  password: process.env.DB_PASSWORD,
  database: process.env.DB_NAME,
  connectionLimit: 10
});

async function query(sql, params) {
  const connection = await pool.getConnection();
  try {
    const [rows] = await connection.query(sql, params);
    return rows;
  } finally {
    connection.release();
  }
}

module.exports = { query };

关键点解析：

使用mysql2/promise库实现异步操作
设置连接池限制防止资源耗尽
自动释放连接资源

2. 账目模型实现

// src/models/account.js
const { query } = require('../db');

class Account {
  static async create(data) {
    const sql = 'INSERT INTO accounts SET ?';
    const result = await query(sql, [data]);
    return result.insertId;
  }

  static async findById(id) {
    const sql = 'SELECT * FROM accounts WHERE id = ?';
    const [rows] = await query(sql, [id]);
    return rows[0];
  }

  static async findAll() {
    const sql = 'SELECT * FROM accounts';
    const [rows] = await query(sql);
    return rows;
  }
}

module.exports = Account;

关键点解析：

使用参数化查询防止SQL注入
封装CRUD操作提高可维护性
简单的封装模式适合小型系统

3. 路由实现

// src/routes/account.js
const express = require('express');
const router = express.Router();
const Account = require('../models/account');

router.get('/accounts', async (req, res) => {
  const accounts = await Account.findAll();
  res.json(accounts);
});

router.post('/accounts', async (req, res) => {
  const { name, amount, category } = req.body;
  const account = await Account.create({
    name,
    amount,
    category
  });
  res.status(201).json({ id: account });
});

module.exports = router;

关键点解析：

使用Express Router组织路由
异步处理确保正确响应
基础的CRUD接口实现

五、完整案例

1. 启动文件

// src/server.js
const express = require('express');
const accountRoutes = require('./routes/account');

const app = express();
app.use(express.json());

app.use('/api', accountRoutes);

const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
  console.log(`Server running on port ${PORT}`);
});

2. 数据库表结构

-- 创建账目表
CREATE TABLE accounts (
  id INT AUTO_INCREMENT PRIMARY KEY,
  name VARCHAR(255) NOT NULL,
  amount DECIMAL(10,2) NOT NULL,
  category VARCHAR(50) NOT NULL,
  created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

-- 创建索引
CREATE INDEX idx_category ON accounts(category);

3. 完整运行流程

启动服务：node src/server.js
使用Postman发送POST请求：
- URL: http://localhost:3000/api/accounts
- Body:
```
{
  "name": "早餐",
  "amount": 15.8,
  "category": "餐饮"
}
```
获取所有账目：GET http://localhost:3000/api/accounts

六、源码解析

1. 连接池配置

connectionLimit: 10

设置连接池最大连接数为10，适用于中小型系统。在高并发场景下，可考虑使用async/await结合for await循环实现更精细的连接管理。

2. 查询封装

const [rows] = await connection.query(sql, params);

使用参数化查询防止SQL注入，同时通过[rows]获取查询结果。注意query方法返回的是一个Promise，需要正确处理异步操作。

3. 异常处理

try {
  const [rows] = await connection.query(sql, params);
  return rows;
} finally {
  connection.release();
}

通过try-finally确保连接释放，避免连接泄漏。在生产环境中建议添加更完善的错误处理机制。

七、进阶使用

1. 复杂查询优化

// 查询某类别的总金额
const sql = 'SELECT category, SUM(amount) as total FROM accounts GROUP BY category';

使用GROUP BY进行聚合查询，结合索引可以显著提升性能。建议对category字段建立索引。

2. 事务处理

async function transfer(fromId, toId, amount) {
  const connection = await pool.getConnection();
  try {
    await connection.beginTransaction();
    
    const fromAccount = await Account.findById(fromId);
    if (!fromAccount || fromAccount.amount < amount) {
      throw new Error('Insufficient funds');
    }
    
    await Account.update({ id: fromId, amount: fromAccount.amount - amount });
    await Account.update({ id: toId, amount: toAccount.amount + amount });
    
    await connection.commit();
  } catch (error) {
    await connection.rollback();
    throw error;
  } finally {
    connection.release();
  }
}

事务处理确保了数据一致性，适用于需要保证操作原子性的场景。

3. 服务端渲染

// 前端页面
const express = require('express');
const app = express();
app.set('views', './views');
app.set('view engine', 'ejs');

app.get('/', (req, res) => {
  res.render('index');
});

结合EJS模板引擎可以实现服务端渲染，适合需要展示账目统计的场景。

八、性能与工程实践

1. 性能优化策略

优化措施	说明
连接池配置	合理设置连接池大小，避免资源浪费
索引优化	对常用查询字段建立索引
缓存机制	对频繁查询的数据进行缓存
异步处理	避免阻塞式操作

2. 安全注意事项

使用参数化查询防止SQL注入
设置严格的CORS策略
对敏感数据进行加密存储
定期更新依赖库

3. 异常处理规范

router.post('/accounts', async (req, res) => {
  try {
    const { name, amount, category } = req.body;
    const account = await Account.create({
      name,
      amount,
      category
    });
    res.status(201).json({ id: account });
  } catch (error) {
    console.error(error);
    res.status(500).json({ error: 'Internal server error' });
  }
});

统一的异常处理机制可以提高系统的健壮性。

九、常见问题与踩坑

1. 常见错误

错误示例：

const [rows] = await connection.query(sql);

问题：未使用参数化查询，容易导致SQL注入。

解决办法：使用参数化查询：

const [rows] = await connection.query(sql, [params]);

2. 索引问题

错误示例：

SELECT * FROM accounts WHERE category = '餐饮';

问题：未对category字段建立索引，导致全表扫描。

解决办法：创建索引：

CREATE INDEX idx_category ON accounts(category);

3. 异步错误处理

错误示例：

async function doSomething() {
  await someAsyncFunction();
  // 未处理错误
}

问题：未处理未捕获的Promise拒绝。

解决办法：使用try-catch块：

try {
  await someAsyncFunction();
} catch (error) {
  console.error(error);
}

十、最佳实践

1. 推荐实践

使用连接池管理数据库连接
对关键字段建立索引
使用参数化查询防止SQL注入
对异常进行统一处理
定期清理无用数据

2. 不推荐实践

直接使用mysql而非mysql2/promise
在事务中进行大量数据操作
未处理未捕获的Promise拒绝
未对敏感数据进行加密

十一、总结

通过记账本系统案例，我们深入探讨了Node.js在数据持久化方面的实践。在实际开发中，需要根据业务需求选择合适的实现方式：

对于小型系统，直接使用连接池和参数化查询即可
对于复杂业务，应考虑使用ORM框架
对于高并发场景，需要进行性能优化和分布式处理

在开发过程中，需要特别注意：

异常处理的完整性
数据库连接的管理
安全性的保障
性能的优化

通过合理的设计和实践，可以构建出稳定、高效的记账本系统。记住，技术选型要根据具体业务需求，避免过度设计。

Node.js案例 - 记账本

Node.js案例 - 记账本

一、背景与问题

二、基本原理

1. Node.js的事件驱动架构

2. 数据库连接池原理

三、环境准备

1. 开发环境配置

2. 项目结构设计

四、核心实现

1. 数据库连接池实现

2. 账目模型实现

3. 路由实现

五、完整案例

1. 启动文件

2. 数据库表结构

3. 完整运行流程

六、源码解析

1. 连接池配置

2. 查询封装

3. 异常处理

七、进阶使用

1. 复杂查询优化

2. 事务处理

3. 服务端渲染

八、性能与工程实践

1. 性能优化策略

2. 安全注意事项

3. 异常处理规范

九、常见问题与踩坑

1. 常见错误

2. 索引问题

3. 异步错误处理

十、最佳实践

1. 推荐实践

2. 不推荐实践

十一、总结

评论已关闭

推荐阅读