1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use actix_web::{web, App, HttpRequest, HttpResponse, HttpServer};
use serde::{Deserialize, Serialize};
use crate::raft::raft_enum::Which;
use crate::raft::resp::{resp_find_leader, resp_confirm_leader, resp_heart_beat, resp_peer_urls, resp_peers_vote, resp_leader_change, resp_snapshot_ids, resp_append_entry, resp_query_id};

#[derive(Serialize, Deserialize)]
pub struct Msg {
    which: Which,
    data: String,
}

pub async fn handle_post(params: web::Form<Msg>) -> actix_web::Result<HttpResponse> {
    let which = &params.which;
    let data = &params.data;
    let res = dispatch_post(which, data).await;

    Ok(HttpResponse::Ok()
        .content_type("text/plain")
        .body(res))
}
async fn dispatch_post(which: &Which, data: &str) -> String {
    let res = match which {
        Which::ask_leader => resp_find_leader().await,
        Which::confirm_leader => resp_confirm_leader().await,
        Which::heart_beat => resp_heart_beat().await,
        Which::peer_urls => resp_peer_urls(data).await,
        Which::peer_vote=> resp_peers_vote(data).await,
        Which::leader_change=> resp_leader_change(data).await,
        Which::snapshot_ids=> resp_snapshot_ids().await,
        Which::append_entry=> resp_append_entry(data).await,
        Which::query_id=> resp_query_id(data).await,
        _ => Err(anyhow!("not match")),
    };
    match res {
        Ok(s) => s,
        Err(e) => {
            let res = format!("raft error from leader: {}", e);
            res
        }
    }
}