执行 SQL 脚本 - SQL

在软件开发中，执行 SQL 脚本是很常见的需求。有时候，我们需要在数据库中创建表、插入数据等操作，而这些操作往往需要通过 SQL 语句来实现。本文将介绍如何在不同的编程语言中执行 SQL 脚本，并提供一些示例代码帮助读者更好地理解。

Python

在 Python 中，可以使用 sqlite3 模块来执行 SQL 脚本。下面是一个简单的例子，假设要创建一个名为 users 的表，其中包含 id 和 name 两个字段。

import sqlite3

conn = sqlite3.connect('example.db')
c = conn.cursor()

# 创建表
c.execute('''
    CREATE TABLE users
    (id INT PRIMARY KEY NOT NULL,
     name TEXT NOT NULL);
''')

# 插入数据
c.execute('INSERT INTO users (id, name) VALUES (?, ?)', (1, 'Alice'))
c.execute('INSERT INTO users (id, name) VALUES (?, ?)', (2, 'Bob'))

# 提交更改
conn.commit()

# 关闭连接
conn.close()

这里我们首先使用 sqlite3 模块连接到数据库，并获取一个 cursor 对象，然后通过调用 execute 方法来执行 SQL 语句。注意，为了避免 SQL 注入攻击，我们在执行 SQL 语句时使用了参数化查询。最后，记得提交更改并关闭数据库连接。

Java

在 Java 中，可以使用 JDBC（Java Database Connectivity）来执行 SQL 脚本。下面是一个示例代码，假设要在 MySQL 数据库中创建一个名为 users 的表。

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;

public class Main {
    public static void main(String[] args) {
        try {
            // 连接到数据库
            Connection conn = DriverManager.getConnection(
                    "jdbc:mysql://localhost:3306/mydb",
                    "myuser",
                    "mypass");

            // 创建 Statement 对象
            Statement stmt = conn.createStatement();

            // 执行 SQL 语句
            String sql = "CREATE TABLE users " +
                    "(id INT PRIMARY KEY NOT NULL, " +
                    " name VARCHAR(255) NOT NULL)";
            stmt.executeUpdate(sql);

            // 关闭连接
            stmt.close();
            conn.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

这里我们首先使用 JDBC 连接到数据库，然后创建一个 Statement 对象来执行 SQL 语句。注意，为了避免 SQL 注入攻击，我们可以使用 PreparedStatement 来替代 Statement，并使用问号占位符代替实际的参数。

Node.js

在 Node.js 中，可以使用 mysql 或 pg 模块来执行 SQL 脚本。下面是一个示例代码，假设要在 PostgreSQL 数据库中创建一个名为 users 的表。

const { Pool } = require('pg')

const pool = new Pool({
    user: 'postgres',
    host: 'localhost',
    database: 'mydb',
    password: 'mypassword',
    port: 5432,
})

pool.query(`
    CREATE TABLE users (
        id INT PRIMARY KEY NOT NULL,
        name VARCHAR(255) NOT NULL
    )
`)
    .then(res => {
        console.log('Table created successfully')
        pool.end()
    })
    .catch(err => {
        console.error(err)
        pool.end()
    })

这里我们首先使用 pg 模块创建了一个 Pool 对象，然后调用 query 方法来执行 SQL 语句。注意，为了避免 SQL 注入攻击，我们在执行 SQL 语句时使用了参数化查询。

总结

执行 SQL 脚本是程序员日常工作的一部分，我们可以使用不同的编程语言和库来实现这一目标。无论选择什么样的工具，都要注意安全性，避免 SQL 注入攻击。