adding new method signature plus test

Author: salaboy

sdk/src/main/java/io/dapr/client/AbstractDaprClient.java

@@ -48,11 +48,7 @@
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Collections;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -509,6 +505,24 @@ public Mono<Void> saveState(String storeName, String key, String etag, Object va
     return this.saveBulkState(storeName, Collections.singletonList(state));
   }
 
+  /**
+   * {@inheritDoc}
+   */
+  @Override
+  public Mono<Void> saveState(String storeName, String key, String etag, Object value, Map<String, String> meta,
+                              StateOptions options) {
+    if (meta == null) {
+      meta = new HashMap<>();
+    }
+
+    if( value != null){
+      meta.put("contentType", stateSerializer.getContentType());
+    }
+
+    State<?> state = new State<>(key, value, etag, meta, options);
+    return this.saveBulkState(storeName, Collections.singletonList(state));
+  }
+
   /**
    * {@inheritDoc}
    */

sdk/src/main/java/io/dapr/client/DaprClient.java

@@ -498,6 +498,21 @@ Mono<Void> executeStateTransaction(String storeName,
    */
   Mono<Void> saveState(String storeName, String key, String etag, Object value, StateOptions options);
 
+
+  /**
+   * Save/Update a state.
+   *
+   * @param storeName The name of the state store.
+   * @param key            The key of the state.
+   * @param etag           The etag to be used.
+   * @param value          The value of the state.
+   * @param meta           The metadata to be set to the state.
+   * @param options        The Options to use for each state.
+   * @return a Mono plan of type Void.
+   */
+  Mono<Void> saveState(String storeName, String key, String etag, Object value, Map<String, String> meta,
+                       StateOptions options);
+
   /**
    * Delete a state.
    *

sdk/src/test/java/io/dapr/client/DaprClientGrpcTest.java

@@ -1241,6 +1241,27 @@ public void saveStateNoOptionsTest() {
     result.block();
   }
 
+  @Test
+  public void saveStateWithMetaTest() {
+    String key = "key1";
+    String etag = "ETag1";
+    String value = "State value";
+    Map<String, String> metadata = new HashMap<>();
+    metadata.put("custom", "customValue");
+    ArgumentCaptor<DaprProtos.SaveStateRequest> argument = ArgumentCaptor.forClass(DaprProtos.SaveStateRequest.class);
+    doAnswer((Answer<Void>) invocation -> {
+      StreamObserver<Empty> observer = (StreamObserver<Empty>) invocation.getArguments()[1];
+      observer.onNext(Empty.getDefaultInstance());
+      observer.onCompleted();
+      return null;
+    }).when(daprStub).saveState(argument.capture(), any());
+
+
+    Mono<Void> result = client.saveState(STATE_STORE_NAME, key, etag, value, metadata,null);
+    result.block();
+    assertEquals("customValue", argument.getValue().getStates(0).getMetadata().get("custom"));
+  }
+
   @Test
   public void saveStateTest() {
     String key = "key1";