From 118de92a59b5d88f762f18a1fc1ccef674976ae9 Mon Sep 17 00:00:00 2001
From: Joan He <johe@adobe.com>
Date: Tue, 2 Feb 2021 12:48:18 -0600
Subject: [PATCH 1/2] B2B-1666: Add Primary Key on Magento Inventory DB Tables

---
 InventoryDistanceBasedSourceSelection/etc/db_schema.xml      | 5 +++++
 .../etc/db_schema_whitelist.json                             | 3 +++
 2 files changed, 8 insertions(+)

diff --git a/InventoryDistanceBasedSourceSelection/etc/db_schema.xml b/InventoryDistanceBasedSourceSelection/etc/db_schema.xml
index 0f30b30bd8d3..dd5582c2c234 100644
--- a/InventoryDistanceBasedSourceSelection/etc/db_schema.xml
+++ b/InventoryDistanceBasedSourceSelection/etc/db_schema.xml
@@ -8,6 +8,8 @@
 <schema xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:noNamespaceSchemaLocation="urn:magento:framework:Setup/Declaration/Schema/etc/schema.xsd">
     <table name="inventory_geoname" resource="default" engine="innodb">
+        <column xsi:type="int" name="entity_id" unsigned="true" nullable="false" identity="true"
+                comment="entity id"/>
         <column xsi:type="varchar" name="country_code" nullable="false" length="64"/>
         <column xsi:type="varchar" name="postcode" nullable="false" length="64"/>
         <column xsi:type="varchar" name="city" nullable="false" length="180"/>
@@ -15,5 +17,8 @@
         <column xsi:type="varchar" name="province" nullable="false" length="64"/>
         <column xsi:type="double" name="latitude" nullable="false" />
         <column xsi:type="double" name="longitude" nullable="false"/>
+        <constraint xsi:type="primary" referenceId="PRIMARY">
+            <column name="entity_id"/>
+        </constraint>
     </table>
 </schema>
diff --git a/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json b/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
index f6abc16afada..4314f110cb98 100644
--- a/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
+++ b/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
@@ -9,6 +9,9 @@
             "latitude": true,
             "longitude": true,
             "postcode": true
+        },
+        "constraint": {
+            "PRIMARY": true
         }
     }
 }
\ No newline at end of file

From 81802abcae451ac2a901ca02b268bd291c639162 Mon Sep 17 00:00:00 2001
From: Joan He <johe@adobe.com>
Date: Wed, 3 Feb 2021 14:16:58 -0600
Subject: [PATCH 2/2] B2B-1666: Add Primary Key on Magento Inventory DB Tables

- Add new column to db_schema_whitelist.json
---
 .../etc/db_schema_whitelist.json                                 | 1 +
 1 file changed, 1 insertion(+)

diff --git a/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json b/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
index 4314f110cb98..63411e22d5b1 100644
--- a/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
+++ b/InventoryDistanceBasedSourceSelection/etc/db_schema_whitelist.json
@@ -1,6 +1,7 @@
 {
     "inventory_geoname": {
         "column": {
+            "entity_id": true,
             "country_code": true,
             "zip_code": true,
             "city": true,