SQL 查询 - PostgreSQL 查询 - 编写和执行查询

SQL 查询 - PostgreSQL 查询 - 编写和执行查询

在现代的软件开发中，数据存储和管理是至关重要的一部分。PostgreSQL 作为一款强大的开源关系型数据库管理系统，因其丰富的功能和高可靠性，被广泛应用于各种项目中。本文将深入探讨如何在 Node.js 环境下编写和执行 PostgreSQL 查询。

1. 环境准备

在开始之前，我们需要确保已经安装了 Node.js 和 PostgreSQL，并且安装了 pg 模块，它是 Node.js 中常用的 PostgreSQL 驱动程序。可以使用以下命令来安装 pg 模块：

npm install pg

2. 连接到 PostgreSQL 数据库

在执行查询之前，我们需要先建立与 PostgreSQL 数据库的连接。以下是一个简单的示例代码：

const { Pool } = require('pg');
// 创建一个连接池
const pool = new Pool({
    user: 'your_user',
    host: 'your_host',
    database: 'your_database',
    password: 'your_password',
    port: 5432, // 默认端口
});
// 测试连接
pool.query('SELECT NOW()', (err, res) => {
    if (err) {
        console.error('连接数据库时出错:', err);
    } else {
        console.log('当前时间:', res.rows[0].now);
    }
    // 关闭连接池
    pool.end();
});

在上述代码中，我们使用 Pool 类创建了一个连接池，它可以管理多个数据库连接。然后，我们执行了一个简单的 SELECT NOW() 查询来测试连接是否成功。

3. 编写和执行简单查询

3.1 查询所有记录

假设我们有一个名为 users 的表，包含 id、name 和 email 字段。以下是查询所有用户记录的示例代码：

const { Pool } = require('pg');
const pool = new Pool({
    user: 'your_user',
    host: 'your_host',
    database: 'your_database',
    password: 'your_password',
    port: 5432,
});
async function getAllUsers() {
    try {
        const query = 'SELECT * FROM users';
        const result = await pool.query(query);
        console.log('所有用户记录:', result.rows);
    } catch (err) {
        console.error('查询用户记录时出错:', err);
    } finally {
        pool.end();
    }
}
getAllUsers();

3.2 根据条件查询记录

如果我们只想查询 name 为 John 的用户记录，可以使用以下代码：

const { Pool } = require('pg');
const pool = new Pool({
    user: 'your_user',
    host: 'your_host',
    database: 'your_database',
    password: 'your_password',
    port: 5432,
});
async function getUsersByName(name) {
    try {
        const query = 'SELECT * FROM users WHERE name = $1';
        const values = [name];
        const result = await pool.query(query, values);
        console.log(`姓名为 ${name} 的用户记录:`, result.rows);
    } catch (err) {
        console.error('查询用户记录时出错:', err);
    } finally {
        pool.end();
    }
}
getUsersByName('John');

在上述代码中，我们使用了参数化查询，通过 $1 占位符来防止 SQL 注入攻击。

4. 插入、更新和删除记录

4.1 插入记录

以下是向 users 表中插入一条新记录的示例代码：

const { Pool } = require('pg');
const pool = new Pool({
    user: 'your_user',
    host: 'your_host',
    database: 'your_database',
    password: 'your_password',
    port: 5432,
});
async function insertUser(name, email) {
    try {
        const query = 'INSERT INTO users (name, email) VALUES ($1, $2) RETURNING *';
        const values = [name, email];
        const result = await pool.query(query, values);
        console.log('插入的用户记录:', result.rows[0]);
    } catch (err) {
        console.error('插入用户记录时出错:', err);
    } finally {
        pool.end();
    }
}
insertUser('Jane', 'jane@example.com');

4.2 更新记录

如果我们想更新 name 为 John 的用户的 email 地址，可以使用以下代码：

const { Pool } = require('pg');
const pool = new Pool({
    user: 'your_user',
    host: 'your_host',
    database: 'your_database',
    password: 'your_password',
    port: 5432,
});
async function updateUserEmail(name, newEmail) {
    try {
        const query = 'UPDATE users SET email = $1 WHERE name = $2 RETURNING *';
        const values = [newEmail, name];
        const result = await pool.query(query, values);
        console.log('更新后的用户记录:', result.rows[0]);
    } catch (err) {
        console.error('更新用户记录时出错:', err);
    } finally {
        pool.end();
    }
}
updateUserEmail('John', 'newjohn@example.com');

4.3 删除记录

以下是删除 name 为 Jane 的用户记录的示例代码：

const { Pool } = require('pg');
const pool = new Pool({
    user: 'your_user',
    host: 'your_host',
    database: 'your_database',
    password: 'your_password',
    port: 5432,
});
async function deleteUser(name) {
    try {
        const query = 'DELETE FROM users WHERE name = $1 RETURNING *';
        const values = [name];
        const result = await pool.query(query, values);
        console.log('删除的用户记录:', result.rows[0]);
    } catch (err) {
        console.error('删除用户记录时出错:', err);
    } finally {
        pool.end();
    }
}
deleteUser('Jane');

5. 总结

通过以上步骤，我们可以在 Node.js 环境下轻松地编写和执行 PostgreSQL 查询。在实际开发中，我们可以根据具体需求灵活运用这些技巧，实现更复杂的数据库操作。