com.gemstone.gemfire.cache.snapshot.RegionGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gemfire-junit Show documentation
SnappyData store based off Pivotal GemFireXD
There is a newer version: 2.0-BETA
/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You
 * may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License. See accompanying
 * LICENSE file.
 */
package com.gemstone.gemfire.cache.snapshot;

import com.examples.snapshot.MyDataSerializer;
import com.examples.snapshot.MyDataSerializer.MyObjectDataSerializable2;
import com.examples.snapshot.MyObject;
import com.examples.snapshot.MyObjectDataSerializable;
import com.examples.snapshot.MyObjectPdx;
import com.examples.snapshot.MyObjectPdx.MyEnumPdx;
import com.examples.snapshot.MyPdxSerializer.MyObjectPdx2;
import com.gemstone.gemfire.DataSerializer;
import com.gemstone.gemfire.cache.Cache;
import com.gemstone.gemfire.cache.EvictionAction;
import com.gemstone.gemfire.cache.EvictionAttributes;
import com.gemstone.gemfire.cache.Region;
import com.gemstone.gemfire.cache.RegionShortcut;

public class RegionGenerator {
  public enum RegionType {
    REPLICATE,
    REPLICATE_PERSISTENT,
    REPLICATE_PERSISTENT_OVERFLOW,
    PARTITION,
    PARTITION_PERSISTENT,
    PARTITION_PERSISTENT_OVERFLOW;
    
    public static RegionType[] persistentValues() {
      return new RegionType[] { 
          REPLICATE_PERSISTENT, 
          REPLICATE_PERSISTENT_OVERFLOW, 
          PARTITION_PERSISTENT, 
          PARTITION_PERSISTENT_OVERFLOW
      };
    }
  }
  
  public enum SerializationType {
    SERIALIZABLE,
    DATA_SERIALIZABLE,
    DATA_SERIALIZER,
    PDX,
    PDX_SERIALIZER;
    
    public static SerializationType[] offlineValues() {
      return new SerializationType[] {
          SERIALIZABLE,
          DATA_SERIALIZABLE,
          DATA_SERIALIZER
      };
    }
  }
  
  public RegionGenerator() {
    DataSerializer.register(MyDataSerializer.class);
  }
  
  public  Region createRegion(Cache c, String diskStore, RegionType type, String name) {
    Region r = c.getRegion(name);
    switch (type) {
    case REPLICATE:
      return c.createRegionFactory(RegionShortcut.REPLICATE).create(name);
      
    case REPLICATE_PERSISTENT:
      return c.createRegionFactory(RegionShortcut.REPLICATE_PERSISTENT)
        .setDiskStoreName(diskStore)
        .create(name);
      
    case REPLICATE_PERSISTENT_OVERFLOW:
      return c.createRegionFactory(RegionShortcut.REPLICATE_PERSISTENT_OVERFLOW)
          .setEvictionAttributes(EvictionAttributes.createLIFOEntryAttributes(1, EvictionAction.OVERFLOW_TO_DISK))
          .setDiskStoreName(diskStore)
          .create(name);
      
    case PARTITION:
      return c.createRegionFactory(RegionShortcut.PARTITION).create(name);

    case PARTITION_PERSISTENT:
      return c.createRegionFactory(RegionShortcut.PARTITION_PERSISTENT)
          .setDiskStoreName(diskStore)
          .create(name);
      
    case PARTITION_PERSISTENT_OVERFLOW:
      return c.createRegionFactory(RegionShortcut.PARTITION_PERSISTENT_OVERFLOW)
          .setEvictionAttributes(EvictionAttributes.createLIFOEntryAttributes(5, EvictionAction.OVERFLOW_TO_DISK))
          .setDiskStoreName(diskStore)
          .create(name);
    }
    throw new IllegalArgumentException();
  }
  
  public MyObject createData(SerializationType type, int f1, String f2) {
    switch (type) {
    case SERIALIZABLE:
      return new MyObject(f1, f2);

    case DATA_SERIALIZABLE:
      return new MyObjectDataSerializable(f1, f2);
      
    case DATA_SERIALIZER:
      return new MyObjectDataSerializable2(f1, f2);

    case PDX:
      return new MyObjectPdx(f1, f2, MyEnumPdx.const1);
    
    case PDX_SERIALIZER:
      return new MyObjectPdx2(f1, f2);
    }
    throw new IllegalArgumentException();
  }
}