From e5970d87e03d12a43f2a232636a2a3078eeab339 Mon Sep 17 00:00:00 2001
From: openGaussDev <wangtianqing2@huawei.com>
Date: Mon, 7 Mar 2022 15:56:47 +0800
Subject: [PATCH]  [] optimize plugins of DBMind

Offering: openGaussDev

More detail: opt

Match-id-9495280ca4e6c28de3c1e6d0150cae65d5e8ead6
---
 .../components/index_advisor/__main__.py      | 24 +++++++++++++++++++
 .../components/opengauss_exporter/__main__.py |  4 +++-
 .../reprocessing_exporter/__main__.py         |  4 +++-
 .../tools/components/sqldiag/__main__.py      |  4 +++-
 .../tools/components/xtuner/__main__.py       |  5 +++-
 5 files changed, 37 insertions(+), 4 deletions(-)
 create mode 100644 src/gausskernel/dbmind/tools/components/index_advisor/__main__.py

diff --git a/src/gausskernel/dbmind/tools/components/index_advisor/__main__.py b/src/gausskernel/dbmind/tools/components/index_advisor/__main__.py
new file mode 100644
index 000000000..82544372c
--- /dev/null
+++ b/src/gausskernel/dbmind/tools/components/index_advisor/__main__.py
@@ -0,0 +1,24 @@
+# Copyright (c) 2022 Huawei Technologies Co.,Ltd.
+#
+# openGauss is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+
+import sys
+import os
+
+try:
+    from dbmind.components.index_advisor import main
+except ImportError:
+    libpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
+    sys.path.append(libpath)
+    from index_advisor import main
+
+main(sys.argv[1:])
diff --git a/src/gausskernel/dbmind/tools/components/opengauss_exporter/__main__.py b/src/gausskernel/dbmind/tools/components/opengauss_exporter/__main__.py
index 6894a1570..8d67b1553 100644
--- a/src/gausskernel/dbmind/tools/components/opengauss_exporter/__main__.py
+++ b/src/gausskernel/dbmind/tools/components/opengauss_exporter/__main__.py
@@ -11,11 +11,13 @@
 # MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 # See the Mulan PSL v2 for more details.
 import sys
+import os
 
 try:
     from dbmind.components.opengauss_exporter import main
 except ImportError:
-    sys.path.append('..')
+    libpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
+    sys.path.append(libpath)
     from opengauss_exporter import main
 
 main(sys.argv[1:])
diff --git a/src/gausskernel/dbmind/tools/components/reprocessing_exporter/__main__.py b/src/gausskernel/dbmind/tools/components/reprocessing_exporter/__main__.py
index 1bb75bd2b..d87752d0d 100644
--- a/src/gausskernel/dbmind/tools/components/reprocessing_exporter/__main__.py
+++ b/src/gausskernel/dbmind/tools/components/reprocessing_exporter/__main__.py
@@ -11,11 +11,13 @@
 # MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 # See the Mulan PSL v2 for more details.
 import sys
+import os
 
 try:
     from dbmind.components.reprocessing_exporter import main
 except ImportError:
-    sys.path.append('..')
+    libpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
+    sys.path.append(libpath)
     from reprocessing_exporter import main
 
 main(sys.argv[1:])
diff --git a/src/gausskernel/dbmind/tools/components/sqldiag/__main__.py b/src/gausskernel/dbmind/tools/components/sqldiag/__main__.py
index 113c50f5f..1f50d4e05 100644
--- a/src/gausskernel/dbmind/tools/components/sqldiag/__main__.py
+++ b/src/gausskernel/dbmind/tools/components/sqldiag/__main__.py
@@ -12,11 +12,13 @@
 # See the Mulan PSL v2 for more details.
 
 import sys
+import os
 
 try:
     from dbmind.components.sqldiag import main
 except ImportError:
-    sys.path.append('..')
+    libpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
+    sys.path.append(libpath)
     from sqldiag import main
 
 main(sys.argv[1:])
diff --git a/src/gausskernel/dbmind/tools/components/xtuner/__main__.py b/src/gausskernel/dbmind/tools/components/xtuner/__main__.py
index 3e2c206be..e388995b8 100644
--- a/src/gausskernel/dbmind/tools/components/xtuner/__main__.py
+++ b/src/gausskernel/dbmind/tools/components/xtuner/__main__.py
@@ -12,11 +12,14 @@
 # See the Mulan PSL v2 for more details.
 
 import sys
+import os
 
 try:
     from dbmind.components.xtuner.tuner.main import main
 except ImportError:
-    sys.path.append('..')
+    libpath = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')
+    sys.path.append(libpath)
+
     from xtuner.tuner.main import main
 
 main(sys.argv[1:])