From d7afc58d0c9cf7f6625a08e77b821a14548fafc0 Mon Sep 17 00:00:00 2001
From: Thinking Chen <cdmikechen@hotmail.com>
Date: Fri, 7 Jan 2022 07:46:51 +0800
Subject: [PATCH] [HUDI-3118] Add default HUDI_DIR in setupKafka.sh (#4460)

---
 hudi-kafka-connect/demo/setupKafka.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/hudi-kafka-connect/demo/setupKafka.sh b/hudi-kafka-connect/demo/setupKafka.sh
index 49155649c..c75b4a931 100755
--- a/hudi-kafka-connect/demo/setupKafka.sh
+++ b/hudi-kafka-connect/demo/setupKafka.sh
@@ -48,6 +48,10 @@ if [ $# -lt 1 ]; then
   exit 0
 fi
 
+if [ ! $HUDI_DIR ]; then
+  export HUDI_DIR=$(dirname "$(dirname $PWD)")
+fi
+
 ## defaults
 rawDataFile=${HUDI_DIR}/docker/demo/data/batch_1.json
 kafkaBrokerHostname=localhost