toolbox/reports/project-disk-use.sh

   1 #!/bin/sh
   2
   3 # Report on project disk use.
   4 # Output can be sent as email to admin with -m
   5 # Automatically runs with nice and ionice (if available)
   6
   7 # Usage: project-disk-use [-m] [top-n-only]
   8
   9 # With -m mail the report to $cfg_admin instead of sending it to stdout
  10
  11 # Shows total disk usage in K bytes for $cfg_reporoot
  12 # The top-n-only (all if not given) repos are then listed by
  13 # decreasing order of disk space.
  14
  15 # Note that any *.git directories that are found in $cfg_reporoot that are
  16 # not listed in $cfg_chroot/etc/group ARE included and have a '!' suffix added.
  17
  18 set -e
  19
  20 datefmt='%Y-%m-%d %H:%M:%S %z'
  21 startdate="$(date "+$datefmt")"
  22
  23 . @basedir@/shlib.sh
  24
  25 mailresult=
  26 if [ "$1" = "-m" ]; then
  27         shift
  28         mailresult=1
  29 fi
  30
  31 hasnice=
  32 ! command -v nice >/dev/null || hasnice=1
  33 hasionice=
  34 ! command -v ionice >/dev/null || hasionice=1
  35
  36 nl='
  37 '
  38
  39 fmtcomma() {
  40         # remove leading 0s
  41         _y="${1%%[1-9]*}"; _x="${1#$_y}"; _x="${_x:-0}"
  42         # add commas after each group of 3
  43         _y=
  44         while [ $_x -gt 999 ]; do
  45                 _y="$(printf '%03d' $(($_x - $_x / 1000 * 1000)))${_y:+,$_y}"
  46                 _x="$(($_x/1000))"
  47         done
  48         [ $_x -eq 0 ] || _y="$_x${_y:+,$_y}"
  49         echo "${_y:-0}"
  50 }
  51
  52 fmtpct() {
  53         if [ -z "$1" ] || [ "$1" = "0" ]; then return; fi
  54         if [ -z "$2" ] || [ "$2" = "0" ]; then return; fi
  55         _fmtpct=$(( ( $1 * 100 + $2 / 2 ) / $2 ))
  56         if [ $_fmtpct -le 100 ]; then
  57                 _of=
  58                 [ -z "$4" ] || _of=" of $(fmtcomma $2)"
  59                 echo "${3:- }($_fmtpct%$_of)"
  60         fi
  61 }
  62
  63 get_use_k() {
  64         _targ="$1"
  65         shift
  66         _cmd="du $var_du_follow -k -s"
  67         if [ -n "$var_du_exclude" ]; then
  68                 while [ -n "$1" ]; do
  69                         _cmd="$_cmd $var_du_exclude \"$1\""
  70                         shift
  71                 done
  72         elif [ $# -ne 0 ]; then
  73                 echo "get_use_k: error: no du exclude option available" >&2
  74                 exit 1
  75         fi
  76         _cmd="$_cmd \"$_targ\" 2>/dev/null | cut -f 1"
  77         [ -z "$hasionice" ] || _cmd="ionice -c 3 $_cmd"
  78         [ -z "$hasnice" ] || _cmd="nice -n 19 $_cmd"
  79         eval "$_cmd"
  80 }
  81
  82 is_active() (
  83         cd "$cfg_reporoot/$1.git" || return 1
  84         check_interval lastchange 2592000 # 30 days
  85 )
  86
  87 projlist="$(cut -d : -f 1 <"$cfg_chroot/etc/group")"
  88
  89 is_listed_proj() {
  90   echo "$projlist" | grep -q -e "^$1$"
  91 }
  92
  93 totaluse="$(get_use_k "$cfg_reporoot")"
  94 if [ -n "$var_du_exclude" ]; then
  95         # Attribute all hard-linked stuff in _recyclebin to non-_recyclebin
  96         inuse="$(get_use_k "$cfg_reporoot" "_recyclebin")"
  97         binned="$(( $totaluse - $inuse ))"
  98 else
  99         # No du exclude option, if binned contains hard links into other repos
 100         # its size may appear larger than it actually is and the kpct will be off
 101         binned="$(get_use_k "$cfg_reporoot/_recyclebin")"
 102         inuse="$(( $totaluse - $binned ))"
 103 fi
 104
 105 cd "$cfg_reporoot"
 106 total=0
 107 ktotal=0
 108 howmany=0
 109 orphans=0
 110 mirrors=0
 111 forks=0
 112 mactive=0
 113 pactive=0
 114 results=
 115 while IFS='' read -r proj; do
 116         proj="${proj#./}"
 117         proj="${proj%.git}"
 118         x=
 119         a=
 120         is_listed_proj "$proj" || { x='!'; orphans="$(( $orphans + 1 ))"; }
 121         case "$proj" in */*) forks="$(( $forks + 1 ))"; esac
 122         mirror="$(get_mirror_type "$proj.git" 2>/dev/null)" || :
 123         [ -z "$mirror" ] || mirrors="$(( $mirrors + 1 ))"
 124         if is_active "$proj"; then
 125                 a='*'
 126                 if [ -n "$mirror" ]; then
 127                         mactive="$(( $mactive + 1 ))"
 128                 else
 129                         pactive="$(( $pactive + 1 ))"
 130                 fi
 131         fi
 132         : ${mirror:=M}
 133         usek="$(get_use_k "$proj.git")"
 134         repok=
 135         [ -L "$proj.git/objects" ] || ! [ -d "$proj.git/objects" ] ||
 136         repok="$(git --git-dir="$cfg_reporoot/$proj.git" config --get girocco.reposizek 2>/dev/null)" || :
 137         repokpct=
 138         case "$repok" in
 139                 [0-9]*)
 140                         repok="${repok%%[!0-9]*}"
 141                         repokpct=$(( ( $repok * 100 + $usek / 2 ) / $usek ))
 142                         if [ $repokpct -le 100 ]; then
 143                                 repokpct="$repokpct%"
 144                         else
 145                                 repokpct="100+"
 146                         fi;;
 147                 *)
 148                         repok=0
 149                         repokpct=-;;
 150         esac
 151         ktotal=$(( $ktotal + $repok ))
 152         total="$(( $total + $usek ))"
 153         howmany="$(( $howmany + 1 ))"
 154         line="$usek $repokpct $mirror $proj$a$x$nl"
 155         results="$results$line"
 156 done <<EOT
 157 $(find . -type d \( -path ./_recyclebin -o -name '*.git' -print \) -prune 2>/dev/null)
 158 EOT
 159
 160 kpct=$(( ( $ktotal * 100 + $inuse / 2 ) / $inuse ))
 161 enddate="$(date "+$datefmt")"
 162 domail=cat
 163 [ -z "$mailresult" ] || domail='mailref "diskuse@$cfg_gitweburl" -s "[$cfg_name] Project Disk Use Report" "$cfg_admin"'
 164 {
 165         cat <<EOT
 166 Project Disk Use Report
 167 =======================
 168
 169        Start Time: $startdate
 170          End Time: $enddate
 171
 172   Repository Root: $cfg_reporoot
 173 Unbinned Disk Use: $(fmtcomma "$inuse") (1024-byte blocks)
 174   reposizek Total: $(fmtcomma "$ktotal") ($kpct%)
 175
 176  Recycle Bin Root: $cfg_reporoot/_recyclebin
 177   Binned Disk Use: $(fmtcomma "$binned") (1024-byte blocks)
 178
 179    Total Disk Use: $(fmtcomma "$totaluse") (1024-byte blocks)
 180
 181  Repository Count: $(fmtcomma "$howmany")
 182             Forks: $(fmtcomma "$forks")$(fmtpct "$forks" "$howmany")
 183           Mirrors: $(fmtcomma "$mirrors")$(fmtpct "$mirrors" "$howmany")
 184          Orphaned: $(fmtcomma "$orphans")
 185  Repository Total: $(fmtcomma "$total") (1024-byte blocks)
 186
 187    *30-Day Active: $(fmtcomma "$(( $pactive + $mactive ))")$(fmtpct "$(( $pactive + $mactive ))" "$howmany")
 188              Push: $(fmtcomma "$pactive")$(fmtpct "$pactive" "$howmany")$(fmtpct "$pactive" "$(( $pactive + $mactive ))" "/" 1)$(fmtpct "$pactive" "$(( $howmany - $mirrors ))" "/" 1)
 189            Mirror: $(fmtcomma "$mactive")$(fmtpct "$mactive" "$howmany")$(fmtpct "$mactive" "$(( $pactive + $mactive ))" "/" 1)$(fmtpct "$mactive" "$mirrors" "/" 1)
 190
 191 $(df -h "$cfg_reporoot")
 192 EOT
 193
 194         if [ $# -lt 1 ] || [ $1 != "0" ]; then
 195                 topn=cat
 196                 message="Individual Repository Use"
 197                 case "${1%%[!0-9]*}" in ?*)
 198                         message="Individual Repository Use (Top $1)"
 199                         topn="head -n ${1%%[!0-9]*}"
 200                 esac
 201                 echo ""
 202                 echo ""
 203                 echo "$message"
 204                 echo "$message" | tr -c '\n' -
 205                 echo ""
 206                 printf '%s' "$results" | sort -k1,1nr -k4,4 |
 207                 while read -r a b c d; do
 208                         printf "%10s %4s  %s  %s\n" "$(fmtcomma "$a")" "$b" "$c" "$d"
 209                 done |
 210                 sed -e 's/ [M-] /   /g' | $topn
 211         fi
 212 } | eval "$domail"