From affc3b7fb2e660716637fb548c462a28d7d7baac Mon Sep 17 00:00:00 2001
From: olivier-tritschler <olivier.tritschler@gmail.com>
Date: Thu, 6 Jun 2024 20:28:08 -0700
Subject: [PATCH] CosmosDB provider

---
 .../scaleoutsoftware/digitaltwin/core/ModelSchema.java   | 2 ++
 .../digitaltwin/core/PersistenceProviderType.java        | 9 +++++++++
 2 files changed, 11 insertions(+)

diff --git a/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/ModelSchema.java b/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/ModelSchema.java
index a606ab6..0379724 100644
--- a/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/ModelSchema.java
+++ b/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/ModelSchema.java
@@ -188,6 +188,7 @@ public ModelSchema(
             case SQLite:
             case SQLServer:
             case DynamoDb:
+            case CosmosDb:
                 enablePersistence           = true;
                 azureDigitalTwinModelName   = null;
                 break;
@@ -242,6 +243,7 @@ public ModelSchema(
             case SQLite:
             case SQLServer:
             case DynamoDb:
+            case CosmosDb:
                 enablePersistence           = true;
                 azureDigitalTwinModelName   = null;
                 break;
diff --git a/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/PersistenceProviderType.java b/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/PersistenceProviderType.java
index 25912b1..672c30b 100644
--- a/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/PersistenceProviderType.java
+++ b/Core/src/main/java/com/scaleoutsoftware/digitaltwin/core/PersistenceProviderType.java
@@ -26,6 +26,11 @@ public enum PersistenceProviderType implements Serializable {
      * Enum for the Azure Digital Twin service.
      */
     AzureDigitalTwinsService("AzureDigitalTwinsService", 1),
+    /**
+     * Enum for Cosmos DB
+     */
+    CosmosDb("Azure Cosmos DB", 6),
+
     /**
      * Enum for DynamoDB
      */
@@ -68,6 +73,8 @@ public static PersistenceProviderType fromString(String name) {
                     return SQLServer;
                 case "DynamoDB":
                     return DynamoDb;
+                case "Azure Cosmos DB":
+                    return CosmosDb;
                 default:
                     return null;
             }
@@ -92,6 +99,8 @@ public static PersistenceProviderType fromOrdinal(int ordinal) {
                 return SQLite;
             case 5:
                 return DynamoDb;
+            case 6:
+                return CosmosDb;
             default:
                 return null;
         }