quickstart.QuarterPartitionResolver Maven / Gradle / Ivy

Go to download
/*
 * 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 quickstart;

import java.util.Calendar;
import java.util.Date;
import java.util.Properties;
import java.util.Set;

import com.gemstone.gemfire.cache.Declarable;
import com.gemstone.gemfire.cache.EntryOperation;
import com.gemstone.gemfire.cache.FixedPartitionResolver;

/**
 * Partitions data into two "quarters."
 * 
 * 
 * @author GemStone Systems, Inc.
 */
public class QuarterPartitionResolver implements FixedPartitionResolver, Declarable{

  @Override
  public String getPartitionName(EntryOperation opDetails, Set targetPartitions) {
    Date date = opDetails.getKey();
    Calendar cal = Calendar.getInstance();
    cal.setTime(date);
    int month = cal.get(Calendar.MONTH);
    if (month == 0 || month == 1 || month == 2 || month == 3 || month == 4 || month == 5) {
      return "Q1";
    }
    else if (month == 6 || month == 7 || month == 8 || month == 9 || month == 10 || month == 11) {
      return "Q2";
    }
    else {
      return "Invalid Quarter";
    }
  }

  @Override
  public String getName() {
    return "QuarterPartitionResolver";
  }

  @Override
  public Object getRoutingObject(EntryOperation opDetails) {
    Date date = opDetails.getKey();
    Calendar cal = Calendar.getInstance();
    cal.setTime(date);
    int month = cal.get(Calendar.MONTH);
    return month;
  }

  @Override
  public void close() {
    // do nothing
  }

  @Override
  public void init(Properties props) {
    // do nothing
  }
}