86 lines
2.6 KiB
Rust
86 lines
2.6 KiB
Rust
/*
|
|
* mCaptcha - A proof of work based DoS protection system
|
|
* Copyright © 2023 Aravinth Manivannan <realravinth@batsense.net>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
use std::collections::BTreeMap;
|
|
use std::collections::BTreeSet;
|
|
|
|
use actix_web::get;
|
|
use actix_web::post;
|
|
use actix_web::web;
|
|
use actix_web::web::Data;
|
|
use actix_web::Responder;
|
|
use openraft::error::Infallible;
|
|
use openraft::BasicNode;
|
|
use openraft::RaftMetrics;
|
|
use web::Json;
|
|
|
|
use crate::app::ExampleApp;
|
|
use crate::ExampleNodeId;
|
|
|
|
// --- Cluster management
|
|
|
|
/// Add a node as **Learner**.
|
|
///
|
|
/// A Learner receives log replication from the leader but does not vote.
|
|
/// This should be done before adding a node as a member into the cluster
|
|
/// (by calling `change-membership`)
|
|
#[post("/add-learner")]
|
|
pub async fn add_learner(
|
|
app: Data<ExampleApp>,
|
|
req: Json<(ExampleNodeId, String)>,
|
|
) -> actix_web::Result<impl Responder> {
|
|
let node_id = req.0 .0;
|
|
let node = BasicNode {
|
|
addr: req.0 .1.clone(),
|
|
};
|
|
let res = app.raft.add_learner(node_id, node, true).await;
|
|
Ok(Json(res))
|
|
}
|
|
|
|
/// Changes specified learners to members, or remove members.
|
|
#[post("/change-membership")]
|
|
pub async fn change_membership(
|
|
app: Data<ExampleApp>,
|
|
req: Json<BTreeSet<ExampleNodeId>>,
|
|
) -> actix_web::Result<impl Responder> {
|
|
let res = app.raft.change_membership(req.0, false).await;
|
|
Ok(Json(res))
|
|
}
|
|
|
|
/// Initialize a single-node cluster.
|
|
#[post("/init")]
|
|
pub async fn init(app: Data<ExampleApp>) -> actix_web::Result<impl Responder> {
|
|
let mut nodes = BTreeMap::new();
|
|
nodes.insert(
|
|
app.id,
|
|
BasicNode {
|
|
addr: app.addr.clone(),
|
|
},
|
|
);
|
|
let res = app.raft.initialize(nodes).await;
|
|
Ok(Json(res))
|
|
}
|
|
|
|
/// Get the latest metrics of the cluster
|
|
#[get("/metrics")]
|
|
pub async fn metrics(app: Data<ExampleApp>) -> actix_web::Result<impl Responder> {
|
|
let metrics = app.raft.metrics().borrow().clone();
|
|
|
|
let res: Result<RaftMetrics<ExampleNodeId, BasicNode>, Infallible> = Ok(metrics);
|
|
Ok(Json(res))
|
|
}
|