Infinispan90HotrodService.java

package org.radargun.service;

import org.infinispan.client.hotrod.configuration.ConfigurationBuilder;
import org.radargun.Service;
import org.radargun.config.Property;
import org.radargun.traits.ContinuousQuery;
import org.radargun.traits.ProvidesTrait;

/**
 * @author Matej Cimbora
 */
@Service(doc = Infinispan60HotrodService.SERVICE_DESCRIPTION)
public class Infinispan90HotrodService extends Infinispan81HotrodService {

   @Property(doc = "Fully qualified Marshaller class. If no class is defined, the default marshaller - JBossMarshaller - is used. " +
           "Valid options are org.infinispan.marshaller.protostuff.ProtostuffMarshaller and org.infinispan.marshaller.kryo.KryoMarshaller")
   protected String marshaller;

   @ProvidesTrait
   public HotRodOperations createOperations() {
      return new Infinispan90HotRodOperations(this);
   }

   @ProvidesTrait
   public ContinuousQuery createContinuousQuery() {
      return new Infinispan90HotrodContinuousQuery(this);
   }

   @Override
   protected ConfigurationBuilder getDefaultHotRodConfig() {
      ConfigurationBuilder defaultHotRodConfig = super.getDefaultHotRodConfig();

      if (marshaller != null) {
         defaultHotRodConfig.marshaller(marshaller);
         log.info("Marshaller is set to " + marshaller);
      }
      return defaultHotRodConfig;
   }
}