commit c30958a32dbdc85244ddac563245b06ed0af8e71 · nekomimi.pet/microcosm-rs

+8

slingshot/src/error.rs

···

       47
       47
        
       }

     

       48
       48
        
       

     

       49
       49
        
       #[derive(Debug, Error)]

     

       50
       50
       +
       pub enum HealthCheckError {

     

       51
       51
       +
           #[error("failed to send checkin: {0}")]

     

       52
       52
       +
           HealthCheckError(#[from] reqwest::Error),

     

       53
       53
       +
       }

     

       54
       54
       +
       

     

       55
       55
       +
       #[derive(Debug, Error)]

     

       50
       56
        
       pub enum MainTaskError {

     

       51
       57
        
           #[error(transparent)]

     

       52
       58
        
           ConsumerTaskError(#[from] ConsumerError),

     
···

       54
       60
        
           ServerTaskError(#[from] ServerError),

     

       55
       61
        
           #[error(transparent)]

     

       56
       62
        
           IdentityTaskError(#[from] IdentityError),

     

       63
       63
       +
           #[error(transparent)]

     

       64
       64
       +
           HealthCheckError(#[from] HealthCheckError),

     

       57
       65
        
           #[error("firehose cache failed to close: {0}")]

     

       58
       66
        
           FirehoseCacheCloseError(foyer::Error),

     

       59
       67
        
       }

+32

slingshot/src/healthcheck.rs

···

       1
       1
       +
       use crate::error::HealthCheckError;

     

       2
       2
       +
       use reqwest::Client;

     

       3
       3
       +
       use std::time::Duration;

     

       4
       4
       +
       use tokio::time::sleep;

     

       5
       5
       +
       use tokio_util::sync::CancellationToken;

     

       6
       6
       +
       

     

       7
       7
       +
       pub async fn healthcheck(

     

       8
       8
       +
           endpoint: String,

     

       9
       9
       +
           shutdown: CancellationToken,

     

       10
       10
       +
       ) -> Result<(), HealthCheckError> {

     

       11
       11
       +
           let client = Client::builder()

     

       12
       12
       +
               .user_agent(format!(

     

       13
       13
       +
                   "microcosm slingshot v{} (dev: @bad-example.com)",

     

       14
       14
       +
                   env!("CARGO_PKG_VERSION")

     

       15
       15
       +
               ))

     

       16
       16
       +
               .no_proxy()

     

       17
       17
       +
               .timeout(Duration::from_secs(10))

     

       18
       18
       +
               .build()?;

     

       19
       19
       +
       

     

       20
       20
       +
           loop {

     

       21
       21
       +
               tokio::select! {

     

       22
       22
       +
                   res = client.get(&endpoint).send() => {

     

       23
       23
       +
                       let _ = res

     

       24
       24
       +
                           .and_then(|r| r.error_for_status())

     

       25
       25
       +
                           .inspect_err(|e| log::error!("failed to send healthcheck: {e}"));

     

       26
       26
       +
                   },

     

       27
       27
       +
                   _ = shutdown.cancelled() => break,

     

       28
       28
       +
               }

     

       29
       29
       +
               sleep(Duration::from_secs(51)).await;

     

       30
       30
       +
           }

     

       31
       31
       +
           Ok(())

     

       32
       32
       +
       }

+2

slingshot/src/lib.rs

···

       1
       1
        
       mod consumer;

     

       2
       2
        
       pub mod error;

     

       3
       3
        
       mod firehose_cache;

     

       4
       4
       +
       mod healthcheck;

     

       4
       5
        
       mod identity;

     

       5
       6
        
       mod record;

     

       6
       7
        
       mod server;

     

       7
       8
        
       

     

       8
       9
        
       pub use consumer::consume;

     

       9
       10
        
       pub use firehose_cache::firehose_cache;

     

       11
       11
       +
       pub use healthcheck::healthcheck;

     

       10
       12
        
       pub use identity::Identity;

     

       11
       13
        
       pub use record::{CachedRecord, ErrorResponseObject, Repo};

     

       12
       14
        
       pub use server::serve;

+14 -1

slingshot/src/main.rs

···

       1
       1
        
       // use foyer::HybridCache;

     

       2
       2
        
       // use foyer::{Engine, DirectFsDeviceOptions, HybridCacheBuilder};

     

       3
       3
        
       use metrics_exporter_prometheus::PrometheusBuilder;

     

       4
       4
       -
       use slingshot::{Identity, Repo, consume, error::MainTaskError, firehose_cache, serve};

     

       4
       4
       +
       use slingshot::{

     

       5
       5
       +
           Identity, Repo, consume, error::MainTaskError, firehose_cache, healthcheck, serve,

     

       6
       6
       +
       };

     

       5
       7
        
       use std::path::PathBuf;

     

       6
       8
        
       

     

       7
       9
        
       use clap::Parser;

     
···

       44
       46
        
           /// recommended in production, but mind the file permissions.

     

       45
       47
        
           #[arg(long)]

     

       46
       48
        
           certs: Option<PathBuf>,

     

       49
       49
       +
           /// an web address to send healtcheck pings to every ~51s or so

     

       50
       50
       +
           #[arg(long)]

     

       51
       51
       +
           healthcheck: Option<String>,

     

       47
       52
        
       }

     

       48
       53
        
       

     

       49
       54
        
       #[tokio::main]

     
···

       126
       131
        
               .await?;

     

       127
       132
        
               Ok(())

     

       128
       133
        
           });

     

       134
       134
       +
       

     

       135
       135
       +
           if let Some(hc) = args.healthcheck {

     

       136
       136
       +
               let healthcheck_shutdown = shutdown.clone();

     

       137
       137
       +
               tasks.spawn(async move {

     

       138
       138
       +
                   healthcheck(hc, healthcheck_shutdown).await?;

     

       139
       139
       +
                   Ok(())

     

       140
       140
       +
               });

     

       141
       141
       +
           }

     

       129
       142
        
       

     

       130
       143
        
           tokio::select! {

     

       131
       144
        
               _ = shutdown.cancelled() => log::warn!("shutdown requested"),

+2 -1

slingshot/src/server.rs

···

       19
       19
        
               Listener, TcpListener,

     

       20
       20
        
               acme::{AutoCert, LETS_ENCRYPT_PRODUCTION},

     

       21
       21
        
           },

     

       22
       22
       -
           middleware::{Cors, Tracing},

     

       22
       22
       +
           middleware::{CatchPanic, Cors, Tracing},

     

       23
       23
        
       };

     

       24
       24
        
       use poem_openapi::{

     

       25
       25
        
           ApiResponse, ContactObject, ExternalDocumentObject, Object, OpenApi, OpenApiService, Tags,

     
···

       758
       758
        
                       .allow_methods([Method::GET])

     

       759
       759
        
                       .allow_credentials(false),

     

       760
       760
        
               )

     

       761
       761
       +
               .with(CatchPanic::new())

     

       761
       762
        
               .with(Tracing);

     

       762
       763
        
           Server::new(listener)

     

       763
       764
        
               .name("slingshot")