mirror of
https://github.com/sadoyan/aralez.git
synced 2026-05-30 03:44:06 +08:00
Merge pull request #31 from Taqman-probe/fix/rate-limit-setting-message
fix: Fix global rate limit and 4xx limit fallback in upstream config log
This commit is contained in:
@@ -222,7 +222,7 @@ async fn clone_compare(upstreams: &UpstreamsDashMap, prev_upstreams: &UpstreamsD
|
|||||||
};
|
};
|
||||||
clone_dashmap_into(upstreams, prev_upstreams);
|
clone_dashmap_into(upstreams, prev_upstreams);
|
||||||
clone_dashmap_into(upstreams, &tosend.upstreams);
|
clone_dashmap_into(upstreams, &tosend.upstreams);
|
||||||
print_upstreams(&tosend.upstreams);
|
print_upstreams(&tosend.upstreams, &tosend.extraparams);
|
||||||
return Some(tosend);
|
return Some(tosend);
|
||||||
};
|
};
|
||||||
None
|
None
|
||||||
|
|||||||
@@ -258,7 +258,7 @@ async fn populate_file_upstreams(config: &mut Configuration, parsed: &Config) {
|
|||||||
clone_dashmap_into(&r, &config.upstreams);
|
clone_dashmap_into(&r, &config.upstreams);
|
||||||
}
|
}
|
||||||
info!("Upstream Config:");
|
info!("Upstream Config:");
|
||||||
print_upstreams(&config.upstreams);
|
print_upstreams(&config.upstreams, &config.extraparams);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
pub fn parce_main_config(path: &str) -> AppConfig {
|
pub fn parce_main_config(path: &str) -> AppConfig {
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use crate::tls::load;
|
use crate::tls::load;
|
||||||
use crate::tls::load::CertificateConfig;
|
use crate::tls::load::CertificateConfig;
|
||||||
use crate::utils::structs::{InnerMap, InnerMapForJson, UpstreamSnapshotForJson, UpstreamsDashMap, UpstreamsIdMap};
|
use crate::utils::structs::{InnerMap, InnerMapForJson, Extraparams, UpstreamSnapshotForJson, UpstreamsDashMap, UpstreamsIdMap};
|
||||||
use dashmap::DashMap;
|
use dashmap::DashMap;
|
||||||
use log::{error, info};
|
use log::{error, info};
|
||||||
use notify::{event::ModifyKind, Config, EventKind, RecommendedWatcher, RecursiveMode, Watcher};
|
use notify::{event::ModifyKind, Config, EventKind, RecommendedWatcher, RecursiveMode, Watcher};
|
||||||
@@ -20,7 +20,7 @@ use std::sync::Arc;
|
|||||||
use std::time::{Duration, Instant};
|
use std::time::{Duration, Instant};
|
||||||
use std::{fs, process, thread, time};
|
use std::{fs, process, thread, time};
|
||||||
|
|
||||||
pub fn print_upstreams(upstreams: &UpstreamsDashMap) {
|
pub fn print_upstreams(upstreams: &UpstreamsDashMap, extraparams: &Extraparams) {
|
||||||
let mut out = String::new();
|
let mut out = String::new();
|
||||||
for host_entry in upstreams.iter() {
|
for host_entry in upstreams.iter() {
|
||||||
writeln!(out, "Hostname: {}", host_entry.key()).unwrap();
|
writeln!(out, "Hostname: {}", host_entry.key()).unwrap();
|
||||||
@@ -35,8 +35,8 @@ pub fn print_upstreams(upstreams: &UpstreamsDashMap) {
|
|||||||
f.is_ssl,
|
f.is_ssl,
|
||||||
f.is_http2,
|
f.is_http2,
|
||||||
f.to_https,
|
f.to_https,
|
||||||
f.rate_limit.unwrap_or(0),
|
f.rate_limit.unwrap_or(extraparams.rate_limit.unwrap_or(0)),
|
||||||
f.x4xx_limit.unwrap_or(0)
|
f.x4xx_limit.unwrap_or(extraparams.x4xx_limit.unwrap_or(0))
|
||||||
)
|
)
|
||||||
.unwrap();
|
.unwrap();
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user