Added content response on write enabled feature to Gateway and Direct TCP

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/AsynReadWithMultipleClients.java

@@ -226,6 +226,7 @@ private void createClients() {
                         .connectionPolicy(configuration.getConnectionPolicy())
                         .consistencyLevel(configuration.getConsistencyLevel())
                         .connectionReuseAcrossClientsEnabled(true)
+                        .returnMinimalResponse(configuration.isReturnMinimalResponse())
                         .buildAsyncClient();
                     List<PojoizedJson> docsToRead = new ArrayList<>();
                     CosmosAsyncDatabase cosmosAsyncDatabase = null;

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/AsyncBenchmark.java

@@ -66,6 +66,7 @@ abstract class AsyncBenchmark<T> {
             .key(cfg.getMasterKey())
             .connectionPolicy(cfg.getConnectionPolicy())
             .consistencyLevel(cfg.getConsistencyLevel())
+            .returnMinimalResponse(cfg.isReturnMinimalResponse())
             .buildAsyncClient();
         configuration = cfg;
         logger = LoggerFactory.getLogger(this.getClass());

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/Configuration.java

@@ -80,6 +80,9 @@ class Configuration {
     @Parameter(names = "-throughput", description = "provisioned throughput for test container")
     private int throughput = 100000;
 
+    @Parameter(names = "-returnMinimalResponse", description = "returns minimal response on document write operations")
+    private boolean returnMinimalResponse = false;
+
     @Parameter(names = "-operation", description = "Type of Workload:\n"
         + "\tReadThroughput- run a READ workload that prints only throughput *\n"
         + "\tReadThroughputWithMultipleClients - run a READ workload that prints throughput and latency for multiple client read.*\n"
@@ -264,6 +267,10 @@ ConsistencyLevel getConsistencyLevel() {
         return consistencyLevel;
     }
 
+    boolean isReturnMinimalResponse() {
+        return returnMinimalResponse;
+    }
+
     String getDatabaseId() {
         return databaseId;
     }

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/Main.java

@@ -27,6 +27,8 @@ public static void main(String[] args) throws Exception {
                 return;
             }
 
+            validateConfiguration(cfg);
+
             if (cfg.isSync()) {
                 syncBenchmark(cfg);
             } else {
@@ -44,6 +46,19 @@ public static void main(String[] args) throws Exception {
         }
     }
 
+    private static void validateConfiguration(Configuration cfg) {
+        switch (cfg.getOperationType()) {
+            case WriteLatency:
+            case WriteThroughput:
+                break;
+            default:
+                if (cfg.isReturnMinimalResponse()) {
+                    throw new IllegalArgumentException("returnMinimalResponse parameter can only be set to true " +
+                        "for write latency and write throughput operations");
+                }
+        }
+    }
+
     private static void syncBenchmark(Configuration cfg) throws Exception {
         LOGGER.info("Sync benchmark ...");
         SyncBenchmark benchmark = null;

sdk/cosmos/azure-cosmos-benchmark/src/main/java/com/azure/cosmos/benchmark/SyncBenchmark.java

@@ -105,6 +105,7 @@ public T apply(T o, Throwable throwable) {
             .key(cfg.getMasterKey())
             .connectionPolicy(cfg.getConnectionPolicy())
             .consistencyLevel(cfg.getConsistencyLevel())
+            .returnMinimalResponse(cfg.isReturnMinimalResponse())
             .buildClient();
 
         try {

sdk/cosmos/azure-cosmos-examples/src/main/java/com/azure/cosmos/examples/ChangeFeed/SampleChangeFeedProcessor.java

@@ -121,6 +121,7 @@ public static CosmosAsyncClient getCosmosClient() {
                 .key(SampleConfigurations.MASTER_KEY)
                 .connectionPolicy(ConnectionPolicy.getDefaultPolicy())
                 .consistencyLevel(ConsistencyLevel.EVENTUAL)
+                .returnMinimalResponse(false)
                 .buildAsyncClient();
     }
 

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

@@ -49,6 +49,7 @@ public final class CosmosAsyncClient implements Closeable {
     private final CosmosKeyCredential cosmosKeyCredential;
     private final boolean sessionCapturingOverride;
     private final boolean enableTransportClientSharing;
+    private final boolean returnMinimalResponse;
 
     CosmosAsyncClient(CosmosClientBuilder builder) {
         this.configs = builder.configs();
@@ -61,6 +62,7 @@ public final class CosmosAsyncClient implements Closeable {
         this.cosmosKeyCredential = builder.getKeyCredential();
         this.sessionCapturingOverride = builder.isSessionCapturingOverrideEnabled();
         this.enableTransportClientSharing = builder.isConnectionReuseAcrossClientsEnabled();
+        this.returnMinimalResponse = builder.isReturnMinimalResponse();
         this.asyncDocumentClient = new AsyncDocumentClient.Builder()
                                        .withServiceEndpoint(this.serviceEndpoint)
                                        .withMasterKeyOrResourceToken(this.keyOrResourceToken)
@@ -71,6 +73,7 @@ public final class CosmosAsyncClient implements Closeable {
                                        .withTokenResolver(this.cosmosAuthorizationTokenResolver)
                                        .withCosmosKeyCredential(this.cosmosKeyCredential)
                                        .withTransportClientSharing(this.enableTransportClientSharing)
+                                       .withReturnMinimalResponse(this.returnMinimalResponse)
                                        .build();
     }
 
@@ -163,6 +166,20 @@ CosmosKeyCredential cosmosKeyCredential() {
         return cosmosKeyCredential;
     }
 
+    /**
+     * Gets the boolean which indicates whether to only return the headers and status code in Cosmos DB response
+     * in case of Create, Update and Delete operations on CosmosItem.
+     *
+     * If true, this removes the resource from response. It reduces networking
+     * and CPU load by not sending the resource back over the network and serializing it
+     * on the client.
+     *
+     * @return a boolean indicating whether resource will be included in the response or not
+     */
+    boolean isReturnMinimalResponse() {
+        return returnMinimalResponse;
+    }
+
     /**
      * CREATE a Database if it does not already exist on the service
      * <p>

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

@@ -66,7 +66,8 @@ public static CosmosClientBuilder cloneCosmosClientBuilder(CosmosClientBuilder b
             .keyCredential(builder.getKeyCredential())
             .permissions(builder.getPermissions())
             .authorizationTokenResolver(builder.getAuthorizationTokenResolver())
-            .resourceToken(builder.getResourceToken());
+            .resourceToken(builder.getResourceToken())
+            .returnMinimalResponse(builder.isReturnMinimalResponse());
 
         return copy;
     }

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

@@ -39,6 +39,7 @@ public class CosmosClientBuilder {
     private CosmosKeyCredential cosmosKeyCredential;
     private boolean sessionCapturingOverrideEnabled;
     private boolean connectionReuseAcrossClientsEnabled;
+    private boolean returnMinimalResponse = true;
 
     /**
      * Instantiates a new Cosmos client builder.
@@ -291,6 +292,39 @@ public CosmosClientBuilder keyCredential(CosmosKeyCredential cosmosKeyCredential
         return this;
     }
 
+    /**
+     * Gets the boolean which indicates whether to only return the headers and status code in Cosmos DB response
+     * in case of Create, Update and Delete operations on CosmosItem.
+     *
+     * If true, this removes the resource from response. It reduces networking
+     * and CPU load by not sending the resource back over the network and serializing it
+     * on the client.
+     *
+     * By-default, this is true.
+     *
+     * @return a boolean indicating whether resource will be included in the response or not
+     */
+    boolean isReturnMinimalResponse() {
+        return returnMinimalResponse;
+    }
+
+    /**
+     * Sets the boolean to only return the headers and status code in Cosmos DB response
+     * in case of Create, Update and Delete operations on CosmosItem.
+     *
+     * If set to true, this removes the resource from response. It reduces networking
+     * and CPU load by not sending the resource back over the network and serializing it on the client.
+     *
+     * By-default, this is true.
+     *
+     * @param returnMinimalResponse a boolean indicating whether resource will be included in the response or not
+     * @return current cosmosClientBuilder
+     */
+    public CosmosClientBuilder returnMinimalResponse(boolean returnMinimalResponse) {
+        this.returnMinimalResponse = returnMinimalResponse;
+        return this;
+    }
+
     /**
      * Builds a cosmos configuration object with the provided properties
      *

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/AsyncDocumentClient.java

@@ -77,6 +77,7 @@ class Builder {
         CosmosKeyCredential cosmosKeyCredential;
         boolean sessionCapturingOverride;
         boolean transportClientSharing;
+        boolean returnMinimalResponse;
 
         public Builder withServiceEndpoint(String serviceEndpoint) {
             try {
@@ -158,6 +159,11 @@ public Builder withCosmosKeyCredential(CosmosKeyCredential cosmosKeyCredential)
             return this;
         }
 
+        public Builder withReturnMinimalResponse(boolean returnMinimalResponse) {
+            this.returnMinimalResponse = returnMinimalResponse;
+            return this;
+        }
+
         /**
          * This method will accept functional interface TokenResolver which helps in generation authorization
          * token per request. AsyncDocumentClient can be successfully initialized with this API without passing any MasterKey, ResourceToken or PermissionFeed.
@@ -187,15 +193,16 @@ public AsyncDocumentClient build() {
                 "cannot buildAsyncClient client without key credential");
 
             RxDocumentClientImpl client = new RxDocumentClientImpl(serviceEndpoint,
-                                                                   masterKeyOrResourceToken,
-                                                                   permissionFeed,
-                                                                   connectionPolicy,
-                                                                   desiredConsistencyLevel,
-                                                                   configs,
-                                                                   cosmosAuthorizationTokenResolver,
-                                                                   cosmosKeyCredential,
-                                                                   sessionCapturingOverride,
-                                                                   transportClientSharing);
+                masterKeyOrResourceToken,
+                permissionFeed,
+                connectionPolicy,
+                desiredConsistencyLevel,
+                configs,
+                cosmosAuthorizationTokenResolver,
+                cosmosKeyCredential,
+                sessionCapturingOverride,
+                transportClientSharing,
+                returnMinimalResponse);
             client.init();
             return client;
         }

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/HttpConstants.java

@@ -306,6 +306,7 @@ public static class SubStatusCodes {
     }
 
     public static class HeaderValues {
-        public static final String NoCache = "no-cache";
+        public static final String NO_CACHE = "no-cache";
+        public static final String PREFER_RETURN_MINIMAL = "return=minimal";
     }
 }

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/ResourceResponse.java

@@ -358,6 +358,17 @@ public String getCosmosResponseDiagnosticString() {
         return cosmosResponseRequestDiagnosticStatistics.toString();
     }
 
+    /**
+     * Gets the ETag from the response headers.
+     *
+     * Null in case of delete operation.
+     *
+     * @return ETag
+     */
+    public String getETag() {
+        return this.response.getResponseHeaders().get(HttpConstants.HttpHeaders.E_TAG);
+    }
+
     long getCurrentQuotaHeader(String headerName) {
         if (this.usageHeaders.size() == 0 &&
                 !StringUtils.isEmpty(this.getMaxResourceQuota()) &&