library/src/main/java/org/apache/apex/malhar/lib/dimensions/CustomTimeBucketRegistry.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */
package org.apache.apex.malhar.lib.dimensions;

import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;

import org.apache.apex.malhar.lib.appdata.schemas.CustomTimeBucket;

import com.google.common.base.Preconditions;

import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
import it.unimi.dsi.fastutil.objects.Object2IntMap;
import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap;

/**
 * @since 3.3.0
 */
public class CustomTimeBucketRegistry implements Serializable
{
  private static final long serialVersionUID = 201509221536L;

  private int currentId;

  private Int2ObjectMap<CustomTimeBucket> idToTimeBucket = new Int2ObjectOpenHashMap<>();
  private Object2IntMap<CustomTimeBucket> timeBucketToId = new Object2IntOpenHashMap<>();
  private Map<String, CustomTimeBucket> textToTimeBucket = new HashMap<>();

  public CustomTimeBucketRegistry()
  {
  }

  public CustomTimeBucketRegistry(int startingId)
  {
    this.currentId = startingId;
  }

  public CustomTimeBucketRegistry(Int2ObjectMap<CustomTimeBucket> idToTimeBucket)
  {
    initialize(idToTimeBucket);
  }

  public CustomTimeBucketRegistry(Int2ObjectMap<CustomTimeBucket> idToTimeBucket, int startingId)
  {
    int tempId = initialize(idToTimeBucket);

    Preconditions.checkArgument(tempId < startingId, "The statingId " + startingId
        + " must be larger than the largest ID " + tempId + " in the given idToTimeBucket mapping");

    this.idToTimeBucket = Preconditions.checkNotNull(idToTimeBucket);
    this.currentId = startingId;
  }

  private int initialize(Int2ObjectMap<CustomTimeBucket> idToTimeBucket)
  {
    Preconditions.checkNotNull(idToTimeBucket);

    int tempId = Integer.MIN_VALUE;

    for (int timeBucketId : idToTimeBucket.keySet()) {
      tempId = Math.max(tempId, timeBucketId);
      CustomTimeBucket customTimeBucket = idToTimeBucket.get(timeBucketId);
      textToTimeBucket.put(customTimeBucket.getText(), customTimeBucket);
      Preconditions.checkNotNull(customTimeBucket);
      timeBucketToId.put(customTimeBucket, timeBucketId);
    }

    return tempId;
  }

  public CustomTimeBucket getTimeBucket(int timeBucketId)
  {
    return idToTimeBucket.get(timeBucketId);
  }

  public Integer getTimeBucketId(CustomTimeBucket timeBucket)
  {
    if (!timeBucketToId.containsKey(timeBucket)) {
      return null;
    }

    return timeBucketToId.get(timeBucket);
  }

  public CustomTimeBucket getTimeBucket(String text)
  {
    return textToTimeBucket.get(text);
  }

  public void register(CustomTimeBucket timeBucket)
  {
    register(timeBucket, currentId);
  }

  public void register(CustomTimeBucket timeBucket, int timeBucketId)
  {
    if (timeBucketToId.containsKey(timeBucket)) {
      throw new IllegalArgumentException("The timeBucket " + timeBucket + " is already registered.");
    }

    if (timeBucketToId.containsValue(timeBucketId)) {
      throw new IllegalArgumentException("The timeBucketId " + timeBucketId + " is already registered.");
    }

    idToTimeBucket.put(timeBucketId, timeBucket);
    timeBucketToId.put(timeBucket, timeBucketId);

    if (timeBucketId >= currentId) {
      currentId = timeBucketId + 1;
    }

    textToTimeBucket.put(timeBucket.getText(), timeBucket);
  }

  @Override
  public String toString()
  {
    return "CustomTimeBucketRegistry{" + "idToTimeBucket=" + idToTimeBucket + '}';
  }

}