CB-22380: Modified Salt script for getting DL storage sizes so that i…

…t can't return empty values in response.
hortonworks · Jul 10, 2023 · 4c9b6bf · 4c9b6bf
1 parent 4f0bb03
commit 4c9b6bf
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/...ain/resources/salt/salt/datalake_metrics/get_database_sizes/scripts/get_database_sizes.sh b/...ain/resources/salt/salt/datalake_metrics/get_database_sizes/scripts/get_database_sizes.sh
@@ -61,8 +61,12 @@ getDataSizesForDatabases() {
     TRIMMED_DB=$(echo "$DB" | tr -d '"')
     if doesDatabaseExist "$TRIMMED_DB" ; then
       CUR_DB_SIZE=$(runPSQLCommand "select sum(pg_table_size(quote_ident(tablename)::regclass)) from pg_tables where schemaname not in ('pg_catalog','information_schema');" "$TRIMMED_DB")
-      RESULT="${RESULT}${DB}:${CUR_DB_SIZE},"
-      doLog "Size of database ${DB} is ${CUR_DB_SIZE} bytes."
+      if [ -z "$CUR_DB_SIZE" ]; then
+        doLog "Unable to get size of database ${DB} even though it seems to exist..."
+      else
+        RESULT="${RESULT}${DB}:${CUR_DB_SIZE},"
+        doLog "Size of database ${DB} is ${CUR_DB_SIZE} bytes."
+      fi
     fi
   done