From e2b8acb8bbfa37058f17686747a1ec42cdc3bae4 Mon Sep 17 00:00:00 2001
From: Andrew Crites <crites@google.com>
Date: Wed, 24 Jul 2024 09:55:21 -0700
Subject: [PATCH] Adds ordered list user state support to fnapi accessor cache.
 (#31923)

---
 .../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 93f89301d158..5b304890b354 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 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;