fixed an error in transcode manager

This commit is contained in:
2026-04-11 09:01:15 -04:00
parent 70a58feee6
commit 7848b7c551
+65 -50
View File
@@ -87,9 +87,9 @@ set_state() {
fi
}
# Initialise state file if missing — ramdisk_setup.sh creates it but just in case
init_state() {
if [[ ! -f "$STATE_FILE" ]]; then
local NOW
NOW=$(date +%s)
cat > "$STATE_FILE" <<EOF
current_target=$RAMDISK_PATH
@@ -103,9 +103,6 @@ EOF
# -----------------------------------------------------------------------------------------------
# FLIP TRACKING
# Tracks how many times symlink has flipped in the current hour window.
# Resets counter when hour window rolls over.
# Notifies if flip count exceeds TRANSCODE_FLIP_WARN.
# -----------------------------------------------------------------------------------------------
track_flip() {
local NOW
@@ -117,7 +114,6 @@ track_flip() {
[[ -z "$FLIP_COUNT" ]] && FLIP_COUNT=0
[[ -z "$HOUR_START" ]] && HOUR_START=$NOW
# Reset counter if hour window has passed
if (( NOW - HOUR_START >= 3600 )); then
FLIP_COUNT=0
set_state "flip_hour_start" "$NOW"
@@ -137,40 +133,46 @@ track_flip() {
# -----------------------------------------------------------------------------------------------
# SYMLINK MANAGEMENT
# Changes the transcode symlink target.
# Only flips if current target differs from desired target.
# Tracks flip for frequency monitoring.
# Sets the symlink to the desired target.
# Only flips if current target differs — tracks flip for frequency monitoring.
# Force mode skips the state check and always sets the symlink (used for recovery).
# -----------------------------------------------------------------------------------------------
set_symlink() {
local target="$1"
local target="$1" force="${2:-false}"
local current_target
current_target=$(get_state "current_target")
if [[ "$current_target" == "$target" ]]; then
if [[ "$force" == false && "$current_target" == "$target" ]]; then
log "Symlink already points to $target — no change"
return 0
fi
if [[ "$DRY_RUN" == true ]]; then
warn "DRY RUN — would flip symlink: $TRANSCODE_LINK$target"
warn "DRY RUN — would set symlink: $TRANSCODE_LINK$target"
return 0
fi
info "$ICON_LINK Flipping symlink: $TRANSCODE_LINK$target"
info "$ICON_LINK Setting symlink: $TRANSCODE_LINK$target"
# Ensure parent directory exists
mkdir -p "$(dirname "$TRANSCODE_LINK")"
if ln -sfn "$target" "$TRANSCODE_LINK"; then
success "$ICON_LINK Symlink updated$target"
success "$ICON_LINK Symlink set$target"
set_state "current_target" "$target"
track_flip
# Only track flip and notify if this was a real direction change
if [[ "$current_target" != "$target" ]]; then
track_flip
if [[ "$target" == "$RAMDISK_PATH" ]]; then
notify "Transcode symlink flipped to RAMDISK on $(hostname) — ramdisk has space" "Transcode Manager" "normal"
else
notify "Transcode symlink flipped to SSD on $(hostname) — ramdisk usage high" "Transcode Manager" "warning"
fi
fi
else
error "Failed to update symlink"
notify "Transcode symlink flip failed on $(hostname)" "Transcode Manager" "warning"
error "Failed to set symlink: $TRANSCODE_LINK$target"
notify "Transcode symlink failed to set on $(hostname)" "Transcode Manager" "warning"
return 1
fi
}
@@ -187,28 +189,35 @@ START=$(date +%s)
# -----------------------------------------------------------------------------------------------
# Symlink integrity check
# If symlink is missing or target doesn't exist — force correct it immediately
# -----------------------------------------------------------------------------------------------
info "$ICON_LINK Checking symlink integrity..."
SYMLINK_OK=true
if [[ ! -L "$TRANSCODE_LINK" ]]; then
error "$ICON_LINK Symlink missing: $TRANSCODE_LINK"
notify "Transcode symlink missing on $(hostname) — attempting recovery" "Transcode Manager" "warning"
# Attempt recovery — point at whichever location exists
if mountpoint -q "$RAMDISK_PATH" 2>/dev/null; then
ln -sfn "$RAMDISK_PATH" "$TRANSCODE_LINK" && success "Symlink recovered → ramdisk"
error "$ICON_LINK Symlink missing: $TRANSCODE_LINK — recovering"
SYMLINK_OK=false
else
ln -sfn "$TRANSCODE_SSD" "$TRANSCODE_LINK" && success "Symlink recovered → SSD"
fi
fi
LINK_TARGET=$(readlink "$TRANSCODE_LINK" 2>/dev/null)
if [[ ! -d "$LINK_TARGET" ]]; then
error "$ICON_LINK Symlink target does not exist: $LINK_TARGET"
notify "Transcode symlink target missing on $(hostname): $LINK_TARGET" "Transcode Manager" "warning"
error "$ICON_LINK Symlink target does not exist: $LINK_TARGET — recovering"
SYMLINK_OK=false
else
info "$ICON_LINK Symlink valid: $TRANSCODE_LINK$LINK_TARGET"
fi
fi
info "$ICON_LINK Current target: $LINK_TARGET"
# Force recovery — point at ramdisk if mounted, SSD otherwise
if [[ "$SYMLINK_OK" == false ]]; then
if mountpoint -q "$RAMDISK_PATH" 2>/dev/null; then
warn "Recovering symlink → ramdisk"
set_symlink "$RAMDISK_PATH" force
else
warn "Ramdisk not mounted — recovering symlink → SSD"
set_symlink "$TRANSCODE_SSD" force
fi
fi
# -----------------------------------------------------------------------------------------------
# Ramdisk mount check
@@ -224,7 +233,7 @@ if ! mountpoint -q "$RAMDISK_PATH" 2>/dev/null; then
fi
if [[ "$RAMDISK_MOUNTED" == true ]]; then
# Verify ramdisk size is correct
# Verify ramdisk size matches expected
MOUNTED_SIZE=$(df -BG "$RAMDISK_PATH" | awk 'NR==2 {print $2}' | tr -d 'G')
EXPECTED_SIZE=$(echo "$RAMDISK_SIZE" | tr -d 'G')
if [[ "$MOUNTED_SIZE" != "$EXPECTED_SIZE" ]]; then
@@ -233,15 +242,17 @@ if [[ "$RAMDISK_MOUNTED" == true ]]; then
success "$ICON_RAM Ramdisk mounted and correct size (${RAMDISK_SIZE})"
fi
# Get current usage
RAMDISK_USED_KB=$(df "$RAMDISK_PATH" | awk 'NR==2 {print $3}')
RAMDISK_AVAIL_KB=$(df "$RAMDISK_PATH" | awk 'NR==2 {print $4}')
RAMDISK_USED_GB=$(awk "BEGIN {printf \"%.1f\", $RAMDISK_USED_KB / 1024 / 1024}")
RAMDISK_AVAIL_GB=$(awk "BEGIN {printf \"%.1f\", $RAMDISK_AVAIL_KB / 1024 / 1024}")
# Get current usage in KB — use raw bytes to avoid scientific notation issues
RAMDISK_USED_KB=$(df "$RAMDISK_PATH" --output=used | tail -1 | tr -d ' ')
RAMDISK_AVAIL_KB=$(df "$RAMDISK_PATH" --output=avail | tail -1 | tr -d ' ')
# Convert to GB for display using awk — avoids bash integer/float issues
RAMDISK_USED_GB=$(awk "BEGIN {printf \"%.2f\", $RAMDISK_USED_KB / 1048576}")
RAMDISK_AVAIL_GB=$(awk "BEGIN {printf \"%.2f\", $RAMDISK_AVAIL_KB / 1048576}")
info "$ICON_RAM Ramdisk usage: ${RAMDISK_USED_GB}GB used / ${RAMDISK_AVAIL_GB}GB available"
# Count sessions in ramdisk and SSD
# File counts and sizes
RAMDISK_FILES=$(find "$RAMDISK_PATH" -type f 2>/dev/null | wc -l)
SSD_FILES=$(find "$TRANSCODE_SSD" -type f 2>/dev/null | wc -l)
RAMDISK_SIZE_HUMAN=$(du -sh "$RAMDISK_PATH" 2>/dev/null | cut -f1)
@@ -251,19 +262,19 @@ if [[ "$RAMDISK_MOUNTED" == true ]]; then
info "$ICON_DISK SSD files: $SSD_FILES ($SSD_SIZE_HUMAN)"
# -----------------------------------------------------------------------------------------------
# Symlink direction decision
# Symlink direction decision — all comparisons via awk to handle float thresholds
# -----------------------------------------------------------------------------------------------
WARN_THRESHOLD=$(awk "BEGIN {print $RAMDISK_WARN_GB * 1024 * 1024}")
LOW_THRESHOLD=$(awk "BEGIN {print $RAMDISK_LOW_GB * 1024 * 1024}")
WARN_EXCEEDED=$(awk "BEGIN {print ($RAMDISK_USED_GB >= $RAMDISK_WARN_GB) ? 1 : 0}")
LOW_RECOVERED=$(awk "BEGIN {print ($RAMDISK_USED_GB < $RAMDISK_LOW_GB) ? 1 : 0}")
if (( RAMDISK_USED_KB >= WARN_THRESHOLD )); then
# Ramdisk is getting full — check SSD has space before flipping
if [[ "$WARN_EXCEEDED" == "1" ]]; then
# Ramdisk is getting full — check SSD has enough space before flipping
info "$ICON_DISK Checking SSD fallback space..."
SSD_AVAIL_KB=$(df "$TRANSCODE_SSD" | awk 'NR==2 {print $4}')
SSD_AVAIL_GB=$(awk "BEGIN {printf \"%.1f\", $SSD_AVAIL_KB / 1024 / 1024}")
SSD_MIN_KB=$(awk "BEGIN {print $RAMDISK_SSD_MIN_GB * 1024 * 1024}")
SSD_AVAIL_KB=$(df "$TRANSCODE_SSD" --output=avail | tail -1 | tr -d ' ')
SSD_AVAIL_GB=$(awk "BEGIN {printf \"%.2f\", $SSD_AVAIL_KB / 1048576}")
SSD_SUFFICIENT=$(awk "BEGIN {print ($SSD_AVAIL_GB >= $RAMDISK_SSD_MIN_GB) ? 1 : 0}")
if (( SSD_AVAIL_KB < SSD_MIN_KB )); then
if [[ "$SSD_SUFFICIENT" == "0" ]]; then
error "$ICON_DISK SSD fallback only has ${SSD_AVAIL_GB}GB free — minimum is ${RAMDISK_SSD_MIN_GB}GB"
notify "Transcode SSD fallback critically low on $(hostname)${SSD_AVAIL_GB}GB free" "Transcode Manager" "warning"
else
@@ -271,15 +282,15 @@ if [[ "$RAMDISK_MOUNTED" == true ]]; then
set_symlink "$TRANSCODE_SSD"
fi
elif (( RAMDISK_USED_KB < LOW_THRESHOLD )); then
elif [[ "$LOW_RECOVERED" == "1" ]]; then
# Ramdisk has recovered — flip back
set_symlink "$RAMDISK_PATH"
else
# In hysteresis gap — hold current state
CURRENT=$(get_state "current_target")
log "Usage ${RAMDISK_USED_GB}GB in hysteresis gap (${RAMDISK_LOW_GB}-${RAMDISK_WARN_GB}GB) — holding current target: $CURRENT"
info "$ICON_RAM Usage in hysteresis gap — holding current symlink target"
log "Usage ${RAMDISK_USED_GB}GB in hysteresis gap (${RAMDISK_LOW_GB}${RAMDISK_WARN_GB}GB) — holding: $CURRENT"
info "$ICON_RAM Usage in hysteresis gap — holding current symlink"
fi
fi
@@ -290,9 +301,13 @@ END=$(date +%s)
# -----------------------------------------------------------------------------------------------
echo ""
echo "━━━━━ $ICON_SUMMARY TRANSCODE MANAGER SUMMARY ━━━━━"
echo "$ICON_RAM Ramdisk: $([[ "$RAMDISK_MOUNTED" == true ]] && echo "${RAMDISK_USED_GB}GB used" || echo "NOT MOUNTED")"
if [[ "$RAMDISK_MOUNTED" == true ]]; then
echo "$ICON_RAM Ramdisk: ${RAMDISK_USED_GB}GB used / ${RAMDISK_AVAIL_GB}GB available"
else
echo "$ICON_RAM Ramdisk: NOT MOUNTED — using SSD"
fi
echo "$ICON_LINK Symlink: $TRANSCODE_LINK$(readlink "$TRANSCODE_LINK" 2>/dev/null || echo "BROKEN")"
echo "$ICON_LINK Flip count: $(get_state "flip_count_hour") this hour (warn at $TRANSCODE_FLIP_WARN)"
echo "$ICON_LINK Flips: $(get_state "flip_count_hour") this hour (warn at $TRANSCODE_FLIP_WARN)"
echo "$ICON_TIME Duration: $(format_duration $((END - START)))"
echo "$ICON_DONE Status: $ICON_SUCCESS DONE"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"