Create cloud scanner log directories on restart

(cherry picked from commit 650a82c)
deepfence · Aug 3, 2024 · e279bf2 · e279bf2
1 parent 7fdd848
commit e279bf2
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/deepfence_agent/plugins/etc/run_shipper.sh b/deepfence_agent/plugins/etc/run_shipper.sh
@@ -1,4 +1,10 @@
-BATCH_SIZE=${1:-100}
-TRUNCATE_SIZE=${2:-10}
+#!/bin/bash
 
-exec $DF_INSTALL_DIR/home/deepfence/bin/shipper --base-path="${DF_INSTALL_DIR:-/}" --truncate-size=$TRUNCATE_SIZE --routes=$DF_INSTALL_DIR/home/deepfence/routes.yaml --batch-size=$BATCH_SIZE
+export BATCH_SIZE=${1:-100}
+export TRUNCATE_SIZE=${2:-10}
+
+if [[ $DF_INSTALL_DIR == "/home/deepfence" ]]; then
+  exec /home/deepfence/bin/shipper --base-path="${DF_INSTALL_DIR:-/}" --truncate-size=$TRUNCATE_SIZE --routes=/home/deepfence/routes.yaml --batch-size=$BATCH_SIZE
+else
+  exec $DF_INSTALL_DIR/home/deepfence/bin/shipper --base-path="${DF_INSTALL_DIR:-/}" --truncate-size=$TRUNCATE_SIZE --routes=$DF_INSTALL_DIR/home/deepfence/routes.yaml --batch-size=$BATCH_SIZE
+fi