feat: with arc
Some checks failed
continuous-integration/drone/push Build is failing

Signed-off-by: kjuulh <contact@kjuulh.io>
This commit is contained in:
Kasper Juul Hermansen 2023-11-27 20:11:49 +01:00
parent 999d81bb7a
commit bda242422d
Signed by: kjuulh
GPG Key ID: 57B6E1465221F912
8 changed files with 41 additions and 48 deletions

View File

@ -1,8 +1,8 @@
use async_trait::async_trait;
use dagger_sdk::Container;
use std::{future::Future, pin::Pin};
use std::{future::Future, pin::Pin, sync::Arc};
pub type DynMiddleware = Box<dyn DaggerMiddleware + Send + Sync>;
pub type DynMiddleware = Arc<dyn DaggerMiddleware + Send + Sync>;
#[async_trait]
pub trait DaggerMiddleware {

View File

@ -1,4 +1,4 @@
use std::path::PathBuf;
use std::{path::PathBuf, sync::Arc};
use async_trait::async_trait;
use dagger_rust::source::RustSource;
@ -12,6 +12,7 @@ use crate::{
use self::architecture::{Architecture, Os};
#[derive(Clone)]
pub enum RustServiceStage {
BeforeDeps(DynMiddleware),
AfterDeps(DynMiddleware),
@ -25,6 +26,7 @@ pub enum RustServiceStage {
AfterRelease(DynMiddleware),
}
#[derive(Clone)]
pub struct RustService {
client: dagger_sdk::Query,
base_image: Option<dagger_sdk::Container>,
@ -141,7 +143,7 @@ impl RustService {
async fn run_stage(
&self,
stages: impl IntoIterator<Item = &Box<dyn DaggerMiddleware + Send + Sync>>,
stages: impl IntoIterator<Item = &Arc<dyn DaggerMiddleware + Send + Sync>>,
container: Container,
) -> eyre::Result<Container> {
let before_deps_stream = stream::iter(stages.into_iter().map(Ok));
@ -317,29 +319,7 @@ impl MainAction for RustService {
let container = self.build_release().await?;
container
.publish(format!("docker.io/kjuulh/{}", self.bin_name))
.await?;
Ok(())
}
}
#[async_trait]
impl PullRequestAction for &mut RustService {
async fn execute_pull_request(&self) -> eyre::Result<()> {
self.build_test().await?;
Ok(())
}
}
#[async_trait]
impl MainAction for &mut RustService {
async fn execute_main(&self) -> eyre::Result<()> {
let container = self.build_release().await?;
container
.publish(format!("docker.io/kjuulh/{}", self.bin_name))
.publish(format!("docker.io//{}", self.bin_name))
.await?;
Ok(())

View File

@ -1,3 +1,5 @@
use std::sync::Arc;
use async_trait::async_trait;
use dagger_sdk::Container;
@ -47,6 +49,8 @@ pub trait AptExt {
impl AptExt for RustService {
fn with_apt(&mut self, deps: &[&str]) -> &mut Self {
self.with_stage(super::RustServiceStage::BeforeDeps(Box::new(Apt::new().extend(deps))))
self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
Apt::new().extend(deps),
)))
}
}

View File

@ -1,3 +1,5 @@
use std::sync::Arc;
use async_trait::async_trait;
use dagger_sdk::Container;
@ -62,8 +64,7 @@ impl CargoBInstall {
#[async_trait]
impl DaggerMiddleware for CargoBInstall {
async fn handle(&self, container: Container) -> eyre::Result<Container> {
let c =
container
let c = container
.with_exec(vec!["wget", &self.get_download_url()])
.with_exec(vec!["tar", "-xvf", &self.get_archive()])
.with_exec(
@ -98,8 +99,8 @@ impl CargoBInstallExt for RustService {
) -> &mut Self {
let crates: Vec<String> = crates.into_iter().map(|s| s.into()).collect();
self.with_stage(super::RustServiceStage::BeforeDeps(
Box::new(CargoBInstall::new(self.get_arch(), self.get_os(), version, crates))
))
self.with_stage(super::RustServiceStage::BeforeDeps(Arc::new(
CargoBInstall::new(self.get_arch(), self.get_os(), version, crates),
)))
}
}

View File

@ -1,3 +1,5 @@
use std::sync::Arc;
use async_trait::async_trait;
use dagger_sdk::Container;
@ -28,7 +30,7 @@ pub trait CargoCleanExt {
impl CargoCleanExt for RustService {
fn with_cargo_clean(&mut self) -> &mut Self {
self.with_stage(super::RustServiceStage::BeforeBuild(Box::new(
self.with_stage(super::RustServiceStage::BeforeBuild(Arc::new(
CargoClean::new(),
)));

View File

@ -1,3 +1,5 @@
use std::sync::Arc;
use async_trait::async_trait;
use dagger_sdk::Container;
@ -32,9 +34,9 @@ pub trait ClapSanityTestExt {
impl ClapSanityTestExt for RustService {
fn with_clap_sanity_test(&mut self) -> &mut Self {
self.with_stage(
super::RustServiceStage::AfterPackage(Box::new(ClapSanityTest::new(&self.bin_name)))
);
self.with_stage(super::RustServiceStage::AfterPackage(Arc::new(
ClapSanityTest::new(&self.bin_name),
)));
self
}

View File

@ -1,3 +1,5 @@
use std::sync::Arc;
use async_trait::async_trait;
use crate::dagger_middleware::DaggerMiddleware;
@ -95,8 +97,8 @@ pub trait MoldActionExt {
impl MoldActionExt for RustService {
fn with_mold(&mut self, version: impl Into<String>) -> &mut Self {
self.with_stage(super::RustServiceStage::AfterDeps(
Box::new(MoldInstall::new(self.get_arch(), self.get_os(), version))
))
self.with_stage(super::RustServiceStage::AfterDeps(Arc::new(
MoldInstall::new(self.get_arch(), self.get_os(), version),
)))
}
}

View File

@ -1,3 +1,5 @@
use std::sync::Arc;
use async_trait::async_trait;
use dagger_sdk::Container;
@ -34,7 +36,7 @@ pub trait SqlxExt {
impl SqlxExt for RustService {
fn with_sqlx(&mut self) -> &mut Self {
self.with_stage(super::RustServiceStage::BeforeBuild(Box::new(Sqlx::new(
self.with_stage(super::RustServiceStage::BeforeBuild(Arc::new(Sqlx::new(
self.client.clone(),
))));