mirror of
https://git.asonix.dog/asonix/pict-rs.git
synced 2024-11-24 18:41:06 +00:00
Simplify old_repo, remove client_pool_size
This commit is contained in:
parent
47751f3875
commit
81c6e73b5f
6 changed files with 61 additions and 52 deletions
|
@ -52,7 +52,6 @@ impl Args {
|
|||
Command::Run(Run {
|
||||
address,
|
||||
api_key,
|
||||
client_pool_size,
|
||||
client_timeout,
|
||||
metrics_prometheus_address,
|
||||
media_preprocess_steps,
|
||||
|
@ -111,7 +110,6 @@ impl Args {
|
|||
};
|
||||
|
||||
let client = Client {
|
||||
pool_size: client_pool_size,
|
||||
timeout: client_timeout,
|
||||
};
|
||||
|
||||
|
@ -493,8 +491,6 @@ struct Server {
|
|||
#[derive(Debug, Default, serde::Serialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
struct Client {
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
pool_size: Option<usize>,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
timeout: Option<u64>,
|
||||
}
|
||||
|
@ -862,14 +858,6 @@ struct Run {
|
|||
#[arg(long)]
|
||||
api_key: Option<String>,
|
||||
|
||||
/// Number of connections the internel HTTP client should maintain in its pool
|
||||
///
|
||||
/// This number defaults to 100, and the total number is multiplied by the number of cores
|
||||
/// available to the program. This means that running on a 2 core system will result in 200
|
||||
/// pooled connections, and running on a 32 core system will result in 3200 pooled connections.
|
||||
#[arg(long)]
|
||||
client_pool_size: Option<usize>,
|
||||
|
||||
/// How long (in seconds) the internel HTTP client should wait for responses
|
||||
///
|
||||
/// This number defaults to 30
|
||||
|
|
|
@ -10,7 +10,6 @@ pub(crate) struct Defaults {
|
|||
server: ServerDefaults,
|
||||
client: ClientDefaults,
|
||||
tracing: TracingDefaults,
|
||||
old_repo: SledDefaults,
|
||||
media: MediaDefaults,
|
||||
repo: RepoDefaults,
|
||||
store: StoreDefaults,
|
||||
|
@ -27,7 +26,6 @@ struct ServerDefaults {
|
|||
#[derive(Clone, Debug, serde::Serialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
struct ClientDefaults {
|
||||
pool_size: usize,
|
||||
timeout: u64,
|
||||
}
|
||||
|
||||
|
@ -138,7 +136,7 @@ enum RepoDefaults {
|
|||
|
||||
#[derive(Clone, Debug, serde::Serialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
struct SledDefaults {
|
||||
pub(super) struct SledDefaults {
|
||||
path: PathBuf,
|
||||
cache_capacity: u64,
|
||||
export_path: PathBuf,
|
||||
|
@ -183,10 +181,7 @@ impl Default for ServerDefaults {
|
|||
|
||||
impl Default for ClientDefaults {
|
||||
fn default() -> Self {
|
||||
ClientDefaults {
|
||||
pool_size: 100,
|
||||
timeout: 30,
|
||||
}
|
||||
ClientDefaults { timeout: 30 }
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -351,6 +346,16 @@ impl From<crate::config::commandline::Filesystem> for crate::config::primitives:
|
|||
}
|
||||
}
|
||||
|
||||
impl From<SledDefaults> for crate::config::file::Sled {
|
||||
fn from(defaults: SledDefaults) -> Self {
|
||||
crate::config::file::Sled {
|
||||
path: defaults.path,
|
||||
cache_capacity: defaults.cache_capacity,
|
||||
export_path: defaults.export_path,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl From<crate::config::commandline::Sled> for crate::config::file::Sled {
|
||||
fn from(value: crate::config::commandline::Sled) -> Self {
|
||||
let defaults = SledDefaults::default();
|
||||
|
|
|
@ -18,7 +18,7 @@ pub(crate) struct ConfigFile {
|
|||
#[serde(default)]
|
||||
pub(crate) metrics: Metrics,
|
||||
|
||||
pub(crate) old_repo: Sled,
|
||||
old_repo: OldRepo,
|
||||
|
||||
pub(crate) media: Media,
|
||||
|
||||
|
@ -27,6 +27,15 @@ pub(crate) struct ConfigFile {
|
|||
pub(crate) store: Store,
|
||||
}
|
||||
|
||||
impl ConfigFile {
|
||||
pub(crate) fn old_repo_path(&self) -> Option<&PathBuf> {
|
||||
self.old_repo.path.as_ref().or_else(|| match self.repo {
|
||||
Repo::Sled(ref sled) => Some(&sled.path),
|
||||
_ => None,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
#[serde(tag = "type")]
|
||||
|
@ -106,8 +115,6 @@ pub(crate) struct Server {
|
|||
|
||||
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
||||
pub(crate) struct Client {
|
||||
pub(crate) pool_size: usize,
|
||||
|
||||
pub(crate) timeout: u64,
|
||||
}
|
||||
|
||||
|
@ -417,6 +424,12 @@ impl Media {
|
|||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub(crate) struct OldRepo {
|
||||
pub(crate) path: Option<PathBuf>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
||||
#[serde(rename_all = "snake_case")]
|
||||
pub(crate) struct Sled {
|
||||
|
|
27
src/lib.rs
27
src/lib.rs
|
@ -1598,11 +1598,10 @@ fn transform_error(error: actix_form_data::Error) -> actix_web::Error {
|
|||
error
|
||||
}
|
||||
|
||||
fn build_client(config: &Configuration) -> Result<ClientWithMiddleware, Error> {
|
||||
fn build_client() -> Result<ClientWithMiddleware, Error> {
|
||||
let client = reqwest::Client::builder()
|
||||
.user_agent("pict-rs v0.5.0-main")
|
||||
.use_rustls_tls()
|
||||
.pool_max_idle_per_host(config.client.pool_size)
|
||||
.build()
|
||||
.map_err(UploadError::BuildClient)?;
|
||||
|
||||
|
@ -1979,7 +1978,7 @@ impl PictRsConfiguration {
|
|||
pub async fn run(self) -> color_eyre::Result<()> {
|
||||
let PictRsConfiguration { config, operation } = self;
|
||||
|
||||
let client = build_client(&config)?;
|
||||
let client = build_client()?;
|
||||
|
||||
match operation {
|
||||
Operation::Run => (),
|
||||
|
@ -2073,14 +2072,21 @@ impl PictRsConfiguration {
|
|||
let store = FileStore::build(path, arc_repo.clone()).await?;
|
||||
|
||||
if arc_repo.get("migrate-0.4").await?.is_none() {
|
||||
if let Some(old_repo) = repo_04::open(&config.old_repo)? {
|
||||
repo::migrate_04(old_repo, arc_repo.clone(), store.clone(), config.clone())
|
||||
if let Some(path) = config.old_repo_path() {
|
||||
if let Some(old_repo) = repo_04::open(path)? {
|
||||
repo::migrate_04(
|
||||
old_repo,
|
||||
arc_repo.clone(),
|
||||
store.clone(),
|
||||
config.clone(),
|
||||
)
|
||||
.await?;
|
||||
arc_repo
|
||||
.set("migrate-0.4", Arc::from(b"migrated".to_vec()))
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
match repo {
|
||||
Repo::Sled(sled_repo) => {
|
||||
|
@ -2129,14 +2135,21 @@ impl PictRsConfiguration {
|
|||
.build(client.clone());
|
||||
|
||||
if arc_repo.get("migrate-0.4").await?.is_none() {
|
||||
if let Some(old_repo) = repo_04::open(&config.old_repo)? {
|
||||
repo::migrate_04(old_repo, arc_repo.clone(), store.clone(), config.clone())
|
||||
if let Some(path) = config.old_repo_path() {
|
||||
if let Some(old_repo) = repo_04::open(path)? {
|
||||
repo::migrate_04(
|
||||
old_repo,
|
||||
arc_repo.clone(),
|
||||
store.clone(),
|
||||
config.clone(),
|
||||
)
|
||||
.await?;
|
||||
arc_repo
|
||||
.set("migrate-0.4", Arc::from(b"migrated".to_vec()))
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
match repo {
|
||||
Repo::Sled(sled_repo) => {
|
||||
|
|
|
@ -1,24 +1,17 @@
|
|||
use crate::{
|
||||
config,
|
||||
details::Details,
|
||||
repo::{Alias, DeleteToken},
|
||||
};
|
||||
use futures_core::Stream;
|
||||
use std::{fmt::Debug, sync::Arc};
|
||||
use std::{fmt::Debug, path::PathBuf, sync::Arc};
|
||||
|
||||
pub(crate) use self::sled::SledRepo;
|
||||
|
||||
mod sled;
|
||||
|
||||
#[tracing::instrument]
|
||||
pub(crate) fn open(config: &config::Sled) -> color_eyre::Result<Option<SledRepo>> {
|
||||
let config::Sled {
|
||||
path,
|
||||
cache_capacity,
|
||||
export_path: _,
|
||||
} = config;
|
||||
|
||||
SledRepo::build(path.clone(), *cache_capacity)
|
||||
pub(crate) fn open(path: &PathBuf) -> color_eyre::Result<Option<SledRepo>> {
|
||||
SledRepo::build(path.clone())
|
||||
}
|
||||
|
||||
#[derive(Debug, thiserror::Error)]
|
||||
|
|
|
@ -109,8 +109,8 @@ impl OldDetails {
|
|||
|
||||
impl SledRepo {
|
||||
#[tracing::instrument]
|
||||
pub(crate) fn build(path: PathBuf, cache_capacity: u64) -> color_eyre::Result<Option<Self>> {
|
||||
let Some(db) = Self::open(path, cache_capacity)? else {
|
||||
pub(crate) fn build(path: PathBuf) -> color_eyre::Result<Option<Self>> {
|
||||
let Some(db) = Self::open(path)? else {
|
||||
return Ok(None);
|
||||
};
|
||||
|
||||
|
@ -129,7 +129,7 @@ impl SledRepo {
|
|||
}))
|
||||
}
|
||||
|
||||
fn open(mut path: PathBuf, cache_capacity: u64) -> Result<Option<Db>, SledError> {
|
||||
fn open(mut path: PathBuf) -> Result<Option<Db>, SledError> {
|
||||
path.push("v0.4.0-alpha.1");
|
||||
|
||||
if let Err(e) = std::fs::metadata(&path) {
|
||||
|
@ -138,10 +138,7 @@ impl SledRepo {
|
|||
}
|
||||
}
|
||||
|
||||
let db = ::sled::Config::new()
|
||||
.cache_capacity(cache_capacity)
|
||||
.path(path)
|
||||
.open()?;
|
||||
let db = ::sled::Config::new().path(path).open()?;
|
||||
|
||||
Ok(Some(db))
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue