예제 #1
0
 @Override
 protected OperationFactory createOperationFactory() {
   CacheOperationProvider operationProvider = getOperationProvider(parameters.name);
   CacheService service = getService(getServiceName());
   ExpiryPolicy expiryPolicy = (ExpiryPolicy) service.toObject(parameters.expiryPolicy);
   return operationProvider.createGetAllOperationFactory(
       (Set<Data>) parameters.keys, expiryPolicy);
 }
 @Override
 public void beforeRun() throws Exception {
   //        //migrate CacheConfigs first
   CacheService service = getService();
   for (CacheConfig config : configs) {
     service.createCacheConfigIfAbsent(config);
   }
 }
  @Test
  public void test_CacheReplicationOperation_serialization() throws Exception {
    TestHazelcastInstanceFactory factory = new TestHazelcastInstanceFactory(1);
    HazelcastInstance hazelcastInstance = factory.newHazelcastInstance();

    try {
      CachingProvider provider =
          HazelcastServerCachingProvider.createCachingProvider(hazelcastInstance);
      CacheManager manager = provider.getCacheManager();

      CompleteConfiguration configuration = new MutableConfiguration();
      Cache cache1 = manager.createCache("cache1", configuration);
      Cache cache2 = manager.createCache("cache2", configuration);
      Cache cache3 = manager.createCache("cache3", configuration);

      for (int i = 0; i < 1000; i++) {
        cache1.put("key" + i, i);
        cache2.put("key" + i, i);
        cache3.put("key" + i, i);
      }

      HazelcastInstanceProxy proxy = (HazelcastInstanceProxy) hazelcastInstance;

      Field original = HazelcastInstanceProxy.class.getDeclaredField("original");
      original.setAccessible(true);

      HazelcastInstanceImpl impl = (HazelcastInstanceImpl) original.get(proxy);
      NodeEngineImpl nodeEngine = impl.node.nodeEngine;
      CacheService cacheService = nodeEngine.getService(CacheService.SERVICE_NAME);

      int partitionCount = nodeEngine.getPartitionService().getPartitionCount();

      for (int partitionId = 0; partitionId < partitionCount; partitionId++) {
        CachePartitionSegment segment = cacheService.getSegment(partitionId);

        CacheReplicationOperation operation = new CacheReplicationOperation(segment, 1);
        Data serialized = service.toData(operation);
        try {
          service.toObject(serialized);
        } catch (Exception e) {
          throw new Exception("Partition: " + partitionId, e);
        }
      }

    } finally {
      factory.shutdownAll();
    }
  }
  @Override
  public void run() throws Exception {
    CacheService service = getService();
    for (Map.Entry<String, Map<Data, CacheRecord>> entry : data.entrySet()) {
      ICacheRecordStore cache = service.getOrCreateCache(entry.getKey(), getPartitionId());
      Map<Data, CacheRecord> map = entry.getValue();

      Iterator<Map.Entry<Data, CacheRecord>> iter = map.entrySet().iterator();
      while (iter.hasNext()) {
        Map.Entry<Data, CacheRecord> next = iter.next();
        Data key = next.getKey();
        CacheRecord record = next.getValue();
        iter.remove();
        cache.setRecord(key, record);
      }
    }
    data.clear();
  }
 private ICacheRecordStore getRecordStore(
     CacheService cacheService, String cacheName, int partitionId) {
   try {
     return cacheService.getOrCreateRecordStore("/hz/" + cacheName, partitionId);
   } catch (Exception e) {
     fail("CacheRecordStore not yet initialized!!!");
   }
   return null;
 }