Fix bazel tests again

Author: chenj-hub

src/main/java/build/buildfarm/instance/shard/ServerInstance.java

@@ -247,7 +247,7 @@ public class ServerInstance extends NodeInstance {
   private Cache<RequestMetadata, Boolean> recentCacheServedExecutions;
 
   private final Random rand = new Random();
-  private final Writes writes;
+  private final Writes writes = new Writes(this::writeInstanceSupplier);
   private final int maxCpu;
   private final int maxRequeueAttempts;
 
@@ -382,7 +382,6 @@ public ServerInstance(
     this.actionCacheFetchService = actionCacheFetchService;
     backplane.setOnUnsubscribe(this::stop);
 
-    this.writes = new Writes(writeInstanceCacheLoader());
     initializeCaches();
 
     remoteInputStreamFactory =
@@ -1248,35 +1247,9 @@ public void onSuccess(List<String> workersList) {
     protected abstract void onQueue(Deque<String> workers);
   }
 
-  private CacheLoader<BlobWriteKey, Instance> writeInstanceCacheLoader() {
-    return new CacheLoader<BlobWriteKey, Instance>() {
-      @SuppressWarnings("NullableProblems")
-      @Override
-      public Instance load(BlobWriteKey key) {
-        String instance = null;
-        // Per the REAPI the identifier should end up as a unique UUID per a
-        // client level - adding bytes to further mitigate collisions and not
-        // store the entire BlobWriteKey.
-        String blobKey = key.getIdentifier() + "." + key.getDigest().getSizeBytes();
-        try {
-          instance = backplane.getWriteInstance(blobKey);
-          if (instance != null) {
-            return workerStub(instance);
-          }
-        } catch (IOException e) {
-          log.log(Level.WARNING, "error getting write instance for " + instance, e);
-        }
-
-        instance = getRandomWorker();
-        try {
-          backplane.setWriteInstance(blobKey, instance);
-          log.log(Level.INFO, "set write-instance: " + blobKey + " -> " + instance); // TODO: [jmarino]: remove
-        } catch (IOException e) {
-          log.log(Level.WARNING, "error getting write instance for " + instance, e);
-        }
-        return workerStub(instance);
-      }
-    };
+  private Instance writeInstanceSupplier() {
+    String worker = getRandomWorker();
+    return workerStub(worker);
   }
 
   String getRandomWorker() {