1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
//
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();
    }
}