要使用Axum通过共享状态的pgconnection,您需要按照以下步骤进行操作:
[dependencies]
axum = "0.2"
tokio = { version = "1", features = ["full"] }
postgres = "0.17"
use std::sync::Arc;
use axum::prelude::*;
use axum::AddExtensionLayer;
use tokio_postgres::{Client, NoTls};
#[tokio::main]
async fn main() {
// 创建数据库连接
let (client, connection) = tokio_postgres::connect("host=localhost user=postgres", NoTls)
.await
.expect("Failed to connect to database");
// 转换为Arc以进行共享
let client = Arc::new(client);
// 创建Axum应用程序
let app = route("/", get(handler)).layer(AddExtensionLayer::new(client));
// 启动服务器
axum::Server::bind(&"127.0.0.1:3000".parse().unwrap())
.serve(app.into_make_service())
.await
.unwrap();
}
async fn handler(
client: extract::Extension>,
) -> Result {
let client = client.0.clone();
// 执行数据库查询
let rows = client.query("SELECT * FROM users", &[]).await.unwrap();
// 处理查询结果
let mut result = String::new();
for row in &rows {
let id: i32 = row.get(0);
let name: &str = row.get(1);
result.push_str(&format!("id: {}, name: {}\n", id, name));
}
Ok(result)
}
在这个示例中,我们首先创建一个与PostgreSQL数据库的连接。然后,我们使用Arc
将连接转换为一个可共享的引用。接下来,我们创建了一个Axum应用程序,并使用AddExtensionLayer
将Arc
添加到应用程序的状态中。最后,我们定义了一个处理程序函数,该函数从状态中提取连接,并执行数据库查询。
请确保根据实际的数据库连接字符串进行更改。此示例仅适用于本地PostgreSQL数据库。
上一篇:Axum帖子的更高级生命周期错误