Merged cosmos trigger async response and cosmos trigger response types

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/BridgeInternal.java

@@ -467,11 +467,6 @@ public static CosmosAsyncPermission createCosmosAsyncPermission(String id, Cosmo
         return new CosmosAsyncPermission(id, user);
     }
 
-    @Warning(value = INTERNAL_USE_ONLY_WARNING)
-    public static CosmosAsyncTrigger createCosmosAsyncTrigger(String id, CosmosAsyncContainer container) {
-        return new CosmosAsyncTrigger(id, container);
-    }
-
     @Warning(value = INTERNAL_USE_ONLY_WARNING)
     public static CosmosAsyncUserDefinedFunction createCosmosAsyncUserDefinedFunction(String id, CosmosAsyncContainer container) {
         return new CosmosAsyncUserDefinedFunction(id, container);

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosAsyncDatabase.java

@@ -544,7 +544,7 @@ public CosmosAsyncContainer getContainer(String id) {
      */
     public Mono<CosmosUserResponse> createUser(CosmosUserProperties userProperties) {
         return getDocClientWrapper().createUser(this.getLink(), ModelBridgeInternal.getV2User(userProperties), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncUserResponse(response)).single();
+                                    .map(response -> ModelBridgeInternal.createCosmosUserResponse(response)).single();
     }
 
 
@@ -561,7 +561,7 @@ public Mono<CosmosUserResponse> createUser(CosmosUserProperties userProperties)
      */
     public Mono<CosmosUserResponse> upsertUser(CosmosUserProperties userProperties) {
         return getDocClientWrapper().upsertUser(this.getLink(), ModelBridgeInternal.getV2User(userProperties), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncUserResponse(response)).single();
+                                    .map(response -> ModelBridgeInternal.createCosmosUserResponse(response)).single();
     }
 
     /**

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosAsyncScripts.java

@@ -6,7 +6,7 @@
 import com.azure.cosmos.implementation.Trigger;
 import com.azure.cosmos.implementation.UserDefinedFunction;
 import com.azure.cosmos.models.CosmosStoredProcedureResponse;
-import com.azure.cosmos.models.CosmosAsyncTriggerResponse;
+import com.azure.cosmos.models.CosmosTriggerResponse;
 import com.azure.cosmos.models.CosmosUserDefinedFunctionResponse;
 import com.azure.cosmos.models.CosmosStoredProcedureProperties;
 import com.azure.cosmos.models.CosmosStoredProcedureRequestOptions;
@@ -304,12 +304,12 @@ public CosmosAsyncUserDefinedFunction getUserDefinedFunction(String id) {
      * @param properties the cosmos trigger properties
      * @return an {@link Mono} containing the single resource response with the created trigger or an error.
      */
-    public Mono<CosmosAsyncTriggerResponse> createTrigger(CosmosTriggerProperties properties) {
+    public Mono<CosmosTriggerResponse> createTrigger(CosmosTriggerProperties properties) {
         Trigger trigger = new Trigger(ModelBridgeInternal.toJsonFromJsonSerializable(ModelBridgeInternal.getResource(properties)));
 
         return database.getDocClientWrapper()
                    .createTrigger(container.getLink(), trigger, null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncTriggerResponse(response, this.container))
+                   .map(response -> ModelBridgeInternal.createCosmosTriggerResponse(response))
                    .single();
     }
 

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosAsyncTrigger.java

@@ -4,7 +4,7 @@
 
 import com.azure.cosmos.implementation.Paths;
 import com.azure.cosmos.implementation.Trigger;
-import com.azure.cosmos.models.CosmosAsyncTriggerResponse;
+import com.azure.cosmos.models.CosmosTriggerResponse;
 import com.azure.cosmos.models.CosmosTriggerProperties;
 import com.azure.cosmos.models.ModelBridgeInternal;
 import reactor.core.publisher.Mono;
@@ -51,11 +51,11 @@ CosmosAsyncTrigger setId(String id) {
      *
      * @return an {@link Mono} containing the single resource response for the read cosmos trigger or an error.
      */
-    public Mono<CosmosAsyncTriggerResponse> read() {
+    public Mono<CosmosTriggerResponse> read() {
         return container.getDatabase()
                    .getDocClientWrapper()
                    .readTrigger(getLink(), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncTriggerResponse(response, container))
+                   .map(response -> ModelBridgeInternal.createCosmosTriggerResponse(response))
                    .single();
     }
 
@@ -70,12 +70,12 @@ public Mono<CosmosAsyncTriggerResponse> read() {
      * @param triggerSettings the cosmos trigger properties.
      * @return an {@link Mono} containing the single resource response with the replaced cosmos trigger or an error.
      */
-    public Mono<CosmosAsyncTriggerResponse> replace(CosmosTriggerProperties triggerSettings) {
+    public Mono<CosmosTriggerResponse> replace(CosmosTriggerProperties triggerSettings) {
         return container.getDatabase()
                    .getDocClientWrapper()
                    .replaceTrigger(new Trigger(ModelBridgeInternal.toJsonFromJsonSerializable(
                        ModelBridgeInternal.getResource(triggerSettings))), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncTriggerResponse(response, container))
+                   .map(response -> ModelBridgeInternal.createCosmosTriggerResponse(response))
                    .single();
     }
 
@@ -88,11 +88,11 @@ public Mono<CosmosAsyncTriggerResponse> replace(CosmosTriggerProperties triggerS
      *
      * @return an {@link Mono} containing the single resource response for the deleted cosmos trigger or an error.
      */
-    public Mono<CosmosAsyncTriggerResponse> delete() {
+    public Mono<CosmosTriggerResponse> delete() {
         return container.getDatabase()
                    .getDocClientWrapper()
                    .deleteTrigger(getLink(), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncTriggerResponse(response, container))
+                   .map(response -> ModelBridgeInternal.createCosmosTriggerResponse(response))
                    .single();
     }
 

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosAsyncUser.java

@@ -57,8 +57,8 @@ CosmosAsyncUser setId(String id) {
      */
     public Mono<CosmosUserResponse> read() {
         return this.database.getDocClientWrapper()
-                   .readUser(getLink(), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncUserResponse(response)).single();
+                            .readUser(getLink(), null)
+                            .map(response -> ModelBridgeInternal.createCosmosUserResponse(response)).single();
     }
 
     /**
@@ -69,8 +69,8 @@ public Mono<CosmosUserResponse> read() {
      */
     public Mono<CosmosUserResponse> replace(CosmosUserProperties userSettings) {
         return this.database.getDocClientWrapper()
-                   .replaceUser(ModelBridgeInternal.getV2User(userSettings), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncUserResponse(response)).single();
+                            .replaceUser(ModelBridgeInternal.getV2User(userSettings), null)
+                            .map(response -> ModelBridgeInternal.createCosmosUserResponse(response)).single();
     }
 
     /**
@@ -80,8 +80,8 @@ public Mono<CosmosUserResponse> replace(CosmosUserProperties userSettings) {
      */
     public Mono<CosmosUserResponse> delete() {
         return this.database.getDocClientWrapper()
-                   .deleteUser(getLink(), null)
-                   .map(response -> ModelBridgeInternal.createCosmosAsyncUserResponse(response)).single();
+                            .deleteUser(getLink(), null)
+                            .map(response -> ModelBridgeInternal.createCosmosUserResponse(response)).single();
     }
 
     /**

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosScripts.java

@@ -4,15 +4,13 @@
 package com.azure.cosmos;
 
 import com.azure.cosmos.models.CosmosStoredProcedureResponse;
-import com.azure.cosmos.models.CosmosAsyncTriggerResponse;
+import com.azure.cosmos.models.CosmosTriggerResponse;
 import com.azure.cosmos.models.CosmosUserDefinedFunctionResponse;
 import com.azure.cosmos.models.CosmosStoredProcedureProperties;
 import com.azure.cosmos.models.CosmosStoredProcedureRequestOptions;
 import com.azure.cosmos.models.CosmosTriggerProperties;
-import com.azure.cosmos.models.CosmosTriggerResponse;
 import com.azure.cosmos.models.CosmosUserDefinedFunctionProperties;
 import com.azure.cosmos.models.FeedOptions;
-import com.azure.cosmos.models.ModelBridgeInternal;
 import com.azure.cosmos.models.SqlQuerySpec;
 import com.azure.cosmos.util.CosmosPagedFlux;
 import com.azure.cosmos.util.CosmosPagedIterable;
@@ -309,11 +307,9 @@ CosmosUserDefinedFunctionResponse mapUDFResponseAndBlock(
      * @param responseMono the response mono
      * @return the cosmos sync trigger response
      */
-    CosmosTriggerResponse mapTriggerResponseAndBlock(Mono<CosmosAsyncTriggerResponse> responseMono) {
+    CosmosTriggerResponse mapTriggerResponseAndBlock(Mono<CosmosTriggerResponse> responseMono) {
         try {
-            return responseMono
-                       .map(this::convertResponse)
-                       .block();
+            return responseMono.block();
         } catch (Exception ex) {
             final Throwable throwable = Exceptions.unwrap(ex);
             if (throwable instanceof CosmosException) {
@@ -324,21 +320,6 @@ CosmosTriggerResponse mapTriggerResponseAndBlock(Mono<CosmosAsyncTriggerResponse
         }
     }
 
-    /**
-     * Convert response cosmos sync trigger response.
-     *
-     * @param response the response
-     * @return the cosmos sync trigger response
-     */
-    CosmosTriggerResponse convertResponse(CosmosAsyncTriggerResponse response) {
-        if (response.getTrigger() != null) {
-            return ModelBridgeInternal.createCosmosTriggerResponse(response,
-                                             getTrigger(response.getTrigger().getId()));
-        } else {
-            return ModelBridgeInternal.createCosmosTriggerResponse(response, null);
-        }
-    }
-
     private <T> CosmosPagedIterable<T> getCosmosPagedIterable(CosmosPagedFlux<T> cosmosPagedFlux) {
         return UtilBridgeInternal.createCosmosPagedIterable(cosmosPagedFlux);
     }

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/CosmosTrigger.java

@@ -3,8 +3,8 @@
 
 package com.azure.cosmos;
 
-import com.azure.cosmos.models.CosmosTriggerProperties;
 import com.azure.cosmos.models.CosmosTriggerResponse;
+import com.azure.cosmos.models.CosmosTriggerProperties;
 
 /**
  * The type Cosmos sync trigger.

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosTriggerResponse.java

@@ -1,46 +1,34 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
-
 package com.azure.cosmos.models;
 
-import com.azure.cosmos.CosmosTrigger;
+import com.azure.cosmos.implementation.ResourceResponse;
+import com.azure.cosmos.implementation.Trigger;
+import com.azure.cosmos.implementation.apachecommons.lang.StringUtils;
 
 /**
- * The type Cosmos sync trigger response.
+ * The type Cosmos async trigger response.
  */
 public class CosmosTriggerResponse extends CosmosResponse<CosmosTriggerProperties> {
 
-    private final CosmosTrigger syncTrigger;
-    private final CosmosAsyncTriggerResponse asyncResponse;
+    private final CosmosTriggerProperties cosmosTriggerProperties;
 
-    /**
-     * Instantiates a new Cosmos sync trigger response.
-     *
-     * @param asyncResponse the async response
-     * @param syncTrigger the sync trigger
-     */
-    CosmosTriggerResponse(CosmosAsyncTriggerResponse asyncResponse,
-                          CosmosTrigger syncTrigger) {
-        super(asyncResponse.resourceResponseWrapper, asyncResponse.getProperties());
-        this.asyncResponse = asyncResponse;
-        this.syncTrigger = syncTrigger;
+    CosmosTriggerResponse(ResourceResponse<Trigger> response) {
+        super(response);
+        String bodyAsString = response.getBodyAsString();
+        if (StringUtils.isEmpty(bodyAsString)) {
+            cosmosTriggerProperties = null;
+        } else {
+            cosmosTriggerProperties = new CosmosTriggerProperties(bodyAsString);
+        }
     }
 
     /**
-     * Gets cosmos trigger properties.
+     * Gets the cosmos trigger properties or null
      *
-     * @return the cosmos trigger properties
+     * @return {@link CosmosTriggerProperties}
      */
     public CosmosTriggerProperties getProperties() {
-        return asyncResponse.getProperties();
-    }
-
-    /**
-     * Gets cosmos sync trigger.
-     *
-     * @return the cosmos sync trigger
-     */
-    public CosmosTrigger getTrigger() {
-        return syncTrigger;
+        return cosmosTriggerProperties;
     }
 }

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/ModelBridgeInternal.java

@@ -3,8 +3,6 @@
 
 package com.azure.cosmos.models;
 
-import com.azure.cosmos.CosmosAsyncContainer;
-import com.azure.cosmos.CosmosTrigger;
 import com.azure.cosmos.implementation.Conflict;
 import com.azure.cosmos.implementation.ConsistencyPolicy;
 import com.azure.cosmos.implementation.CosmosItemProperties;
@@ -113,9 +111,8 @@ public static CosmosPermissionProperties createCosmosPermissionProperties(String
     }
 
     @Warning(value = INTERNAL_USE_ONLY_WARNING)
-    public static CosmosAsyncTriggerResponse createCosmosAsyncTriggerResponse(ResourceResponse<Trigger> response,
-                                                                              CosmosAsyncContainer container) {
-        return new CosmosAsyncTriggerResponse(response, container);
+    public static CosmosTriggerResponse createCosmosTriggerResponse(ResourceResponse<Trigger> response) {
+        return new CosmosTriggerResponse(response);
     }
 
     @Warning(value = INTERNAL_USE_ONLY_WARNING)
@@ -124,7 +121,7 @@ public static CosmosUserDefinedFunctionResponse createCosmosUserDefinedFunctionR
     }
 
     @Warning(value = INTERNAL_USE_ONLY_WARNING)
-    public static CosmosUserResponse createCosmosAsyncUserResponse(ResourceResponse<User> response) {
+    public static CosmosUserResponse createCosmosUserResponse(ResourceResponse<User> response) {
         return new CosmosUserResponse(response);
     }
 
@@ -133,12 +130,6 @@ public static <T> CosmosItemResponse<T> createCosmosItemResponse(CosmosAsyncItem
         return new CosmosItemResponse<>(response);
     }
 
-    @Warning(value = INTERNAL_USE_ONLY_WARNING)
-    public static CosmosTriggerResponse createCosmosTriggerResponse(CosmosAsyncTriggerResponse asyncResponse,
-                                        CosmosTrigger syncTrigger) {
-        return new CosmosTriggerResponse(asyncResponse, syncTrigger);
-    }
-
     @Warning(value = INTERNAL_USE_ONLY_WARNING)
     public static List<CosmosConflictProperties> getCosmosConflictPropertiesFromV2Results(List<Conflict> results) {
         return CosmosConflictProperties.getFromV2Results(results);