feat: add daemon module with cross-platform process lifecycle
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
129
src/jianda_proxy/daemon.py
Normal file
129
src/jianda_proxy/daemon.py
Normal file
@@ -0,0 +1,129 @@
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import signal
|
||||
from jianda_proxy.config import get_pid_path
|
||||
|
||||
|
||||
def read_pid():
|
||||
path = get_pid_path()
|
||||
if not os.path.exists(path):
|
||||
return None
|
||||
try:
|
||||
with open(path) as f:
|
||||
return int(f.read().strip())
|
||||
except (ValueError, OSError):
|
||||
return None
|
||||
|
||||
|
||||
def write_pid(pid):
|
||||
path = get_pid_path()
|
||||
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||
with open(path, "w") as f:
|
||||
f.write(str(pid))
|
||||
|
||||
|
||||
def remove_pid():
|
||||
path = get_pid_path()
|
||||
if os.path.exists(path):
|
||||
os.remove(path)
|
||||
|
||||
|
||||
def is_process_alive(pid):
|
||||
if os.name == "nt":
|
||||
import ctypes
|
||||
import ctypes.wintypes
|
||||
|
||||
kernel32 = ctypes.windll.kernel32
|
||||
PROCESS_QUERY_LIMITED_INFORMATION = 0x1000
|
||||
STILL_ACTIVE = 259
|
||||
handle = kernel32.OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, False, pid)
|
||||
if not handle:
|
||||
return False
|
||||
try:
|
||||
exit_code = ctypes.wintypes.DWORD()
|
||||
kernel32.GetExitCodeProcess(handle, ctypes.byref(exit_code))
|
||||
return exit_code.value == STILL_ACTIVE
|
||||
finally:
|
||||
kernel32.CloseHandle(handle)
|
||||
else:
|
||||
try:
|
||||
os.kill(pid, 0)
|
||||
return True
|
||||
except ProcessLookupError:
|
||||
return False
|
||||
except PermissionError:
|
||||
return True
|
||||
|
||||
|
||||
def is_running():
|
||||
pid = read_pid()
|
||||
if pid is None:
|
||||
return False
|
||||
return is_process_alive(pid)
|
||||
|
||||
|
||||
def stop_daemon():
|
||||
pid = read_pid()
|
||||
if pid is None:
|
||||
return "not_running"
|
||||
|
||||
if not is_process_alive(pid):
|
||||
remove_pid()
|
||||
return "not_running"
|
||||
|
||||
if os.name == "nt":
|
||||
os.system(f"taskkill /PID {pid} /T")
|
||||
else:
|
||||
try:
|
||||
os.kill(pid, signal.SIGTERM)
|
||||
except ProcessLookupError:
|
||||
remove_pid()
|
||||
return "stopped"
|
||||
|
||||
for _ in range(6):
|
||||
time.sleep(0.5)
|
||||
if not is_process_alive(pid):
|
||||
remove_pid()
|
||||
return "stopped"
|
||||
|
||||
if os.name == "nt":
|
||||
os.system(f"taskkill /F /PID {pid} /T")
|
||||
else:
|
||||
try:
|
||||
os.kill(pid, signal.SIGKILL)
|
||||
except ProcessLookupError:
|
||||
pass
|
||||
|
||||
remove_pid()
|
||||
return "stopped"
|
||||
|
||||
|
||||
def get_status():
|
||||
pid = read_pid()
|
||||
if pid is None or not is_process_alive(pid):
|
||||
return {"running": False, "pid": None}
|
||||
return {"running": True, "pid": pid}
|
||||
|
||||
|
||||
def daemonize():
|
||||
if os.name == "nt":
|
||||
import subprocess
|
||||
|
||||
creation_flags = subprocess.CREATE_NEW_PROCESS_GROUP | subprocess.DETACHED_PROCESS
|
||||
subprocess.Popen(
|
||||
[sys.executable] + sys.argv + ["--_daemon_child"],
|
||||
creationflags=creation_flags,
|
||||
close_fds=True,
|
||||
stdin=subprocess.DEVNULL,
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL,
|
||||
)
|
||||
else:
|
||||
pid = os.fork()
|
||||
if pid > 0:
|
||||
os._exit(0)
|
||||
os.setsid()
|
||||
pid = os.fork()
|
||||
if pid > 0:
|
||||
os._exit(0)
|
||||
96
tests/test_daemon.py
Normal file
96
tests/test_daemon.py
Normal file
@@ -0,0 +1,96 @@
|
||||
import os
|
||||
import tempfile
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock
|
||||
from jianda_proxy.daemon import (
|
||||
read_pid,
|
||||
write_pid,
|
||||
remove_pid,
|
||||
is_process_alive,
|
||||
is_running,
|
||||
stop_daemon,
|
||||
get_status,
|
||||
)
|
||||
|
||||
|
||||
class TestPidFile:
|
||||
def test_write_and_read_pid(self, tmp_path, monkeypatch):
|
||||
pid_file = str(tmp_path / "test.pid")
|
||||
monkeypatch.setattr("jianda_proxy.daemon.get_pid_path", lambda: pid_file)
|
||||
write_pid(12345)
|
||||
assert read_pid() == 12345
|
||||
|
||||
def test_read_missing_pid_returns_none(self, tmp_path, monkeypatch):
|
||||
pid_file = str(tmp_path / "nonexistent.pid")
|
||||
monkeypatch.setattr("jianda_proxy.daemon.get_pid_path", lambda: pid_file)
|
||||
assert read_pid() is None
|
||||
|
||||
def test_remove_pid(self, tmp_path, monkeypatch):
|
||||
pid_file = str(tmp_path / "test.pid")
|
||||
monkeypatch.setattr("jianda_proxy.daemon.get_pid_path", lambda: pid_file)
|
||||
write_pid(12345)
|
||||
remove_pid()
|
||||
assert read_pid() is None
|
||||
|
||||
|
||||
class TestProcessCheck:
|
||||
@patch("os.kill")
|
||||
def test_alive_on_unix(self, mock_kill):
|
||||
mock_kill.return_value = None
|
||||
assert is_process_alive(12345) is True
|
||||
|
||||
@patch("os.kill", side_effect=ProcessLookupError)
|
||||
def test_dead_on_unix(self, mock_kill):
|
||||
assert is_process_alive(12345) is False
|
||||
|
||||
@patch("os.kill", side_effect=PermissionError)
|
||||
def test_permission_means_alive(self, mock_kill):
|
||||
assert is_process_alive(12345) is True
|
||||
|
||||
|
||||
class TestIsRunning:
|
||||
@patch("jianda_proxy.daemon.is_process_alive", return_value=True)
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=12345)
|
||||
def test_running_when_pid_and_alive(self, mock_read, mock_alive):
|
||||
assert is_running() is True
|
||||
|
||||
@patch("jianda_proxy.daemon.is_process_alive", return_value=False)
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=12345)
|
||||
def test_not_running_when_pid_dead(self, mock_read, mock_alive):
|
||||
assert is_running() is False
|
||||
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=None)
|
||||
def test_not_running_when_no_pid(self, mock_read):
|
||||
assert is_running() is False
|
||||
|
||||
|
||||
class TestStopDaemon:
|
||||
@patch("jianda_proxy.daemon.is_process_alive", return_value=True)
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=12345)
|
||||
@patch("time.sleep")
|
||||
@patch("os.kill")
|
||||
def test_stop_sends_sigterm(self, mock_kill, mock_sleep, mock_read, mock_alive):
|
||||
mock_alive.side_effect = [True, False]
|
||||
stop_daemon()
|
||||
mock_kill.assert_called_once_with(12345, 15) # SIGTERM=15
|
||||
|
||||
@patch("jianda_proxy.daemon.remove_pid")
|
||||
@patch("jianda_proxy.daemon.is_process_alive", return_value=False)
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=None)
|
||||
def test_stop_noop_when_not_running(self, mock_read, mock_alive, mock_remove):
|
||||
result = stop_daemon()
|
||||
assert result == "not_running"
|
||||
|
||||
|
||||
class TestGetStatus:
|
||||
@patch("jianda_proxy.daemon.is_process_alive", return_value=True)
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=12345)
|
||||
def test_status_running(self, mock_read, mock_alive):
|
||||
status = get_status()
|
||||
assert status["running"] is True
|
||||
assert status["pid"] == 12345
|
||||
|
||||
@patch("jianda_proxy.daemon.read_pid", return_value=None)
|
||||
def test_status_not_running(self, mock_read):
|
||||
status = get_status()
|
||||
assert status["running"] is False
|
||||
Reference in New Issue
Block a user