Merge pull request #211 from ChuckPa/chuckpa/add-log

v1.11.03 -Add summary logfile messages

DBRepair.sh

@@ -1,13 +1,13 @@
-#!/bin/sh
+#!/bin/bash
 #########################################################################
 # Database Repair Utility for Plex Media Server.                        #
 # Maintainer: ChuckPa                                                   #
-# Version:    v1.11.00                                                  #
-# Date:       23-May-2025                                               #
+# Version:    v1.11.03                                                  #
+# Date:       31-May-2025                                               #
 #########################################################################
 
 # Version for display purposes
-Version="v1.11.00"
+Version="v1.11.03"
 
 # Have the databases passed integrity checks
 CheckedDB=0
@@ -219,11 +219,11 @@ FreeSpaceAvailable() {
   [ "$1" != "" ] && Multiplier=$1
 
   # Available space where DB resides
-  SpaceAvailable=$(df $DFFLAGS "$AppSuppDir" | tail -1 | awk '{print $4}')
+  SpaceAvailable=$(df $DFFLAGS "$DBDIR" | tail -1 | awk '{print $4}')
 
   # Get size of DB and blobs, Minimally needing sum of both
-  LibSize="$(stat $STATFMT $STATBYTES "$CPPL.db")"
-  BlobsSize="$(stat $STATFMT $STATBYTES "$CPPL.blobs.db")"
+  LibSize="$(stat $STATFMT $STATBYTES "${DBDIR}/$CPPL.db")"
+  BlobsSize="$(stat $STATFMT $STATBYTES "${DBDIR}/$CPPL.blobs.db")"
   SpaceNeeded=$((LibSize + BlobsSize))
 
   # Compute need (minimum $Multiplier existing; current, backup, temp and room to write new)
@@ -277,21 +277,25 @@ ConfirmYesNo() {
   Answer=""
   while [ "$Answer" != "Y" ] && [ "$Answer" != "N" ]
   do
-    printf "%s (Y/N) ? " "$1"
-    read Input
+    if [ $Scripted -eq 1 ]; then
+      Answer=Y
+    else
+      printf "%s (Y/N) ? " "$1"
+      read Input
 
-    # EOF = No
-    case "$Input" in
-      YES|YE|Y|yes|ye|y)
-        Answer=Y
-        ;;
-      NO|N|no|n)
-        Answer=N
-        ;;
-      *)
-        Answer=""
-        ;;
-    esac
+      # EOF = No
+      case "$Input" in
+        YES|YE|Y|yes|ye|y)
+          Answer=Y
+          ;;
+        NO|N|no|n)
+          Answer=N
+          ;;
+        *)
+          Answer=""
+          ;;
+      esac
+    fi
 
     # Unrecognized
     if [ "$Answer" != "Y" ] && [ "$Answer" != "N" ]; then
@@ -976,11 +980,6 @@ DoRepair() {
       return 1
     fi
 
-    # Temporary DB actions
-    Output "Performing DB cleanup tasks."
-    WriteLog "DB Cleanup tasks."
-    "$PLEX_SQLITE" $CPPL.db "DELETE from statistics_bandwidth where account_id is NULL;"
-
     # Continue
     Output "Exporting current databases using timestamp: $TimeStamp"
     Fail=0
@@ -1098,7 +1097,7 @@ DoRepair() {
       [ -e $CPPL.blobs.db ] && mv $CPPL.blobs.db "$TMPDIR/$CPPL.blobs.db-BACKUP-$TimeStamp"
 
       Output "Making repaired databases active"
-      WriteLog "Making repaired databases active"
+      WriteLog "Repair  - Making repaired databases active"
       mv "$TMPDIR/$CPPL.db-REPAIR-$TimeStamp"       $CPPL.db
       mv "$TMPDIR/$CPPL.blobs.db-REPAIR-$TimeStamp" $CPPL.blobs.db
 
@@ -1739,6 +1738,73 @@ DoPrunePhotoTranscoder() {
 
 }
 
+##### Special function
+
+# Remove bloat records from PMS database on a full disk
+DoDeflate() {
+
+  if IsRunning; then
+    Output  "Please stop PMS first and try again."
+    return 1
+  fi
+
+  # Get into DBDIR
+  cd "$DBDIR"
+
+  # Run this loop until we get a partial block
+  Removed=100000000
+  Limit=100000000
+  TotalRemoved=0
+
+  # Turn it off
+  Result="$("$PLEX_SQLITE" "$CPPL.db" 'PRAGMA journal_mode = off;')"
+
+  InitialSize=$(stat $STATFMT $STATBYTES $CPPL.db)
+
+  SQL="PRAGMA journal_mode = off;"
+  SQL="$SQL Begin transaction; "
+  SQL="$SQL DELETE from statistics_bandwidth where rowid in ("
+  SQL="$SQL select rowid from statistics_bandwidth where account_id is null limit 100000000 );"
+  SQL="$SQL Commit;"
+  SQL="$SQL select changes();"
+
+  Count=1
+  while [ $Removed -eq $Limit ]
+  do
+    Output "Removing - block $Count"
+    Removed=$("$PLEX_SQLITE" "$CPPL.db" "$SQL" | tail -1)
+    TotalRemoved=$((TotalRemoved + Removed))
+    Count=$((Count+1))
+  done
+
+  Output "Removed $TotalRemoved records."
+  Output "Vacuuming DB to reclaim working space."
+  "$PLEX_SQLITE" "$CPPL.db" "vacuum;"
+  Result=$?
+
+  if [ $Result -ne 0 ]; then
+    Output "ERROR:  Unable to shrink database size. Error $Result"
+    WriteLog "Deflate - ERROR.  Unable to shrink database size. ERROR $Result"
+    WriteLog "Deflate - FAIL."
+    return 1
+  fi
+
+  # Check size
+  Size=$(stat $STATFMT $STATBYTES $CPPL.db)
+  Output "Initial Database size = $((InitialSize / 1000000)) MB"
+  Output "Final Database size   = $((Size / 1000000)) MB"
+
+  # Turn journal mode back on
+  Result="$("$PLEX_SQLITE" "$CPPL.db" "PRAGMA journal_mode = WAL;")"
+
+  # Now write abbreviated log entries
+  WriteLog "Deflate - Report."
+  WriteLog "Deflate - Initial Database size = $((InitialSize / 1000000)) MB"
+  WriteLog "Deflate - Final Database size   = $((Size / 1000000)) MB"
+  WriteLog "Deflate - Records Removed       = $TotalRemoved"
+  WriteLog "Deflate - PASS."
+}
+
 
 #############################################################
 #         Main utility begins here                          #
@@ -1958,8 +2024,10 @@ do
 
       echo ""
       echo " 88 - 'update'    - Check for updates."
-      echo " 99 - 'quit'      - Quit immediately.  Keep all temporary files."
-      echo "      'exit'      - Exit with cleanup options."
+      echo " 98 - 'quit'      - Quit immediately.  Keep all temporary files."
+      echo " 99 - 'exit'      - Exit with cleanup options."
+      echo " "
+      echo "911 - 'deflate'   - Deflate a bloated PMS database"
     fi
 
     if [ $Scripted -eq 0 ]; then
@@ -2030,7 +2098,7 @@ do
         if ! FreeSpaceAvailable; then
           WriteLog "Auto    - FAIL - Insufficient free space on $AppSuppDir"
           Output   "Error:   Unable to run automatic sequence.  Insufficient free space available on $AppSuppDir"
-          Output   "         Space needed = $SpaceNeeded MB,  Space available = $SpaveAvailable MB"
+          Output   "         Space needed = $SpaceNeeded MB,  Space available = $SpaceAvailable MB"
           continue
         fi
 
@@ -2348,7 +2416,7 @@ do
       ;;
 
       # Quit
-      99|quit)
+      98|quit)
 
         Output "Retaining all temporary work files."
         WriteLog "Exit    - Retain temp files."
@@ -2356,7 +2424,7 @@ do
         ;;
 
       # Orderly Exit
-      exit)
+      99|exit)
 
         # If forced exit set,  exit and retain
         if [ $Exit -eq 1 ]; then
@@ -2388,6 +2456,21 @@ do
         exit 0
         ;;
 
+      # Deflate
+      911|defl*)
+
+        Output "This command operates directly on your existing PMS DB."
+        Output "There are no backups however the DB is protected by transaction blocking."
+        Output "DO NOT INTERRUPT once started but you can if you must (It will resume where it left off)."
+        Output " "
+        if ! ConfirmYesNo "Ok to begin deflating the databases?" ; then
+          Output "No action taken."
+          WriteLog "Deflate - No action taken."
+        else
+          DoDeflate
+        fi
+        ;;
+
       # Unknown command
       *)
         WriteLog "Unknown command:  '$Input'"

ReleaseNotes

@@ -8,6 +8,28 @@
 ![Maintenance](https://img.shields.io/badge/Maintained-Yes-green.svg)
 
 # Release Info:
+v1.11.03                  - Add log entries for DEFLATE
+
+v1.11.02                  - Special release
+
+  1. Database bloat       - (Enh) A better method was found to handle bloat processing.
+                            This new method works on SSD/HDDs where the disk is full and PMS can't run.
+                            There is no free space required to perform this task.
+
+                            Each "block" being removed is 100 million records.
+                            The CPU will have one core maxed at 100% for the duration.
+
+                            Before & after info will be printed after final cleanup is finished.
+
+                            New command name:  Deflate  ( # 911 )
+
+v1.11.01
+
+  1. Database bloat       - (New) Database bloat processing now surrounded by TRANSACTION/COMMIT to speed processing.
+  2. Space Needed         - (Fix) Handle cases where "Databases" directory is not on the same volume as AppSuppDir.
+  3. Menu options         - (New) Add option 98 = quit. Rename 99 = exit.
+  4. Reporting            - (Fix) When insufficient space available, Now print out how much is needed
+
 v1.11.00
 
   1. Rename Utility       - Rename this tool to be compliant with Plex inc. Trademark Policy.