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
55
56
57
58
59
60
61
62
63
64
65
66

use super::Router;
use crate::channel_types::SmallSender;
use crate::{
    error::{NodeError, NodeReason},
    NodeReplyResult, RouterReply,
};
use ockam_core::{Address, Result};

pub(super) async fn exec(
    router: &mut Router,
    addr: &Address,
    detached: bool,
    reply: &SmallSender<NodeReplyResult>,
) -> Result<()> {
    trace!("Stopping worker '{}'", addr);

    // Resolve any secondary address to the primary address
    let primary_address = match router.map.addr_map.get(addr) {
        Some(p) => p.clone(),
        None => {
            reply
                .send(RouterReply::no_such_address(addr.clone()))
                .await
                .map_err(|_| NodeError::NodeState(NodeReason::Unknown).internal())?;

            return Ok(());
        }
    };

    // Get the internal address record
    let record = match router.map.internal.get_mut(&primary_address) {
        Some(r) => r,
        None => {
            // Actually should not happen
            reply
                .send(RouterReply::no_such_address(addr.clone()))
                .await
                .map_err(|_| NodeError::NodeState(NodeReason::Unknown).internal())?;

            return Ok(());
        }
    };

    // Remove all secondary addresses
    for addr in record.address_set().iter() {
        router.map.addr_map.remove(addr);
    }

    reply
        .send(RouterReply::ok())
        .await
        .map_err(|_| NodeError::NodeState(NodeReason::Unknown).internal())?;

    // If we are dropping a real worker, then we simply close the
    // mailbox channel to trigger a graceful worker self-shutdown.
    //
    // For detached workers (i.e. Context's without a mailbox relay
    // running) we simply drop the record
    if !detached {
        record.sender_drop();
    } else {
        router.map.free_address(primary_address);
    }

    Ok(())
}