added get_one, get_popular datasource methods, added respective routes

This commit is contained in:
2024-09-26 13:11:55 -04:00
parent 9bd2cf373a
commit e3e81f6685
6 changed files with 125 additions and 32 deletions

View File

@ -1,10 +1,13 @@
use axum::Router;
use config::config;
use sqlx::{postgres::PgPoolOptions, PgPool};
use std::fs::File;
use std::sync::Arc;
use std::time::Duration;
use tokio::net::TcpListener;
use tokio::signal;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
use tower_http::trace::{self, TraceLayer};
use tracing_subscriber::{filter, layer::SubscriberExt, prelude::*, util::SubscriberInitExt};
mod config;
mod datasources;
@ -16,22 +19,57 @@ pub struct AppState {
#[tokio::main]
async fn main() {
// setting up configuration
let _ = config();
// setting up logging
let stdout_log = tracing_subscriber::fmt::layer().pretty();
let file = File::create("debug.log");
let file = match file {
Ok(file) => file,
Err(error) => panic!("Error: {:?}", error),
};
let debug_log = tracing_subscriber::fmt::layer().with_writer(Arc::new(file));
let metrics_layer = filter::LevelFilter::INFO;
tracing_subscriber::registry()
.with(
tracing_subscriber::EnvFilter::try_from_default_env().unwrap_or_else(|_| {
format!(
"{}=debug,tower_http=debug,axum=trace",
env!("CARGO_CRATE_NAME")
)
.into()
}),
stdout_log
// Add an `INFO` filter to the stdout logging layer
.with_filter(filter::LevelFilter::INFO)
// Combine the filtered `stdout_log` layer with the
// `debug_log` layer, producing a new `Layered` layer.
.and_then(debug_log)
// Add a filter to *both* layers that rejects spans and
// events whose targets start with `metrics`.
.with_filter(filter::filter_fn(|metadata| {
!metadata.target().starts_with("metrics")
})),
)
.with(
// Add a filter to the metrics label that *only* enables
// events whose targets start with `metrics`.
metrics_layer.with_filter(filter::filter_fn(|metadata| {
metadata.target().starts_with("metrics")
})),
)
.with(tracing_subscriber::fmt::layer())
.init();
let db_connection_str = std::env::var("DATABASE_URL").unwrap();
// .unwrap_or_else(|_| "postgres://postgres:password@localhost".to_string());
// tracing_subscriber::registry()
// .with(
// tracing_subscriber::EnvFilter::try_from_default_env().unwrap_or_else(|_| {
// format!(
// "{}=debug,tower_http=debug,axum=trace",
// env!("CARGO_CRATE_NAME")
// )
// .into()
// }),
// )
// .with(tracing_subscriber::fmt::layer())
// .init();
// grabbing the database url from our env variables
let db_connection_str = std::env::var("DATABASE_URL")
.unwrap_or_else(|_| "postgres://postgres:password@localhost".to_string());
// set up connection pool
let pool = PgPoolOptions::new()
@ -46,7 +84,12 @@ async fn main() {
// build our application with some routes
let app = Router::new()
.nest("/", routes::root::RootRoute::routes())
.nest("/posts", routes::posts::PostsRoute::routes(&app_state));
.nest("/posts", routes::posts::PostsRoute::routes(&app_state))
.layer(
TraceLayer::new_for_http()
.make_span_with(trace::DefaultMakeSpan::new().level(tracing::Level::INFO))
.on_response(trace::DefaultOnResponse::new().level(tracing::Level::INFO)),
);
// .nest(
// "/comments",
// routes::comments::CommentsRoute::routes(&app_state),