From 190212c8ac54cf79c6e4c0740a3577041d6a4ef7 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Thu, 11 Jan 2024 12:35:52 +0530
Subject: [PATCH] Fix #11556: Add support for Db2 for IBM i (#14680)

---
 ingestion/setup.py                                              | 1 +
 .../content/v1.3.x-SNAPSHOT/connectors/database/db2/index.md    | 2 ++
 .../entity/services/connections/database/db2Connection.json     | 2 +-
 .../src/main/resources/ui/public/locales/en-US/Database/Db2.md  | 2 ++
 4 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/ingestion/setup.py b/ingestion/setup.py
index 9f44a9e8012..0d5831e816d 100644
--- a/ingestion/setup.py
+++ b/ingestion/setup.py
@@ -157,6 +157,7 @@ plugins: Dict[str, Set[str]] = {
         VERSIONS["azure-identity"],
     },
     "db2": {"ibm-db-sa~=0.3"},
+    "db2-ibmi": {"sqlalchemy-ibmi~=0.9.3"},
     "databricks": {VERSIONS["sqlalchemy-databricks"], VERSIONS["databricks-sdk"]},
     "datalake-azure": {
         VERSIONS["azure-storage-blob"],
diff --git a/openmetadata-docs/content/v1.3.x-SNAPSHOT/connectors/database/db2/index.md b/openmetadata-docs/content/v1.3.x-SNAPSHOT/connectors/database/db2/index.md
index 3349bb1daa2..7488e5b3f33 100644
--- a/openmetadata-docs/content/v1.3.x-SNAPSHOT/connectors/database/db2/index.md
+++ b/openmetadata-docs/content/v1.3.x-SNAPSHOT/connectors/database/db2/index.md
@@ -105,6 +105,8 @@ Executing the profiler workflow or data quality tests, will require the user to
 - **database**: Database of the data source.
 - **Host and Port**: Enter the fully qualified hostname and port number for your DB2 deployment in the Host and Port field.
 
+Note: In case you are using Db2 for IBM i, then from advanced config you need choose the `ibmi` scheme.
+
 {% partial file="/v1.3/connectors/database/advanced-configuration.md" /%}
 
 {% /extraContent %}
diff --git a/openmetadata-spec/src/main/resources/json/schema/entity/services/connections/database/db2Connection.json b/openmetadata-spec/src/main/resources/json/schema/entity/services/connections/database/db2Connection.json
index 6fe788a20be..fc43747ffee 100644
--- a/openmetadata-spec/src/main/resources/json/schema/entity/services/connections/database/db2Connection.json
+++ b/openmetadata-spec/src/main/resources/json/schema/entity/services/connections/database/db2Connection.json
@@ -15,7 +15,7 @@
     "db2Scheme": {
       "description": "SQLAlchemy driver scheme options.",
       "type": "string",
-      "enum": ["db2+ibm_db"],
+      "enum": ["db2+ibm_db", "ibmi"],
       "default": "db2+ibm_db"
     }
   },
diff --git a/openmetadata-ui/src/main/resources/ui/public/locales/en-US/Database/Db2.md b/openmetadata-ui/src/main/resources/ui/public/locales/en-US/Database/Db2.md
index e8b1f2cb7d4..8c16513bc94 100644
--- a/openmetadata-ui/src/main/resources/ui/public/locales/en-US/Database/Db2.md
+++ b/openmetadata-ui/src/main/resources/ui/public/locales/en-US/Database/Db2.md
@@ -39,6 +39,8 @@ $$section
 ### Scheme $(id="scheme")
 
 SQLAlchemy driver scheme options.
+
+Note: In case you are using Db2 for IBM i, then from advanced config you can choose the `ibmi` scheme.
 $$
 
 $$section