How to share SQLite connection? #964

ekanna · 2022-04-25T11:59:49Z

ekanna
Apr 25, 2022

How to share SQLite connection ?

mod utils;
mod jwt;
mod stipe;

use axum::{
    routing::{get, post},
    response::IntoResponse,
    Router,
    extract::{Extension}, Json,
};

use std::net::SocketAddr;

use rusqlite::{params, Connection};

struct Person {
    name: String,
    place: String,
}

#[tokio::main]
async fn main() {
    // initialize tracing
    tracing_subscriber::fmt::init();

    // Initialise Database
    let conn = Connection::open("kanna.db").unwrap();
    init_db(&conn).unwrap();

    let app = Router::new()
        .route("/", get(root))
        .route("/api/create", post(create))
// How to share connection?
        .layer(Extension(&conn));
        

    let addr = SocketAddr::from(([0, 0, 0, 0], 8080));
    println!("running at http://0.0.0.0:8080");
    tracing::info!("listening on {}", addr);
    axum::Server::bind(&addr).serve(app.into_make_service()).await.unwrap();
}


fn init_db(conn: &Connection) -> Result<(), rusqlite::Error> {
    dbg!("Initialising sqlite database");
    let sql_schema = include_str!("schema.sql");
    conn.execute(sql_schema, [])?;
    Ok(())
}

// This is not working
async fn create(Extension(conn): Extension<&Connection>, Json(p): Json<Person>) -> impl IntoResponse {
    conn.execute(
        "INSERT INTO profile (name, place) VALUES (?1, ?2)",
        params![&p.name, &p.place],
    );
}

// basic handler that responds with a static string
async fn root() -> impl IntoResponse {
    "Hello, World!"
}