Adds ordered list user state support to fnapi accessor cache. (#31923) (

#32013) Co-authored-by: Andrew Crites <[email protected]>
apache · Jul 29, 2024 · 7fbbe0b · 7fbbe0b
1 parent bb828dc
commit 7fbbe0b
Showing 1 changed file with 1 addition and 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
@@ -813,14 +813,8 @@ public WatermarkHoldState bindWatermark(
   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;