summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorjonnnh <tou_tou@126.com>2019-07-19 11:05:17 +0800
committerjonnnh <tou_tou@126.com>2019-07-19 11:05:17 +0800
commitaaa0baf5b879e5a8be290bce76e57bc61808c936 (patch)
tree2a7e25edc6c35c9f92b6ea9d9021c839983d56b2
parentc4721f8a59d261fe808cab8b4d9291abad929a27 (diff)
upgrade
-rw-r--r--server/src/main.rs34
1 files changed, 20 insertions, 14 deletions
diff --git a/server/src/main.rs b/server/src/main.rs
index 75a48865..72bacf32 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -2,14 +2,14 @@ extern crate lemmy_server;
#[macro_use]
extern crate diesel_migrations;
-use std::time::{Instant, Duration};
-use std::env;
-use actix_web::*;
use actix::prelude::*;
use actix_files::NamedFile;
+use actix_web::*;
use actix_web_actors::ws;
-use lemmy_server::websocket::server::*;
use lemmy_server::db::establish_connection;
+use lemmy_server::websocket::server::*;
+use std::env;
+use std::time::{Duration, Instant};
embed_migrations!();
@@ -18,15 +18,19 @@ const HEARTBEAT_INTERVAL: Duration = Duration::from_secs(5);
/// How long before lack of client response causes a timeout
const CLIENT_TIMEOUT: Duration = Duration::from_secs(10);
-
/// Entry point for our route
-fn chat_route(req: HttpRequest, stream: web::Payload, chat_server: web::Data<Addr<ChatServer>>) -> Result<HttpResponse, Error> {
+fn chat_route(
+ req: HttpRequest,
+ stream: web::Payload,
+ chat_server: web::Data<Addr<ChatServer>>,
+) -> Result<HttpResponse, Error> {
ws::start(
WSSession {
cs_addr: chat_server.get_ref().to_owned(),
id: 0,
hb: Instant::now(),
- ip: req.connection_info()
+ ip: req
+ .connection_info()
.remote()
.unwrap_or("127.0.0.1:12345")
.split(":")
@@ -155,8 +159,10 @@ impl WSSession {
println!("Websocket Client heartbeat failed, disconnecting!");
// notify chat server
- act.cs_addr
- .do_send(Disconnect { id: act.id, ip: act.ip.to_owned() });
+ act.cs_addr.do_send(Disconnect {
+ id: act.id,
+ ip: act.ip.to_owned(),
+ });
// stop actor
ctx.stop();
@@ -182,17 +188,17 @@ fn main() {
let server = ChatServer::default().start();
// Create Http server with websocket support
HttpServer::new(move || {
-
App::new()
.data(server.clone())
.service(web::resource("/api/v1/ws").to(chat_route))
-// .service(web::resource("/api/v1/rest").route(web::post().to(||{})))
+ // .service(web::resource("/api/v1/rest").route(web::post().to(||{})))
.service(web::resource("/").to(index))
// static resources
.service(actix_files::Files::new("/static", front_end_dir()))
- }).bind("0.0.0.0:8536")
- .unwrap()
- .start();
+ })
+ .bind("0.0.0.0:8536")
+ .unwrap()
+ .start();
println!("Started http server: 0.0.0.0:8536");
let _ = sys.run();