如何根据Rocket服务器的状态以不同的值响应?

时间:2019-08-21 15:50:27

标签: rust rust-rocket

我想根据服务器的状态以不同的值响应“相同的请求”。

我期望这些响应会循环发生:

  • 有人发送了GET /请求,并以“ Hello,World!”响应。
  • 然后他们发送一个GET /请求,并以“ Hello,Rust!”响应。
  • 然后他们发送一个GET /请求,并以“你好,火箭!”作为响应。
  • 然后他们发送一个GET /请求,并以“ Hello,State!”响应。

我不想使用在main中(在Rocket路由处理程序中)初始化的变量。我想使用一个变量,其值可以在main中(在Rocket路由处理程序中)更改。

此代码编译无误,但实际行为是它始终响应“ Hello,World!”。

#![feature(proc_macro_hygiene)]
#![feature(decl_macro)]

#[macro_use]
extern crate rocket;

#[macro_use]
extern crate lazy_static;

use std::sync::{Arc, Mutex};
use rocket::State;

use std::thread;
use std::time::Duration;

lazy_static! {
    static ref RESPONSE: Arc<Mutex<String>>
         = Arc::new(Mutex::new(String::from("Hello, World!")));
}

#[get("/")]
fn get_response(state: State<Arc<Mutex<String>>>) -> String {
    let response = state.lock().unwrap();
    let ret: String = String::from(response.as_str());
    ret
}

fn main() {
    let managed_response: Arc<Mutex<String>> = RESPONSE.clone();
    rocket::ignite()
        .manage(managed_response)
        .mount("/", routes![get_response])
        .launch();

    let mut server_state = 0;
    loop {
        // Pseudo transition of the state of server
        thread::sleep(Duration::from_secs(5));
        let mut response = RESPONSE.lock().unwrap();
        match server_state {
            0 => *response = String::from("Hello, Rust!"), // state 0
            1 => *response = String::from("Hello, Rocket!"), // state 1
            2 => *response = String::from("Hello, State!"), // state 2
            _ => panic!(),
        }
        server_state += 1;
        if server_state >= 3 {
            server_state = 0;
        }
    }
}

1 个答案:

答案 0 :(得分:0)

通过反复试验,我自己解决了这个问题。 这是代码。 如果您还有其他更好的解决方案,我希望您教我。

#![feature(proc_macro_hygiene)]
#![feature(decl_macro)]

#[macro_use]
extern crate rocket;

#[macro_use]
extern crate lazy_static;

use std::sync::{Arc, Mutex};

use std::thread;
use std::time::Duration;

lazy_static! {
    static ref RESPONSE: Arc<Mutex<String>> = Arc::new(Mutex::new(String::from("Hello, World!")));
}

#[get("/")]
fn get_response() -> String {
    let response = RESPONSE.lock().unwrap();
    let ret: String = String::from(response.as_str());
    ret
}

fn main_code() {
    let mut server_state = 0;
    loop {
        thread::sleep(Duration::from_secs(5));
        let mut response = RESPONSE.lock().unwrap();
        match server_state {
            0 => *response = String::from("Hello, Rust!"),
            1 => *response = String::from("Hello, Rocket!"),
            2 => *response = String::from("Hello, State!"),
            _ => panic!(),
        }
        server_state += 1;
        if server_state >= 3 {
            server_state = 0;
        }
    }
}

fn main() {
    let handle = thread::spawn(|| {
        main_code();
    });
    rocket::ignite()
        .mount("/", routes![get_response])
        .launch();
}