Guides > Databases > Postgresql > Rust

本教程将指导你如何在 Sealos DevBox 项目中使用 Rust 语言连接和操作 PostgreSQL 数据库。

准备工作

在开始之前，请确保你已经：

安装依赖

在项目的 Cargo.toml 文件中，需要添加以下依赖项：

Cargo.toml

[dependencies]
tokio = { version = "1.28", features = ["full"] }
sqlx = { version = "0.6", features = ["runtime-tokio-rustls", "postgres"] }
dotenv = "0.15"

主要依赖包括：

tokio：Rust 的异步运行时环境
sqlx：异步数据库操作工具包
dotenv：环境变量配置管理工具

配置数据库连接

设置环境变量

首先，在项目根目录创建 .env 文件，配置数据库连接信息：

.env

DATABASE_URL=postgres://your_username:your_password@your_database_host:5432/your_database_name

注意：请使用从 Sealos 桌面数据库应用中获取的实际 PostgreSQL 连接信息替换上述占位符。

编写主程序代码

创建 src/main.rs 文件，实现数据库操作逻辑：

src/main.rs

use sqlx::postgres::PgPoolOptions;
use sqlx::Row;
use dotenv::dotenv;
use std::env;
 
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
    // Load environment variables from .env file
    dotenv().ok();
 
    // Get the database URL from the environment
    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
 
    // Create a connection pool
    let pool = PgPoolOptions::new()
        .max_connections(5)
        .connect(&database_url)
        .await?;
 
    // Create the employees table if it doesn't exist
    sqlx::query(
        r#"
        CREATE TABLE IF NOT EXISTS employees (
            id SERIAL PRIMARY KEY,
            name VARCHAR(100) NOT NULL,
            position VARCHAR(100) NOT NULL
        )
        "#,
    )
    .execute(&pool)
    .await?;
 
    println!("Table created successfully");
 
    // Insert a sample employee
    let new_employee = sqlx::query(
        r#"
        INSERT INTO employees (name, position)
        VALUES ($1, $2)
        RETURNING id, name, position
        "#,
    )
    .bind("John Doe")
    .bind("Developer")
    .fetch_one(&pool)
    .await?;
 
    println!(
        "Inserted employee: ID: {}, Name: {}, Position: {}",
        new_employee.get::<i32, _>("id"),
        new_employee.get::<String, _>("name"),
        new_employee.get::<String, _>("position")
    );
 
    // Query all employees
    let employees = sqlx::query("SELECT id, name, position FROM employees")
        .fetch_all(&pool)
        .await?;
 
    println!("All employees:");
    for employee in employees {
        println!(
            "ID: {}, Name: {}, Position: {}",
            employee.get::<i32, _>("id"),
            employee.get::<String, _>("name"),
            employee.get::<String, _>("position")
        );
    }
 
    Ok(())
}

代码的主要组成部分：

引入依赖：导入 sqlx、dotenv 和 std::env 等必要模块。
配置主函数：使用 #[tokio::main] 标注启用异步运行时。
加载配置：读取 .env 文件中的数据库连接参数。
创建连接池：通过 PgPoolOptions 实现数据库连接池管理。
初始化数据表：创建示例表结构 (如果不存在)。
数据操作示例：演示基本的数据插入操作。
查询数据：展示如何查询并显示数据记录。

运行程序

在终端中执行以下命令运行程序：

cargo run

该命令将编译并执行示例程序，展示完整的数据库操作流程。

最佳实践

妥善管理数据库凭据，使用环境变量存储敏感信息。
使用连接池优化数据库连接性能和资源利用。
采用预处理语句 (如示例中的 sqlx::query) 防止 SQL 注入风险。
充分利用 Rust 的 Result 类型进行错误处理。
合理使用 async/await 提升异步操作效率。

常见问题排查

如果遇到连接问题，请检查以下几点：

确认 .env 文件中的数据库连接信息是否正确。
验证 PostgreSQL 数据库服务是否正常运行且可访问。
检查 DevBox 环境的网络连接状态。
确保 Cargo.toml 中的依赖配置完整且正确。

更多关于 Rust 操作 PostgreSQL 的详细信息，请参考 sqlx 官方文档。

Rust

本页导航