com.google.cloud.dataflow.sdk.transforms.AppliedPTransform Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2015 Google Inc.
 *
 * 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 com.google.cloud.dataflow.sdk.transforms;

import com.google.cloud.dataflow.sdk.values.PInput;
import com.google.cloud.dataflow.sdk.values.POutput;
import com.google.common.base.MoreObjects;
import com.google.common.base.Objects;

/**
 * Represents the application of a {@link PTransform} to a specific input to produce
 * a specific output.
 *
 * @param  transform input type
 * @param  transform output type
 * @param  transform type
 */
public class AppliedPTransform
    > {

  private final String fullName;
  private final InputT input;
  private final OutputT output;
  private final TransformT transform;

  private AppliedPTransform(String fullName, InputT input, OutputT output, TransformT transform) {
    this.input = input;
    this.output = output;
    this.transform = transform;
    this.fullName = fullName;
  }

  public static >
  AppliedPTransform of(
      String fullName, InputT input, OutputT output, TransformT transform) {
    return new AppliedPTransform(fullName, input, output, transform);
  }

  public String getFullName() {
    return fullName;
  }

  public InputT getInput() {
    return input;
  }

  public OutputT getOutput() {
    return output;
  }

  public TransformT getTransform() {
    return transform;
  }

  @Override
  public int hashCode() {
    return Objects.hashCode(getFullName(), getInput(), getOutput(), getTransform());
  }

  @Override
  public boolean equals(Object other) {
    if (other instanceof AppliedPTransform) {
      AppliedPTransform that = (AppliedPTransform) other;
      return Objects.equal(this.getFullName(), that.getFullName())
          && Objects.equal(this.getInput(), that.getInput())
          && Objects.equal(this.getOutput(), that.getOutput())
          && Objects.equal(this.getTransform(), that.getTransform());
    } else {
      return false;
    }
  }

  @Override
  public String toString() {
    return MoreObjects.toStringHelper(getClass())
        .add("fullName", getFullName())
        .add("input", getInput())
        .add("output", getOutput())
        .add("transform", getTransform())
        .toString();
  }
}