maelstrom-worker 0.14.0

The Maelstrom worker. This process executes jobs as directed by the broker.
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

pub use crate::{
    check_open_file_limit,
    config::CacheDir,
    dispatcher::Message,
    types::{DispatcherReceiver as Receiver, DispatcherSender as Sender},
    Tasks,
};
pub use maelstrom_util::cache::GotArtifact;

use crate::dispatcher::{ArtifactFetcher, BrokerSender};
use anyhow::Result;
use maelstrom_util::{
    config::common::{CacheSize, InlineLimit, Slots},
    root::RootBuf,
    sync::EventSender,
};
use slog::Logger;
use tokio::{
    sync::mpsc::{self},
    task::JoinSet,
};

pub fn channel() -> (Sender, Receiver) {
    mpsc::unbounded_channel()
}

#[allow(clippy::too_many_arguments)]
pub fn start_task(
    artifact_fetcher: impl ArtifactFetcher + Send + Sync + 'static,
    broker_sender: impl BrokerSender + Send + Sync + 'static,
    cache_root: RootBuf<CacheDir>,
    cache_size: CacheSize,
    done: EventSender,
    inline_limit: InlineLimit,
    log: &Logger,
    receiver: Receiver,
    sender: Sender,
    slots: Slots,
    tasks: JoinSet<Result<()>>,
) -> Result<Tasks> {
    crate::start_dispatcher_task_common(
        move |_| artifact_fetcher,
        broker_sender,
        cache_size,
        cache_root,
        receiver,
        sender,
        Some(done),
        inline_limit,
        log,
        false, /* log_initial_cache_message_at_info */
        slots,
        tasks,
    )
}