Server: Switch to tracing subscriber

This commit is contained in:
asonix 2021-01-17 15:08:34 -06:00
parent 4183da60ed
commit 5987f2f544
5 changed files with 24 additions and 19 deletions

View file

@ -16,7 +16,6 @@ anyhow = "1.0.35"
background-jobs = "0.8.0"
base64 = "0.13.0"
chrono = { version = "0.4.19", features = ["serde"] }
env_logger = "0.8.2"
event-listener = "2.5.1"
futures = "0.3.11"
html-minifier = "3.0.8"
@ -37,6 +36,7 @@ sha2 = "0.9"
sled = { version = "0.34.6", features = ["compression"] }
structopt = "0.3"
thiserror = "1.0"
tracing-subscriber = { version = "0.2.15", features = ["env-filter"] }
url = { version = "2.2.0", features = ["serde"] }
uuid = { version = "0.8.1", features = ["serde", "v4"] }

View file

@ -76,6 +76,7 @@ impl SignatureVerify for RsaVerifier {
})?;
if !matches!(algorithm, Some(Algorithm::Hs2019)) {
log::error!("Bad Algorithm");
return Err(VerifyError);
}
@ -83,6 +84,7 @@ impl SignatureVerify for RsaVerifier {
let public_key = if let Some(key) = apub.public_key_for_id(&key_id)? {
key
} else {
log::warn!("No public key for ID {}", key_id);
spawner.download_apub(hyaenidae_profiles::OnBehalfOf::Server, key_id, vec![]);
return Err(VerifyError.into());
};

View file

@ -47,10 +47,10 @@ pub(super) fn build(
pub(super) struct Spawn(QueueHandle);
impl Spawn {
pub(crate) fn ingest(&self, any_base: AnyBase, key_owner: Url) {
pub(crate) fn ingest(&self, any_base: AnyBase, key_id: Url) {
if let Err(e) = self.0.queue(Ingest {
any_base,
key_owner: Some(key_owner),
key_id: Some(key_id),
stack: vec![],
}) {
log::error!("Failed to queue ingest: {}", e);
@ -90,7 +90,7 @@ impl Spawner for Spawn {
fn process(&self, any_base: AnyBase, stack: Vec<AnyBase>) {
if let Err(e) = self.0.queue(Ingest {
any_base,
key_owner: None,
key_id: None,
stack,
}) {
log::error!("Failed to queue process job: {}", e);
@ -151,7 +151,7 @@ struct DownloadApub {
#[derive(Clone, serde::Deserialize, serde::Serialize)]
struct Ingest {
any_base: AnyBase,
key_owner: Option<Url>,
key_id: Option<Url>,
stack: Vec<AnyBase>,
}
@ -413,7 +413,7 @@ impl ActixJob for Ingest {
}
state
.profiles
.ingest(self.any_base, self.key_owner, self.stack)
.ingest(self.any_base, self.key_id, self.stack)
.await?;
Ok(())
})

View file

@ -46,7 +46,10 @@ async fn main() -> anyhow::Result<()> {
std::env::set_var("RUST_LOG", "info");
}
}
env_logger::init();
tracing_subscriber::fmt()
.with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
.init();
let secret_key = if let Some(secret_key) = &config.secret_key {
secret_key.inner.to_owned()

View file

@ -569,23 +569,23 @@ impl Profile {
let inner = profiles.by_id(profile_id)?.req()?;
let banner = match inner.banner() {
Some(banner_id) => {
let file = files.by_id(banner_id)?.req()?;
if let Some(key) = file.pictrs_key() {
Some(key.to_owned())
} else {
None
}
let fallible = || -> Result<String, Error> {
let file = files.by_id(banner_id)?.req()?;
let key = file.pictrs_key().req()?;
Ok(key.to_owned())
};
(fallible)().ok()
}
None => None,
};
let icon = match inner.icon() {
Some(icon_id) => {
let file = files.by_id(icon_id)?.req()?;
if let Some(key) = file.pictrs_key() {
Some(key.to_owned())
} else {
None
}
let fallible = || -> Result<String, Error> {
let file = files.by_id(icon_id)?.req()?;
let key = file.pictrs_key().req()?;
Ok(key.to_owned())
};
(fallible)().ok()
}
None => None,
};