(beam) branch release-2.58.0 updated: Adds ordered list user state support to fnapi accessor cache. (#31923) (#32013)

[jrmccluskey]

This is an automated email from the ASF dual-hosted git repository.

jrmccluskey pushed a commit to branch release-2.58.0
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/release-2.58.0 by this push:
     new 7fbbe0bf26a Adds ordered list user state support to fnapi accessor 
cache. (#31923) (#32013)
7fbbe0bf26a is described below

commit 7fbbe0bf26ae6bf1112efcd0ec5814b03c7b9f0a
Author: Shunping Huang <shunp...@google.com>
AuthorDate: Mon Jul 29 17:43:25 2024 -0400

    Adds ordered list user state support to fnapi accessor cache. (#31923) 
(#32013)
    
    Co-authored-by: Andrew Crites <cri...@google.com>
---
 .../java/org/apache/beam/fn/harness/state/FnApiStateAccessor.java | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git 
a/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/state/FnApiStateAccessor.java
 
b/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/state/FnApiStateAccessor.java
index 93f89301d15..5b304890b35 100644
--- 
a/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/state/FnApiStateAccessor.java
+++ 
b/sdks/java/harness/src/main/java/org/apache/beam/fn/harness/state/FnApiStateAccessor.java
@@ -813,14 +813,8 @@ public class FnApiStateAccessor<K> implements 
SideInputReader, StateBinder {
   private Cache<?, ?> getCacheFor(StateKey stateKey) {
     switch (stateKey.getTypeCase()) {
       case BAG_USER_STATE:
-        for (CacheToken token : cacheTokens.get()) {
-          if (!token.hasUserState()) {
-            continue;
-          }
-          return Caches.subCache(processWideCache, token, stateKey);
-        }
-        break;
       case MULTIMAP_KEYS_USER_STATE:
+      case ORDERED_LIST_USER_STATE:
         for (CacheToken token : cacheTokens.get()) {
           if (!token.hasUserState()) {
             continue;