org.infinispan.server.tasks.DistributedServerTaskRunner Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of infinispan-server-runtime Show documentation

Infinispan Server

The newest version!

package org.infinispan.server.tasks;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;

import org.infinispan.Cache;
import org.infinispan.commons.CacheException;
import org.infinispan.manager.ClusterExecutor;
import org.infinispan.remoting.transport.Address;
import org.infinispan.security.Security;
import org.infinispan.security.actions.SecurityActions;
import org.infinispan.tasks.TaskContext;
import org.infinispan.util.function.TriConsumer;

/**
 * @author Michal Szynkiewicz, [email protected]
 */
public final class DistributedServerTaskRunner implements ServerTaskRunner {

   private static final DistributedServerTaskRunner INSTANCE = new DistributedServerTaskRunner();

   private DistributedServerTaskRunner() { }

   public static DistributedServerTaskRunner getInstance() {
      return INSTANCE;
   }

   @Override
   public  CompletableFuture execute(ServerTaskWrapper task, TaskContext context) {
      String cacheName = context.getCache().map(Cache::getName).orElse(null);

      ClusterExecutor clusterExecutor = SecurityActions.getClusterExecutor(context.getCacheManager());

      List results = new ArrayList<>();
      TriConsumer triConsumer = (a, v, t) -> {
         if (t != null) {
            throw new CacheException(t);
         }
         synchronized (results) {
            results.add(v);
         }
      };
      CompletableFuture future = Security.doAs(context.subject(), () -> clusterExecutor.submitConsumer(
            new DistributedServerTask<>(task.getName(), cacheName, context),
            triConsumer
      ));
      return (CompletableFuture) future.thenApply(ignore -> results);
   }
}