Adding DB conncetion to subscription

Signed-off-by: Anthony M. Bonafide <AnthonyMBonafide@gmail.com>

Cargo.toml

@@ -15,11 +15,20 @@ actix-web = "4"
 tokio = { version = "1", features = ["macros", "rt-multi-thread"] }
 serde = { version = "1", features = ["derive"] }
 config = "0.14"
+uuid = { version = "1", features = ["v4"] }
+chrono = { version = "0.4.22", default-features = false, features = ["clock"] }
 
 [dependencies.sqlx]
 verion = "0.7"
 default-features = false
-features = ["runtime-tokio-rustls", "macros", "postgres", "uuid","chrono","migrate"]
+features = [
+  "runtime-tokio-rustls",
+  "macros",
+  "postgres",
+  "uuid",
+  "chrono",
+  "migrate",
+]
 
 [dev-dependencies]
 reqwest = "0.12"

src/main.rs

@@ -1,11 +1,17 @@
 use std::net::TcpListener;
 
+use sqlx::{Connection, PgConnection};
 use zero2prod::{configuration::get_configuration, run};
 
 #[tokio::main]
 async fn main() -> Result<(), std::io::Error> {
     let config = get_configuration().expect("Failed to read configuration");
     let address = format!("127.0.0.1:{}", config.application_port);
+    let config = get_configuration().expect("Failed to get configuration");
+    let connection = PgConnection::connect(&config.database.connection_string())
+        .await
+        .expect("Failed to connect to database");
+
     let listener = TcpListener::bind(address)?;
-    run(listener)?.await
+    run(listener, connection)?.await
 }

src/routes/subscriptions.rs

@@ -1,11 +1,29 @@
 use actix_web::{web, HttpResponse};
+use chrono::Utc;
+use sqlx::PgConnection;
+use uuid::Uuid;
 
 #[derive(serde::Deserialize, serde::Serialize)]
 pub struct FormData {
     name: String,
     email: String,
 }
 
-pub async fn subscribe(_form: web::Form<FormData>) -> HttpResponse {
+pub async fn subscribe(
+    form: web::Form<FormData>,
+    connection: web::Data<PgConnection>,
+) -> HttpResponse {
+    sqlx::query!(
+        r#"
+    INSERT INTO subscriptions (id, email, name, subscribed_at)
+    VALUES ($1, $2, $3, $4)
+    "#,
+        Uuid::new_v4(),
+        form.email,
+        form.name,
+        Utc::now()
+    )
+    .execute(connection.get_ref())
+    .await;
     HttpResponse::Ok().finish()
 }

src/startup.rs

@@ -1,12 +1,15 @@
 use crate::routes::{health_check, subscribe};
 use actix_web::{dev::Server, web, App, HttpServer};
+use sqlx::PgConnection;
 use std::net::TcpListener;
 
-pub fn run(tcp_listener: TcpListener) -> Result<Server, std::io::Error> {
-    let server = HttpServer::new(|| {
+pub fn run(tcp_listener: TcpListener, connection: PgConnection) -> Result<Server, std::io::Error> {
+    let connection = web::Data::new(connection);
+    let server = HttpServer::new(move || {
         App::new()
             .route("/health_check", web::get().to(health_check))
             .route("/subscribe", web::post().to(subscribe))
+            .app_data(connection.clone())
     })
     .listen(tcp_listener)?
     .run();

tests/health_check.rs

@@ -79,7 +79,13 @@ async fn subscribe_returns_400_for_missing_form_data() {
 fn spawn_app() -> String {
     let listener = TcpListener::bind("127.0.0.1:0").expect("Failed to bind port");
     let port = listener.local_addr().unwrap().port();
-    let server = run(listener).expect("Failed to bind address");
+    let config = get_configuration().expect("Failed to get configuration");
+    let connection = PgConnection::connect(&config.database.connection_string())
+        .await
+        .expect("Failed to connect to database");
+
+    let server = run(listener, connection).expect("Failed to bind address");
+
     let _server_run = tokio::spawn(server);
     std::mem::drop(_server_run);
     format!("http://127.0.0.1:{}", port)