daemon-base 0.1.1

A flexible and configurable Rust daemon library with lifecycle management, logging, callbacks, and optional async support. Works on Linux, macOS, and Windows.
Documentation 
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

//! Concurrency models for the daemon library.
//!
//! This module provides functionality to support both multi-threaded and
//! asynchronous operations.

use std::thread;
use log::info;
use crate::error::DaemonError;

/// Represents a concurrency model.
pub enum ConcurrencyModel {
    MultiThreaded,
    Async,
}

/// Executes a task using the specified concurrency model.
pub fn execute_task<F>(model: ConcurrencyModel, task: F) -> Result<(), DaemonError>
where
    F: FnOnce() -> Result<(), DaemonError> + Send + 'static,
{
    match model {
        ConcurrencyModel::MultiThreaded => {
            info!("Executing task in multi-threaded mode.");
            let handle = thread::spawn(move || {
                if let Err(e) = task() {
                    log::error!("Task failed: {}", e);
                }
            });
            handle.join().map_err(|_| DaemonError::CustomError("Thread panicked".to_string()))?;
        }
        ConcurrencyModel::Async => {
            #[cfg(feature = "async")]
            {
                info!("Executing task in async mode.");
                let runtime = tokio::runtime::Runtime::new()
                    .map_err(|e| DaemonError::CustomError(format!("Failed to create runtime: {}", e)))?;
                runtime.block_on(async {
                    if let Err(e) = task() {
                        log::error!("Task failed: {}", e);
                    }
                });
            }
            #[cfg(not(feature = "async"))]
            {
                return Err(DaemonError::CustomError("Async feature is not enabled".to_string()));
            }
        }
    }
    Ok(())
}