src/main.rs

   1 #[macro_use]
   2 extern crate diesel_migrations;
   3
   4 use actix::prelude::*;
   5 use actix_web::*;
   6 use diesel::{
   7   r2d2::{ConnectionManager, Pool},
   8   PgConnection,
   9 };
  10 use lemmy_api::match_websocket_operation;
  11 use lemmy_apub::activity_queue::create_activity_queue;
  12 use lemmy_db_queries::get_database_url_from_env;
  13 use lemmy_routes::{feeds, images, nodeinfo, webfinger};
  14 use lemmy_server::{code_migrations::run_advanced_migrations, scheduled_tasks};
  15 use lemmy_structs::blocking;
  16 use lemmy_utils::{
  17   rate_limit::{rate_limiter::RateLimiter, RateLimit},
  18   settings::Settings,
  19   LemmyError,
  20 };
  21 use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
  22 use reqwest::Client;
  23 use std::{sync::Arc, thread};
  24 use tokio::sync::Mutex;
  25
  26 embed_migrations!();
  27
  28 #[actix_web::main]
  29 async fn main() -> Result<(), LemmyError> {
  30   env_logger::init();
  31   let settings = Settings::get();
  32
  33   // Set up the r2d2 connection pool
  34   let db_url = match get_database_url_from_env() {
  35     Ok(url) => url,
  36     Err(_) => settings.get_database_url(),
  37   };
  38   let manager = ConnectionManager::<PgConnection>::new(&db_url);
  39   let pool = Pool::builder()
  40     .max_size(settings.database.pool_size)
  41     .build(manager)
  42     .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
  43
  44   // Run the migrations from code
  45   blocking(&pool, move |conn| {
  46     embedded_migrations::run(conn)?;
  47     run_advanced_migrations(conn)?;
  48     Ok(()) as Result<(), LemmyError>
  49   })
  50   .await??;
  51
  52   let pool2 = pool.clone();
  53   thread::spawn(move || {
  54     scheduled_tasks::setup(pool2);
  55   });
  56
  57   // Set up the rate limiter
  58   let rate_limiter = RateLimit {
  59     rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
  60   };
  61
  62   println!(
  63     "Starting http server at {}:{}",
  64     settings.bind, settings.port
  65   );
  66
  67   let activity_queue = create_activity_queue();
  68   let chat_server = ChatServer::startup(
  69     pool.clone(),
  70     rate_limiter.clone(),
  71     |c, i, o, d| Box::pin(match_websocket_operation(c, i, o, d)),
  72     Client::default(),
  73     activity_queue.clone(),
  74   )
  75   .start();
  76
  77   // Create Http server with websocket support
  78   HttpServer::new(move || {
  79     let context = LemmyContext::create(
  80       pool.clone(),
  81       chat_server.to_owned(),
  82       Client::default(),
  83       activity_queue.to_owned(),
  84     );
  85     let rate_limiter = rate_limiter.clone();
  86     App::new()
  87       .wrap(middleware::Logger::default())
  88       .data(context)
  89       // The routes
  90       .configure(|cfg| lemmy_api::routes::config(cfg, &rate_limiter))
  91       .configure(lemmy_apub::routes::config)
  92       .configure(feeds::config)
  93       .configure(|cfg| images::config(cfg, &rate_limiter))
  94       .configure(nodeinfo::config)
  95       .configure(webfinger::config)
  96   })
  97   .bind((settings.bind, settings.port))?
  98   .run()
  99   .await?;
 100
 101   Ok(())
 102 }