Files
Unraid_Intel-ARC_Deployment/tdarr_node_killer.sh
Admin9705 3e1fd2e289 Update tdarr_node_killer.sh
going back
2024-08-17 07:22:04 -04:00

129 lines
5.2 KiB
Bash

#!/bin/bash
# ============================================================
# Script Name: Plex Transcoding & Docker Management Script
# Author: Admin9705 & Reddit
# Description:
# This script monitors Plex transcoding activity on a specified
# GPU device. If Plex is detected using the GPU, it resets a
# restart counter. If Plex does not use the GPU for a defined
# number of checks, it increments a counter. Once the counter
# reaches a threshold, the script restarts a specified Docker
# container. The script is designed to run indefinitely and
# includes logging for monitoring and troubleshooting.
# ============================================================
# NOTE: Setup for AV1 with Intel ARC; could be modified for
# Other Cards
# ============================================================
# ----------------- User Configuration Area -----------------
# Modify the following variables based on your environment
# Docker container name to monitor and restart
DOCKER_CONTAINER_NAME="380-128-N1"
# Path to the GPU device being used by Plex for transcoding
TRANSCODING_DEVICE="/dev/dri/renderD128"
# Initial delay to allow Docker containers to load before starting
initial_delay=60 # Default is 60 seconds, adjust if needed
# ------------------------------------------------------------
# Maximum number of checks before deciding Plex is not confirmed
MAX_CHECKS=10
# Threshold for t_restart before the Docker container is restarted
t_restart_threshold=60 # Default is 60, adjust if needed
# Initialize variables
t_restart=0
plex0_count=0
# Initial delay before starting the script's main loop
echo "[$(date)] Initial delay: Waiting $initial_delay seconds for Docker containers to load..."
sleep $initial_delay
while true; do
# Initialize variables at the start of each full run
numbercheck=0
numberplex=0
while true; do
# Increment numbercheck with each iteration
numbercheck=$((numbercheck + 1))
# Check if the device is being accessed by Plex
check=$(lsof | grep "$TRANSCODING_DEVICE")
if echo "$check" | grep -q "Plex"; then
numberplex=$((numberplex + 1))
echo "[$(date)] plex1 detected: Plex is currently transcoding."
plex0_count=0 # Reset plex0_count if Plex is detected
t_restart=0 # Reset t_restart when Plex transcoding is detected
# Check if numberplex has reached 4 to confirm Plex usage
if [ "$numberplex" -ge 4 ]; then
echo "[$(date)] PLEX Confirmed: Plex has reached the required number of transcoding events."
# Stop the Docker container if it's running
if docker ps | grep -q "$DOCKER_CONTAINER_NAME"; then
echo "[$(date)] Stopping Docker container $DOCKER_CONTAINER_NAME..."
docker stop "$DOCKER_CONTAINER_NAME"
else
echo "[$(date)] Docker container $DOCKER_CONTAINER_NAME is not running. Skipping stop command."
fi
sleep 2 # Wait for 2 seconds before restarting the loop
break
fi
else
echo "[$(date)] plex0 detected: Plex is not currently transcoding."
plex0_count=$((plex0_count + 1))
fi
# Proceed only after plex0 has been echoed 3 times consecutively
if [ "$plex0_count" -ge 3 ]; then
echo "[$(date)] PLEX not confirmed: Plex has not been detected for 3 consecutive checks."
# Check if the Docker container is running before incrementing t_restart
if ! docker ps | grep -q "$DOCKER_CONTAINER_NAME"; then
t_restart=$((t_restart + 1)) # Increment t_restart only if container is not running
echo "[$(date)] t_restart count: $t_restart"
# Calculate and echo the estimated time to reach the restart threshold
remaining_increments=$((t_restart_threshold - t_restart))
estimated_time=$((remaining_increments * 5)) # 5 seconds per increment (as sleep 5 is used)
echo "[$(date)] Estimated time to reach t_restart threshold: $estimated_time seconds"
else
echo "[$(date)] Docker container $DOCKER_CONTAINER_NAME is already running. t_restart not incremented."
fi
# Sleep for 5 seconds for visual purposes
sleep 5
# If t_restart hits the threshold, restart the Docker container
if [ "$t_restart" -ge "$t_restart_threshold" ]; then
echo "[$(date)] Restarting Docker container $DOCKER_CONTAINER_NAME..."
docker restart "$DOCKER_CONTAINER_NAME"
t_restart=0 # Reset t_restart after restarting
fi
break
fi
# If numbercheck reaches MAX_CHECKS without plex0_count hitting 3, exit loop
if [ "$numbercheck" -ge "$MAX_CHECKS" ]; then
echo "[$(date)] MAX_CHECKS reached without Plex confirmation. Exiting loop."
sleep 5
break
fi
# Optional: Add a sleep delay if you want to pause between iterations
# sleep 1
done
# Restart the entire script after the 5-second delay
done