Fix issue with adding multiple instances of chatserver.

1 files changed, 6 insertions, 0 deletions

diff --git a/server/src/main.rs b/server/src/main.rs
index 3ac07233..636182aa 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -2,11 +2,13 @@ extern crate lemmy_server;
 #[macro_use]
 extern crate diesel_migrations;
 
+use actix::prelude::*;
 use actix_web::*;
 use diesel::r2d2::{ConnectionManager, Pool};
 use diesel::PgConnection;
 use lemmy_server::routes::{federation, feeds, index, nodeinfo, webfinger, websocket};
 use lemmy_server::settings::Settings;
+use lemmy_server::websocket::server::*;
 use std::io;
 
 embed_migrations!();
@@ -27,6 +29,9 @@ async fn main() -> io::Result<()> {
   let conn = pool.get().unwrap();
   embedded_migrations::run(&conn).unwrap();
 
+  // Set up websocket server
+  let server = ChatServer::startup(pool.clone()).start();
+
   println!(
     "Starting http server at {}:{}",
     settings.bind, settings.port
@@ -37,6 +42,7 @@ async fn main() -> io::Result<()> {
     App::new()
       .wrap(middleware::Logger::default())
       .data(pool.clone())
+      .data(server.clone())
       // The routes
       .configure(federation::config)
       .configure(feeds::config)