mirror of
https://github.com/actix/examples
synced 2024-11-23 22:41:07 +01:00
actix examples in actix release version
This commit is contained in:
parent
ad58e559a0
commit
3ebde8e7c2
4
.gitignore
vendored
4
.gitignore
vendored
@ -1,10 +1,10 @@
|
|||||||
# Generated by Cargo
|
# Generated by Cargo
|
||||||
# will have compiled files and executables
|
# will have compiled files and executables
|
||||||
/target/
|
/*/target/
|
||||||
|
|
||||||
# Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
|
# Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
|
||||||
# More information here http://doc.crates.io/guide.html#cargotoml-vs-cargolock
|
# More information here http://doc.crates.io/guide.html#cargotoml-vs-cargolock
|
||||||
Cargo.lock
|
/*/Cargo.lock
|
||||||
|
|
||||||
# These are backup files generated by rustfmt
|
# These are backup files generated by rustfmt
|
||||||
**/*.rs.bk
|
**/*.rs.bk
|
||||||
|
10
basics/Cargo.toml
Normal file
10
basics/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "basics"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
futures = "0.1"
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
20
basics/README.md
Normal file
20
basics/README.md
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
# basics
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/basics
|
||||||
|
cargo run
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
- [http://localhost:8080/index.html](http://localhost:8080/index.html)
|
||||||
|
- [http://localhost:8080/async/bob](http://localhost:8080/async/bob)
|
||||||
|
- [http://localhost:8080/user/bob/](http://localhost:8080/user/bob/) plain/text download
|
||||||
|
- [http://localhost:8080/test](http://localhost:8080/test) (return status switch GET or POST or other)
|
||||||
|
- [http://localhost:8080/static/index.html](http://localhost:8080/static/index.html)
|
||||||
|
- [http://localhost:8080/static/notexit](http://localhost:8080/static/notexit) display 404 page
|
136
basics/src/main.rs
Normal file
136
basics/src/main.rs
Normal file
@ -0,0 +1,136 @@
|
|||||||
|
#![allow(unused_variables)]
|
||||||
|
#![cfg_attr(feature="cargo-clippy", allow(needless_pass_by_value))]
|
||||||
|
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate futures;
|
||||||
|
use futures::Stream;
|
||||||
|
|
||||||
|
use std::{io, env};
|
||||||
|
use actix_web::{error, fs, pred, server,
|
||||||
|
App, HttpRequest, HttpResponse, Result, Error};
|
||||||
|
use actix_web::http::{header, Method, StatusCode};
|
||||||
|
use actix_web::middleware::{self, RequestSession};
|
||||||
|
use futures::future::{FutureResult, result};
|
||||||
|
|
||||||
|
/// favicon handler
|
||||||
|
fn favicon(req: HttpRequest) -> Result<fs::NamedFile> {
|
||||||
|
Ok(fs::NamedFile::open("../static/favicon.ico")?)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// simple index handler
|
||||||
|
fn index(mut req: HttpRequest) -> Result<HttpResponse> {
|
||||||
|
println!("{:?}", req);
|
||||||
|
|
||||||
|
// example of ...
|
||||||
|
if let Ok(ch) = req.poll() {
|
||||||
|
if let futures::Async::Ready(Some(d)) = ch {
|
||||||
|
println!("{}", String::from_utf8_lossy(d.as_ref()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// session
|
||||||
|
let mut counter = 1;
|
||||||
|
if let Some(count) = req.session().get::<i32>("counter")? {
|
||||||
|
println!("SESSION value: {}", count);
|
||||||
|
counter = count + 1;
|
||||||
|
req.session().set("counter", counter)?;
|
||||||
|
} else {
|
||||||
|
req.session().set("counter", counter)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// response
|
||||||
|
Ok(HttpResponse::build(StatusCode::OK)
|
||||||
|
.content_type("text/html; charset=utf-8")
|
||||||
|
.body(include_str!("../static/welcome.html")))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 404 handler
|
||||||
|
fn p404(req: HttpRequest) -> Result<fs::NamedFile> {
|
||||||
|
Ok(fs::NamedFile::open("./static/404.html")?
|
||||||
|
.set_status_code(StatusCode::NOT_FOUND))
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// async handler
|
||||||
|
fn index_async(req: HttpRequest) -> FutureResult<HttpResponse, Error>
|
||||||
|
{
|
||||||
|
println!("{:?}", req);
|
||||||
|
|
||||||
|
result(Ok(HttpResponse::Ok()
|
||||||
|
.content_type("text/html")
|
||||||
|
.body(format!("Hello {}!", req.match_info().get("name").unwrap()))))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// handler with path parameters like `/user/{name}/`
|
||||||
|
fn with_param(req: HttpRequest) -> HttpResponse
|
||||||
|
{
|
||||||
|
println!("{:?}", req);
|
||||||
|
|
||||||
|
HttpResponse::Ok()
|
||||||
|
.content_type("test/plain")
|
||||||
|
.body(format!("Hello {}!", req.match_info().get("name").unwrap()))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
env::set_var("RUST_LOG", "actix_web=debug");
|
||||||
|
env::set_var("RUST_BACKTRACE", "1");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("basic-example");
|
||||||
|
|
||||||
|
let addr = server::new(
|
||||||
|
|| App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
// cookie session middleware
|
||||||
|
.middleware(middleware::SessionStorage::new(
|
||||||
|
middleware::CookieSessionBackend::signed(&[0; 32]).secure(false)
|
||||||
|
))
|
||||||
|
// register favicon
|
||||||
|
.resource("/favicon.ico", |r| r.f(favicon))
|
||||||
|
// register simple route, handle all methods
|
||||||
|
.resource("/index.html", |r| r.f(index))
|
||||||
|
// with path parameters
|
||||||
|
.resource("/user/{name}/", |r| r.method(Method::GET).f(with_param))
|
||||||
|
// async handler
|
||||||
|
.resource("/async/{name}", |r| r.method(Method::GET).a(index_async))
|
||||||
|
.resource("/test", |r| r.f(|req| {
|
||||||
|
match *req.method() {
|
||||||
|
Method::GET => HttpResponse::Ok(),
|
||||||
|
Method::POST => HttpResponse::MethodNotAllowed(),
|
||||||
|
_ => HttpResponse::NotFound(),
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
.resource("/error.html", |r| r.f(|req| {
|
||||||
|
error::InternalError::new(
|
||||||
|
io::Error::new(io::ErrorKind::Other, "test"), StatusCode::OK)
|
||||||
|
}))
|
||||||
|
// static files
|
||||||
|
.handler("/static/", fs::StaticFiles::new("../static/"))
|
||||||
|
// redirect
|
||||||
|
.resource("/", |r| r.method(Method::GET).f(|req| {
|
||||||
|
println!("{:?}", req);
|
||||||
|
HttpResponse::Found()
|
||||||
|
.header(header::LOCATION, "/index.html")
|
||||||
|
.finish()
|
||||||
|
}))
|
||||||
|
// default
|
||||||
|
.default_resource(|r| {
|
||||||
|
// 404 for GET request
|
||||||
|
r.method(Method::GET).f(p404);
|
||||||
|
|
||||||
|
// all requests that are not `GET`
|
||||||
|
r.route().filter(pred::Not(pred::Get())).f(
|
||||||
|
|req| HttpResponse::MethodNotAllowed());
|
||||||
|
}))
|
||||||
|
|
||||||
|
.bind("127.0.0.1:8080").expect("Can not bind to 127.0.0.1:8080")
|
||||||
|
.shutdown_timeout(0) // <- Set shutdown timeout to 0 seconds (default 60s)
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Starting http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
7
basics/static/404.html
Normal file
7
basics/static/404.html
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
<!DOCTYPE html><html><head><title>actix - basics</title>
|
||||||
|
<link rel="shortcut icon" type="image/x-icon" href="/favicon.ico" /></head>
|
||||||
|
<body>
|
||||||
|
<a href="index.html">back to home</a>
|
||||||
|
<h1>404</h1>
|
||||||
|
</body>
|
||||||
|
</html>
|
6
basics/static/welcome.html
Normal file
6
basics/static/welcome.html
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
<!DOCTYPE html><html><head><title>actix - basics</title>
|
||||||
|
<link rel="shortcut icon" type="image/x-icon" href="/favicon.ico" /></head>
|
||||||
|
<body>
|
||||||
|
<h1>Welcome <img width="30px" height="30px" src="/static/actixLogo.png" /></h1>
|
||||||
|
</body>
|
||||||
|
</html>
|
1
diesel/.env
Normal file
1
diesel/.env
Normal file
@ -0,0 +1 @@
|
|||||||
|
DATABASE_URL=file:test.db
|
19
diesel/Cargo.toml
Normal file
19
diesel/Cargo.toml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
[package]
|
||||||
|
name = "diesel-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
|
||||||
|
futures = "0.1"
|
||||||
|
uuid = { version = "0.5", features = ["serde", "v4"] }
|
||||||
|
serde = "1.0"
|
||||||
|
serde_json = "1.0"
|
||||||
|
serde_derive = "1.0"
|
||||||
|
|
||||||
|
diesel = { version = "^1.1.0", features = ["sqlite", "r2d2"] }
|
||||||
|
r2d2 = "0.8"
|
||||||
|
dotenv = "0.10"
|
43
diesel/README.md
Normal file
43
diesel/README.md
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
# diesel
|
||||||
|
|
||||||
|
Diesel's `Getting Started` guide using SQLite for Actix web
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### init database sqlite
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cargo install diesel_cli --no-default-features --features sqlite
|
||||||
|
cd actix-web/examples/diesel
|
||||||
|
echo "DATABASE_URL=file:test.db" > .env
|
||||||
|
diesel migration run
|
||||||
|
```
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# if ubuntu : sudo apt-get install libsqlite3-dev
|
||||||
|
# if fedora : sudo dnf install libsqlite3x-devel
|
||||||
|
cd actix-web/examples/diesel
|
||||||
|
cargo run (or ``cargo watch -x run``)
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
[http://127.0.0.1:8080/NAME](http://127.0.0.1:8080/NAME)
|
||||||
|
|
||||||
|
### sqlite client
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# if ubuntu : sudo apt-get install sqlite3
|
||||||
|
# if fedora : sudo dnf install sqlite3x
|
||||||
|
sqlite3 test.db
|
||||||
|
sqlite> .tables
|
||||||
|
sqlite> select * from users;
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
## Postgresql
|
||||||
|
|
||||||
|
You will also find another complete example of diesel+postgresql on [https://github.com/TechEmpower/FrameworkBenchmarks/tree/master/frameworks/Rust/actix](https://github.com/TechEmpower/FrameworkBenchmarks/tree/master/frameworks/Rust/actix)
|
1
diesel/migrations/20170124012402_create_users/down.sql
Normal file
1
diesel/migrations/20170124012402_create_users/down.sql
Normal file
@ -0,0 +1 @@
|
|||||||
|
DROP TABLE users
|
4
diesel/migrations/20170124012402_create_users/up.sql
Normal file
4
diesel/migrations/20170124012402_create_users/up.sql
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
CREATE TABLE users (
|
||||||
|
id VARCHAR NOT NULL PRIMARY KEY,
|
||||||
|
name VARCHAR NOT NULL
|
||||||
|
)
|
55
diesel/src/db.rs
Normal file
55
diesel/src/db.rs
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
//! Db executor actor
|
||||||
|
use uuid;
|
||||||
|
use diesel;
|
||||||
|
use actix_web::*;
|
||||||
|
use actix::prelude::*;
|
||||||
|
use diesel::prelude::*;
|
||||||
|
use diesel::r2d2::{Pool, ConnectionManager};
|
||||||
|
|
||||||
|
use models;
|
||||||
|
use schema;
|
||||||
|
|
||||||
|
/// This is db executor actor. We are going to run 3 of them in parallel.
|
||||||
|
pub struct DbExecutor(pub Pool<ConnectionManager<SqliteConnection>>);
|
||||||
|
|
||||||
|
/// This is only message that this actor can handle, but it is easy to extend number of
|
||||||
|
/// messages.
|
||||||
|
pub struct CreateUser {
|
||||||
|
pub name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Message for CreateUser {
|
||||||
|
type Result = Result<models::User, Error>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Actor for DbExecutor {
|
||||||
|
type Context = SyncContext<Self>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Handler<CreateUser> for DbExecutor {
|
||||||
|
type Result = Result<models::User, Error>;
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: CreateUser, _: &mut Self::Context) -> Self::Result {
|
||||||
|
use self::schema::users::dsl::*;
|
||||||
|
|
||||||
|
let uuid = format!("{}", uuid::Uuid::new_v4());
|
||||||
|
let new_user = models::NewUser {
|
||||||
|
id: &uuid,
|
||||||
|
name: &msg.name,
|
||||||
|
};
|
||||||
|
|
||||||
|
let conn: &SqliteConnection = &self.0.get().unwrap();
|
||||||
|
|
||||||
|
diesel::insert_into(users)
|
||||||
|
.values(&new_user)
|
||||||
|
.execute(conn)
|
||||||
|
.expect("Error inserting person");
|
||||||
|
|
||||||
|
let mut items = users
|
||||||
|
.filter(id.eq(&uuid))
|
||||||
|
.load::<models::User>(conn)
|
||||||
|
.expect("Error loading person");
|
||||||
|
|
||||||
|
Ok(items.pop().unwrap())
|
||||||
|
}
|
||||||
|
}
|
78
diesel/src/main.rs
Normal file
78
diesel/src/main.rs
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
//! Actix web diesel example
|
||||||
|
//!
|
||||||
|
//! Diesel does not support tokio, so we have to run it in separate threads.
|
||||||
|
//! Actix supports sync actors by default, so we going to create sync actor that use diesel.
|
||||||
|
//! Technically sync actors are worker style actors, multiple of them
|
||||||
|
//! can run in parallel and process messages from same queue.
|
||||||
|
extern crate serde;
|
||||||
|
extern crate serde_json;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate serde_derive;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate diesel;
|
||||||
|
extern crate r2d2;
|
||||||
|
extern crate uuid;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use actix::prelude::*;
|
||||||
|
use actix_web::{http, server, middleware,
|
||||||
|
App, Path, State, HttpResponse, AsyncResponder, FutureResponse};
|
||||||
|
|
||||||
|
use diesel::prelude::*;
|
||||||
|
use diesel::r2d2::{ Pool, ConnectionManager };
|
||||||
|
use futures::future::Future;
|
||||||
|
|
||||||
|
mod db;
|
||||||
|
mod models;
|
||||||
|
mod schema;
|
||||||
|
|
||||||
|
use db::{CreateUser, DbExecutor};
|
||||||
|
|
||||||
|
|
||||||
|
/// State with DbExecutor address
|
||||||
|
struct AppState {
|
||||||
|
db: Addr<Syn, DbExecutor>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Async request handler
|
||||||
|
fn index(name: Path<String>, state: State<AppState>) -> FutureResponse<HttpResponse> {
|
||||||
|
// send async `CreateUser` message to a `DbExecutor`
|
||||||
|
state.db.send(CreateUser{name: name.into_inner()})
|
||||||
|
.from_err()
|
||||||
|
.and_then(|res| {
|
||||||
|
match res {
|
||||||
|
Ok(user) => Ok(HttpResponse::Ok().json(user)),
|
||||||
|
Err(_) => Ok(HttpResponse::InternalServerError().into())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("diesel-example");
|
||||||
|
|
||||||
|
// Start 3 db executor actors
|
||||||
|
let manager = ConnectionManager::<SqliteConnection>::new("test.db");
|
||||||
|
let pool = r2d2::Pool::builder().build(manager).expect("Failed to create pool.");
|
||||||
|
|
||||||
|
let addr = SyncArbiter::start(3, move || {
|
||||||
|
DbExecutor(pool.clone())
|
||||||
|
});
|
||||||
|
|
||||||
|
// Start http server
|
||||||
|
server::new(move || {
|
||||||
|
App::with_state(AppState{db: addr.clone()})
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/{name}", |r| r.method(http::Method::GET).with2(index))})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
14
diesel/src/models.rs
Normal file
14
diesel/src/models.rs
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
use super::schema::users;
|
||||||
|
|
||||||
|
#[derive(Serialize, Queryable)]
|
||||||
|
pub struct User {
|
||||||
|
pub id: String,
|
||||||
|
pub name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Insertable)]
|
||||||
|
#[table_name = "users"]
|
||||||
|
pub struct NewUser<'a> {
|
||||||
|
pub id: &'a str,
|
||||||
|
pub name: &'a str,
|
||||||
|
}
|
6
diesel/src/schema.rs
Normal file
6
diesel/src/schema.rs
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
table! {
|
||||||
|
users (id) {
|
||||||
|
id -> Text,
|
||||||
|
name -> Text,
|
||||||
|
}
|
||||||
|
}
|
BIN
diesel/test.db
Normal file
BIN
diesel/test.db
Normal file
Binary file not shown.
9
hello-world/Cargo.toml
Normal file
9
hello-world/Cargo.toml
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
[package]
|
||||||
|
name = "hello-world"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
28
hello-world/src/main.rs
Normal file
28
hello-world/src/main.rs
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use actix_web::{App, HttpRequest, server, middleware};
|
||||||
|
|
||||||
|
|
||||||
|
fn index(_req: HttpRequest) -> &'static str {
|
||||||
|
"Hello world!"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("hello-world");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/index.html", |r| r.f(|_| "Hello world!"))
|
||||||
|
.resource("/", |r| r.f(index)))
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
10
http-proxy/Cargo.toml
Normal file
10
http-proxy/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "http-proxy"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
futures = "0.1"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = { version = "^0.5", features=["alpn"] }
|
58
http-proxy/src/main.rs
Normal file
58
http-proxy/src/main.rs
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use futures::{Future, Stream};
|
||||||
|
use actix_web::{
|
||||||
|
client, server, middleware,
|
||||||
|
App, AsyncResponder, Body, HttpRequest, HttpResponse, HttpMessage, Error};
|
||||||
|
|
||||||
|
|
||||||
|
/// Stream client request response and then send body to a server response
|
||||||
|
fn index(_req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
client::ClientRequest::get("https://www.rust-lang.org/en-US/")
|
||||||
|
.finish().unwrap()
|
||||||
|
.send()
|
||||||
|
.map_err(Error::from) // <- convert SendRequestError to an Error
|
||||||
|
.and_then(
|
||||||
|
|resp| resp.body() // <- this is MessageBody type, resolves to complete body
|
||||||
|
.from_err() // <- convert PayloadError to a Error
|
||||||
|
.and_then(|body| { // <- we got complete body, now send as server response
|
||||||
|
Ok(HttpResponse::Ok().body(body))
|
||||||
|
}))
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// streaming client request to a streaming server response
|
||||||
|
fn streaming(_req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
// send client request
|
||||||
|
client::ClientRequest::get("https://www.rust-lang.org/en-US/")
|
||||||
|
.finish().unwrap()
|
||||||
|
.send() // <- connect to host and send request
|
||||||
|
.map_err(Error::from) // <- convert SendRequestError to an Error
|
||||||
|
.and_then(|resp| { // <- we received client response
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
// read one chunk from client response and send this chunk to a server response
|
||||||
|
// .from_err() converts PayloadError to a Error
|
||||||
|
.body(Body::Streaming(Box::new(resp.from_err()))))
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("http-proxy");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/streaming", |r| r.f(streaming))
|
||||||
|
.resource("/", |r| r.f(index)))
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
17
json/Cargo.toml
Normal file
17
json/Cargo.toml
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
[package]
|
||||||
|
name = "json-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
bytes = "0.4"
|
||||||
|
futures = "0.1"
|
||||||
|
env_logger = "*"
|
||||||
|
|
||||||
|
serde = "1.0"
|
||||||
|
serde_json = "1.0"
|
||||||
|
serde_derive = "1.0"
|
||||||
|
json = "*"
|
||||||
|
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
48
json/README.md
Normal file
48
json/README.md
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
# json
|
||||||
|
|
||||||
|
Json's `Getting Started` guide using json (serde-json or json-rust) for Actix web
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/json
|
||||||
|
cargo run
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
With [Postman](https://www.getpostman.com/) or [Rested](moz-extension://60daeb1c-5b1b-4afd-9842-0579ed34dfcb/dist/index.html)
|
||||||
|
|
||||||
|
- POST / (embed serde-json):
|
||||||
|
|
||||||
|
- method : ``POST``
|
||||||
|
- url : ``http://127.0.0.1:8080/``
|
||||||
|
- header : ``Content-Type`` = ``application/json``
|
||||||
|
- body (raw) : ``{"name": "Test user", "number": 100}``
|
||||||
|
|
||||||
|
- POST /manual (manual serde-json):
|
||||||
|
|
||||||
|
- method : ``POST``
|
||||||
|
- url : ``http://127.0.0.1:8080/manual``
|
||||||
|
- header : ``Content-Type`` = ``application/json``
|
||||||
|
- body (raw) : ``{"name": "Test user", "number": 100}``
|
||||||
|
|
||||||
|
- POST /mjsonrust (manual json-rust):
|
||||||
|
|
||||||
|
- method : ``POST``
|
||||||
|
- url : ``http://127.0.0.1:8080/mjsonrust``
|
||||||
|
- header : ``Content-Type`` = ``application/json``
|
||||||
|
- body (raw) : ``{"name": "Test user", "number": 100}`` (you can also test ``{notjson}``)
|
||||||
|
|
||||||
|
### python client
|
||||||
|
|
||||||
|
- ``pip install aiohttp``
|
||||||
|
- ``python client.py``
|
||||||
|
|
||||||
|
if ubuntu :
|
||||||
|
|
||||||
|
- ``pip3 install aiohttp``
|
||||||
|
- ``python3 client.py``
|
18
json/client.py
Normal file
18
json/client.py
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
# This script could be used for actix-web multipart example test
|
||||||
|
# just start server and run client.py
|
||||||
|
|
||||||
|
import json
|
||||||
|
import asyncio
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
|
async def req():
|
||||||
|
resp = await aiohttp.ClientSession().request(
|
||||||
|
"post", 'http://localhost:8080/',
|
||||||
|
data=json.dumps({"name": "Test user", "number": 100}),
|
||||||
|
headers={"content-type": "application/json"})
|
||||||
|
print(str(resp))
|
||||||
|
print(await resp.text())
|
||||||
|
assert 200 == resp.status
|
||||||
|
|
||||||
|
|
||||||
|
asyncio.get_event_loop().run_until_complete(req())
|
110
json/src/main.rs
Normal file
110
json/src/main.rs
Normal file
@ -0,0 +1,110 @@
|
|||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate bytes;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate serde_json;
|
||||||
|
#[macro_use] extern crate serde_derive;
|
||||||
|
#[macro_use] extern crate json;
|
||||||
|
|
||||||
|
use actix_web::{
|
||||||
|
middleware, http, error, server,
|
||||||
|
App, AsyncResponder, HttpRequest, HttpResponse, HttpMessage, Error, Json};
|
||||||
|
|
||||||
|
use bytes::BytesMut;
|
||||||
|
use futures::{Future, Stream};
|
||||||
|
use json::JsonValue;
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
|
struct MyObj {
|
||||||
|
name: String,
|
||||||
|
number: i32,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This handler uses `HttpRequest::json()` for loading json object.
|
||||||
|
fn index(req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
req.json()
|
||||||
|
.from_err() // convert all errors into `Error`
|
||||||
|
.and_then(|val: MyObj| {
|
||||||
|
println!("model: {:?}", val);
|
||||||
|
Ok(HttpResponse::Ok().json(val)) // <- send response
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This handler uses json extractor
|
||||||
|
fn extract_item(item: Json<MyObj>) -> HttpResponse {
|
||||||
|
println!("model: {:?}", &item);
|
||||||
|
HttpResponse::Ok().json(item.0) // <- send response
|
||||||
|
}
|
||||||
|
|
||||||
|
const MAX_SIZE: usize = 262_144; // max payload size is 256k
|
||||||
|
|
||||||
|
/// This handler manually load request payload and parse json object
|
||||||
|
fn index_manual(req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
// HttpRequest is stream of Bytes objects
|
||||||
|
req
|
||||||
|
// `Future::from_err` acts like `?` in that it coerces the error type from
|
||||||
|
// the future into the final error type
|
||||||
|
.from_err()
|
||||||
|
|
||||||
|
// `fold` will asynchronously read each chunk of the request body and
|
||||||
|
// call supplied closure, then it resolves to result of closure
|
||||||
|
.fold(BytesMut::new(), move |mut body, chunk| {
|
||||||
|
// limit max size of in-memory payload
|
||||||
|
if (body.len() + chunk.len()) > MAX_SIZE {
|
||||||
|
Err(error::ErrorBadRequest("overflow"))
|
||||||
|
} else {
|
||||||
|
body.extend_from_slice(&chunk);
|
||||||
|
Ok(body)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
// `Future::and_then` can be used to merge an asynchronous workflow with a
|
||||||
|
// synchronous workflow
|
||||||
|
.and_then(|body| {
|
||||||
|
// body is loaded, now we can deserialize serde-json
|
||||||
|
let obj = serde_json::from_slice::<MyObj>(&body)?;
|
||||||
|
Ok(HttpResponse::Ok().json(obj)) // <- send response
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This handler manually load request payload and parse json-rust
|
||||||
|
fn index_mjsonrust(req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
req.concat2()
|
||||||
|
.from_err()
|
||||||
|
.and_then(|body| {
|
||||||
|
// body is loaded, now we can deserialize json-rust
|
||||||
|
let result = json::parse(std::str::from_utf8(&body).unwrap()); // return Result
|
||||||
|
let injson: JsonValue = match result { Ok(v) => v, Err(e) => object!{"err" => e.to_string() } };
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
.content_type("application/json")
|
||||||
|
.body(injson.dump()))
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("json-example");
|
||||||
|
|
||||||
|
server::new(|| {
|
||||||
|
App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/extractor", |r| {
|
||||||
|
r.method(http::Method::POST)
|
||||||
|
.with(extract_item)
|
||||||
|
.limit(4096); // <- limit size of the payload
|
||||||
|
})
|
||||||
|
.resource("/manual", |r| r.method(http::Method::POST).f(index_manual))
|
||||||
|
.resource("/mjsonrust", |r| r.method(http::Method::POST).f(index_mjsonrust))
|
||||||
|
.resource("/", |r| r.method(http::Method::POST).f(index))})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.shutdown_timeout(1)
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
16
juniper/Cargo.toml
Normal file
16
juniper/Cargo.toml
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
[package]
|
||||||
|
name = "juniper-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["pyros2097 <pyros2097@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
|
||||||
|
futures = "0.1"
|
||||||
|
serde = "1.0"
|
||||||
|
serde_json = "1.0"
|
||||||
|
serde_derive = "1.0"
|
||||||
|
|
||||||
|
juniper = "0.9.2"
|
15
juniper/README.md
Normal file
15
juniper/README.md
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
# juniper
|
||||||
|
|
||||||
|
Juniper integration for Actix web
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/juniper
|
||||||
|
cargo run (or ``cargo watch -x run``)
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
[http://127.0.0.1:8080/graphiql](http://127.0.0.1:8080/graphiql)
|
108
juniper/src/main.rs
Normal file
108
juniper/src/main.rs
Normal file
@ -0,0 +1,108 @@
|
|||||||
|
//! Actix web juniper example
|
||||||
|
//!
|
||||||
|
//! A simple example integrating juniper in actix-web
|
||||||
|
extern crate serde;
|
||||||
|
extern crate serde_json;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate serde_derive;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate juniper;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use actix::prelude::*;
|
||||||
|
use actix_web::{
|
||||||
|
middleware, http, server,
|
||||||
|
App, AsyncResponder, HttpRequest, HttpResponse, FutureResponse, Error, State, Json};
|
||||||
|
use juniper::http::graphiql::graphiql_source;
|
||||||
|
use juniper::http::GraphQLRequest;
|
||||||
|
use futures::future::Future;
|
||||||
|
|
||||||
|
mod schema;
|
||||||
|
|
||||||
|
use schema::Schema;
|
||||||
|
use schema::create_schema;
|
||||||
|
|
||||||
|
struct AppState {
|
||||||
|
executor: Addr<Syn, GraphQLExecutor>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize, Deserialize)]
|
||||||
|
pub struct GraphQLData(GraphQLRequest);
|
||||||
|
|
||||||
|
impl Message for GraphQLData {
|
||||||
|
type Result = Result<String, Error>;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct GraphQLExecutor {
|
||||||
|
schema: std::sync::Arc<Schema>
|
||||||
|
}
|
||||||
|
|
||||||
|
impl GraphQLExecutor {
|
||||||
|
fn new(schema: std::sync::Arc<Schema>) -> GraphQLExecutor {
|
||||||
|
GraphQLExecutor {
|
||||||
|
schema: schema,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Actor for GraphQLExecutor {
|
||||||
|
type Context = SyncContext<Self>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Handler<GraphQLData> for GraphQLExecutor {
|
||||||
|
type Result = Result<String, Error>;
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: GraphQLData, _: &mut Self::Context) -> Self::Result {
|
||||||
|
let res = msg.0.execute(&self.schema, &());
|
||||||
|
let res_text = serde_json::to_string(&res)?;
|
||||||
|
Ok(res_text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn graphiql(_req: HttpRequest<AppState>) -> Result<HttpResponse, Error> {
|
||||||
|
let html = graphiql_source("http://127.0.0.1:8080/graphql");
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
.content_type("text/html; charset=utf-8")
|
||||||
|
.body(html))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn graphql(st: State<AppState>, data: Json<GraphQLData>) -> FutureResponse<HttpResponse> {
|
||||||
|
st.executor.send(data.0)
|
||||||
|
.from_err()
|
||||||
|
.and_then(|res| {
|
||||||
|
match res {
|
||||||
|
Ok(user) => Ok(HttpResponse::Ok()
|
||||||
|
.content_type("application/json")
|
||||||
|
.body(user)),
|
||||||
|
Err(_) => Ok(HttpResponse::InternalServerError().into())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("juniper-example");
|
||||||
|
|
||||||
|
let schema = std::sync::Arc::new(create_schema());
|
||||||
|
let addr = SyncArbiter::start(3, move || {
|
||||||
|
GraphQLExecutor::new(schema.clone())
|
||||||
|
});
|
||||||
|
|
||||||
|
// Start http server
|
||||||
|
server::new(move || {
|
||||||
|
App::with_state(AppState{executor: addr.clone()})
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/graphql", |r| r.method(http::Method::POST).with2(graphql))
|
||||||
|
.resource("/graphiql", |r| r.method(http::Method::GET).h(graphiql))})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
58
juniper/src/schema.rs
Normal file
58
juniper/src/schema.rs
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
use juniper::FieldResult;
|
||||||
|
use juniper::RootNode;
|
||||||
|
|
||||||
|
#[derive(GraphQLEnum)]
|
||||||
|
enum Episode {
|
||||||
|
NewHope,
|
||||||
|
Empire,
|
||||||
|
Jedi,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(GraphQLObject)]
|
||||||
|
#[graphql(description = "A humanoid creature in the Star Wars universe")]
|
||||||
|
struct Human {
|
||||||
|
id: String,
|
||||||
|
name: String,
|
||||||
|
appears_in: Vec<Episode>,
|
||||||
|
home_planet: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(GraphQLInputObject)]
|
||||||
|
#[graphql(description = "A humanoid creature in the Star Wars universe")]
|
||||||
|
struct NewHuman {
|
||||||
|
name: String,
|
||||||
|
appears_in: Vec<Episode>,
|
||||||
|
home_planet: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct QueryRoot;
|
||||||
|
|
||||||
|
graphql_object!(QueryRoot: () |&self| {
|
||||||
|
field human(&executor, id: String) -> FieldResult<Human> {
|
||||||
|
Ok(Human{
|
||||||
|
id: "1234".to_owned(),
|
||||||
|
name: "Luke".to_owned(),
|
||||||
|
appears_in: vec![Episode::NewHope],
|
||||||
|
home_planet: "Mars".to_owned(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
pub struct MutationRoot;
|
||||||
|
|
||||||
|
graphql_object!(MutationRoot: () |&self| {
|
||||||
|
field createHuman(&executor, new_human: NewHuman) -> FieldResult<Human> {
|
||||||
|
Ok(Human{
|
||||||
|
id: "1234".to_owned(),
|
||||||
|
name: new_human.name,
|
||||||
|
appears_in: new_human.appears_in,
|
||||||
|
home_planet: new_human.home_planet,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
pub type Schema = RootNode<'static, QueryRoot, MutationRoot>;
|
||||||
|
|
||||||
|
pub fn create_schema() -> Schema {
|
||||||
|
Schema::new(QueryRoot {}, MutationRoot {})
|
||||||
|
}
|
14
multipart/Cargo.toml
Normal file
14
multipart/Cargo.toml
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
[package]
|
||||||
|
name = "multipart-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "multipart"
|
||||||
|
path = "src/main.rs"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "*"
|
||||||
|
futures = "0.1"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
24
multipart/README.md
Normal file
24
multipart/README.md
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
# multipart
|
||||||
|
|
||||||
|
Multipart's `Getting Started` guide for Actix web
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/multipart
|
||||||
|
cargo run (or ``cargo watch -x run``)
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### client
|
||||||
|
|
||||||
|
- ``pip install aiohttp``
|
||||||
|
- ``python client.py``
|
||||||
|
- you must see in server console multipart fields
|
||||||
|
|
||||||
|
if ubuntu :
|
||||||
|
|
||||||
|
- ``pip3 install aiohttp``
|
||||||
|
- ``python3 client.py``
|
34
multipart/client.py
Normal file
34
multipart/client.py
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
# This script could be used for actix-web multipart example test
|
||||||
|
# just start server and run client.py
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
|
async def req1():
|
||||||
|
with aiohttp.MultipartWriter() as writer:
|
||||||
|
writer.append('test')
|
||||||
|
writer.append_json({'passed': True})
|
||||||
|
|
||||||
|
resp = await aiohttp.ClientSession().request(
|
||||||
|
"post", 'http://localhost:8080/multipart',
|
||||||
|
data=writer, headers=writer.headers)
|
||||||
|
print(resp)
|
||||||
|
assert 200 == resp.status
|
||||||
|
|
||||||
|
|
||||||
|
async def req2():
|
||||||
|
with aiohttp.MultipartWriter() as writer:
|
||||||
|
writer.append('test')
|
||||||
|
writer.append_json({'passed': True})
|
||||||
|
writer.append(open('src/main.rs'))
|
||||||
|
|
||||||
|
resp = await aiohttp.ClientSession().request(
|
||||||
|
"post", 'http://localhost:8080/multipart',
|
||||||
|
data=writer, headers=writer.headers)
|
||||||
|
print(resp)
|
||||||
|
assert 200 == resp.status
|
||||||
|
|
||||||
|
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
loop.run_until_complete(req1())
|
||||||
|
loop.run_until_complete(req2())
|
61
multipart/src/main.rs
Normal file
61
multipart/src/main.rs
Normal file
@ -0,0 +1,61 @@
|
|||||||
|
#![allow(unused_variables)]
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate futures;
|
||||||
|
|
||||||
|
use actix::*;
|
||||||
|
use actix_web::{
|
||||||
|
http, middleware, multipart, server,
|
||||||
|
App, AsyncResponder, HttpRequest, HttpResponse, HttpMessage, Error};
|
||||||
|
|
||||||
|
use futures::{Future, Stream};
|
||||||
|
use futures::future::{result, Either};
|
||||||
|
|
||||||
|
|
||||||
|
fn index(req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>>
|
||||||
|
{
|
||||||
|
println!("{:?}", req);
|
||||||
|
|
||||||
|
req.multipart() // <- get multipart stream for current request
|
||||||
|
.from_err() // <- convert multipart errors
|
||||||
|
.and_then(|item| { // <- iterate over multipart items
|
||||||
|
match item {
|
||||||
|
// Handle multipart Field
|
||||||
|
multipart::MultipartItem::Field(field) => {
|
||||||
|
println!("==== FIELD ==== {:?}", field);
|
||||||
|
|
||||||
|
// Field in turn is stream of *Bytes* object
|
||||||
|
Either::A(
|
||||||
|
field.map_err(Error::from)
|
||||||
|
.map(|chunk| {
|
||||||
|
println!("-- CHUNK: \n{}",
|
||||||
|
std::str::from_utf8(&chunk).unwrap());})
|
||||||
|
.finish())
|
||||||
|
},
|
||||||
|
multipart::MultipartItem::Nested(mp) => {
|
||||||
|
// Or item could be nested Multipart stream
|
||||||
|
Either::B(result(Ok(())))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.finish() // <- Stream::finish() combinator from actix
|
||||||
|
.map(|_| HttpResponse::Ok().into())
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
let _ = env_logger::init();
|
||||||
|
let sys = actix::System::new("multipart-example");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
.middleware(middleware::Logger::default()) // <- logger
|
||||||
|
.resource("/multipart", |r| r.method(http::Method::POST).a(index)))
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Starting http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
16
protobuf/Cargo.toml
Normal file
16
protobuf/Cargo.toml
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
[package]
|
||||||
|
name = "protobuf-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["kingxsp <jin_hb_zh@126.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
bytes = "0.4"
|
||||||
|
futures = "0.1"
|
||||||
|
failure = "0.1"
|
||||||
|
env_logger = "*"
|
||||||
|
|
||||||
|
prost = "0.2.0"
|
||||||
|
prost-derive = "0.2.0"
|
||||||
|
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
66
protobuf/client.py
Normal file
66
protobuf/client.py
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
# just start server and run client.py
|
||||||
|
|
||||||
|
# wget https://github.com/google/protobuf/releases/download/v3.5.1/protobuf-python-3.5.1.zip
|
||||||
|
# unzip protobuf-python-3.5.1.zip.1
|
||||||
|
# cd protobuf-3.5.1/python/
|
||||||
|
# python3.6 setup.py install
|
||||||
|
|
||||||
|
# pip3.6 install --upgrade pip
|
||||||
|
# pip3.6 install aiohttp
|
||||||
|
|
||||||
|
#!/usr/bin/env python
|
||||||
|
import test_pb2
|
||||||
|
import traceback
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
|
def op():
|
||||||
|
try:
|
||||||
|
obj = test_pb2.MyObj()
|
||||||
|
obj.number = 9
|
||||||
|
obj.name = 'USB'
|
||||||
|
|
||||||
|
#Serialize
|
||||||
|
sendDataStr = obj.SerializeToString()
|
||||||
|
#print serialized string value
|
||||||
|
print('serialized string:', sendDataStr)
|
||||||
|
#------------------------#
|
||||||
|
# message transmission #
|
||||||
|
#------------------------#
|
||||||
|
receiveDataStr = sendDataStr
|
||||||
|
receiveData = test_pb2.MyObj()
|
||||||
|
|
||||||
|
#Deserialize
|
||||||
|
receiveData.ParseFromString(receiveDataStr)
|
||||||
|
print('pares serialize string, return: devId = ', receiveData.number, ', name = ', receiveData.name)
|
||||||
|
except(Exception, e):
|
||||||
|
print(Exception, ':', e)
|
||||||
|
print(traceback.print_exc())
|
||||||
|
errInfo = sys.exc_info()
|
||||||
|
print(errInfo[0], ':', errInfo[1])
|
||||||
|
|
||||||
|
|
||||||
|
async def fetch(session):
|
||||||
|
obj = test_pb2.MyObj()
|
||||||
|
obj.number = 9
|
||||||
|
obj.name = 'USB'
|
||||||
|
async with session.post('http://localhost:8080/', data=obj.SerializeToString(),
|
||||||
|
headers={"content-type": "application/protobuf"}) as resp:
|
||||||
|
print(resp.status)
|
||||||
|
data = await resp.read()
|
||||||
|
receiveObj = test_pb2.MyObj()
|
||||||
|
receiveObj.ParseFromString(data)
|
||||||
|
print(receiveObj)
|
||||||
|
|
||||||
|
async def go(loop):
|
||||||
|
obj = test_pb2.MyObj()
|
||||||
|
obj.number = 9
|
||||||
|
obj.name = 'USB'
|
||||||
|
async with aiohttp.ClientSession(loop=loop) as session:
|
||||||
|
await fetch(session)
|
||||||
|
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
loop.run_until_complete(go(loop))
|
||||||
|
loop.close()
|
57
protobuf/src/main.rs
Normal file
57
protobuf/src/main.rs
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate bytes;
|
||||||
|
extern crate futures;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate failure;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate prost;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate prost_derive;
|
||||||
|
|
||||||
|
use futures::Future;
|
||||||
|
use actix_web::{
|
||||||
|
http, middleware, server,
|
||||||
|
App, AsyncResponder, HttpRequest, HttpResponse, Error};
|
||||||
|
|
||||||
|
mod protobuf;
|
||||||
|
use protobuf::ProtoBufResponseBuilder;
|
||||||
|
|
||||||
|
|
||||||
|
#[derive(Clone, Debug, PartialEq, Message)]
|
||||||
|
pub struct MyObj {
|
||||||
|
#[prost(int32, tag="1")]
|
||||||
|
pub number: i32,
|
||||||
|
#[prost(string, tag="2")]
|
||||||
|
pub name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// This handler uses `ProtoBufMessage` for loading protobuf object.
|
||||||
|
fn index(req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
protobuf::ProtoBufMessage::new(req)
|
||||||
|
.from_err() // convert all errors into `Error`
|
||||||
|
.and_then(|val: MyObj| {
|
||||||
|
println!("model: {:?}", val);
|
||||||
|
Ok(HttpResponse::Ok().protobuf(val)?) // <- send response
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("protobuf-example");
|
||||||
|
|
||||||
|
server::new(|| {
|
||||||
|
App::new()
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/", |r| r.method(http::Method::POST).f(index))})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.shutdown_timeout(1)
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
168
protobuf/src/protobuf.rs
Normal file
168
protobuf/src/protobuf.rs
Normal file
@ -0,0 +1,168 @@
|
|||||||
|
use bytes::{Bytes, BytesMut};
|
||||||
|
use futures::{Poll, Future, Stream};
|
||||||
|
|
||||||
|
use bytes::IntoBuf;
|
||||||
|
use prost::Message;
|
||||||
|
use prost::DecodeError as ProtoBufDecodeError;
|
||||||
|
use prost::EncodeError as ProtoBufEncodeError;
|
||||||
|
|
||||||
|
use actix_web::http::header::{CONTENT_TYPE, CONTENT_LENGTH};
|
||||||
|
use actix_web::{Responder, HttpMessage, HttpRequest, HttpResponse};
|
||||||
|
use actix_web::dev::HttpResponseBuilder;
|
||||||
|
use actix_web::error::{Error, PayloadError, ResponseError};
|
||||||
|
|
||||||
|
|
||||||
|
#[derive(Fail, Debug)]
|
||||||
|
pub enum ProtoBufPayloadError {
|
||||||
|
/// Payload size is bigger than 256k
|
||||||
|
#[fail(display="Payload size is bigger than 256k")]
|
||||||
|
Overflow,
|
||||||
|
/// Content type error
|
||||||
|
#[fail(display="Content type error")]
|
||||||
|
ContentType,
|
||||||
|
/// Serialize error
|
||||||
|
#[fail(display="ProtoBud serialize error: {}", _0)]
|
||||||
|
Serialize(#[cause] ProtoBufEncodeError),
|
||||||
|
/// Deserialize error
|
||||||
|
#[fail(display="ProtoBud deserialize error: {}", _0)]
|
||||||
|
Deserialize(#[cause] ProtoBufDecodeError),
|
||||||
|
/// Payload error
|
||||||
|
#[fail(display="Error that occur during reading payload: {}", _0)]
|
||||||
|
Payload(#[cause] PayloadError),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ResponseError for ProtoBufPayloadError {
|
||||||
|
|
||||||
|
fn error_response(&self) -> HttpResponse {
|
||||||
|
match *self {
|
||||||
|
ProtoBufPayloadError::Overflow => HttpResponse::PayloadTooLarge().into(),
|
||||||
|
_ => HttpResponse::BadRequest().into(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<PayloadError> for ProtoBufPayloadError {
|
||||||
|
fn from(err: PayloadError) -> ProtoBufPayloadError {
|
||||||
|
ProtoBufPayloadError::Payload(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<ProtoBufDecodeError> for ProtoBufPayloadError {
|
||||||
|
fn from(err: ProtoBufDecodeError) -> ProtoBufPayloadError {
|
||||||
|
ProtoBufPayloadError::Deserialize(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct ProtoBuf<T: Message>(pub T);
|
||||||
|
|
||||||
|
impl<T: Message> Responder for ProtoBuf<T> {
|
||||||
|
type Item = HttpResponse;
|
||||||
|
type Error = Error;
|
||||||
|
|
||||||
|
fn respond_to(self, _: HttpRequest) -> Result<HttpResponse, Error> {
|
||||||
|
let mut buf = Vec::new();
|
||||||
|
self.0.encode(&mut buf)
|
||||||
|
.map_err(|e| Error::from(ProtoBufPayloadError::Serialize(e)))
|
||||||
|
.and_then(|()| {
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
.content_type("application/protobuf")
|
||||||
|
.body(buf)
|
||||||
|
.into())
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct ProtoBufMessage<T, U: Message + Default>{
|
||||||
|
limit: usize,
|
||||||
|
ct: &'static str,
|
||||||
|
req: Option<T>,
|
||||||
|
fut: Option<Box<Future<Item=U, Error=ProtoBufPayloadError>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T, U: Message + Default> ProtoBufMessage<T, U> {
|
||||||
|
|
||||||
|
/// Create `ProtoBufMessage` for request.
|
||||||
|
pub fn new(req: T) -> Self {
|
||||||
|
ProtoBufMessage{
|
||||||
|
limit: 262_144,
|
||||||
|
req: Some(req),
|
||||||
|
fut: None,
|
||||||
|
ct: "application/protobuf",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Change max size of payload. By default max size is 256Kb
|
||||||
|
pub fn limit(mut self, limit: usize) -> Self {
|
||||||
|
self.limit = limit;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Set allowed content type.
|
||||||
|
///
|
||||||
|
/// By default *application/protobuf* content type is used. Set content type
|
||||||
|
/// to empty string if you want to disable content type check.
|
||||||
|
pub fn content_type(mut self, ct: &'static str) -> Self {
|
||||||
|
self.ct = ct;
|
||||||
|
self
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T, U: Message + Default + 'static> Future for ProtoBufMessage<T, U>
|
||||||
|
where T: HttpMessage + Stream<Item=Bytes, Error=PayloadError> + 'static
|
||||||
|
{
|
||||||
|
type Item = U;
|
||||||
|
type Error = ProtoBufPayloadError;
|
||||||
|
|
||||||
|
fn poll(&mut self) -> Poll<U, ProtoBufPayloadError> {
|
||||||
|
if let Some(req) = self.req.take() {
|
||||||
|
if let Some(len) = req.headers().get(CONTENT_LENGTH) {
|
||||||
|
if let Ok(s) = len.to_str() {
|
||||||
|
if let Ok(len) = s.parse::<usize>() {
|
||||||
|
if len > self.limit {
|
||||||
|
return Err(ProtoBufPayloadError::Overflow);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
return Err(ProtoBufPayloadError::Overflow);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// check content-type
|
||||||
|
if !self.ct.is_empty() && req.content_type() != self.ct {
|
||||||
|
return Err(ProtoBufPayloadError::ContentType)
|
||||||
|
}
|
||||||
|
|
||||||
|
let limit = self.limit;
|
||||||
|
let fut = req.from_err()
|
||||||
|
.fold(BytesMut::new(), move |mut body, chunk| {
|
||||||
|
if (body.len() + chunk.len()) > limit {
|
||||||
|
Err(ProtoBufPayloadError::Overflow)
|
||||||
|
} else {
|
||||||
|
body.extend_from_slice(&chunk);
|
||||||
|
Ok(body)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.and_then(|body| Ok(<U>::decode(&mut body.into_buf())?));
|
||||||
|
self.fut = Some(Box::new(fut));
|
||||||
|
}
|
||||||
|
|
||||||
|
self.fut.as_mut().expect("ProtoBufBody could not be used second time").poll()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
pub trait ProtoBufResponseBuilder {
|
||||||
|
|
||||||
|
fn protobuf<T: Message>(&mut self, value: T) -> Result<HttpResponse, Error>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ProtoBufResponseBuilder for HttpResponseBuilder {
|
||||||
|
|
||||||
|
fn protobuf<T: Message>(&mut self, value: T) -> Result<HttpResponse, Error> {
|
||||||
|
self.header(CONTENT_TYPE, "application/protobuf");
|
||||||
|
|
||||||
|
let mut body = Vec::new();
|
||||||
|
value.encode(&mut body).map_err(|e| ProtoBufPayloadError::Serialize(e))?;
|
||||||
|
Ok(self.body(body))
|
||||||
|
}
|
||||||
|
}
|
6
protobuf/test.proto
Normal file
6
protobuf/test.proto
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
syntax = "proto3";
|
||||||
|
|
||||||
|
message MyObj {
|
||||||
|
int32 number = 1;
|
||||||
|
string name = 2;
|
||||||
|
}
|
76
protobuf/test_pb2.py
Normal file
76
protobuf/test_pb2.py
Normal file
@ -0,0 +1,76 @@
|
|||||||
|
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||||
|
# source: test.proto
|
||||||
|
|
||||||
|
import sys
|
||||||
|
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
|
||||||
|
from google.protobuf import descriptor as _descriptor
|
||||||
|
from google.protobuf import message as _message
|
||||||
|
from google.protobuf import reflection as _reflection
|
||||||
|
from google.protobuf import symbol_database as _symbol_database
|
||||||
|
from google.protobuf import descriptor_pb2
|
||||||
|
# @@protoc_insertion_point(imports)
|
||||||
|
|
||||||
|
_sym_db = _symbol_database.Default()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||||
|
name='test.proto',
|
||||||
|
package='',
|
||||||
|
syntax='proto3',
|
||||||
|
serialized_pb=_b('\n\ntest.proto\"%\n\x05MyObj\x12\x0e\n\x06number\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\tb\x06proto3')
|
||||||
|
)
|
||||||
|
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
_MYOBJ = _descriptor.Descriptor(
|
||||||
|
name='MyObj',
|
||||||
|
full_name='MyObj',
|
||||||
|
filename=None,
|
||||||
|
file=DESCRIPTOR,
|
||||||
|
containing_type=None,
|
||||||
|
fields=[
|
||||||
|
_descriptor.FieldDescriptor(
|
||||||
|
name='number', full_name='MyObj.number', index=0,
|
||||||
|
number=1, type=5, cpp_type=1, label=1,
|
||||||
|
has_default_value=False, default_value=0,
|
||||||
|
message_type=None, enum_type=None, containing_type=None,
|
||||||
|
is_extension=False, extension_scope=None,
|
||||||
|
options=None),
|
||||||
|
_descriptor.FieldDescriptor(
|
||||||
|
name='name', full_name='MyObj.name', index=1,
|
||||||
|
number=2, type=9, cpp_type=9, label=1,
|
||||||
|
has_default_value=False, default_value=_b("").decode('utf-8'),
|
||||||
|
message_type=None, enum_type=None, containing_type=None,
|
||||||
|
is_extension=False, extension_scope=None,
|
||||||
|
options=None),
|
||||||
|
],
|
||||||
|
extensions=[
|
||||||
|
],
|
||||||
|
nested_types=[],
|
||||||
|
enum_types=[
|
||||||
|
],
|
||||||
|
options=None,
|
||||||
|
is_extendable=False,
|
||||||
|
syntax='proto3',
|
||||||
|
extension_ranges=[],
|
||||||
|
oneofs=[
|
||||||
|
],
|
||||||
|
serialized_start=14,
|
||||||
|
serialized_end=51,
|
||||||
|
)
|
||||||
|
|
||||||
|
DESCRIPTOR.message_types_by_name['MyObj'] = _MYOBJ
|
||||||
|
|
||||||
|
MyObj = _reflection.GeneratedProtocolMessageType('MyObj', (_message.Message,), dict(
|
||||||
|
DESCRIPTOR = _MYOBJ,
|
||||||
|
__module__ = 'test_pb2'
|
||||||
|
# @@protoc_insertion_point(class_scope:MyObj)
|
||||||
|
))
|
||||||
|
_sym_db.RegisterMessage(MyObj)
|
||||||
|
|
||||||
|
|
||||||
|
# @@protoc_insertion_point(module_scope)
|
19
r2d2/Cargo.toml
Normal file
19
r2d2/Cargo.toml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
[package]
|
||||||
|
name = "r2d2-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
|
||||||
|
futures = "0.1"
|
||||||
|
uuid = { version = "0.5", features = ["serde", "v4"] }
|
||||||
|
serde = "1.0"
|
||||||
|
serde_json = "1.0"
|
||||||
|
serde_derive = "1.0"
|
||||||
|
|
||||||
|
r2d2 = "*"
|
||||||
|
r2d2_sqlite = "*"
|
||||||
|
rusqlite = "*"
|
41
r2d2/src/db.rs
Normal file
41
r2d2/src/db.rs
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
//! Db executor actor
|
||||||
|
use std::io;
|
||||||
|
use uuid;
|
||||||
|
use actix_web::*;
|
||||||
|
use actix::prelude::*;
|
||||||
|
use r2d2::Pool;
|
||||||
|
use r2d2_sqlite::SqliteConnectionManager;
|
||||||
|
|
||||||
|
|
||||||
|
/// This is db executor actor. We are going to run 3 of them in parallel.
|
||||||
|
pub struct DbExecutor(pub Pool<SqliteConnectionManager>);
|
||||||
|
|
||||||
|
/// This is only message that this actor can handle, but it is easy to extend number of
|
||||||
|
/// messages.
|
||||||
|
pub struct CreateUser {
|
||||||
|
pub name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Message for CreateUser {
|
||||||
|
type Result = Result<String, io::Error>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Actor for DbExecutor {
|
||||||
|
type Context = SyncContext<Self>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Handler<CreateUser> for DbExecutor {
|
||||||
|
type Result = Result<String, io::Error>;
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: CreateUser, _: &mut Self::Context) -> Self::Result {
|
||||||
|
let conn = self.0.get().unwrap();
|
||||||
|
|
||||||
|
let uuid = format!("{}", uuid::Uuid::new_v4());
|
||||||
|
conn.execute("INSERT INTO users (id, name) VALUES ($1, $2)",
|
||||||
|
&[&uuid, &msg.name]).unwrap();
|
||||||
|
|
||||||
|
Ok(conn.query_row("SELECT name FROM users WHERE id=$1", &[&uuid], |row| {
|
||||||
|
row.get(0)
|
||||||
|
}).map_err(|_| io::Error::new(io::ErrorKind::Other, "db error"))?)
|
||||||
|
}
|
||||||
|
}
|
65
r2d2/src/main.rs
Normal file
65
r2d2/src/main.rs
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
//! Actix web r2d2 example
|
||||||
|
extern crate serde;
|
||||||
|
extern crate serde_json;
|
||||||
|
extern crate uuid;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate r2d2;
|
||||||
|
extern crate r2d2_sqlite;
|
||||||
|
extern crate rusqlite;
|
||||||
|
|
||||||
|
use actix::prelude::*;
|
||||||
|
use actix_web::{
|
||||||
|
middleware, http, server, App, AsyncResponder, HttpRequest, HttpResponse, Error};
|
||||||
|
use futures::future::Future;
|
||||||
|
use r2d2_sqlite::SqliteConnectionManager;
|
||||||
|
|
||||||
|
mod db;
|
||||||
|
use db::{CreateUser, DbExecutor};
|
||||||
|
|
||||||
|
|
||||||
|
/// State with DbExecutor address
|
||||||
|
struct State {
|
||||||
|
db: Addr<Syn, DbExecutor>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Async request handler
|
||||||
|
fn index(req: HttpRequest<State>) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
let name = &req.match_info()["name"];
|
||||||
|
|
||||||
|
req.state().db.send(CreateUser{name: name.to_owned()})
|
||||||
|
.from_err()
|
||||||
|
.and_then(|res| {
|
||||||
|
match res {
|
||||||
|
Ok(user) => Ok(HttpResponse::Ok().json(user)),
|
||||||
|
Err(_) => Ok(HttpResponse::InternalServerError().into())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.responder()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=debug");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("r2d2-example");
|
||||||
|
|
||||||
|
// r2d2 pool
|
||||||
|
let manager = SqliteConnectionManager::file("test.db");
|
||||||
|
let pool = r2d2::Pool::new(manager).unwrap();
|
||||||
|
|
||||||
|
// Start db executor actors
|
||||||
|
let addr = SyncArbiter::start(3, move || DbExecutor(pool.clone()));
|
||||||
|
|
||||||
|
// Start http server
|
||||||
|
server::new(move || {
|
||||||
|
App::with_state(State{db: addr.clone()})
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/{name}", |r| r.method(http::Method::GET).a(index))})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
BIN
r2d2/test.db
Normal file
BIN
r2d2/test.db
Normal file
Binary file not shown.
10
redis-session/Cargo.toml
Normal file
10
redis-session/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "redis-session"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
actix-redis = { version = "0.3", features = ["web"] }
|
48
redis-session/src/main.rs
Normal file
48
redis-session/src/main.rs
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
#![allow(unused_variables)]
|
||||||
|
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate actix_redis;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use actix_web::{server, App, HttpRequest, HttpResponse, Result};
|
||||||
|
use actix_web::middleware::{Logger, SessionStorage, RequestSession};
|
||||||
|
use actix_redis::RedisSessionBackend;
|
||||||
|
|
||||||
|
|
||||||
|
/// simple handler
|
||||||
|
fn index(mut req: HttpRequest) -> Result<HttpResponse> {
|
||||||
|
println!("{:?}", req);
|
||||||
|
|
||||||
|
// session
|
||||||
|
if let Some(count) = req.session().get::<i32>("counter")? {
|
||||||
|
println!("SESSION value: {}", count);
|
||||||
|
req.session().set("counter", count+1)?;
|
||||||
|
} else {
|
||||||
|
req.session().set("counter", 1)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok("Welcome!".into())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info,actix_redis=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("basic-example");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(Logger::default())
|
||||||
|
// cookie session middleware
|
||||||
|
.middleware(SessionStorage::new(
|
||||||
|
RedisSessionBackend::new("127.0.0.1:6379", &[0; 32])
|
||||||
|
))
|
||||||
|
// register simple route, handle all methods
|
||||||
|
.resource("/", |r| r.f(index)))
|
||||||
|
.bind("0.0.0.0:8080").unwrap()
|
||||||
|
.threads(1)
|
||||||
|
.start();
|
||||||
|
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
10
state/Cargo.toml
Normal file
10
state/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "state"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
futures = "0.1"
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
15
state/README.md
Normal file
15
state/README.md
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
# state
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/state
|
||||||
|
cargo run
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
- [http://localhost:8080/](http://localhost:8080/)
|
77
state/src/main.rs
Normal file
77
state/src/main.rs
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
#![cfg_attr(feature="cargo-clippy", allow(needless_pass_by_value))]
|
||||||
|
//! There are two level of statefulness in actix-web. Application has state
|
||||||
|
//! that is shared across all handlers within same Application.
|
||||||
|
//! And individual handler can have state.
|
||||||
|
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use std::cell::Cell;
|
||||||
|
|
||||||
|
use actix::prelude::*;
|
||||||
|
use actix_web::{
|
||||||
|
http, server, ws, middleware, App, HttpRequest, HttpResponse};
|
||||||
|
|
||||||
|
/// Application state
|
||||||
|
struct AppState {
|
||||||
|
counter: Cell<usize>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// simple handle
|
||||||
|
fn index(req: HttpRequest<AppState>) -> HttpResponse {
|
||||||
|
println!("{:?}", req);
|
||||||
|
req.state().counter.set(req.state().counter.get() + 1);
|
||||||
|
|
||||||
|
HttpResponse::Ok().body(format!("Num of requests: {}", req.state().counter.get()))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// `MyWebSocket` counts how many messages it receives from peer,
|
||||||
|
/// websocket-client.py could be used for tests
|
||||||
|
struct MyWebSocket {
|
||||||
|
counter: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Actor for MyWebSocket {
|
||||||
|
type Context = ws::WebsocketContext<Self, AppState>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl StreamHandler<ws::Message, ws::ProtocolError> for MyWebSocket {
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
|
||||||
|
self.counter += 1;
|
||||||
|
println!("WS({}): {:?}", self.counter, msg);
|
||||||
|
match msg {
|
||||||
|
ws::Message::Ping(msg) => ctx.pong(&msg),
|
||||||
|
ws::Message::Text(text) => ctx.text(text),
|
||||||
|
ws::Message::Binary(bin) => ctx.binary(bin),
|
||||||
|
ws::Message::Close(_) => {
|
||||||
|
ctx.stop();
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("ws-example");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::with_state(AppState{counter: Cell::new(0)})
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
// websocket route
|
||||||
|
.resource(
|
||||||
|
"/ws/", |r|
|
||||||
|
r.method(http::Method::GET).f(
|
||||||
|
|req| ws::start(req, MyWebSocket{counter: 0})))
|
||||||
|
// register simple handler, handle all methods
|
||||||
|
.resource("/", |r| r.f(index)))
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
BIN
static/actixLogo.png
Normal file
BIN
static/actixLogo.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 13 KiB |
BIN
static/favicon.ico
Normal file
BIN
static/favicon.ico
Normal file
Binary file not shown.
After Width: | Height: | Size: 1.1 KiB |
90
static/index.html
Normal file
90
static/index.html
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<meta charset="utf-8" />
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<script src="http://ajax.googleapis.com/ajax/libs/jquery/1.4.2/jquery.min.js">
|
||||||
|
</script>
|
||||||
|
<script language="javascript" type="text/javascript">
|
||||||
|
$(function() {
|
||||||
|
var conn = null;
|
||||||
|
function log(msg) {
|
||||||
|
var control = $('#log');
|
||||||
|
control.html(control.html() + msg + '<br/>');
|
||||||
|
control.scrollTop(control.scrollTop() + 1000);
|
||||||
|
}
|
||||||
|
function connect() {
|
||||||
|
disconnect();
|
||||||
|
var wsUri = (window.location.protocol=='https:'&&'wss://'||'ws://')+window.location.host + '/ws/';
|
||||||
|
conn = new WebSocket(wsUri);
|
||||||
|
log('Connecting...');
|
||||||
|
conn.onopen = function() {
|
||||||
|
log('Connected.');
|
||||||
|
update_ui();
|
||||||
|
};
|
||||||
|
conn.onmessage = function(e) {
|
||||||
|
log('Received: ' + e.data);
|
||||||
|
};
|
||||||
|
conn.onclose = function() {
|
||||||
|
log('Disconnected.');
|
||||||
|
conn = null;
|
||||||
|
update_ui();
|
||||||
|
};
|
||||||
|
}
|
||||||
|
function disconnect() {
|
||||||
|
if (conn != null) {
|
||||||
|
log('Disconnecting...');
|
||||||
|
conn.close();
|
||||||
|
conn = null;
|
||||||
|
update_ui();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function update_ui() {
|
||||||
|
var msg = '';
|
||||||
|
if (conn == null) {
|
||||||
|
$('#status').text('disconnected');
|
||||||
|
$('#connect').html('Connect');
|
||||||
|
} else {
|
||||||
|
$('#status').text('connected (' + conn.protocol + ')');
|
||||||
|
$('#connect').html('Disconnect');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
$('#connect').click(function() {
|
||||||
|
if (conn == null) {
|
||||||
|
connect();
|
||||||
|
} else {
|
||||||
|
disconnect();
|
||||||
|
}
|
||||||
|
update_ui();
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
$('#send').click(function() {
|
||||||
|
var text = $('#text').val();
|
||||||
|
log('Sending: ' + text);
|
||||||
|
conn.send(text);
|
||||||
|
$('#text').val('').focus();
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
$('#text').keyup(function(e) {
|
||||||
|
if (e.keyCode === 13) {
|
||||||
|
$('#send').click();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h3>Chat!</h3>
|
||||||
|
<div>
|
||||||
|
<button id="connect">Connect</button> | Status:
|
||||||
|
<span id="status">disconnected</span>
|
||||||
|
</div>
|
||||||
|
<div id="log"
|
||||||
|
style="width:20em;height:15em;overflow:auto;border:1px solid black">
|
||||||
|
</div>
|
||||||
|
<form id="chatform" onsubmit="return false;">
|
||||||
|
<input id="text" type="text" />
|
||||||
|
<input id="send" type="button" value="Send" />
|
||||||
|
</form>
|
||||||
|
</body>
|
||||||
|
</html>
|
10
template_tera/Cargo.toml
Normal file
10
template_tera/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "template-tera"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
tera = "*"
|
17
template_tera/README.md
Normal file
17
template_tera/README.md
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
# template_tera
|
||||||
|
|
||||||
|
Minimal example of using the template [tera](https://github.com/Keats/tera) that displays a form.
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/template_tera
|
||||||
|
cargo run (or ``cargo watch -x run``)
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
- [http://localhost:8080](http://localhost:8080)
|
48
template_tera/src/main.rs
Normal file
48
template_tera/src/main.rs
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
#[macro_use]
|
||||||
|
extern crate tera;
|
||||||
|
|
||||||
|
use actix_web::{
|
||||||
|
http, error, middleware, server, App, HttpRequest, HttpResponse, Error};
|
||||||
|
|
||||||
|
|
||||||
|
struct State {
|
||||||
|
template: tera::Tera, // <- store tera template in application state
|
||||||
|
}
|
||||||
|
|
||||||
|
fn index(req: HttpRequest<State>) -> Result<HttpResponse, Error> {
|
||||||
|
let s = if let Some(name) = req.query().get("name") { // <- submitted form
|
||||||
|
let mut ctx = tera::Context::new();
|
||||||
|
ctx.add("name", &name.to_owned());
|
||||||
|
ctx.add("text", &"Welcome!".to_owned());
|
||||||
|
req.state().template.render("user.html", &ctx)
|
||||||
|
.map_err(|_| error::ErrorInternalServerError("Template error"))?
|
||||||
|
} else {
|
||||||
|
req.state().template.render("index.html", &tera::Context::new())
|
||||||
|
.map_err(|_| error::ErrorInternalServerError("Template error"))?
|
||||||
|
};
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
.content_type("text/html")
|
||||||
|
.body(s))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("tera-example");
|
||||||
|
|
||||||
|
server::new(|| {
|
||||||
|
let tera = compile_templates!(concat!(env!("CARGO_MANIFEST_DIR"), "/templates/**/*"));
|
||||||
|
|
||||||
|
App::with_state(State{template: tera})
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/", |r| r.method(http::Method::GET).f(index))})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
17
template_tera/templates/index.html
Normal file
17
template_tera/templates/index.html
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8" />
|
||||||
|
<title>Actix web</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Welcome!</h1>
|
||||||
|
<p>
|
||||||
|
<h3>What is your name?</h3>
|
||||||
|
<form>
|
||||||
|
<input type="text" name="name" /><br/>
|
||||||
|
<p><input type="submit"></p>
|
||||||
|
</form>
|
||||||
|
</p>
|
||||||
|
</body>
|
||||||
|
</html>
|
13
template_tera/templates/user.html
Normal file
13
template_tera/templates/user.html
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8" />
|
||||||
|
<title>Actix web</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Hi, {{ name }}!</h1>
|
||||||
|
<p>
|
||||||
|
{{ text }}
|
||||||
|
</p>
|
||||||
|
</body>
|
||||||
|
</html>
|
14
tls/Cargo.toml
Normal file
14
tls/Cargo.toml
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
[package]
|
||||||
|
name = "tls-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "server"
|
||||||
|
path = "src/main.rs"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = { version = "^0.5", features=["alpn"] }
|
||||||
|
openssl = { version="0.10" }
|
16
tls/README.md
Normal file
16
tls/README.md
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
# tls example
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/tls
|
||||||
|
cargo run (or ``cargo watch -x run``)
|
||||||
|
# Started http server: 127.0.0.1:8443
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
- curl: ``curl -v https://127.0.0.1:8443/index.html --compress -k``
|
||||||
|
- browser: [https://127.0.0.1:8443/index.html](https://127.0.0.1:8080/index.html)
|
31
tls/cert.pem
Normal file
31
tls/cert.pem
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
-----BEGIN CERTIFICATE-----
|
||||||
|
MIIFPjCCAyYCCQDvLYiYD+jqeTANBgkqhkiG9w0BAQsFADBhMQswCQYDVQQGEwJV
|
||||||
|
UzELMAkGA1UECAwCQ0ExCzAJBgNVBAcMAlNGMRAwDgYDVQQKDAdDb21wYW55MQww
|
||||||
|
CgYDVQQLDANPcmcxGDAWBgNVBAMMD3d3dy5leGFtcGxlLmNvbTAeFw0xODAxMjUx
|
||||||
|
NzQ2MDFaFw0xOTAxMjUxNzQ2MDFaMGExCzAJBgNVBAYTAlVTMQswCQYDVQQIDAJD
|
||||||
|
QTELMAkGA1UEBwwCU0YxEDAOBgNVBAoMB0NvbXBhbnkxDDAKBgNVBAsMA09yZzEY
|
||||||
|
MBYGA1UEAwwPd3d3LmV4YW1wbGUuY29tMIICIjANBgkqhkiG9w0BAQEFAAOCAg8A
|
||||||
|
MIICCgKCAgEA2WzIA2IpVR9Tb9EFhITlxuhE5rY2a3S6qzYNzQVgSFggxXEPn8k1
|
||||||
|
sQEcer5BfAP986Sck3H0FvB4Bt/I8PwOtUCmhwcc8KtB5TcGPR4fjXnrpC+MIK5U
|
||||||
|
NLkwuyBDKziYzTdBj8kUFX1WxmvEHEgqToPOZfBgsS71cJAR/zOWraDLSRM54jXy
|
||||||
|
voLZN4Ti9rQagQrvTQ44Vz5ycDQy7UxtbUGh1CVv69vNVr7/SOOh/Nw5FNOZWLWr
|
||||||
|
odGyoec5wh9iqRZgRqiTUc6Lt7V2RWc2X2gjwST2UfI+U46Ip3oaQ7ZD4eAkoqND
|
||||||
|
xdniBZAykVG3c/99ux4BAESTF8fsNch6UticBxYMuTu+ouvP0psfI9wwwNliJDmA
|
||||||
|
CRUTB9AgRynbL1AzhqQoDfsb98IZfjfNOpwnwuLwpMAPhbgd5KNdZaIJ4Hb6/stI
|
||||||
|
yFElOExxd3TAxF2Gshd/lq1JcNHAZ1DSXV5MvOWT/NWgXwbIzUgQ8eIi+HuDYX2U
|
||||||
|
UuaB6R8tbd52H7rbUv6HrfinuSlKWqjSYLkiKHkwUpoMw8y9UycRSzs1E9nPwPTO
|
||||||
|
vRXb0mNCQeBCV9FvStNVXdCUTT8LGPv87xSD2pmt7LijlE6mHLG8McfcWkzA69un
|
||||||
|
CEHIFAFDimTuN7EBljc119xWFTcHMyoZAfFF+oTqwSbBGImruCxnaJECAwEAATAN
|
||||||
|
BgkqhkiG9w0BAQsFAAOCAgEApavsgsn7SpPHfhDSN5iZs1ILZQRewJg0Bty0xPfk
|
||||||
|
3tynSW6bNH3nSaKbpsdmxxomthNSQgD2heOq1By9YzeOoNR+7Pk3s4FkASnf3ToI
|
||||||
|
JNTUasBFFfaCG96s4Yvs8KiWS/k84yaWuU8c3Wb1jXs5Rv1qE1Uvuwat1DSGXSoD
|
||||||
|
JNluuIkCsC4kWkyq5pWCGQrabWPRTWsHwC3PTcwSRBaFgYLJaR72SloHB1ot02zL
|
||||||
|
d2age9dmFRFLLCBzP+D7RojBvL37qS/HR+rQ4SoQwiVc/JzaeqSe7ZbvEH9sZYEu
|
||||||
|
ALowJzgbwro7oZflwTWunSeSGDSltkqKjvWvZI61pwfHKDahUTmZ5h2y67FuGEaC
|
||||||
|
CIOUI8dSVSPKITxaq3JL4ze2e9/0Lt7hj19YK2uUmtMAW5Tirz4Yx5lyGH9U8Wur
|
||||||
|
y/X8VPxTc4A9TMlJgkyz0hqvhbPOT/zSWB10zXh0glKAsSBryAOEDxV1UygmSir7
|
||||||
|
YV8Qaq+oyKUTMc1MFq5vZ07M51EPaietn85t8V2Y+k/8XYltRp32NxsypxAJuyxh
|
||||||
|
g/ko6RVTrWa1sMvz/F9LFqAdKiK5eM96lh9IU4xiLg4ob8aS/GRAA8oIFkZFhLrt
|
||||||
|
tOwjIUPmEPyHWFi8dLpNuQKYalLYhuwZftG/9xV+wqhKGZO9iPrpHSYBRTap8w2y
|
||||||
|
1QU=
|
||||||
|
-----END CERTIFICATE-----
|
51
tls/key.pem
Normal file
51
tls/key.pem
Normal file
@ -0,0 +1,51 @@
|
|||||||
|
-----BEGIN RSA PRIVATE KEY-----
|
||||||
|
MIIJKAIBAAKCAgEA2WzIA2IpVR9Tb9EFhITlxuhE5rY2a3S6qzYNzQVgSFggxXEP
|
||||||
|
n8k1sQEcer5BfAP986Sck3H0FvB4Bt/I8PwOtUCmhwcc8KtB5TcGPR4fjXnrpC+M
|
||||||
|
IK5UNLkwuyBDKziYzTdBj8kUFX1WxmvEHEgqToPOZfBgsS71cJAR/zOWraDLSRM5
|
||||||
|
4jXyvoLZN4Ti9rQagQrvTQ44Vz5ycDQy7UxtbUGh1CVv69vNVr7/SOOh/Nw5FNOZ
|
||||||
|
WLWrodGyoec5wh9iqRZgRqiTUc6Lt7V2RWc2X2gjwST2UfI+U46Ip3oaQ7ZD4eAk
|
||||||
|
oqNDxdniBZAykVG3c/99ux4BAESTF8fsNch6UticBxYMuTu+ouvP0psfI9wwwNli
|
||||||
|
JDmACRUTB9AgRynbL1AzhqQoDfsb98IZfjfNOpwnwuLwpMAPhbgd5KNdZaIJ4Hb6
|
||||||
|
/stIyFElOExxd3TAxF2Gshd/lq1JcNHAZ1DSXV5MvOWT/NWgXwbIzUgQ8eIi+HuD
|
||||||
|
YX2UUuaB6R8tbd52H7rbUv6HrfinuSlKWqjSYLkiKHkwUpoMw8y9UycRSzs1E9nP
|
||||||
|
wPTOvRXb0mNCQeBCV9FvStNVXdCUTT8LGPv87xSD2pmt7LijlE6mHLG8McfcWkzA
|
||||||
|
69unCEHIFAFDimTuN7EBljc119xWFTcHMyoZAfFF+oTqwSbBGImruCxnaJECAwEA
|
||||||
|
AQKCAgAME3aoeXNCPxMrSri7u4Xnnk71YXl0Tm9vwvjRQlMusXZggP8VKN/KjP0/
|
||||||
|
9AE/GhmoxqPLrLCZ9ZE1EIjgmZ9Xgde9+C8rTtfCG2RFUL7/5J2p6NonlocmxoJm
|
||||||
|
YkxYwjP6ce86RTjQWL3RF3s09u0inz9/efJk5O7M6bOWMQ9VZXDlBiRY5BYvbqUR
|
||||||
|
6FeSzD4MnMbdyMRoVBeXE88gTvZk8xhB6DJnLzYgc0tKiRoeKT0iYv5JZw25VyRM
|
||||||
|
ycLzfTrFmXCPfB1ylb483d9Ly4fBlM8nkx37PzEnAuukIawDxsPOb9yZC+hfvNJI
|
||||||
|
7NFiMN+3maEqG2iC00w4Lep4skHY7eHUEUMl+Wjr+koAy2YGLWAwHZQTm7iXn9Ab
|
||||||
|
L6adL53zyCKelRuEQOzbeosJAqS+5fpMK0ekXyoFIuskj7bWuIoCX7K/kg6q5IW+
|
||||||
|
vC2FrlsrbQ79GztWLVmHFO1I4J9M5r666YS0qdh8c+2yyRl4FmSiHfGxb3eOKpxQ
|
||||||
|
b6uI97iZlkxPF9LYUCSc7wq0V2gGz+6LnGvTHlHrOfVXqw/5pLAKhXqxvnroDTwz
|
||||||
|
0Ay/xFF6ei/NSxBY5t8ztGCBm45wCU3l8pW0X6dXqwUipw5b4MRy1VFRu6rqlmbL
|
||||||
|
OPSCuLxqyqsigiEYsBgS/icvXz9DWmCQMPd2XM9YhsHvUq+R4QKCAQEA98EuMMXI
|
||||||
|
6UKIt1kK2t/3OeJRyDd4iv/fCMUAnuPjLBvFE4cXD/SbqCxcQYqb+pue3PYkiTIC
|
||||||
|
71rN8OQAc5yKhzmmnCE5N26br/0pG4pwEjIr6mt8kZHmemOCNEzvhhT83nfKmV0g
|
||||||
|
9lNtuGEQMiwmZrpUOF51JOMC39bzcVjYX2Cmvb7cFbIq3lR0zwM+aZpQ4P8LHCIu
|
||||||
|
bgHmwbdlkLyIULJcQmHIbo6nPFB3ZZE4mqmjwY+rA6Fh9rgBa8OFCfTtrgeYXrNb
|
||||||
|
IgZQ5U8GoYRPNC2ot0vpTinraboa/cgm6oG4M7FW1POCJTl+/ktHEnKuO5oroSga
|
||||||
|
/BSg7hCNFVaOhwKCAQEA4Kkys0HtwEbV5mY/NnvUD5KwfXX7BxoXc9lZ6seVoLEc
|
||||||
|
KjgPYxqYRVrC7dB2YDwwp3qcRTi/uBAgFNm3iYlDzI4xS5SeaudUWjglj7BSgXE2
|
||||||
|
iOEa7EwcvVPluLaTgiWjlzUKeUCNNHWSeQOt+paBOT+IgwRVemGVpAgkqQzNh/nP
|
||||||
|
tl3p9aNtgzEm1qVlPclY/XUCtf3bcOR+z1f1b4jBdn0leu5OhnxkC+Htik+2fTXD
|
||||||
|
jt6JGrMkanN25YzsjnD3Sn+v6SO26H99wnYx5oMSdmb8SlWRrKtfJHnihphjG/YY
|
||||||
|
l1cyorV6M/asSgXNQfGJm4OuJi0I4/FL2wLUHnU+JwKCAQEAzh4WipcRthYXXcoj
|
||||||
|
gMKRkMOb3GFh1OpYqJgVExtudNTJmZxq8GhFU51MR27Eo7LycMwKy2UjEfTOnplh
|
||||||
|
Us2qZiPtW7k8O8S2m6yXlYUQBeNdq9IuuYDTaYD94vsazscJNSAeGodjE+uGvb1q
|
||||||
|
1wLqE87yoE7dUInYa1cOA3+xy2/CaNuviBFJHtzOrSb6tqqenQEyQf6h9/12+DTW
|
||||||
|
t5pSIiixHrzxHiFqOoCLRKGToQB+71rSINwTf0nITNpGBWmSj5VcC3VV3TG5/XxI
|
||||||
|
fPlxV2yhD5WFDPVNGBGvwPDSh4jSMZdZMSNBZCy4XWFNSKjGEWoK4DFYed3DoSt9
|
||||||
|
5IG1YwKCAQA63ntHl64KJUWlkwNbboU583FF3uWBjee5VqoGKHhf3CkKMxhtGqnt
|
||||||
|
+oN7t5VdUEhbinhqdx1dyPPvIsHCS3K1pkjqii4cyzNCVNYa2dQ00Qq+QWZBpwwc
|
||||||
|
3GAkz8rFXsGIPMDa1vxpU6mnBjzPniKMcsZ9tmQDppCEpBGfLpio2eAA5IkK8eEf
|
||||||
|
cIDB3CM0Vo94EvI76CJZabaE9IJ+0HIJb2+jz9BJ00yQBIqvJIYoNy9gP5Xjpi+T
|
||||||
|
qV/tdMkD5jwWjHD3AYHLWKUGkNwwkAYFeqT/gX6jpWBP+ZRPOp011X3KInJFSpKU
|
||||||
|
DT5GQ1Dux7EMTCwVGtXqjO8Ym5wjwwsfAoIBAEcxlhIW1G6BiNfnWbNPWBdh3v/K
|
||||||
|
5Ln98Rcrz8UIbWyl7qNPjYb13C1KmifVG1Rym9vWMO3KuG5atK3Mz2yLVRtmWAVc
|
||||||
|
fxzR57zz9MZFDun66xo+Z1wN3fVxQB4CYpOEI4Lb9ioX4v85hm3D6RpFukNtRQEc
|
||||||
|
Gfr4scTjJX4jFWDp0h6ffMb8mY+quvZoJ0TJqV9L9Yj6Ksdvqez/bdSraev97bHQ
|
||||||
|
4gbQxaTZ6WjaD4HjpPQefMdWp97Metg0ZQSS8b8EzmNFgyJ3XcjirzwliKTAQtn6
|
||||||
|
I2sd0NCIooelrKRD8EJoDUwxoOctY7R97wpZ7/wEHU45cBCbRV3H4JILS5c=
|
||||||
|
-----END RSA PRIVATE KEY-----
|
49
tls/src/main.rs
Normal file
49
tls/src/main.rs
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
#![allow(unused_variables)]
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate openssl;
|
||||||
|
|
||||||
|
use openssl::ssl::{SslMethod, SslAcceptor, SslFiletype};
|
||||||
|
use actix_web::{
|
||||||
|
http, middleware, server, App, HttpRequest, HttpResponse, Error};
|
||||||
|
|
||||||
|
|
||||||
|
/// simple handle
|
||||||
|
fn index(req: HttpRequest) -> Result<HttpResponse, Error> {
|
||||||
|
println!("{:?}", req);
|
||||||
|
Ok(HttpResponse::Ok()
|
||||||
|
.content_type("text/plain")
|
||||||
|
.body("Welcome!"))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
if ::std::env::var("RUST_LOG").is_err() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
}
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("ws-example");
|
||||||
|
|
||||||
|
// load ssl keys
|
||||||
|
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
|
||||||
|
builder.set_private_key_file("key.pem", SslFiletype::PEM).unwrap();
|
||||||
|
builder.set_certificate_chain_file("cert.pem").unwrap();
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
// register simple handler, handle all methods
|
||||||
|
.resource("/index.html", |r| r.f(index))
|
||||||
|
// with path parameters
|
||||||
|
.resource("/", |r| r.method(http::Method::GET).f(|req| {
|
||||||
|
HttpResponse::Found()
|
||||||
|
.header("LOCATION", "/index.html")
|
||||||
|
.finish()
|
||||||
|
})))
|
||||||
|
.bind("127.0.0.1:8443").unwrap()
|
||||||
|
.start_ssl(builder).unwrap();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8443");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
10
unix-socket/Cargo.toml
Normal file
10
unix-socket/Cargo.toml
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
[package]
|
||||||
|
name = "unix-socket"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Messense Lv <messense@icloud.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "0.5"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
tokio-uds = "0.1"
|
14
unix-socket/README.md
Normal file
14
unix-socket/README.md
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
## Unix domain socket example
|
||||||
|
|
||||||
|
```bash
|
||||||
|
$ curl --unix-socket /tmp/actix-uds.socket http://localhost/
|
||||||
|
Hello world!
|
||||||
|
```
|
||||||
|
|
||||||
|
Although this will only one thread for handling incoming connections
|
||||||
|
according to the
|
||||||
|
[documentation](https://actix.github.io/actix-web/actix_web/struct.HttpServer.html#method.start_incoming).
|
||||||
|
|
||||||
|
And it does not delete the socket file (`/tmp/actix-uds.socket`) when stopping
|
||||||
|
the server so it will fail to start next time you run it unless you delete
|
||||||
|
the socket file manually.
|
32
unix-socket/src/main.rs
Normal file
32
unix-socket/src/main.rs
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate tokio_uds;
|
||||||
|
|
||||||
|
use actix::*;
|
||||||
|
use actix_web::{middleware, server, App, HttpRequest};
|
||||||
|
use tokio_uds::UnixListener;
|
||||||
|
|
||||||
|
|
||||||
|
fn index(_req: HttpRequest) -> &'static str {
|
||||||
|
"Hello world!"
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("unix-socket");
|
||||||
|
|
||||||
|
let listener = UnixListener::bind(
|
||||||
|
"/tmp/actix-uds.socket", Arbiter::handle()).expect("bind failed");
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/index.html", |r| r.f(|_| "Hello world!"))
|
||||||
|
.resource("/", |r| r.f(index)))
|
||||||
|
.start_incoming(listener.incoming(), false);
|
||||||
|
|
||||||
|
println!("Started http server: /tmp/actix-uds.socket");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
15
web-cors/README.md
Normal file
15
web-cors/README.md
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
# Actix Web CORS example
|
||||||
|
|
||||||
|
## start
|
||||||
|
1 - backend server
|
||||||
|
```bash
|
||||||
|
$ cd web-cors/backend
|
||||||
|
$ cargo run
|
||||||
|
```
|
||||||
|
2 - frontend server
|
||||||
|
```bash
|
||||||
|
$ cd web-cors/frontend
|
||||||
|
$ npm install
|
||||||
|
$ npm run dev
|
||||||
|
```
|
||||||
|
then open browser 'http://localhost:1234/'
|
4
web-cors/backend/.gitignore
vendored
Normal file
4
web-cors/backend/.gitignore
vendored
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
|
||||||
|
/target/
|
||||||
|
**/*.rs.bk
|
||||||
|
Cargo.lock
|
16
web-cors/backend/Cargo.toml
Normal file
16
web-cors/backend/Cargo.toml
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
[package]
|
||||||
|
name = "actix-web-cors"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["krircc <krircc@aliyun.com>"]
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
serde = "1.0"
|
||||||
|
serde_derive = "1.0"
|
||||||
|
serde_json = "1.0"
|
||||||
|
http = "0.1"
|
||||||
|
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
||||||
|
dotenv = "0.10"
|
||||||
|
env_logger = "0.5"
|
||||||
|
futures = "0.1"
|
43
web-cors/backend/src/main.rs
Normal file
43
web-cors/backend/src/main.rs
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
#[macro_use] extern crate serde_derive;
|
||||||
|
extern crate serde;
|
||||||
|
extern crate serde_json;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use std::env;
|
||||||
|
use actix_web::{http, middleware, server, App};
|
||||||
|
|
||||||
|
mod user;
|
||||||
|
use user::info;
|
||||||
|
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
|
||||||
|
let sys = actix::System::new("Actix-web-CORS");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
.resource("/user/info", |r| {
|
||||||
|
middleware::cors::Cors::build()
|
||||||
|
.allowed_origin("http://localhost:1234")
|
||||||
|
.allowed_methods(vec!["GET", "POST"])
|
||||||
|
.allowed_headers(
|
||||||
|
vec![http::header::AUTHORIZATION,
|
||||||
|
http::header::ACCEPT,
|
||||||
|
http::header::CONTENT_TYPE])
|
||||||
|
.max_age(3600)
|
||||||
|
.finish().expect("Can not create CORS middleware")
|
||||||
|
.register(r);
|
||||||
|
r.method(http::Method::POST).a(info);
|
||||||
|
}))
|
||||||
|
.bind("127.0.0.1:8000").unwrap()
|
||||||
|
.shutdown_timeout(200)
|
||||||
|
.start();
|
||||||
|
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
19
web-cors/backend/src/user.rs
Normal file
19
web-cors/backend/src/user.rs
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
use actix_web::{AsyncResponder, Error, HttpMessage, HttpResponse, HttpRequest};
|
||||||
|
use futures::Future;
|
||||||
|
|
||||||
|
|
||||||
|
#[derive(Deserialize,Serialize, Debug)]
|
||||||
|
struct Info {
|
||||||
|
username: String,
|
||||||
|
email: String,
|
||||||
|
password: String,
|
||||||
|
confirm_password: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn info(req: HttpRequest) -> Box<Future<Item=HttpResponse, Error=Error>> {
|
||||||
|
req.json()
|
||||||
|
.from_err()
|
||||||
|
.and_then(|res: Info| {
|
||||||
|
Ok(HttpResponse::Ok().json(res))
|
||||||
|
}).responder()
|
||||||
|
}
|
3
web-cors/frontend/.babelrc
Normal file
3
web-cors/frontend/.babelrc
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
{
|
||||||
|
"presets": ["env"]
|
||||||
|
}
|
14
web-cors/frontend/.gitignore
vendored
Normal file
14
web-cors/frontend/.gitignore
vendored
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
.DS_Store
|
||||||
|
node_modules/
|
||||||
|
/dist/
|
||||||
|
.cache
|
||||||
|
npm-debug.log*
|
||||||
|
yarn-debug.log*
|
||||||
|
yarn-error.log*
|
||||||
|
|
||||||
|
# Editor directories and files
|
||||||
|
.idea
|
||||||
|
*.suo
|
||||||
|
*.ntvs*
|
||||||
|
*.njsproj
|
||||||
|
*.sln
|
13
web-cors/frontend/index.html
Normal file
13
web-cors/frontend/index.html
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<meta name="viewport" content="width=device-width,initial-scale=1.0">
|
||||||
|
<title>webapp</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div id="app"></div>
|
||||||
|
<script src="./src/main.js"></script>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
22
web-cors/frontend/package.json
Normal file
22
web-cors/frontend/package.json
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
{
|
||||||
|
"name": "actix-web-cors",
|
||||||
|
"version": "0.1.0",
|
||||||
|
"description": "webapp",
|
||||||
|
"main": "main.js",
|
||||||
|
"scripts": {
|
||||||
|
"dev": "rm -rf dist/ && NODE_ENV=development parcel index.html",
|
||||||
|
"build": "NODE_ENV=production parcel build index.html",
|
||||||
|
"test": "echo \"Error: no test specified\" && exit 1"
|
||||||
|
},
|
||||||
|
"license": "ISC",
|
||||||
|
"dependencies": {
|
||||||
|
"vue": "^2.5.13",
|
||||||
|
"vue-router": "^3.0.1",
|
||||||
|
"axios": "^0.17.1"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"babel-preset-env": "^1.6.1",
|
||||||
|
"parcel-bundler": "^1.4.1",
|
||||||
|
"parcel-plugin-vue": "^1.5.0"
|
||||||
|
}
|
||||||
|
}
|
145
web-cors/frontend/src/app.vue
Normal file
145
web-cors/frontend/src/app.vue
Normal file
@ -0,0 +1,145 @@
|
|||||||
|
<template>
|
||||||
|
<div id="app">
|
||||||
|
<div id="content">
|
||||||
|
<div id="title">
|
||||||
|
<a to="#">SignUp</a>
|
||||||
|
</div>
|
||||||
|
<input type="text" name="username" placeholder="Username" v-model="Username" required />
|
||||||
|
<input type="text" name="email" placeholder="E-mail" v-model="Email" required />
|
||||||
|
<input type="password" name="password" placeholder="Password" v-model="Password" required/>
|
||||||
|
<input type="password" name="confirm_password" placeholder="Confirm password" v-model="ConfirmPassword" required/><br/>
|
||||||
|
|
||||||
|
<button id="submit" @click="signup">Sign up</button>
|
||||||
|
|
||||||
|
<div id="user-info">
|
||||||
|
<p>Click Above 'Sign up' Button <br> Then Get Your Signup Info!</p>
|
||||||
|
<p>email : {{ email }}</p>
|
||||||
|
<p>username :{{ username }}</p>
|
||||||
|
<p>password : {{ password }}</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import axios from 'axios'
|
||||||
|
export default {
|
||||||
|
name: 'app',
|
||||||
|
data () {
|
||||||
|
return {
|
||||||
|
Username: '',
|
||||||
|
Email: '',
|
||||||
|
Password: '',
|
||||||
|
ConfirmPassword: '',
|
||||||
|
|
||||||
|
email: '',
|
||||||
|
username: '',
|
||||||
|
password: ''
|
||||||
|
}
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
signup () {
|
||||||
|
var username = this.Username
|
||||||
|
var email = this.Email
|
||||||
|
var password = this.Password
|
||||||
|
var confirm_password = this.ConfirmPassword
|
||||||
|
console.log(email)
|
||||||
|
axios.post('http://localhost:8000/user/info', {
|
||||||
|
username: username,
|
||||||
|
email: email,
|
||||||
|
password: password,
|
||||||
|
confirm_password: confirm_password
|
||||||
|
})
|
||||||
|
.then(response => {
|
||||||
|
console.log(response.data)
|
||||||
|
this.email = response.data.email
|
||||||
|
this.username = response.data.username
|
||||||
|
this.password = response.data.password
|
||||||
|
})
|
||||||
|
.catch(e => {
|
||||||
|
console.log(e)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
#content {
|
||||||
|
width: 250px;
|
||||||
|
margin: 0 auto;
|
||||||
|
padding-top: 33px;
|
||||||
|
}
|
||||||
|
#title {
|
||||||
|
padding: 0.5rem 0;
|
||||||
|
font-size: 22px;
|
||||||
|
font-weight: bold;
|
||||||
|
background-color:bisque;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
input[type="text"],
|
||||||
|
input[type="password"] {
|
||||||
|
margin: 6px auto auto;
|
||||||
|
width: 250px;
|
||||||
|
height: 36px;
|
||||||
|
border: none;
|
||||||
|
border-bottom: 1px solid #AAA;
|
||||||
|
font-size: 16px;
|
||||||
|
}
|
||||||
|
#submit {
|
||||||
|
margin: 10px 0 20px 0;
|
||||||
|
width: 250px;
|
||||||
|
height: 33px;
|
||||||
|
background-color:bisque;
|
||||||
|
border: none;
|
||||||
|
border-radius: 2px;
|
||||||
|
font-family: 'Roboto', sans-serif;
|
||||||
|
font-weight: bold;
|
||||||
|
text-transform: uppercase;
|
||||||
|
transition: 0.1s ease;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
input[type="checkbox"] {
|
||||||
|
margin-top: 11px;
|
||||||
|
}
|
||||||
|
dialog {
|
||||||
|
top: 50%;
|
||||||
|
width: 80%;
|
||||||
|
border: 5px solid rgba(0, 0, 0, 0.3);
|
||||||
|
}
|
||||||
|
dialog::backdrop{
|
||||||
|
position: fixed;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
bottom: 0;
|
||||||
|
background-color: rgba(0, 0, 0, 0.7);
|
||||||
|
}
|
||||||
|
#closeDialog {
|
||||||
|
display: inline-block;
|
||||||
|
border-radius: 3px;
|
||||||
|
border: none;
|
||||||
|
font-size: 1rem;
|
||||||
|
padding: 0.4rem 0.8em;
|
||||||
|
background: #eb9816;
|
||||||
|
border-bottom: 1px solid #f1b75c;
|
||||||
|
color: white;
|
||||||
|
font-weight: bold;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
#closeDialog:hover, #closeDialog:focus {
|
||||||
|
opacity: 0.92;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
#user-info {
|
||||||
|
width: 250px;
|
||||||
|
margin: 0 auto;
|
||||||
|
padding-top: 44px;
|
||||||
|
}
|
||||||
|
@media only screen and (min-width: 600px) {
|
||||||
|
#content {
|
||||||
|
margin: 0 auto;
|
||||||
|
padding-top: 100px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
11
web-cors/frontend/src/main.js
Normal file
11
web-cors/frontend/src/main.js
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
import Vue from 'vue'
|
||||||
|
import App from './app'
|
||||||
|
|
||||||
|
new Vue({
|
||||||
|
el: '#app',
|
||||||
|
render: h => h(App)
|
||||||
|
})
|
||||||
|
|
||||||
|
if (module.hot) {
|
||||||
|
module.hot.accept();
|
||||||
|
}
|
28
websocket-chat/Cargo.toml
Normal file
28
websocket-chat/Cargo.toml
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
[package]
|
||||||
|
name = "websocket-example"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "server"
|
||||||
|
path = "src/main.rs"
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "client"
|
||||||
|
path = "src/client.rs"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
rand = "*"
|
||||||
|
bytes = "0.4"
|
||||||
|
byteorder = "1.1"
|
||||||
|
futures = "0.1"
|
||||||
|
tokio-io = "0.1"
|
||||||
|
tokio-core = "0.1"
|
||||||
|
env_logger = "*"
|
||||||
|
|
||||||
|
serde = "1.0"
|
||||||
|
serde_json = "1.0"
|
||||||
|
serde_derive = "1.0"
|
||||||
|
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
32
websocket-chat/README.md
Normal file
32
websocket-chat/README.md
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
# Websocket chat example
|
||||||
|
|
||||||
|
This is extension of the
|
||||||
|
[actix chat example](https://github.com/actix/actix/tree/master/examples/chat)
|
||||||
|
|
||||||
|
Added features:
|
||||||
|
|
||||||
|
* Browser WebSocket client
|
||||||
|
* Chat server runs in separate thread
|
||||||
|
* Tcp listener runs in separate thread
|
||||||
|
|
||||||
|
## Server
|
||||||
|
|
||||||
|
Chat server listens for incoming tcp connections. Server can access several types of message:
|
||||||
|
|
||||||
|
* `\list` - list all available rooms
|
||||||
|
* `\join name` - join room, if room does not exist, create new one
|
||||||
|
* `\name name` - set session name
|
||||||
|
* `some message` - just string, send message to all peers in same room
|
||||||
|
* client has to send heartbeat `Ping` messages, if server does not receive a heartbeat message for 10 seconds connection gets dropped
|
||||||
|
|
||||||
|
To start server use command: `cargo run --bin server`
|
||||||
|
|
||||||
|
## Client
|
||||||
|
|
||||||
|
Client connects to server. Reads input from stdin and sends to server.
|
||||||
|
|
||||||
|
To run client use command: `cargo run --bin client`
|
||||||
|
|
||||||
|
## WebSocket Browser Client
|
||||||
|
|
||||||
|
Open url: [http://localhost:8080/](http://localhost:8080/)
|
72
websocket-chat/client.py
Executable file
72
websocket-chat/client.py
Executable file
@ -0,0 +1,72 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""websocket cmd client for wssrv.py example."""
|
||||||
|
import argparse
|
||||||
|
import asyncio
|
||||||
|
import signal
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
|
|
||||||
|
def start_client(loop, url):
|
||||||
|
name = input('Please enter your name: ')
|
||||||
|
|
||||||
|
# send request
|
||||||
|
ws = yield from aiohttp.ClientSession().ws_connect(url, autoclose=False, autoping=False)
|
||||||
|
|
||||||
|
# input reader
|
||||||
|
def stdin_callback():
|
||||||
|
line = sys.stdin.buffer.readline().decode('utf-8')
|
||||||
|
if not line:
|
||||||
|
loop.stop()
|
||||||
|
else:
|
||||||
|
ws.send_str(name + ': ' + line)
|
||||||
|
loop.add_reader(sys.stdin.fileno(), stdin_callback)
|
||||||
|
|
||||||
|
@asyncio.coroutine
|
||||||
|
def dispatch():
|
||||||
|
while True:
|
||||||
|
msg = yield from ws.receive()
|
||||||
|
|
||||||
|
if msg.type == aiohttp.WSMsgType.TEXT:
|
||||||
|
print('Text: ', msg.data.strip())
|
||||||
|
elif msg.type == aiohttp.WSMsgType.BINARY:
|
||||||
|
print('Binary: ', msg.data)
|
||||||
|
elif msg.type == aiohttp.WSMsgType.PING:
|
||||||
|
ws.pong()
|
||||||
|
elif msg.type == aiohttp.WSMsgType.PONG:
|
||||||
|
print('Pong received')
|
||||||
|
else:
|
||||||
|
if msg.type == aiohttp.WSMsgType.CLOSE:
|
||||||
|
yield from ws.close()
|
||||||
|
elif msg.type == aiohttp.WSMsgType.ERROR:
|
||||||
|
print('Error during receive %s' % ws.exception())
|
||||||
|
elif msg.type == aiohttp.WSMsgType.CLOSED:
|
||||||
|
pass
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
yield from dispatch()
|
||||||
|
|
||||||
|
|
||||||
|
ARGS = argparse.ArgumentParser(
|
||||||
|
description="websocket console client for wssrv.py example.")
|
||||||
|
ARGS.add_argument(
|
||||||
|
'--host', action="store", dest='host',
|
||||||
|
default='127.0.0.1', help='Host name')
|
||||||
|
ARGS.add_argument(
|
||||||
|
'--port', action="store", dest='port',
|
||||||
|
default=8080, type=int, help='Port number')
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
args = ARGS.parse_args()
|
||||||
|
if ':' in args.host:
|
||||||
|
args.host, port = args.host.split(':', 1)
|
||||||
|
args.port = int(port)
|
||||||
|
|
||||||
|
url = 'http://{}:{}/ws/'.format(args.host, args.port)
|
||||||
|
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
loop.add_signal_handler(signal.SIGINT, loop.stop)
|
||||||
|
asyncio.Task(start_client(loop, url))
|
||||||
|
loop.run_forever()
|
153
websocket-chat/src/client.rs
Normal file
153
websocket-chat/src/client.rs
Normal file
@ -0,0 +1,153 @@
|
|||||||
|
#[macro_use] extern crate actix;
|
||||||
|
extern crate bytes;
|
||||||
|
extern crate byteorder;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate tokio_io;
|
||||||
|
extern crate tokio_core;
|
||||||
|
extern crate serde;
|
||||||
|
extern crate serde_json;
|
||||||
|
#[macro_use] extern crate serde_derive;
|
||||||
|
|
||||||
|
use std::{io, net, process, thread};
|
||||||
|
use std::str::FromStr;
|
||||||
|
use std::time::Duration;
|
||||||
|
use futures::Future;
|
||||||
|
use tokio_io::AsyncRead;
|
||||||
|
use tokio_io::io::WriteHalf;
|
||||||
|
use tokio_io::codec::FramedRead;
|
||||||
|
use tokio_core::net::TcpStream;
|
||||||
|
use actix::prelude::*;
|
||||||
|
|
||||||
|
mod codec;
|
||||||
|
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
let sys = actix::System::new("chat-client");
|
||||||
|
|
||||||
|
// Connect to server
|
||||||
|
let addr = net::SocketAddr::from_str("127.0.0.1:12345").unwrap();
|
||||||
|
Arbiter::handle().spawn(
|
||||||
|
TcpStream::connect(&addr, Arbiter::handle())
|
||||||
|
.and_then(|stream| {
|
||||||
|
let addr: Addr<Syn, _> = ChatClient::create(|ctx| {
|
||||||
|
let (r, w) = stream.split();
|
||||||
|
ChatClient::add_stream(FramedRead::new(r, codec::ClientChatCodec), ctx);
|
||||||
|
ChatClient{
|
||||||
|
framed: actix::io::FramedWrite::new(
|
||||||
|
w, codec::ClientChatCodec, ctx)}});
|
||||||
|
|
||||||
|
// start console loop
|
||||||
|
thread::spawn(move|| {
|
||||||
|
loop {
|
||||||
|
let mut cmd = String::new();
|
||||||
|
if io::stdin().read_line(&mut cmd).is_err() {
|
||||||
|
println!("error");
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
addr.do_send(ClientCommand(cmd));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
futures::future::ok(())
|
||||||
|
})
|
||||||
|
.map_err(|e| {
|
||||||
|
println!("Can not connect to server: {}", e);
|
||||||
|
process::exit(1)
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
println!("Running chat client");
|
||||||
|
sys.run();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
struct ChatClient {
|
||||||
|
framed: actix::io::FramedWrite<WriteHalf<TcpStream>, codec::ClientChatCodec>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Message)]
|
||||||
|
struct ClientCommand(String);
|
||||||
|
|
||||||
|
impl Actor for ChatClient {
|
||||||
|
type Context = Context<Self>;
|
||||||
|
|
||||||
|
fn started(&mut self, ctx: &mut Context<Self>) {
|
||||||
|
// start heartbeats otherwise server will disconnect after 10 seconds
|
||||||
|
self.hb(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn stopped(&mut self, _: &mut Context<Self>) {
|
||||||
|
println!("Disconnected");
|
||||||
|
|
||||||
|
// Stop application on disconnect
|
||||||
|
Arbiter::system().do_send(actix::msgs::SystemExit(0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChatClient {
|
||||||
|
fn hb(&self, ctx: &mut Context<Self>) {
|
||||||
|
ctx.run_later(Duration::new(1, 0), |act, ctx| {
|
||||||
|
act.framed.write(codec::ChatRequest::Ping);
|
||||||
|
act.hb(ctx);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl actix::io::WriteHandler<io::Error> for ChatClient {}
|
||||||
|
|
||||||
|
/// Handle stdin commands
|
||||||
|
impl Handler<ClientCommand> for ChatClient {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: ClientCommand, _: &mut Context<Self>) {
|
||||||
|
let m = msg.0.trim();
|
||||||
|
if m.is_empty() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// we check for /sss type of messages
|
||||||
|
if m.starts_with('/') {
|
||||||
|
let v: Vec<&str> = m.splitn(2, ' ').collect();
|
||||||
|
match v[0] {
|
||||||
|
"/list" => {
|
||||||
|
self.framed.write(codec::ChatRequest::List);
|
||||||
|
},
|
||||||
|
"/join" => {
|
||||||
|
if v.len() == 2 {
|
||||||
|
self.framed.write(codec::ChatRequest::Join(v[1].to_owned()));
|
||||||
|
} else {
|
||||||
|
println!("!!! room name is required");
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => println!("!!! unknown command"),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
self.framed.write(codec::ChatRequest::Message(m.to_owned()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Server communication
|
||||||
|
|
||||||
|
impl StreamHandler<codec::ChatResponse, io::Error> for ChatClient {
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: codec::ChatResponse, _: &mut Context<Self>) {
|
||||||
|
match msg {
|
||||||
|
codec::ChatResponse::Message(ref msg) => {
|
||||||
|
println!("message: {}", msg);
|
||||||
|
}
|
||||||
|
codec::ChatResponse::Joined(ref msg) => {
|
||||||
|
println!("!!! joined: {}", msg);
|
||||||
|
}
|
||||||
|
codec::ChatResponse::Rooms(rooms) => {
|
||||||
|
println!("\n!!! Available rooms:");
|
||||||
|
for room in rooms {
|
||||||
|
println!("{}", room);
|
||||||
|
}
|
||||||
|
println!("");
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
123
websocket-chat/src/codec.rs
Normal file
123
websocket-chat/src/codec.rs
Normal file
@ -0,0 +1,123 @@
|
|||||||
|
#![allow(dead_code)]
|
||||||
|
use std::io;
|
||||||
|
use serde_json as json;
|
||||||
|
use byteorder::{BigEndian , ByteOrder};
|
||||||
|
use bytes::{BytesMut, BufMut};
|
||||||
|
use tokio_io::codec::{Encoder, Decoder};
|
||||||
|
|
||||||
|
/// Client request
|
||||||
|
#[derive(Serialize, Deserialize, Debug, Message)]
|
||||||
|
#[serde(tag="cmd", content="data")]
|
||||||
|
pub enum ChatRequest {
|
||||||
|
/// List rooms
|
||||||
|
List,
|
||||||
|
/// Join rooms
|
||||||
|
Join(String),
|
||||||
|
/// Send message
|
||||||
|
Message(String),
|
||||||
|
/// Ping
|
||||||
|
Ping
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Server response
|
||||||
|
#[derive(Serialize, Deserialize, Debug, Message)]
|
||||||
|
#[serde(tag="cmd", content="data")]
|
||||||
|
pub enum ChatResponse {
|
||||||
|
Ping,
|
||||||
|
|
||||||
|
/// List of rooms
|
||||||
|
Rooms(Vec<String>),
|
||||||
|
|
||||||
|
/// Joined
|
||||||
|
Joined(String),
|
||||||
|
|
||||||
|
/// Message
|
||||||
|
Message(String),
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Codec for Client -> Server transport
|
||||||
|
pub struct ChatCodec;
|
||||||
|
|
||||||
|
impl Decoder for ChatCodec
|
||||||
|
{
|
||||||
|
type Item = ChatRequest;
|
||||||
|
type Error = io::Error;
|
||||||
|
|
||||||
|
fn decode(&mut self, src: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
|
||||||
|
let size = {
|
||||||
|
if src.len() < 2 {
|
||||||
|
return Ok(None)
|
||||||
|
}
|
||||||
|
BigEndian::read_u16(src.as_ref()) as usize
|
||||||
|
};
|
||||||
|
|
||||||
|
if src.len() >= size + 2 {
|
||||||
|
src.split_to(2);
|
||||||
|
let buf = src.split_to(size);
|
||||||
|
Ok(Some(json::from_slice::<ChatRequest>(&buf)?))
|
||||||
|
} else {
|
||||||
|
Ok(None)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Encoder for ChatCodec
|
||||||
|
{
|
||||||
|
type Item = ChatResponse;
|
||||||
|
type Error = io::Error;
|
||||||
|
|
||||||
|
fn encode(&mut self, msg: ChatResponse, dst: &mut BytesMut) -> Result<(), Self::Error> {
|
||||||
|
let msg = json::to_string(&msg).unwrap();
|
||||||
|
let msg_ref: &[u8] = msg.as_ref();
|
||||||
|
|
||||||
|
dst.reserve(msg_ref.len() + 2);
|
||||||
|
dst.put_u16::<BigEndian>(msg_ref.len() as u16);
|
||||||
|
dst.put(msg_ref);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// Codec for Server -> Client transport
|
||||||
|
pub struct ClientChatCodec;
|
||||||
|
|
||||||
|
impl Decoder for ClientChatCodec
|
||||||
|
{
|
||||||
|
type Item = ChatResponse;
|
||||||
|
type Error = io::Error;
|
||||||
|
|
||||||
|
fn decode(&mut self, src: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
|
||||||
|
let size = {
|
||||||
|
if src.len() < 2 {
|
||||||
|
return Ok(None)
|
||||||
|
}
|
||||||
|
BigEndian::read_u16(src.as_ref()) as usize
|
||||||
|
};
|
||||||
|
|
||||||
|
if src.len() >= size + 2 {
|
||||||
|
src.split_to(2);
|
||||||
|
let buf = src.split_to(size);
|
||||||
|
Ok(Some(json::from_slice::<ChatResponse>(&buf)?))
|
||||||
|
} else {
|
||||||
|
Ok(None)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Encoder for ClientChatCodec
|
||||||
|
{
|
||||||
|
type Item = ChatRequest;
|
||||||
|
type Error = io::Error;
|
||||||
|
|
||||||
|
fn encode(&mut self, msg: ChatRequest, dst: &mut BytesMut) -> Result<(), Self::Error> {
|
||||||
|
let msg = json::to_string(&msg).unwrap();
|
||||||
|
let msg_ref: &[u8] = msg.as_ref();
|
||||||
|
|
||||||
|
dst.reserve(msg_ref.len() + 2);
|
||||||
|
dst.put_u16::<BigEndian>(msg_ref.len() as u16);
|
||||||
|
dst.put(msg_ref);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
209
websocket-chat/src/main.rs
Normal file
209
websocket-chat/src/main.rs
Normal file
@ -0,0 +1,209 @@
|
|||||||
|
#![allow(unused_variables)]
|
||||||
|
extern crate rand;
|
||||||
|
extern crate bytes;
|
||||||
|
extern crate byteorder;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate tokio_io;
|
||||||
|
extern crate tokio_core;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate serde;
|
||||||
|
extern crate serde_json;
|
||||||
|
#[macro_use] extern crate serde_derive;
|
||||||
|
|
||||||
|
#[macro_use]
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
|
||||||
|
use std::time::Instant;
|
||||||
|
|
||||||
|
use actix::*;
|
||||||
|
use actix_web::server::HttpServer;
|
||||||
|
use actix_web::{http, fs, ws, App, HttpRequest, HttpResponse, Error};
|
||||||
|
|
||||||
|
mod codec;
|
||||||
|
mod server;
|
||||||
|
mod session;
|
||||||
|
|
||||||
|
/// This is our websocket route state, this state is shared with all route instances
|
||||||
|
/// via `HttpContext::state()`
|
||||||
|
struct WsChatSessionState {
|
||||||
|
addr: Addr<Syn, server::ChatServer>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Entry point for our route
|
||||||
|
fn chat_route(req: HttpRequest<WsChatSessionState>) -> Result<HttpResponse, Error> {
|
||||||
|
ws::start(
|
||||||
|
req,
|
||||||
|
WsChatSession {
|
||||||
|
id: 0,
|
||||||
|
hb: Instant::now(),
|
||||||
|
room: "Main".to_owned(),
|
||||||
|
name: None})
|
||||||
|
}
|
||||||
|
|
||||||
|
struct WsChatSession {
|
||||||
|
/// unique session id
|
||||||
|
id: usize,
|
||||||
|
/// Client must send ping at least once per 10 seconds, otherwise we drop connection.
|
||||||
|
hb: Instant,
|
||||||
|
/// joined room
|
||||||
|
room: String,
|
||||||
|
/// peer name
|
||||||
|
name: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Actor for WsChatSession {
|
||||||
|
type Context = ws::WebsocketContext<Self, WsChatSessionState>;
|
||||||
|
|
||||||
|
/// Method is called on actor start.
|
||||||
|
/// We register ws session with ChatServer
|
||||||
|
fn started(&mut self, ctx: &mut Self::Context) {
|
||||||
|
// register self in chat server. `AsyncContext::wait` register
|
||||||
|
// future within context, but context waits until this future resolves
|
||||||
|
// before processing any other events.
|
||||||
|
// HttpContext::state() is instance of WsChatSessionState, state is shared across all
|
||||||
|
// routes within application
|
||||||
|
let addr: Addr<Syn, _> = ctx.address();
|
||||||
|
ctx.state().addr.send(server::Connect{addr: addr.recipient()})
|
||||||
|
.into_actor(self)
|
||||||
|
.then(|res, act, ctx| {
|
||||||
|
match res {
|
||||||
|
Ok(res) => act.id = res,
|
||||||
|
// something is wrong with chat server
|
||||||
|
_ => ctx.stop(),
|
||||||
|
}
|
||||||
|
fut::ok(())
|
||||||
|
}).wait(ctx);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn stopping(&mut self, ctx: &mut Self::Context) -> Running {
|
||||||
|
// notify chat server
|
||||||
|
ctx.state().addr.do_send(server::Disconnect{id: self.id});
|
||||||
|
Running::Stop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handle messages from chat server, we simply send it to peer websocket
|
||||||
|
impl Handler<session::Message> for WsChatSession {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: session::Message, ctx: &mut Self::Context) {
|
||||||
|
ctx.text(msg.0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// WebSocket message handler
|
||||||
|
impl StreamHandler<ws::Message, ws::ProtocolError> for WsChatSession {
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
|
||||||
|
println!("WEBSOCKET MESSAGE: {:?}", msg);
|
||||||
|
match msg {
|
||||||
|
ws::Message::Ping(msg) => ctx.pong(&msg),
|
||||||
|
ws::Message::Pong(msg) => self.hb = Instant::now(),
|
||||||
|
ws::Message::Text(text) => {
|
||||||
|
let m = text.trim();
|
||||||
|
// we check for /sss type of messages
|
||||||
|
if m.starts_with('/') {
|
||||||
|
let v: Vec<&str> = m.splitn(2, ' ').collect();
|
||||||
|
match v[0] {
|
||||||
|
"/list" => {
|
||||||
|
// Send ListRooms message to chat server and wait for response
|
||||||
|
println!("List rooms");
|
||||||
|
ctx.state().addr.send(server::ListRooms)
|
||||||
|
.into_actor(self)
|
||||||
|
.then(|res, _, ctx| {
|
||||||
|
match res {
|
||||||
|
Ok(rooms) => {
|
||||||
|
for room in rooms {
|
||||||
|
ctx.text(room);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => println!("Something is wrong"),
|
||||||
|
}
|
||||||
|
fut::ok(())
|
||||||
|
}).wait(ctx)
|
||||||
|
// .wait(ctx) pauses all events in context,
|
||||||
|
// so actor wont receive any new messages until it get list
|
||||||
|
// of rooms back
|
||||||
|
},
|
||||||
|
"/join" => {
|
||||||
|
if v.len() == 2 {
|
||||||
|
self.room = v[1].to_owned();
|
||||||
|
ctx.state().addr.do_send(
|
||||||
|
server::Join{id: self.id, name: self.room.clone()});
|
||||||
|
|
||||||
|
ctx.text("joined");
|
||||||
|
} else {
|
||||||
|
ctx.text("!!! room name is required");
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"/name" => {
|
||||||
|
if v.len() == 2 {
|
||||||
|
self.name = Some(v[1].to_owned());
|
||||||
|
} else {
|
||||||
|
ctx.text("!!! name is required");
|
||||||
|
}
|
||||||
|
},
|
||||||
|
_ => ctx.text(format!("!!! unknown command: {:?}", m)),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
let msg = if let Some(ref name) = self.name {
|
||||||
|
format!("{}: {}", name, m)
|
||||||
|
} else {
|
||||||
|
m.to_owned()
|
||||||
|
};
|
||||||
|
// send message to chat server
|
||||||
|
ctx.state().addr.do_send(
|
||||||
|
server::Message{id: self.id,
|
||||||
|
msg: msg,
|
||||||
|
room: self.room.clone()})
|
||||||
|
}
|
||||||
|
},
|
||||||
|
ws::Message::Binary(bin) =>
|
||||||
|
println!("Unexpected binary"),
|
||||||
|
ws::Message::Close(_) => {
|
||||||
|
ctx.stop();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
let _ = env_logger::init();
|
||||||
|
let sys = actix::System::new("websocket-example");
|
||||||
|
|
||||||
|
// Start chat server actor in separate thread
|
||||||
|
let server: Addr<Syn, _> = Arbiter::start(|_| server::ChatServer::default());
|
||||||
|
|
||||||
|
// Start tcp server in separate thread
|
||||||
|
let srv = server.clone();
|
||||||
|
Arbiter::new("tcp-server").do_send::<msgs::Execute>(
|
||||||
|
msgs::Execute::new(move || {
|
||||||
|
session::TcpServer::new("127.0.0.1:12345", srv);
|
||||||
|
Ok(())
|
||||||
|
}));
|
||||||
|
|
||||||
|
// Create Http server with websocket support
|
||||||
|
HttpServer::new(
|
||||||
|
move || {
|
||||||
|
// Websocket sessions state
|
||||||
|
let state = WsChatSessionState { addr: server.clone() };
|
||||||
|
|
||||||
|
App::with_state(state)
|
||||||
|
// redirect to websocket.html
|
||||||
|
.resource("/", |r| r.method(http::Method::GET).f(|_| {
|
||||||
|
HttpResponse::Found()
|
||||||
|
.header("LOCATION", "/static/websocket.html")
|
||||||
|
.finish()
|
||||||
|
}))
|
||||||
|
// websocket
|
||||||
|
.resource("/ws/", |r| r.route().f(chat_route))
|
||||||
|
// static resources
|
||||||
|
.handler("/static/", fs::StaticFiles::new("static/"))
|
||||||
|
})
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
197
websocket-chat/src/server.rs
Normal file
197
websocket-chat/src/server.rs
Normal file
@ -0,0 +1,197 @@
|
|||||||
|
//! `ChatServer` is an actor. It maintains list of connection client session.
|
||||||
|
//! And manages available rooms. Peers send messages to other peers in same
|
||||||
|
//! room through `ChatServer`.
|
||||||
|
|
||||||
|
use std::cell::RefCell;
|
||||||
|
use std::collections::{HashMap, HashSet};
|
||||||
|
use rand::{self, Rng, ThreadRng};
|
||||||
|
use actix::prelude::*;
|
||||||
|
|
||||||
|
use session;
|
||||||
|
|
||||||
|
/// Message for chat server communications
|
||||||
|
|
||||||
|
/// New chat session is created
|
||||||
|
#[derive(Message)]
|
||||||
|
#[rtype(usize)]
|
||||||
|
pub struct Connect {
|
||||||
|
pub addr: Recipient<Syn, session::Message>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Session is disconnected
|
||||||
|
#[derive(Message)]
|
||||||
|
pub struct Disconnect {
|
||||||
|
pub id: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Send message to specific room
|
||||||
|
#[derive(Message)]
|
||||||
|
pub struct Message {
|
||||||
|
/// Id of the client session
|
||||||
|
pub id: usize,
|
||||||
|
/// Peer message
|
||||||
|
pub msg: String,
|
||||||
|
/// Room name
|
||||||
|
pub room: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// List of available rooms
|
||||||
|
pub struct ListRooms;
|
||||||
|
|
||||||
|
impl actix::Message for ListRooms {
|
||||||
|
type Result = Vec<String>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Join room, if room does not exists create new one.
|
||||||
|
#[derive(Message)]
|
||||||
|
pub struct Join {
|
||||||
|
/// Client id
|
||||||
|
pub id: usize,
|
||||||
|
/// Room name
|
||||||
|
pub name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// `ChatServer` manages chat rooms and responsible for coordinating chat session.
|
||||||
|
/// implementation is super primitive
|
||||||
|
pub struct ChatServer {
|
||||||
|
sessions: HashMap<usize, Recipient<Syn, session::Message>>,
|
||||||
|
rooms: HashMap<String, HashSet<usize>>,
|
||||||
|
rng: RefCell<ThreadRng>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ChatServer {
|
||||||
|
fn default() -> ChatServer {
|
||||||
|
// default room
|
||||||
|
let mut rooms = HashMap::new();
|
||||||
|
rooms.insert("Main".to_owned(), HashSet::new());
|
||||||
|
|
||||||
|
ChatServer {
|
||||||
|
sessions: HashMap::new(),
|
||||||
|
rooms: rooms,
|
||||||
|
rng: RefCell::new(rand::thread_rng()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChatServer {
|
||||||
|
/// Send message to all users in the room
|
||||||
|
fn send_message(&self, room: &str, message: &str, skip_id: usize) {
|
||||||
|
if let Some(sessions) = self.rooms.get(room) {
|
||||||
|
for id in sessions {
|
||||||
|
if *id != skip_id {
|
||||||
|
if let Some(addr) = self.sessions.get(id) {
|
||||||
|
let _ = addr.do_send(session::Message(message.to_owned()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Make actor from `ChatServer`
|
||||||
|
impl Actor for ChatServer {
|
||||||
|
/// We are going to use simple Context, we just need ability to communicate
|
||||||
|
/// with other actors.
|
||||||
|
type Context = Context<Self>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handler for Connect message.
|
||||||
|
///
|
||||||
|
/// Register new session and assign unique id to this session
|
||||||
|
impl Handler<Connect> for ChatServer {
|
||||||
|
type Result = usize;
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: Connect, _: &mut Context<Self>) -> Self::Result {
|
||||||
|
println!("Someone joined");
|
||||||
|
|
||||||
|
// notify all users in same room
|
||||||
|
self.send_message(&"Main".to_owned(), "Someone joined", 0);
|
||||||
|
|
||||||
|
// register session with random id
|
||||||
|
let id = self.rng.borrow_mut().gen::<usize>();
|
||||||
|
self.sessions.insert(id, msg.addr);
|
||||||
|
|
||||||
|
// auto join session to Main room
|
||||||
|
self.rooms.get_mut(&"Main".to_owned()).unwrap().insert(id);
|
||||||
|
|
||||||
|
// send id back
|
||||||
|
id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handler for Disconnect message.
|
||||||
|
impl Handler<Disconnect> for ChatServer {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: Disconnect, _: &mut Context<Self>) {
|
||||||
|
println!("Someone disconnected");
|
||||||
|
|
||||||
|
let mut rooms: Vec<String> = Vec::new();
|
||||||
|
|
||||||
|
// remove address
|
||||||
|
if self.sessions.remove(&msg.id).is_some() {
|
||||||
|
// remove session from all rooms
|
||||||
|
for (name, sessions) in &mut self.rooms {
|
||||||
|
if sessions.remove(&msg.id) {
|
||||||
|
rooms.push(name.to_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// send message to other users
|
||||||
|
for room in rooms {
|
||||||
|
self.send_message(&room, "Someone disconnected", 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handler for Message message.
|
||||||
|
impl Handler<Message> for ChatServer {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: Message, _: &mut Context<Self>) {
|
||||||
|
self.send_message(&msg.room, msg.msg.as_str(), msg.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handler for `ListRooms` message.
|
||||||
|
impl Handler<ListRooms> for ChatServer {
|
||||||
|
type Result = MessageResult<ListRooms>;
|
||||||
|
|
||||||
|
fn handle(&mut self, _: ListRooms, _: &mut Context<Self>) -> Self::Result {
|
||||||
|
let mut rooms = Vec::new();
|
||||||
|
|
||||||
|
for key in self.rooms.keys() {
|
||||||
|
rooms.push(key.to_owned())
|
||||||
|
}
|
||||||
|
|
||||||
|
MessageResult(rooms)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Join room, send disconnect message to old room
|
||||||
|
/// send join message to new room
|
||||||
|
impl Handler<Join> for ChatServer {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: Join, _: &mut Context<Self>) {
|
||||||
|
let Join {id, name} = msg;
|
||||||
|
let mut rooms = Vec::new();
|
||||||
|
|
||||||
|
// remove session from all rooms
|
||||||
|
for (n, sessions) in &mut self.rooms {
|
||||||
|
if sessions.remove(&id) {
|
||||||
|
rooms.push(n.to_owned());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// send message to other users
|
||||||
|
for room in rooms {
|
||||||
|
self.send_message(&room, "Someone disconnected", 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.rooms.get_mut(&name).is_none() {
|
||||||
|
self.rooms.insert(name.clone(), HashSet::new());
|
||||||
|
}
|
||||||
|
self.send_message(&name, "Someone connected", id);
|
||||||
|
self.rooms.get_mut(&name).unwrap().insert(id);
|
||||||
|
}
|
||||||
|
}
|
207
websocket-chat/src/session.rs
Normal file
207
websocket-chat/src/session.rs
Normal file
@ -0,0 +1,207 @@
|
|||||||
|
//! `ClientSession` is an actor, it manages peer tcp connection and
|
||||||
|
//! proxies commands from peer to `ChatServer`.
|
||||||
|
use std::{io, net};
|
||||||
|
use std::str::FromStr;
|
||||||
|
use std::time::{Instant, Duration};
|
||||||
|
use futures::Stream;
|
||||||
|
use tokio_io::AsyncRead;
|
||||||
|
use tokio_io::io::WriteHalf;
|
||||||
|
use tokio_io::codec::FramedRead;
|
||||||
|
use tokio_core::net::{TcpStream, TcpListener};
|
||||||
|
|
||||||
|
use actix::prelude::*;
|
||||||
|
|
||||||
|
use server::{self, ChatServer};
|
||||||
|
use codec::{ChatRequest, ChatResponse, ChatCodec};
|
||||||
|
|
||||||
|
|
||||||
|
/// Chat server sends this messages to session
|
||||||
|
#[derive(Message)]
|
||||||
|
pub struct Message(pub String);
|
||||||
|
|
||||||
|
/// `ChatSession` actor is responsible for tcp peer communications.
|
||||||
|
pub struct ChatSession {
|
||||||
|
/// unique session id
|
||||||
|
id: usize,
|
||||||
|
/// this is address of chat server
|
||||||
|
addr: Addr<Syn, ChatServer>,
|
||||||
|
/// Client must send ping at least once per 10 seconds, otherwise we drop connection.
|
||||||
|
hb: Instant,
|
||||||
|
/// joined room
|
||||||
|
room: String,
|
||||||
|
/// Framed wrapper
|
||||||
|
framed: actix::io::FramedWrite<WriteHalf<TcpStream>, ChatCodec>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Actor for ChatSession {
|
||||||
|
/// For tcp communication we are going to use `FramedContext`.
|
||||||
|
/// It is convenient wrapper around `Framed` object from `tokio_io`
|
||||||
|
type Context = Context<Self>;
|
||||||
|
|
||||||
|
fn started(&mut self, ctx: &mut Self::Context) {
|
||||||
|
// we'll start heartbeat process on session start.
|
||||||
|
self.hb(ctx);
|
||||||
|
|
||||||
|
// register self in chat server. `AsyncContext::wait` register
|
||||||
|
// future within context, but context waits until this future resolves
|
||||||
|
// before processing any other events.
|
||||||
|
let addr: Addr<Syn, _> = ctx.address();
|
||||||
|
self.addr.send(server::Connect{addr: addr.recipient()})
|
||||||
|
.into_actor(self)
|
||||||
|
.then(|res, act, ctx| {
|
||||||
|
match res {
|
||||||
|
Ok(res) => act.id = res,
|
||||||
|
// something is wrong with chat server
|
||||||
|
_ => ctx.stop(),
|
||||||
|
}
|
||||||
|
actix::fut::ok(())
|
||||||
|
}).wait(ctx);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn stopping(&mut self, ctx: &mut Self::Context) -> Running {
|
||||||
|
// notify chat server
|
||||||
|
self.addr.do_send(server::Disconnect{id: self.id});
|
||||||
|
Running::Stop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl actix::io::WriteHandler<io::Error> for ChatSession {}
|
||||||
|
|
||||||
|
/// To use `Framed` we have to define Io type and Codec
|
||||||
|
impl StreamHandler<ChatRequest, io::Error> for ChatSession {
|
||||||
|
|
||||||
|
/// This is main event loop for client requests
|
||||||
|
fn handle(&mut self, msg: ChatRequest, ctx: &mut Context<Self>) {
|
||||||
|
match msg {
|
||||||
|
ChatRequest::List => {
|
||||||
|
// Send ListRooms message to chat server and wait for response
|
||||||
|
println!("List rooms");
|
||||||
|
self.addr.send(server::ListRooms)
|
||||||
|
.into_actor(self)
|
||||||
|
.then(|res, act, ctx| {
|
||||||
|
match res {
|
||||||
|
Ok(rooms) => {
|
||||||
|
act.framed.write(ChatResponse::Rooms(rooms));
|
||||||
|
},
|
||||||
|
_ => println!("Something is wrong"),
|
||||||
|
}
|
||||||
|
actix::fut::ok(())
|
||||||
|
}).wait(ctx)
|
||||||
|
// .wait(ctx) pauses all events in context,
|
||||||
|
// so actor wont receive any new messages until it get list of rooms back
|
||||||
|
},
|
||||||
|
ChatRequest::Join(name) => {
|
||||||
|
println!("Join to room: {}", name);
|
||||||
|
self.room = name.clone();
|
||||||
|
self.addr.do_send(server::Join{id: self.id, name: name.clone()});
|
||||||
|
self.framed.write(ChatResponse::Joined(name));
|
||||||
|
},
|
||||||
|
ChatRequest::Message(message) => {
|
||||||
|
// send message to chat server
|
||||||
|
println!("Peer message: {}", message);
|
||||||
|
self.addr.do_send(
|
||||||
|
server::Message{id: self.id,
|
||||||
|
msg: message, room:
|
||||||
|
self.room.clone()})
|
||||||
|
}
|
||||||
|
// we update heartbeat time on ping from peer
|
||||||
|
ChatRequest::Ping =>
|
||||||
|
self.hb = Instant::now(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handler for Message, chat server sends this message, we just send string to peer
|
||||||
|
impl Handler<Message> for ChatSession {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: Message, ctx: &mut Context<Self>) {
|
||||||
|
// send message to peer
|
||||||
|
self.framed.write(ChatResponse::Message(msg.0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Helper methods
|
||||||
|
impl ChatSession {
|
||||||
|
|
||||||
|
pub fn new(addr: Addr<Syn,ChatServer>,
|
||||||
|
framed: actix::io::FramedWrite<WriteHalf<TcpStream>, ChatCodec>) -> ChatSession {
|
||||||
|
ChatSession {id: 0, addr: addr, hb: Instant::now(),
|
||||||
|
room: "Main".to_owned(), framed: framed}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// helper method that sends ping to client every second.
|
||||||
|
///
|
||||||
|
/// also this method check heartbeats from client
|
||||||
|
fn hb(&self, ctx: &mut Context<Self>) {
|
||||||
|
ctx.run_later(Duration::new(1, 0), |act, ctx| {
|
||||||
|
// check client heartbeats
|
||||||
|
if Instant::now().duration_since(act.hb) > Duration::new(10, 0) {
|
||||||
|
// heartbeat timed out
|
||||||
|
println!("Client heartbeat failed, disconnecting!");
|
||||||
|
|
||||||
|
// notify chat server
|
||||||
|
act.addr.do_send(server::Disconnect{id: act.id});
|
||||||
|
|
||||||
|
// stop actor
|
||||||
|
ctx.stop();
|
||||||
|
}
|
||||||
|
|
||||||
|
act.framed.write(ChatResponse::Ping);
|
||||||
|
// if we can not send message to sink, sink is closed (disconnected)
|
||||||
|
act.hb(ctx);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// Define tcp server that will accept incoming tcp connection and create
|
||||||
|
/// chat actors.
|
||||||
|
pub struct TcpServer {
|
||||||
|
chat: Addr<Syn, ChatServer>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TcpServer {
|
||||||
|
pub fn new(s: &str, chat: Addr<Syn, ChatServer>) {
|
||||||
|
// Create server listener
|
||||||
|
let addr = net::SocketAddr::from_str("127.0.0.1:12345").unwrap();
|
||||||
|
let listener = TcpListener::bind(&addr, Arbiter::handle()).unwrap();
|
||||||
|
|
||||||
|
// Our chat server `Server` is an actor, first we need to start it
|
||||||
|
// and then add stream on incoming tcp connections to it.
|
||||||
|
// TcpListener::incoming() returns stream of the (TcpStream, net::SocketAddr) items
|
||||||
|
// So to be able to handle this events `Server` actor has to implement
|
||||||
|
// stream handler `StreamHandler<(TcpStream, net::SocketAddr), io::Error>`
|
||||||
|
let _: () = TcpServer::create(|ctx| {
|
||||||
|
ctx.add_message_stream(listener.incoming()
|
||||||
|
.map_err(|_| ())
|
||||||
|
.map(|(t, a)| TcpConnect(t, a)));
|
||||||
|
TcpServer{chat: chat}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Make actor from `Server`
|
||||||
|
impl Actor for TcpServer {
|
||||||
|
/// Every actor has to provide execution `Context` in which it can run.
|
||||||
|
type Context = Context<Self>;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Message)]
|
||||||
|
struct TcpConnect(TcpStream, net::SocketAddr);
|
||||||
|
|
||||||
|
/// Handle stream of TcpStream's
|
||||||
|
impl Handler<TcpConnect> for TcpServer {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: TcpConnect, _: &mut Context<Self>) {
|
||||||
|
// For each incoming connection we create `ChatSession` actor
|
||||||
|
// with out chat server address.
|
||||||
|
let server = self.chat.clone();
|
||||||
|
let _: () = ChatSession::create(|ctx| {
|
||||||
|
let (r, w) = msg.0.split();
|
||||||
|
ChatSession::add_stream(FramedRead::new(r, ChatCodec), ctx);
|
||||||
|
ChatSession::new(server, actix::io::FramedWrite::new(w, ChatCodec, ctx))
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
90
websocket-chat/static/websocket.html
Normal file
90
websocket-chat/static/websocket.html
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<meta charset="utf-8" />
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<script src="http://ajax.googleapis.com/ajax/libs/jquery/1.4.2/jquery.min.js">
|
||||||
|
</script>
|
||||||
|
<script language="javascript" type="text/javascript">
|
||||||
|
$(function() {
|
||||||
|
var conn = null;
|
||||||
|
function log(msg) {
|
||||||
|
var control = $('#log');
|
||||||
|
control.html(control.html() + msg + '<br/>');
|
||||||
|
control.scrollTop(control.scrollTop() + 1000);
|
||||||
|
}
|
||||||
|
function connect() {
|
||||||
|
disconnect();
|
||||||
|
var wsUri = (window.location.protocol=='https:'&&'wss://'||'ws://')+window.location.host + '/ws/';
|
||||||
|
conn = new WebSocket(wsUri);
|
||||||
|
log('Connecting...');
|
||||||
|
conn.onopen = function() {
|
||||||
|
log('Connected.');
|
||||||
|
update_ui();
|
||||||
|
};
|
||||||
|
conn.onmessage = function(e) {
|
||||||
|
log('Received: ' + e.data);
|
||||||
|
};
|
||||||
|
conn.onclose = function() {
|
||||||
|
log('Disconnected.');
|
||||||
|
conn = null;
|
||||||
|
update_ui();
|
||||||
|
};
|
||||||
|
}
|
||||||
|
function disconnect() {
|
||||||
|
if (conn != null) {
|
||||||
|
log('Disconnecting...');
|
||||||
|
conn.close();
|
||||||
|
conn = null;
|
||||||
|
update_ui();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
function update_ui() {
|
||||||
|
var msg = '';
|
||||||
|
if (conn == null) {
|
||||||
|
$('#status').text('disconnected');
|
||||||
|
$('#connect').html('Connect');
|
||||||
|
} else {
|
||||||
|
$('#status').text('connected (' + conn.protocol + ')');
|
||||||
|
$('#connect').html('Disconnect');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
$('#connect').click(function() {
|
||||||
|
if (conn == null) {
|
||||||
|
connect();
|
||||||
|
} else {
|
||||||
|
disconnect();
|
||||||
|
}
|
||||||
|
update_ui();
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
$('#send').click(function() {
|
||||||
|
var text = $('#text').val();
|
||||||
|
log('Sending: ' + text);
|
||||||
|
conn.send(text);
|
||||||
|
$('#text').val('').focus();
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
$('#text').keyup(function(e) {
|
||||||
|
if (e.keyCode === 13) {
|
||||||
|
$('#send').click();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h3>Chat!</h3>
|
||||||
|
<div>
|
||||||
|
<button id="connect">Connect</button> | Status:
|
||||||
|
<span id="status">disconnected</span>
|
||||||
|
</div>
|
||||||
|
<div id="log"
|
||||||
|
style="width:20em;height:15em;overflow:auto;border:1px solid black">
|
||||||
|
</div>
|
||||||
|
<form id="chatform" onsubmit="return false;">
|
||||||
|
<input id="text" type="text" />
|
||||||
|
<input id="send" type="button" value="Send" />
|
||||||
|
</form>
|
||||||
|
</body>
|
||||||
|
</html>
|
19
websocket/Cargo.toml
Normal file
19
websocket/Cargo.toml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
[package]
|
||||||
|
name = "websocket"
|
||||||
|
version = "0.1.0"
|
||||||
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "server"
|
||||||
|
path = "src/main.rs"
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "client"
|
||||||
|
path = "src/client.rs"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
env_logger = "*"
|
||||||
|
futures = "0.1"
|
||||||
|
tokio-core = "0.1"
|
||||||
|
actix = "0.5"
|
||||||
|
actix-web = "^0.5"
|
27
websocket/README.md
Normal file
27
websocket/README.md
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
# websocket
|
||||||
|
|
||||||
|
Simple echo websocket server.
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
### server
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd actix-web/examples/websocket
|
||||||
|
cargo run
|
||||||
|
# Started http server: 127.0.0.1:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### web client
|
||||||
|
|
||||||
|
- [http://localhost:8080/ws/index.html](http://localhost:8080/ws/index.html)
|
||||||
|
|
||||||
|
### python client
|
||||||
|
|
||||||
|
- ``pip install aiohttp``
|
||||||
|
- ``python websocket-client.py``
|
||||||
|
|
||||||
|
if ubuntu :
|
||||||
|
|
||||||
|
- ``pip3 install aiohttp``
|
||||||
|
- ``python3 websocket-client.py``
|
113
websocket/src/client.rs
Normal file
113
websocket/src/client.rs
Normal file
@ -0,0 +1,113 @@
|
|||||||
|
//! Simple websocket client.
|
||||||
|
|
||||||
|
#![allow(unused_variables)]
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
extern crate futures;
|
||||||
|
extern crate tokio_core;
|
||||||
|
|
||||||
|
use std::{io, thread};
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
|
use actix::*;
|
||||||
|
use futures::Future;
|
||||||
|
use actix_web::ws::{Message, ProtocolError, Client, ClientWriter};
|
||||||
|
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
let _ = env_logger::init();
|
||||||
|
let sys = actix::System::new("ws-example");
|
||||||
|
|
||||||
|
Arbiter::handle().spawn(
|
||||||
|
Client::new("http://127.0.0.1:8080/ws/")
|
||||||
|
.connect()
|
||||||
|
.map_err(|e| {
|
||||||
|
println!("Error: {}", e);
|
||||||
|
()
|
||||||
|
})
|
||||||
|
.map(|(reader, writer)| {
|
||||||
|
let addr: Addr<Syn, _> = ChatClient::create(|ctx| {
|
||||||
|
ChatClient::add_stream(reader, ctx);
|
||||||
|
ChatClient(writer)
|
||||||
|
});
|
||||||
|
|
||||||
|
// start console loop
|
||||||
|
thread::spawn(move|| {
|
||||||
|
loop {
|
||||||
|
let mut cmd = String::new();
|
||||||
|
if io::stdin().read_line(&mut cmd).is_err() {
|
||||||
|
println!("error");
|
||||||
|
return
|
||||||
|
}
|
||||||
|
addr.do_send(ClientCommand(cmd));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
()
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
struct ChatClient(ClientWriter);
|
||||||
|
|
||||||
|
#[derive(Message)]
|
||||||
|
struct ClientCommand(String);
|
||||||
|
|
||||||
|
impl Actor for ChatClient {
|
||||||
|
type Context = Context<Self>;
|
||||||
|
|
||||||
|
fn started(&mut self, ctx: &mut Context<Self>) {
|
||||||
|
// start heartbeats otherwise server will disconnect after 10 seconds
|
||||||
|
self.hb(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn stopped(&mut self, _: &mut Context<Self>) {
|
||||||
|
println!("Disconnected");
|
||||||
|
|
||||||
|
// Stop application on disconnect
|
||||||
|
Arbiter::system().do_send(actix::msgs::SystemExit(0));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChatClient {
|
||||||
|
fn hb(&self, ctx: &mut Context<Self>) {
|
||||||
|
ctx.run_later(Duration::new(1, 0), |act, ctx| {
|
||||||
|
act.0.ping("");
|
||||||
|
act.hb(ctx);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handle stdin commands
|
||||||
|
impl Handler<ClientCommand> for ChatClient {
|
||||||
|
type Result = ();
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: ClientCommand, ctx: &mut Context<Self>) {
|
||||||
|
self.0.text(msg.0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handle server websocket messages
|
||||||
|
impl StreamHandler<Message, ProtocolError> for ChatClient {
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: Message, ctx: &mut Context<Self>) {
|
||||||
|
match msg {
|
||||||
|
Message::Text(txt) => println!("Server: {:?}", txt),
|
||||||
|
_ => ()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn started(&mut self, ctx: &mut Context<Self>) {
|
||||||
|
println!("Connected");
|
||||||
|
}
|
||||||
|
|
||||||
|
fn finished(&mut self, ctx: &mut Context<Self>) {
|
||||||
|
println!("Server disconnected");
|
||||||
|
ctx.stop()
|
||||||
|
}
|
||||||
|
}
|
66
websocket/src/main.rs
Normal file
66
websocket/src/main.rs
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
//! Simple echo websocket server.
|
||||||
|
//! Open `http://localhost:8080/ws/index.html` in browser
|
||||||
|
//! or [python console client](https://github.com/actix/actix-web/blob/master/examples/websocket-client.py)
|
||||||
|
//! could be used for testing.
|
||||||
|
|
||||||
|
#![allow(unused_variables)]
|
||||||
|
extern crate actix;
|
||||||
|
extern crate actix_web;
|
||||||
|
extern crate env_logger;
|
||||||
|
|
||||||
|
use actix::prelude::*;
|
||||||
|
use actix_web::{
|
||||||
|
http, middleware, server, fs, ws, App, HttpRequest, HttpResponse, Error};
|
||||||
|
|
||||||
|
/// do websocket handshake and start `MyWebSocket` actor
|
||||||
|
fn ws_index(r: HttpRequest) -> Result<HttpResponse, Error> {
|
||||||
|
ws::start(r, MyWebSocket)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// websocket connection is long running connection, it easier
|
||||||
|
/// to handle with an actor
|
||||||
|
struct MyWebSocket;
|
||||||
|
|
||||||
|
impl Actor for MyWebSocket {
|
||||||
|
type Context = ws::WebsocketContext<Self>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Handler for `ws::Message`
|
||||||
|
impl StreamHandler<ws::Message, ws::ProtocolError> for MyWebSocket {
|
||||||
|
|
||||||
|
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
|
||||||
|
// process websocket messages
|
||||||
|
println!("WS: {:?}", msg);
|
||||||
|
match msg {
|
||||||
|
ws::Message::Ping(msg) => ctx.pong(&msg),
|
||||||
|
ws::Message::Text(text) => ctx.text(text),
|
||||||
|
ws::Message::Binary(bin) => ctx.binary(bin),
|
||||||
|
ws::Message::Close(_) => {
|
||||||
|
ctx.stop();
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
::std::env::set_var("RUST_LOG", "actix_web=info");
|
||||||
|
env_logger::init();
|
||||||
|
let sys = actix::System::new("ws-example");
|
||||||
|
|
||||||
|
server::new(
|
||||||
|
|| App::new()
|
||||||
|
// enable logger
|
||||||
|
.middleware(middleware::Logger::default())
|
||||||
|
// websocket route
|
||||||
|
.resource("/ws/", |r| r.method(http::Method::GET).f(ws_index))
|
||||||
|
// static files
|
||||||
|
.handler("/", fs::StaticFiles::new("../static/")
|
||||||
|
.index_file("index.html")))
|
||||||
|
// start http server on 127.0.0.1:8080
|
||||||
|
.bind("127.0.0.1:8080").unwrap()
|
||||||
|
.start();
|
||||||
|
|
||||||
|
println!("Started http server: 127.0.0.1:8080");
|
||||||
|
let _ = sys.run();
|
||||||
|
}
|
72
websocket/websocket-client.py
Executable file
72
websocket/websocket-client.py
Executable file
@ -0,0 +1,72 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""websocket cmd client for wssrv.py example."""
|
||||||
|
import argparse
|
||||||
|
import asyncio
|
||||||
|
import signal
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
|
|
||||||
|
def start_client(loop, url):
|
||||||
|
name = input('Please enter your name: ')
|
||||||
|
|
||||||
|
# send request
|
||||||
|
ws = yield from aiohttp.ClientSession().ws_connect(url, autoclose=False, autoping=False)
|
||||||
|
|
||||||
|
# input reader
|
||||||
|
def stdin_callback():
|
||||||
|
line = sys.stdin.buffer.readline().decode('utf-8')
|
||||||
|
if not line:
|
||||||
|
loop.stop()
|
||||||
|
else:
|
||||||
|
ws.send_str(name + ': ' + line)
|
||||||
|
loop.add_reader(sys.stdin.fileno(), stdin_callback)
|
||||||
|
|
||||||
|
@asyncio.coroutine
|
||||||
|
def dispatch():
|
||||||
|
while True:
|
||||||
|
msg = yield from ws.receive()
|
||||||
|
|
||||||
|
if msg.type == aiohttp.WSMsgType.TEXT:
|
||||||
|
print('Text: ', msg.data.strip())
|
||||||
|
elif msg.type == aiohttp.WSMsgType.BINARY:
|
||||||
|
print('Binary: ', msg.data)
|
||||||
|
elif msg.type == aiohttp.WSMsgType.PING:
|
||||||
|
ws.pong()
|
||||||
|
elif msg.type == aiohttp.WSMsgType.PONG:
|
||||||
|
print('Pong received')
|
||||||
|
else:
|
||||||
|
if msg.type == aiohttp.WSMsgType.CLOSE:
|
||||||
|
yield from ws.close()
|
||||||
|
elif msg.type == aiohttp.WSMsgType.ERROR:
|
||||||
|
print('Error during receive %s' % ws.exception())
|
||||||
|
elif msg.type == aiohttp.WSMsgType.CLOSED:
|
||||||
|
pass
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
yield from dispatch()
|
||||||
|
|
||||||
|
|
||||||
|
ARGS = argparse.ArgumentParser(
|
||||||
|
description="websocket console client for wssrv.py example.")
|
||||||
|
ARGS.add_argument(
|
||||||
|
'--host', action="store", dest='host',
|
||||||
|
default='127.0.0.1', help='Host name')
|
||||||
|
ARGS.add_argument(
|
||||||
|
'--port', action="store", dest='port',
|
||||||
|
default=8080, type=int, help='Port number')
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
args = ARGS.parse_args()
|
||||||
|
if ':' in args.host:
|
||||||
|
args.host, port = args.host.split(':', 1)
|
||||||
|
args.port = int(port)
|
||||||
|
|
||||||
|
url = 'http://{}:{}/ws/'.format(args.host, args.port)
|
||||||
|
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
loop.add_signal_handler(signal.SIGINT, loop.stop)
|
||||||
|
asyncio.Task(start_client(loop, url))
|
||||||
|
loop.run_forever()
|
Loading…
Reference in New Issue
Block a user