mirror of
https://github.com/sadoyan/aralez.git
synced 2026-04-29 22:38:36 +08:00
6cc72c8b48
1. Nested upstreams with params 2. SSL upstream support 3. Upstreams move to yaml format 4. Command line start arguments
62 lines
2.4 KiB
Rust
62 lines
2.4 KiB
Rust
use crate::utils::tools::*;
|
|
use axum::body::Body;
|
|
use axum::http::{Response, StatusCode};
|
|
use axum::response::IntoResponse;
|
|
use axum::routing::{delete, get, head, post, put};
|
|
use axum::Router;
|
|
use futures::channel::mpsc::Sender;
|
|
use futures::SinkExt;
|
|
use tokio::net::TcpListener;
|
|
|
|
// struct UpstreamData {
|
|
// servers: UpstreamsDashMap,
|
|
// }
|
|
|
|
#[allow(unused_mut)]
|
|
pub async fn run_server(mut toreturn: Sender<UpstreamsDashMap>) {
|
|
let mut tr = toreturn.clone();
|
|
let app = Router::new()
|
|
.route("/{*wildcard}", get(getconfig))
|
|
.route("/{*wildcard}", post(getconfig))
|
|
.route("/{*wildcard}", put(getconfig))
|
|
.route("/{*wildcard}", head(getconfig))
|
|
.route("/{*wildcard}", delete(getconfig))
|
|
.route(
|
|
"/conf",
|
|
post(|up: String| async move {
|
|
let serverlist = crate::utils::parceyaml::load_yaml_to_dashmap(up.as_str(), "content");
|
|
let _ = tr.send(serverlist).await.unwrap();
|
|
Response::builder().status(StatusCode::CREATED).body(Body::from("Config, conf file, updated!\n")).unwrap()
|
|
})
|
|
.with_state("state"),
|
|
);
|
|
let listener = TcpListener::bind("0.0.0.0:3000").await.unwrap();
|
|
println!("Axum API server running on port 3000");
|
|
axum::serve(listener, app).await.unwrap();
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
async fn getconfig() -> impl IntoResponse {
|
|
"Hello from Axum API inside Pingora!\n".to_string();
|
|
Response::builder().status(StatusCode::BAD_GATEWAY).body(Body::from("No live upstream found!\n")).unwrap()
|
|
}
|
|
// curl -XPOST -H 'Content-Type: application/json' --data-binary @./push.json 127.0.0.1:3000/json
|
|
// curl -XPOST --data-binary @./etc/upstreams.txt 127.0.0.1:3000/conf
|
|
|
|
/*
|
|
async fn config(Json(payload): Json<HashMap<String, UpstreamData>>) -> impl IntoResponse {
|
|
let upstreams = DashMap::new();
|
|
for (key, value) in payload {
|
|
upstreams.insert(key, (value.servers, AtomicUsize::new(value.counter)));
|
|
}
|
|
println!("{:?}", upstreams);
|
|
Response::builder().status(StatusCode::CREATED).body(Body::from("Config updated!\n")).unwrap()
|
|
}
|
|
async fn parse_upstreams(up: String) -> impl IntoResponse {
|
|
println!("Parsing: {}", up);
|
|
let serverlist = read_upstreams_from_file(up.as_str());
|
|
println!("{:?}", serverlist);
|
|
Response::builder().status(StatusCode::CREATED).body(Body::from("Config updated!\n")).unwrap()
|
|
}
|
|
*/
|