org.infinispan.remoting.transport.ValidSingleResponseCollector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of infinispan-embedded

Infinispan Embedded All-in-One module

There is a newer version: 9.1.7.Final

package org.infinispan.remoting.transport;

import org.infinispan.commons.util.Experimental;
import org.infinispan.remoting.RpcException;
import org.infinispan.remoting.responses.CacheNotFoundResponse;
import org.infinispan.remoting.responses.ExceptionResponse;
import org.infinispan.remoting.responses.Response;
import org.infinispan.remoting.responses.ValidResponse;
import org.infinispan.util.logging.Log;
import org.infinispan.util.logging.LogFactory;

/**
 * @author Dan Berindei
 * @since 9.1
 */
@Experimental
public abstract class ValidSingleResponseCollector implements ResponseCollector {
   protected static final Log log = LogFactory.getLog(ValidSingleResponseCollector.class);

   @Override
   public final T addResponse(Address sender, Response response) {
      if (response instanceof ValidResponse) {
         return withValidResponse(sender, ((ValidResponse) response));
      } else if (response instanceof ExceptionResponse) {
         return withException(sender, ((ExceptionResponse) response).getException());
      } else if (response instanceof CacheNotFoundResponse) {
         return targetNotFound(sender);
      } else {
         // Should never happen
         return withException(sender, new RpcException("Unknown response type: " + response));
      }
   }

   @Override
   public final T finish() {
      // addResponse returned null, that means we want the final result to be null.
      return null;
   }

   protected T withException(Address sender, Exception exception) {
      throw ResponseCollectors.wrapRemoteException(sender, exception);
   }

   protected abstract T withValidResponse(Address sender, ValidResponse response);

   protected abstract T targetNotFound(Address sender);

}