gflow 0.4.18

A lightweight, single-node job scheduler written in Rust.
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

use anyhow::Result;
use std::time::Duration;
use tmux_interface::{KillSession, Tmux};

pub async fn handle_down() -> Result<()> {
    // systemd user service takes priority when installed and active.
    if super::systemd::unit_installed().unwrap_or(false)
        && super::systemd::is_active().unwrap_or(false)
    {
        super::systemd::stop()?;
        println!("gflowd stopped.");
        return Ok(());
    }

    // Direct (flock/pidfile) mode first.
    if let Some(pid) = super::lifecycle::direct_daemon_pid() {
        // Re-verify identity immediately before signalling so a PID that was
        // recycled to an unrelated process since the liveness probe is never
        // signalled (flock auto-released on crash; identity check on reuse).
        if !super::lifecycle::verify_before_signal(pid) {
            super::lifecycle::remove_daemon_lock();
            println!("gflowd is not running (stale lock cleaned up).");
            return Ok(());
        }
        unsafe {
            libc::kill(pid as libc::pid_t, libc::SIGTERM);
        }
        // Wait for graceful shutdown (the daemon kills managed jobs and
        // saves state), then escalate to SIGKILL if needed.
        let mut exited = false;
        for _ in 0..50 {
            if !super::lifecycle::process_alive(pid) {
                exited = true;
                break;
            }
            tokio::time::sleep(Duration::from_millis(100)).await;
        }
        if !exited {
            eprintln!(
                "gflowd (PID {}) did not exit within 5s, sending SIGKILL",
                pid
            );
            unsafe {
                libc::kill(pid as libc::pid_t, libc::SIGKILL);
            }
        }
        super::lifecycle::remove_daemon_lock();
        println!("gflowd stopped.");
        return Ok(());
    }

    if let Err(e) =
        Tmux::with_command(KillSession::new().target_session(super::TMUX_SESSION_NAME)).output()
    {
        eprintln!("Failed to stop gflowd: {e}");
    } else {
        println!("gflowd stopped.");
    }
    Ok(())
}