54 lines
1.5 KiB
Rust
54 lines
1.5 KiB
Rust
//
|
|
use actix::prelude::*;
|
|
|
|
mod router;
|
|
mod server;
|
|
|
|
use git_next_repo_actor::messages::WebhookNotification;
|
|
|
|
use std::net::SocketAddr;
|
|
|
|
pub use router::AddWebhookRecipient;
|
|
pub use router::WebhookRouter;
|
|
use tracing::Instrument;
|
|
|
|
#[derive(Debug)]
|
|
pub struct WebhookActor {
|
|
socket_addr: SocketAddr,
|
|
span: tracing::Span,
|
|
spawn_handle: Option<actix::SpawnHandle>,
|
|
message_receiver: Recipient<WebhookNotification>,
|
|
}
|
|
impl WebhookActor {
|
|
pub fn new(socket_addr: SocketAddr, message_receiver: Recipient<WebhookNotification>) -> Self {
|
|
let span = tracing::info_span!("WebhookActor");
|
|
Self {
|
|
socket_addr,
|
|
span,
|
|
message_receiver,
|
|
spawn_handle: None,
|
|
}
|
|
}
|
|
}
|
|
impl Actor for WebhookActor {
|
|
type Context = actix::Context<Self>;
|
|
fn started(&mut self, ctx: &mut Self::Context) {
|
|
let _gaurd = self.span.enter();
|
|
let address: Recipient<WebhookNotification> = self.message_receiver.clone();
|
|
let server = server::start(self.socket_addr, address);
|
|
let spawn_handle = ctx.spawn(server.in_current_span().into_actor(self));
|
|
self.spawn_handle.replace(spawn_handle);
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Message)]
|
|
#[rtype(result = "()")]
|
|
pub struct ShutdownWebhook;
|
|
impl Handler<ShutdownWebhook> for WebhookActor {
|
|
type Result = ();
|
|
|
|
fn handle(&mut self, _msg: ShutdownWebhook, ctx: &mut Self::Context) -> Self::Result {
|
|
self.spawn_handle.take();
|
|
ctx.stop();
|
|
}
|
|
}
|