org.apache.lens.cube.metadata.BaseDimAttribute Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lens-cube Show documentation
OLAP cube queries
The newest version!
/**
 * 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.lens.cube.metadata;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;

import java.util.*;

import org.apache.hadoop.hive.metastore.api.FieldSchema;

import com.google.common.base.Optional;

import lombok.Getter;
import lombok.ToString;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@ToString(callSuper=true, includeFieldNames=true)
public class BaseDimAttribute extends CubeDimAttribute {
  @Getter private final String type;
  @Getter private Optional numOfDistinctValues = Optional.absent();
  @Getter private Optional> values = Optional.absent();

  public BaseDimAttribute(FieldSchema column) {
    this(column, null, null, null, null);
  }

  public BaseDimAttribute(FieldSchema column, String displayString, Date startTime, Date endTime, Double cost) {
    this(column, displayString, startTime, endTime, cost, null);
  }

  public BaseDimAttribute(FieldSchema column, String displayString, Date startTime, Date endTime, Double cost,
      Long numOfDistinctValues) {
    this(column, displayString, startTime, endTime, cost, numOfDistinctValues, null);
  }

  public BaseDimAttribute(FieldSchema column, String displayString, Date startTime, Date endTime, Double cost,
                          Long numOfDistinctValues, List values) {
    this(column, displayString, startTime, endTime, cost, numOfDistinctValues, values, new HashMap());
  }

  public BaseDimAttribute(FieldSchema column, String displayString, Date startTime, Date endTime, Double cost,
    Long numOfDistinctValues, List values, Map tags) {
    super(column.getName(), column.getComment(), displayString, startTime, endTime, cost, tags);
    this.type = column.getType();
    checkNotNull(type);
    Optional optionalNumOfDistnctValues = Optional.fromNullable(numOfDistinctValues);
    if (optionalNumOfDistnctValues.isPresent()) {
      this.numOfDistinctValues = optionalNumOfDistnctValues;
      checkArgument(this.numOfDistinctValues.get() > 0);
    }
    Optional> optionalValues = Optional.fromNullable(values);
    if (optionalValues.isPresent()) {
      this.values = optionalValues;
      if (!this.values.get().isEmpty()) {
        // always put num distinct values same of values size, if values are specified.
        this.numOfDistinctValues = Optional.of(Long.valueOf(optionalValues.get().size()));
      }
    }
  }

  @Override
  public void addProperties(Map props) {
    super.addProperties(props);
    props.put(MetastoreUtil.getDimTypePropertyKey(getName()), type);
    if (numOfDistinctValues.isPresent()) {
      props.put(MetastoreUtil.getDimNumOfDistinctValuesPropertyKey(getName()),
          String.valueOf(numOfDistinctValues.get()));
    }
    if (values.isPresent() && !this.values.get().isEmpty()) {
      props.put(MetastoreUtil.getInlineDimensionValuesKey(getName()), MetastoreUtil.getStr(values.get()));
    }
  }

  /**
   * This is used only for serializing
   *
   * @param name
   * @param props
   */
  public BaseDimAttribute(String name, Map props) {
    super(name, props);
    this.type = getDimType(name, props);
    this.numOfDistinctValues = getDimNumOfDistinctValues(name, props);
    this.values = getValues(name, props);

  }

  static String getDimType(String name, Map props) {
    return props.get(MetastoreUtil.getDimTypePropertyKey(name));
  }

  static Optional getDimNumOfDistinctValues(String name, Map props) {
    if (props.containsKey(MetastoreUtil.getDimNumOfDistinctValuesPropertyKey(name))) {
      try {
        return Optional.of(Long.parseLong((props.get(MetastoreUtil.getDimNumOfDistinctValuesPropertyKey(name)))));
      } catch (NumberFormatException ne) {
        log.error("NumberFormat exception while parsing the num of distinct vlaues {}",
            props.get(MetastoreUtil.getDimNumOfDistinctValuesPropertyKey(name)));
      }
    }
    return Optional.absent();
  }

  static Optional> getValues(String name, Map props) {
    if (props.containsKey(MetastoreUtil.getInlineDimensionValuesKey(name))) {
      String valueStr = props.get(MetastoreUtil.getInlineDimensionValuesKey(name));
      return Optional.of(Arrays.asList(valueStr.split(",")));
    }
    return Optional.absent();
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = super.hashCode();
    result = prime * result + ((getType() == null) ? 0 : getType().toLowerCase().hashCode());
    if (numOfDistinctValues.isPresent()) {
      result = prime * result + numOfDistinctValues.get().hashCode();
    }
    if (values.isPresent()) {
      result = prime * result + values.get().hashCode();
    }
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (!super.equals(obj)) {
      return false;
    }
    BaseDimAttribute other = (BaseDimAttribute) obj;
    if (this.getType() == null) {
      if (other.getType() != null) {
        return false;
      }
    } else if (!this.getType().equalsIgnoreCase(other.getType())) {
      return false;
    } else if (!this.numOfDistinctValues.equals(other.numOfDistinctValues)) {
      return false;
    } else if (!this.getValues().equals(other.getValues())) {
      return false;
    }
    return true;
  }
}