mirror of https://github.com/rwf2/Rocket.git
54 lines
1.4 KiB
Rust
54 lines
1.4 KiB
Rust
#![feature(proc_macro_hygiene)]
|
|
|
|
#[macro_use] extern crate rocket;
|
|
|
|
use rusqlite::types::ToSql;
|
|
|
|
#[cfg(test)] mod tests;
|
|
|
|
use std::sync::Mutex;
|
|
|
|
use rocket::{Rocket, State, response::Debug};
|
|
|
|
use rusqlite::{Connection, Error};
|
|
|
|
type DbConn = Mutex<Connection>;
|
|
|
|
fn init_database(conn: &Connection) {
|
|
conn.execute("CREATE TABLE entries (
|
|
id INTEGER PRIMARY KEY,
|
|
name TEXT NOT NULL
|
|
)", &[] as &[&dyn ToSql])
|
|
.expect("create entries table");
|
|
|
|
conn.execute("INSERT INTO entries (id, name) VALUES ($1, $2)",
|
|
&[&0 as &dyn ToSql, &"Rocketeer"])
|
|
.expect("insert single entry into entries table");
|
|
}
|
|
|
|
#[get("/")]
|
|
fn hello(db_conn: State<'_, DbConn>) -> Result<String, Debug<Error>> {
|
|
db_conn.lock()
|
|
.expect("db connection lock")
|
|
.query_row("SELECT name FROM entries WHERE id = 0",
|
|
&[] as &[&dyn ToSql], |row| { row.get(0) })
|
|
.map_err(Debug)
|
|
}
|
|
|
|
fn rocket() -> Rocket {
|
|
// Open a new in-memory SQLite database.
|
|
let conn = Connection::open_in_memory().expect("in memory db");
|
|
|
|
// Initialize the `entries` table in the in-memory database.
|
|
init_database(&conn);
|
|
|
|
// Have Rocket manage the database pool.
|
|
rocket::ignite()
|
|
.manage(Mutex::new(conn))
|
|
.mount("/", routes![hello])
|
|
}
|
|
|
|
fn main() {
|
|
let _ = rocket().launch();
|
|
}
|