node脚本导出数据库数据

说在前面

   在数据库管理和数据迁徙的过程中，常常必要将数据库中的表数据和结构举行导出，以便举行备份、共享或在不怜悯况中使用。数据库数据备份的方法有很多，今天让我们用 Node.js 来编写一个脚本，快速从 MySQL 数据库中导出所有表的数据和结构，并保存为单独的 SQL 文件。
  脚本编写

1. 模块引入

1. const mysql = require("mysql");
2. const fs = require("fs");

复制代码

• mysql模块用于毗连和操纵MySQL数据库
  
• fs模块用于文件操纵，将数据写入文件。
  

2. 数据库毗连设置

1. const dbConfig = {
2.   host: "localhost",
3.   user: "root",
4.   password: "password",
5.   database: "test",
6. };
8. const connection = mysql.createConnection(dbConfig);
10. connection.connect((error) => {
11.   if (error) throw error;
12.   console.log("Successfully connected to the database.");
13. });

复制代码

界说了数据库毗连设置对象dbConfig，包罗主机名、用户名、暗码和数据库名等信息。然后使用mysql.createConnection创建数据库毗连，并通过connection.connect方法举行毗连，若毗连乐成则打印相应信息。
3. 导出表数据和结构的函数

1. function exportTableData(tableName, close = false) {
2.   connection.query(
3.     `SHOW CREATE TABLE ${tableName}`,
4.     (error, results, fields) => {
5.       if (error) throw error;
7.       const createTableStatement = results[0]["Create Table"];
9.       connection.query(
10.         `SELECT * FROM ${tableName}`,
11.         (error, results, fields) => {
12.           if (error) throw error;
14.           const insertStatements = results
15.             .map((row) => {
16.               const keys = [],
17.                 values = [];
18.               Object.entries(row).forEach((item) => {
19.                 keys.push(item[0]);
20.                 values.push(`'${item[1]}'`);
21.               });
22.               return `INSERT INTO ${tableName} (${keys.join(
23.                 ","
24.               )}) VALUES (${values.join(",")});`;
25.             })
26.             .join("\n");
28.           const sql = `${createTableStatement};\n\n${insertStatements}`;
30.           fs.writeFile(`${tableName}.sql`, sql, (err) => {
31.             if (err) throw err;
32.             console.log(`Data from ${tableName} exported successfully.`);
33.             if (close) {
34.               connection.end();
35.               console.log("All exported!");
36.             }
37.           });
38.         }
39.       );
40.     }
41.   );
42. }

复制代码

• 这个函数担当表名tableName作为参数，另有一个可选参数close，用于控制是否在导出完成后关闭数据库毗连。
  
• 起首查询表的创建语句：
  

1. connection.query(
2.   `SHOW CREATE TABLE ${tableName}`,
3.   (error, results, fields) => {
4.     //...
5.   }
6. );

复制代码

• 然后查询表中的所有数据：
  

1. connection.query(
2.   `SELECT * FROM ${tableName}`,
3.   (error, results, fields) => {
4.     //...
5.   }
6. );

复制代码

• 对于查询到的数据，将每行数据转换为INSERT INTO语句：
  

1. const insertStatements = results
2. .map((row) => {
3.     const keys = [],
4.       values = [];
5.     Object.entries(row).forEach((item) => {
6.       keys.push(item[0]);
7.       values =append('${item[1]}');
8.     });
9.     return `INSERT INTO ${tableName} (${keys.join(
10.       ","
11.     )}) VALUES (${values.join(",")});`;
12.   })
13. .join("\n");

复制代码

• 末了将表的创建语句和所有数据的INSERT INTO语句组合起来写入到以表名命名的.sql文件中：
  

1. const sql = `${createTableStatement};\n\n${insertStatements}`;
3. fs.writeFile(`${tableName}.sql`, sql, (err) => {
4.   if (err) throw err;
5.   console.log(`Data from ${tableName} exported successfully.`);
6.   if (close) {
7.     connection.end();
8.     console.log("All exported!");
9.   }
10. });

复制代码

4. 执行导出操纵

1. connection.query("SHOW TABLES", (error, results, fields) => {
2.   if (error) throw error;
4.   results.forEach((result, index) => {
5.     const tableName = result[`Tables_in_${dbConfig.database}`];
6.     exportTableData(tableName, index === results.length - 1);
7.   });
8. });

复制代码

通过查询数据库中的所有表，然后对每个表调用exportTableData函数举行数据和结构的导出，当处理到末了一个表时，会根据exportTableData函数的close参数设置来决定是否关闭数据库毗连。
测试

创建数据库

1. CREATE DATABASE test;

复制代码

创建表

1. CREATE TABLE `t_user` (
2.   `id` int NOT NULL AUTO_INCREMENT,
3.   `name` varchar(255) NOT NULL,
4.   `age` int NOT NULL,
5.   PRIMARY KEY (`id`)
6. )

复制代码

插入数据

1. INSERT INTO t_user (name,age) VALUES ('张三', 25);
2. INSERT INTO t_user (name,age) VALUES ('李四', 24);

复制代码

导出数据

公众号

关注公众号『前端也能这么有趣』，获取更多有趣内容。
说在后面