From e583a1ec4fbaa4a0ed060d4dcc30599300dfbcbf Mon Sep 17 00:00:00 2001
From: simonzhangsz <simonzhangsz@outlook.com>
Date: Thu, 21 Dec 2023 17:57:16 +0800
Subject: [PATCH] fix: importdb script

---
 misc/importdb.sh | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/misc/importdb.sh b/misc/importdb.sh
index 1e40d46b..e934135c 100644
--- a/misc/importdb.sh
+++ b/misc/importdb.sh
@@ -8,16 +8,43 @@ UpgradeSQLDir=/usr/local/omc/etc/db/upgrade
 Upgvue3SQLDir=/usr/local/omc/etc/db/upgvue3
 InstallSQLDir=/usr/local/omc/etc/db/install
 
+C_ARG="agt"
+M_ARG="upgrade"
+
 drop_db_sql="drop database IF EXISTS ${DBNAME}"
 create_db_sql="create database IF NOT EXISTS ${DBNAME}"
 
-case "$1" in
+get_args() {
+    getopts "c:m:" option; do
+		case $option in
+			c)
+				C_ARG=$OPTARG
+	        ;;
+
+      		m)
+        		M_ARG=$OPTARG
+        	;;
+
+      		\?)
+        		echo "Invalid option: -$OPTARG" >&2
+        	;;
+    	esac
+  	done
+}
+
+get_args "$@"
+
+echo "C_ARG: $C_ARG"
+echo "M_ARG: $M_ARG"
+
+case "$M_ARG" in
         upgrade)
 			echo "Upgrade database ${DBNAME}"
 			for SQL in ${UpgradeSQLDir}/*.sql; do 
 				echo "Execute SQL script: ${SQL} ..."
 				mysql -u${USER} -p${PASSWORD} -P ${PORT} --protocol tcp -D ${DBNAME} < ${SQL}; 
 			done
+
         ;;
 
         upgvue3)
@@ -100,3 +127,4 @@ esac
 
 
 
+