Infinispan70HotrodService.java

package org.radargun.service;

import org.infinispan.client.hotrod.RemoteCacheManager;
import org.infinispan.protostream.SerializationContext;
import org.radargun.Service;
import org.radargun.traits.ProvidesTrait;
import org.radargun.traits.Queryable;

/**
 * @author Vojtech Juranek <vjuranek@redhat.com>
 */
@Service(doc = Infinispan60HotrodService.SERVICE_DESCRIPTION)
public class Infinispan70HotrodService extends Infinispan60HotrodService {

   public RemoteCacheManager getRemoteManager(boolean forceReturn) {
      return forceReturn ? managerForceReturn : managerNoReturn;
   }

   @ProvidesTrait
   public InfinispanClientListeners createListeners() {
      return new InfinispanClientListeners(this);
   }

   @ProvidesTrait
   public Queryable getQueryable() {
      return new Infinispan70HotrodQueryable(this);
   }

   @Override
   protected void registerMarshallers(SerializationContext context) {
      for (RegisteredClass rc : classes) {
         try {
            context.registerMarshaller(rc.getMarshaller());
         } catch (Exception e) {
            throw new IllegalArgumentException("Could not instantiate marshaller for " + rc.clazz, e);
         }
      }
   }

}