From d79e9312a0856ff13acd27bbbc09beba77491869 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-10-9-2-132.us-west-2.compute.internal>
Date: Wed, 16 Oct 2024 14:34:59 +0000
Subject: [PATCH] Test using adaptive parallelism

---
 migtests/scripts/functions.sh | 32 +++++++++++++++++++-------------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/migtests/scripts/functions.sh b/migtests/scripts/functions.sh
index c37023e7d..93efc108f 100644
--- a/migtests/scripts/functions.sh
+++ b/migtests/scripts/functions.sh
@@ -451,21 +451,27 @@ import_data_to_source_replica() {
 }
 
 import_data_file() {
-	yb-voyager import data file --export-dir ${EXPORT_DIR} \
-		--target-db-host ${TARGET_DB_HOST} \
-		--target-db-port ${TARGET_DB_PORT} \
-		--target-db-user ${TARGET_DB_USER} \
-		--target-db-password ${TARGET_DB_PASSWORD:-''} \
-		--target-db-schema ${TARGET_DB_SCHEMA:-''} \
-		--target-db-name ${TARGET_DB_NAME} \
-		--disable-pb true \
-		--send-diagnostics=false \
-		$* || {
-			cat ${EXPORT_DIR}/metainfo/dataFileDescriptor.json
-			exit 1
-		}
+    args="
+    --export-dir ${EXPORT_DIR}
+    --target-db-host ${TARGET_DB_HOST}
+    --target-db-port ${TARGET_DB_PORT}
+    --target-db-user ${TARGET_DB_USER}
+    --target-db-password ${TARGET_DB_PASSWORD:-''}
+    --target-db-schema ${TARGET_DB_SCHEMA:-''}
+    --target-db-name ${TARGET_DB_NAME}
+    --disable-pb true
+    --send-diagnostics=false
+    "
+
+    # Check if RUN_USING_ADAPTIVE_PARALLELISM is true
+    if [ "${RUN_USING_ADAPTIVE_PARALLELISM}" = "true" ]; then
+        args="${args} --enable-adaptive-parallelism true"
+    fi
+
+    yb-voyager import data file ${args} $*
 }
 
+
 archive_changes() {
 	ENABLE=$(shuf -i 0-1 -n 1)
 	echo "archive changes ENABLE=${ENABLE}"