added get_one, get_popular datasource methods, added respective routes
This commit is contained in:
@ -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),
|
||||
|
Reference in New Issue
Block a user