BuildGroupItemBase.java

/*
 * Copyright © 2014 - 2021 Leipzig University (Database Research Group)
 *
 * 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.
 */
package org.gradoop.flink.model.impl.operators.grouping.functions;

import org.gradoop.common.model.api.entities.Element;
import org.gradoop.common.model.impl.properties.PropertyValue;
import org.gradoop.common.model.impl.properties.PropertyValueList;
import org.gradoop.flink.model.impl.operators.grouping.Grouping;
import org.gradoop.flink.model.impl.operators.grouping.tuples.GroupItem;
import org.gradoop.flink.model.impl.operators.grouping.tuples.LabelGroup;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * Base class for vertex and edge item building.
 */
public class BuildGroupItemBase extends BuildBase {

  /**
   * Stores grouping properties and aggregators for vertex labels.
   */
  private final List<LabelGroup> labelGroups;
  /**
   * Stores the information about the default label group, this is either the vertex or the
   * edge default label group.
   */
  private final LabelGroup defaultLabelGroup;

  /**
   * Stores the grouping values. Used to avoid object instantiation.
   */
  private List<PropertyValue> groupingValues;

  /**
   * Valued constructor.
   *
   * @param useLabel    true if label shall be used for grouping
   * @param labelGroups all vertex or edge label groups
   */
  public BuildGroupItemBase(
    boolean useLabel, List<LabelGroup> labelGroups) {
    super(useLabel);
    this.labelGroups = labelGroups;
    groupingValues = new ArrayList<>();
    LabelGroup standardLabelGroup = null;

    // find and keep the default label group for fast access
    for (LabelGroup labelGroup : labelGroups) {
      if (labelGroup.getGroupingLabel().equals(Grouping.DEFAULT_VERTEX_LABEL_GROUP)) {
        standardLabelGroup = labelGroup;
        break;
      } else if (labelGroup.getGroupingLabel().equals(Grouping.DEFAULT_EDGE_LABEL_GROUP)) {
        standardLabelGroup = labelGroup;
        break;
      }
    }
    defaultLabelGroup = standardLabelGroup;
  }

  /**
   * Sets the basic values for either a vertex or an edge group item.
   *
   * @param groupItem the group item to be set
   * @param element the epgm element
   * @param labelGroup label group to be assigned
   * @throws IOException on failure
   */
  protected void setGroupItem(GroupItem groupItem, Element element, LabelGroup labelGroup)
    throws IOException {
    // stores all, in the label group specified, grouping values of the element, if the element
    // does not have a property a null property value is stored
    for (String groupPropertyKey : labelGroup.getPropertyKeys()) {
      if (element.hasProperty(groupPropertyKey)) {
        groupingValues.add(element.getPropertyValue(groupPropertyKey));
      } else {
        groupingValues.add(PropertyValue.NULL_VALUE);
      }
    }
    // If the label group is the default one and the labels shall be used for grouping the
    // elements labels are kept, otherwise the label given by the group is taken. The default
    // label groups label is empty and if the current label group is a manually specified one its
    // label is also taken.
    if (labelGroup.getGroupingLabel().equals(getDefaultLabelGroup().getGroupingLabel()) &&
      useLabel()) {
      groupItem.setGroupLabel(element.getLabel());
    } else {
      groupItem.setGroupLabel(labelGroup.getGroupLabel());
    }

    groupItem.setAggregateValues(labelGroup.getIncrementValues(element));
    groupItem.setLabelGroup(labelGroup);
    groupItem.setGroupingValues(PropertyValueList.fromPropertyValues(groupingValues));
    groupingValues.clear();
  }

  protected List<LabelGroup> getLabelGroups() {
    return labelGroups;
  }

  protected LabelGroup getDefaultLabelGroup() {
    return defaultLabelGroup;
  }
}