56 lines
1.4 KiB
Bash
56 lines
1.4 KiB
Bash
|
#!/bin/bash
|
||
|
|
# Keep two codex exec jobs running by topping up when fewer are active.
|
||
|
|
# Usage:
|
||
|
|
# tools/codex-keep-two.sh <tmux_session> "Task A" "Task B" ["Task C" ...]
|
||
|
|
# Notes:
|
||
|
|
# - Detects only real `codex exec` processes via ps/awk (avoids self-matches).
|
||
|
|
# - Starts tasks in order, cycling if more top-ups are needed.
|
||
|
|
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
if [ $# -lt 2 ]; then
|
||
|
|
echo "Usage: $0 <tmux_session> \"Task A\" \"Task B\" [\"Task C\" ...]" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
SESSION="$1"; shift
|
||
|
|
TASKS=("$@")
|
||
|
|
if [ ${#TASKS[@]} -eq 0 ]; then
|
||
|
|
echo "Provide at least one task string." >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
CODEX_PROC_PATTERN=${CODEX_PROC_PATTERN:-'codex .* exec'}
|
||
|
|
|
||
|
|
list_running() {
|
||
|
|
if command -v pgrep >/dev/null 2>&1; then
|
||
|
|
pgrep -af -- "$CODEX_PROC_PATTERN" | grep -v 'pgrep' || true
|
||
|
|
else
|
||
|
|
ps -eo pid=,args= | grep -E -- "$CODEX_PROC_PATTERN" | grep -v grep || true
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
count_running() {
|
||
|
|
list_running | wc -l | tr -d ' '\
|
||
|
|
|| echo 0
|
||
|
|
}
|
||
|
|
|
||
|
|
RUNNING=$(count_running)
|
||
|
|
echo "[keep-two] 実際のcodexプロセス数: ${RUNNING}"
|
||
|
|
NEED=$((2 - RUNNING))
|
||
|
|
if [ $NEED -le 0 ]; then
|
||
|
|
echo "[keep-two] already running: $RUNNING (>=2)."
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "[keep-two] running=$RUNNING, starting $NEED top-up task(s)."
|
||
|
|
|
||
|
|
idx=0
|
||
|
|
for ((i=0; i<NEED; i++)); do
|
||
|
|
TASK="${TASKS[$idx]}"; idx=$(((idx+1) % ${#TASKS[@]}))
|
||
|
|
echo "[keep-two] start: $TASK"
|
||
|
|
CODEX_ASYNC_DETACH=1 ./tools/codex-async-notify.sh "$TASK" "$SESSION" >/dev/null 2>&1 || true
|
||
|
|
done
|
||
|
|
|
||
|
|
echo "[keep-two] now running: $(count_running)"
|