org.apache.druid.client.indexing.SamplerResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of druid-server Show documentation
Druid Server
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.druid.client.indexing;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import org.apache.druid.data.input.impl.DimensionSchema;
import org.apache.druid.segment.column.RowSignature;

import java.util.List;
import java.util.Map;
import java.util.Objects;

@JsonInclude(JsonInclude.Include.NON_NULL)
public class SamplerResponse
{
  private final int numRowsRead;
  private final int numRowsIndexed;

  private final List logicalDimensions;
  private final List physicalDimensions;
  private final RowSignature logicalSegmentSchema;
  private final List data;

  @JsonCreator
  public SamplerResponse(
      @JsonProperty("numRowsRead") int numRowsRead,
      @JsonProperty("numRowsIndexed") int numRowsIndexed,
      @JsonProperty("logicalDimensions") List logicalDimensions,
      @JsonProperty("physicalDimensions") List physicalDimensions,
      @JsonProperty("logicalSegmentSchema") RowSignature logicalSegmentSchema,
      @JsonProperty("data") List data
  )
  {
    this.numRowsRead = numRowsRead;
    this.numRowsIndexed = numRowsIndexed;
    this.logicalDimensions = logicalDimensions;
    this.physicalDimensions = physicalDimensions;
    this.logicalSegmentSchema = logicalSegmentSchema;
    this.data = data;
  }

  @JsonProperty
  public int getNumRowsRead()
  {
    return numRowsRead;
  }

  @JsonProperty
  public int getNumRowsIndexed()
  {
    return numRowsIndexed;
  }

  @JsonProperty
  public List getLogicalDimensions()
  {
    return logicalDimensions;
  }

  @JsonProperty
  public List getPhysicalDimensions()
  {
    return physicalDimensions;
  }

  @JsonProperty
  public RowSignature getLogicalSegmentSchema()
  {
    return logicalSegmentSchema;
  }

  @JsonProperty
  public List getData()
  {
    return data;
  }

  @Override
  public boolean equals(Object o)
  {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }
    SamplerResponse that = (SamplerResponse) o;
    return getNumRowsRead() == that.getNumRowsRead() &&
           getNumRowsIndexed() == that.getNumRowsIndexed() &&
           Objects.equals(logicalDimensions, that.logicalDimensions) &&
           Objects.equals(physicalDimensions, that.physicalDimensions) &&
           Objects.equals(logicalSegmentSchema, that.logicalSegmentSchema) &&
           Objects.equals(data, that.data);
  }

  @Override
  public int hashCode()
  {
    return Objects.hash(
        getNumRowsRead(),
        getNumRowsIndexed(),
        logicalDimensions,
        physicalDimensions,
        logicalSegmentSchema,
        data
    );
  }

  @Override
  public String toString()
  {
    return "SamplerResponse{" +
           "numRowsRead=" + numRowsRead +
           ", numRowsIndexed=" + numRowsIndexed +
           ", logicalDimensions=" + logicalDimensions +
           ", physicalDimensions=" + physicalDimensions +
           ", logicalSegmentSchema=" + logicalSegmentSchema +
           ", data=" + data +
           '}';
  }

  @JsonInclude(JsonInclude.Include.NON_NULL)
  public static class SamplerResponseRow
  {
    private final Map input;
    private final Map parsed;
    private final Boolean unparseable;
    private final String error;

    @JsonCreator
    public SamplerResponseRow(
        @JsonProperty("input") Map input,
        @JsonProperty("parsed") Map parsed,
        @JsonProperty("unparseable") Boolean unparseable,
        @JsonProperty("error") String error
    )
    {
      this.input = input;
      this.parsed = parsed;
      this.unparseable = unparseable;
      this.error = error;
    }

    @JsonProperty
    public Map getInput()
    {
      return input;
    }

    @JsonProperty
    public Map getParsed()
    {
      return parsed;
    }

    @JsonProperty
    public Boolean isUnparseable()
    {
      return unparseable;
    }

    @JsonProperty
    public String getError()
    {
      return error;
    }

    public SamplerResponseRow withParsed(Map parsed)
    {
      return new SamplerResponseRow(input, parsed, unparseable, error);
    }

    @Override
    public boolean equals(Object o)
    {
      if (this == o) {
        return true;
      }
      if (o == null || getClass() != o.getClass()) {
        return false;
      }
      SamplerResponseRow that = (SamplerResponseRow) o;
      return Objects.equals(input, that.input) &&
             Objects.equals(parsed, that.parsed) &&
             Objects.equals(unparseable, that.unparseable) &&
             Objects.equals(error, that.error);
    }

    @Override
    public int hashCode()
    {
      return Objects.hash(input, parsed, unparseable, error);
    }

    @Override
    public String toString()
    {
      return "SamplerResponseRow{" +
             "rawInput=" + input +
             ", parsed=" + parsed +
             ", unparseable=" + unparseable +
             ", error='" + error + '\'' +
             '}';
    }
  }
}