adding new method signature plus test (#1570)

salaboy · artur-ciocanu · web-flow · commit 6e9ee5adc4a7 · 2025-10-24T15:00:48.000-07:00
* adding new method signature plus test Signed-off-by: salaboy <Salaboy@gmail.com> * re adding imports Signed-off-by: salaboy <Salaboy@gmail.com> * fixing style Signed-off-by: salaboy <Salaboy@gmail.com> * checking empty metadata Signed-off-by: salaboy <Salaboy@gmail.com> * copy meta for safety and check if key is present Signed-off-by: salaboy <Salaboy@gmail.com> * Centralize Maven dependency version management (#1564) Signed-off-by: salaboy <Salaboy@gmail.com> * Fix dependencies multi app build and add proper test deps (#1572) * Force Jackson version to override the SB Jackson version Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> * Move all the Jackson deps to parent POM. Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> * Ensure app JAR build order Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> * Remove explicit Jackson from sdk-tests module. Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> * Make sure <scope>test</scope> is used for test dependencies. Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> * Remove extra Jackson modules. Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> --------- Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> Signed-off-by: salaboy <Salaboy@gmail.com> * reverting pom Signed-off-by: salaboy <Salaboy@gmail.com> * fix codestyle Signed-off-by: salaboy <Salaboy@gmail.com> * using metaCopy Signed-off-by: salaboy <Salaboy@gmail.com> --------- Signed-off-by: salaboy <Salaboy@gmail.com> Signed-off-by: Artur Ciocanu <artur.ciocanu@gmail.com> Co-authored-by: artur-ciocanu <artur.ciocanu@gmail.com>
diff --git a/sdk/src/main/java/io/dapr/client/AbstractDaprClient.java b/sdk/src/main/java/io/dapr/client/AbstractDaprClient.java
@@ -51,6 +51,7 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.stream.Collectors;
@@ -509,6 +510,27 @@ 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) {
+    Map<String, String> metaCopy = null;
+    if (meta == null) {
+      metaCopy = new HashMap<>();
+    } else {
+      metaCopy = new HashMap<>(meta);
+    }
+
+    if (value != null) {
+      metaCopy.putIfAbsent("contentType", stateSerializer.getContentType());
+    }
+
+    State<?> state = new State<>(key, value, etag, metaCopy, options);
+    return this.saveBulkState(storeName, Collections.singletonList(state));
+  }
+
   /**
    * {@inheritDoc}
    */
diff --git a/sdk/src/main/java/io/dapr/client/DaprClient.java b/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.
    *
diff --git a/sdk/src/test/java/io/dapr/client/DaprClientGrpcTest.java b/sdk/src/test/java/io/dapr/client/DaprClientGrpcTest.java
@@ -1241,6 +1241,64 @@ 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 saveStateWithMetaContentTypeTest() {
+    String key = "key1";
+    String etag = "ETag1";
+    String value = "State value";
+    Map<String, String> metadata = new HashMap<>();
+    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("application/json", argument.getValue().getStates(0).getMetadata().get("contentType"));
+  }
+
+  @Test
+  public void saveStateWithMetaEmptyTest() {
+    String key = "key1";
+    String etag = "ETag1";
+    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, null, null,null);
+    result.block();
+    assertTrue(argument.getValue().getStates(0).getMetadata().keySet().isEmpty());
+  }
+
   @Test
   public void saveStateTest() {
     String key = "key1";
