TFetchSplitBatchRequest.java

/**
 * Autogenerated by Thrift Compiler (0.16.0)
 *
 * DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 *  @generated
 */
package org.apache.doris.thrift;

@SuppressWarnings({"cast", "rawtypes", "serial", "unchecked", "unused"})
@javax.annotation.Generated(value = "Autogenerated by Thrift Compiler (0.16.0)", date = "2026-05-14")
public class TFetchSplitBatchRequest implements org.apache.thrift.TBase<TFetchSplitBatchRequest, TFetchSplitBatchRequest._Fields>, java.io.Serializable, Cloneable, Comparable<TFetchSplitBatchRequest> {
  private static final org.apache.thrift.protocol.TStruct STRUCT_DESC = new org.apache.thrift.protocol.TStruct("TFetchSplitBatchRequest");

  private static final org.apache.thrift.protocol.TField SPLIT_SOURCE_ID_FIELD_DESC = new org.apache.thrift.protocol.TField("split_source_id", org.apache.thrift.protocol.TType.I64, (short)1);
  private static final org.apache.thrift.protocol.TField MAX_NUM_SPLITS_FIELD_DESC = new org.apache.thrift.protocol.TField("max_num_splits", org.apache.thrift.protocol.TType.I32, (short)2);

  private static final org.apache.thrift.scheme.SchemeFactory STANDARD_SCHEME_FACTORY = new TFetchSplitBatchRequestStandardSchemeFactory();
  private static final org.apache.thrift.scheme.SchemeFactory TUPLE_SCHEME_FACTORY = new TFetchSplitBatchRequestTupleSchemeFactory();

  public long split_source_id; // optional
  public int max_num_splits; // optional

  /** The set of fields this struct contains, along with convenience methods for finding and manipulating them. */
  public enum _Fields implements org.apache.thrift.TFieldIdEnum {
    SPLIT_SOURCE_ID((short)1, "split_source_id"),
    MAX_NUM_SPLITS((short)2, "max_num_splits");

    private static final java.util.Map<java.lang.String, _Fields> byName = new java.util.HashMap<java.lang.String, _Fields>();

    static {
      for (_Fields field : java.util.EnumSet.allOf(_Fields.class)) {
        byName.put(field.getFieldName(), field);
      }
    }

    /**
     * Find the _Fields constant that matches fieldId, or null if its not found.
     */
    @org.apache.thrift.annotation.Nullable
    public static _Fields findByThriftId(int fieldId) {
      switch(fieldId) {
        case 1: // SPLIT_SOURCE_ID
          return SPLIT_SOURCE_ID;
        case 2: // MAX_NUM_SPLITS
          return MAX_NUM_SPLITS;
        default:
          return null;
      }
    }

    /**
     * Find the _Fields constant that matches fieldId, throwing an exception
     * if it is not found.
     */
    public static _Fields findByThriftIdOrThrow(int fieldId) {
      _Fields fields = findByThriftId(fieldId);
      if (fields == null) throw new java.lang.IllegalArgumentException("Field " + fieldId + " doesn't exist!");
      return fields;
    }

    /**
     * Find the _Fields constant that matches name, or null if its not found.
     */
    @org.apache.thrift.annotation.Nullable
    public static _Fields findByName(java.lang.String name) {
      return byName.get(name);
    }

    private final short _thriftId;
    private final java.lang.String _fieldName;

    _Fields(short thriftId, java.lang.String fieldName) {
      _thriftId = thriftId;
      _fieldName = fieldName;
    }

    public short getThriftFieldId() {
      return _thriftId;
    }

    public java.lang.String getFieldName() {
      return _fieldName;
    }
  }

  // isset id assignments
  private static final int __SPLIT_SOURCE_ID_ISSET_ID = 0;
  private static final int __MAX_NUM_SPLITS_ISSET_ID = 1;
  private byte __isset_bitfield = 0;
  private static final _Fields optionals[] = {_Fields.SPLIT_SOURCE_ID,_Fields.MAX_NUM_SPLITS};
  public static final java.util.Map<_Fields, org.apache.thrift.meta_data.FieldMetaData> metaDataMap;
  static {
    java.util.Map<_Fields, org.apache.thrift.meta_data.FieldMetaData> tmpMap = new java.util.EnumMap<_Fields, org.apache.thrift.meta_data.FieldMetaData>(_Fields.class);
    tmpMap.put(_Fields.SPLIT_SOURCE_ID, new org.apache.thrift.meta_data.FieldMetaData("split_source_id", org.apache.thrift.TFieldRequirementType.OPTIONAL, 
        new org.apache.thrift.meta_data.FieldValueMetaData(org.apache.thrift.protocol.TType.I64)));
    tmpMap.put(_Fields.MAX_NUM_SPLITS, new org.apache.thrift.meta_data.FieldMetaData("max_num_splits", org.apache.thrift.TFieldRequirementType.OPTIONAL, 
        new org.apache.thrift.meta_data.FieldValueMetaData(org.apache.thrift.protocol.TType.I32)));
    metaDataMap = java.util.Collections.unmodifiableMap(tmpMap);
    org.apache.thrift.meta_data.FieldMetaData.addStructMetaDataMap(TFetchSplitBatchRequest.class, metaDataMap);
  }

  public TFetchSplitBatchRequest() {
  }

  /**
   * Performs a deep copy on <i>other</i>.
   */
  public TFetchSplitBatchRequest(TFetchSplitBatchRequest other) {
    __isset_bitfield = other.__isset_bitfield;
    this.split_source_id = other.split_source_id;
    this.max_num_splits = other.max_num_splits;
  }

  public TFetchSplitBatchRequest deepCopy() {
    return new TFetchSplitBatchRequest(this);
  }

  @Override
  public void clear() {
    setSplitSourceIdIsSet(false);
    this.split_source_id = 0;
    setMaxNumSplitsIsSet(false);
    this.max_num_splits = 0;
  }

  public long getSplitSourceId() {
    return this.split_source_id;
  }

  public TFetchSplitBatchRequest setSplitSourceId(long split_source_id) {
    this.split_source_id = split_source_id;
    setSplitSourceIdIsSet(true);
    return this;
  }

  public void unsetSplitSourceId() {
    __isset_bitfield = org.apache.thrift.EncodingUtils.clearBit(__isset_bitfield, __SPLIT_SOURCE_ID_ISSET_ID);
  }

  /** Returns true if field split_source_id is set (has been assigned a value) and false otherwise */
  public boolean isSetSplitSourceId() {
    return org.apache.thrift.EncodingUtils.testBit(__isset_bitfield, __SPLIT_SOURCE_ID_ISSET_ID);
  }

  public void setSplitSourceIdIsSet(boolean value) {
    __isset_bitfield = org.apache.thrift.EncodingUtils.setBit(__isset_bitfield, __SPLIT_SOURCE_ID_ISSET_ID, value);
  }

  public int getMaxNumSplits() {
    return this.max_num_splits;
  }

  public TFetchSplitBatchRequest setMaxNumSplits(int max_num_splits) {
    this.max_num_splits = max_num_splits;
    setMaxNumSplitsIsSet(true);
    return this;
  }

  public void unsetMaxNumSplits() {
    __isset_bitfield = org.apache.thrift.EncodingUtils.clearBit(__isset_bitfield, __MAX_NUM_SPLITS_ISSET_ID);
  }

  /** Returns true if field max_num_splits is set (has been assigned a value) and false otherwise */
  public boolean isSetMaxNumSplits() {
    return org.apache.thrift.EncodingUtils.testBit(__isset_bitfield, __MAX_NUM_SPLITS_ISSET_ID);
  }

  public void setMaxNumSplitsIsSet(boolean value) {
    __isset_bitfield = org.apache.thrift.EncodingUtils.setBit(__isset_bitfield, __MAX_NUM_SPLITS_ISSET_ID, value);
  }

  public void setFieldValue(_Fields field, @org.apache.thrift.annotation.Nullable java.lang.Object value) {
    switch (field) {
    case SPLIT_SOURCE_ID:
      if (value == null) {
        unsetSplitSourceId();
      } else {
        setSplitSourceId((java.lang.Long)value);
      }
      break;

    case MAX_NUM_SPLITS:
      if (value == null) {
        unsetMaxNumSplits();
      } else {
        setMaxNumSplits((java.lang.Integer)value);
      }
      break;

    }
  }

  @org.apache.thrift.annotation.Nullable
  public java.lang.Object getFieldValue(_Fields field) {
    switch (field) {
    case SPLIT_SOURCE_ID:
      return getSplitSourceId();

    case MAX_NUM_SPLITS:
      return getMaxNumSplits();

    }
    throw new java.lang.IllegalStateException();
  }

  /** Returns true if field corresponding to fieldID is set (has been assigned a value) and false otherwise */
  public boolean isSet(_Fields field) {
    if (field == null) {
      throw new java.lang.IllegalArgumentException();
    }

    switch (field) {
    case SPLIT_SOURCE_ID:
      return isSetSplitSourceId();
    case MAX_NUM_SPLITS:
      return isSetMaxNumSplits();
    }
    throw new java.lang.IllegalStateException();
  }

  @Override
  public boolean equals(java.lang.Object that) {
    if (that instanceof TFetchSplitBatchRequest)
      return this.equals((TFetchSplitBatchRequest)that);
    return false;
  }

  public boolean equals(TFetchSplitBatchRequest that) {
    if (that == null)
      return false;
    if (this == that)
      return true;

    boolean this_present_split_source_id = true && this.isSetSplitSourceId();
    boolean that_present_split_source_id = true && that.isSetSplitSourceId();
    if (this_present_split_source_id || that_present_split_source_id) {
      if (!(this_present_split_source_id && that_present_split_source_id))
        return false;
      if (this.split_source_id != that.split_source_id)
        return false;
    }

    boolean this_present_max_num_splits = true && this.isSetMaxNumSplits();
    boolean that_present_max_num_splits = true && that.isSetMaxNumSplits();
    if (this_present_max_num_splits || that_present_max_num_splits) {
      if (!(this_present_max_num_splits && that_present_max_num_splits))
        return false;
      if (this.max_num_splits != that.max_num_splits)
        return false;
    }

    return true;
  }

  @Override
  public int hashCode() {
    int hashCode = 1;

    hashCode = hashCode * 8191 + ((isSetSplitSourceId()) ? 131071 : 524287);
    if (isSetSplitSourceId())
      hashCode = hashCode * 8191 + org.apache.thrift.TBaseHelper.hashCode(split_source_id);

    hashCode = hashCode * 8191 + ((isSetMaxNumSplits()) ? 131071 : 524287);
    if (isSetMaxNumSplits())
      hashCode = hashCode * 8191 + max_num_splits;

    return hashCode;
  }

  @Override
  public int compareTo(TFetchSplitBatchRequest other) {
    if (!getClass().equals(other.getClass())) {
      return getClass().getName().compareTo(other.getClass().getName());
    }

    int lastComparison = 0;

    lastComparison = java.lang.Boolean.compare(isSetSplitSourceId(), other.isSetSplitSourceId());
    if (lastComparison != 0) {
      return lastComparison;
    }
    if (isSetSplitSourceId()) {
      lastComparison = org.apache.thrift.TBaseHelper.compareTo(this.split_source_id, other.split_source_id);
      if (lastComparison != 0) {
        return lastComparison;
      }
    }
    lastComparison = java.lang.Boolean.compare(isSetMaxNumSplits(), other.isSetMaxNumSplits());
    if (lastComparison != 0) {
      return lastComparison;
    }
    if (isSetMaxNumSplits()) {
      lastComparison = org.apache.thrift.TBaseHelper.compareTo(this.max_num_splits, other.max_num_splits);
      if (lastComparison != 0) {
        return lastComparison;
      }
    }
    return 0;
  }

  @org.apache.thrift.annotation.Nullable
  public _Fields fieldForId(int fieldId) {
    return _Fields.findByThriftId(fieldId);
  }

  public void read(org.apache.thrift.protocol.TProtocol iprot) throws org.apache.thrift.TException {
    scheme(iprot).read(iprot, this);
  }

  public void write(org.apache.thrift.protocol.TProtocol oprot) throws org.apache.thrift.TException {
    scheme(oprot).write(oprot, this);
  }

  @Override
  public java.lang.String toString() {
    java.lang.StringBuilder sb = new java.lang.StringBuilder("TFetchSplitBatchRequest(");
    boolean first = true;

    if (isSetSplitSourceId()) {
      sb.append("split_source_id:");
      sb.append(this.split_source_id);
      first = false;
    }
    if (isSetMaxNumSplits()) {
      if (!first) sb.append(", ");
      sb.append("max_num_splits:");
      sb.append(this.max_num_splits);
      first = false;
    }
    sb.append(")");
    return sb.toString();
  }

  public void validate() throws org.apache.thrift.TException {
    // check for required fields
    // check for sub-struct validity
  }

  private void writeObject(java.io.ObjectOutputStream out) throws java.io.IOException {
    try {
      write(new org.apache.thrift.protocol.TCompactProtocol(new org.apache.thrift.transport.TIOStreamTransport(out)));
    } catch (org.apache.thrift.TException te) {
      throw new java.io.IOException(te);
    }
  }

  private void readObject(java.io.ObjectInputStream in) throws java.io.IOException, java.lang.ClassNotFoundException {
    try {
      // it doesn't seem like you should have to do this, but java serialization is wacky, and doesn't call the default constructor.
      __isset_bitfield = 0;
      read(new org.apache.thrift.protocol.TCompactProtocol(new org.apache.thrift.transport.TIOStreamTransport(in)));
    } catch (org.apache.thrift.TException te) {
      throw new java.io.IOException(te);
    }
  }

  private static class TFetchSplitBatchRequestStandardSchemeFactory implements org.apache.thrift.scheme.SchemeFactory {
    public TFetchSplitBatchRequestStandardScheme getScheme() {
      return new TFetchSplitBatchRequestStandardScheme();
    }
  }

  private static class TFetchSplitBatchRequestStandardScheme extends org.apache.thrift.scheme.StandardScheme<TFetchSplitBatchRequest> {

    public void read(org.apache.thrift.protocol.TProtocol iprot, TFetchSplitBatchRequest struct) throws org.apache.thrift.TException {
      org.apache.thrift.protocol.TField schemeField;
      iprot.readStructBegin();
      while (true)
      {
        schemeField = iprot.readFieldBegin();
        if (schemeField.type == org.apache.thrift.protocol.TType.STOP) { 
          break;
        }
        switch (schemeField.id) {
          case 1: // SPLIT_SOURCE_ID
            if (schemeField.type == org.apache.thrift.protocol.TType.I64) {
              struct.split_source_id = iprot.readI64();
              struct.setSplitSourceIdIsSet(true);
            } else { 
              org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type);
            }
            break;
          case 2: // MAX_NUM_SPLITS
            if (schemeField.type == org.apache.thrift.protocol.TType.I32) {
              struct.max_num_splits = iprot.readI32();
              struct.setMaxNumSplitsIsSet(true);
            } else { 
              org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type);
            }
            break;
          default:
            org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type);
        }
        iprot.readFieldEnd();
      }
      iprot.readStructEnd();

      // check for required fields of primitive type, which can't be checked in the validate method
      struct.validate();
    }

    public void write(org.apache.thrift.protocol.TProtocol oprot, TFetchSplitBatchRequest struct) throws org.apache.thrift.TException {
      struct.validate();

      oprot.writeStructBegin(STRUCT_DESC);
      if (struct.isSetSplitSourceId()) {
        oprot.writeFieldBegin(SPLIT_SOURCE_ID_FIELD_DESC);
        oprot.writeI64(struct.split_source_id);
        oprot.writeFieldEnd();
      }
      if (struct.isSetMaxNumSplits()) {
        oprot.writeFieldBegin(MAX_NUM_SPLITS_FIELD_DESC);
        oprot.writeI32(struct.max_num_splits);
        oprot.writeFieldEnd();
      }
      oprot.writeFieldStop();
      oprot.writeStructEnd();
    }

  }

  private static class TFetchSplitBatchRequestTupleSchemeFactory implements org.apache.thrift.scheme.SchemeFactory {
    public TFetchSplitBatchRequestTupleScheme getScheme() {
      return new TFetchSplitBatchRequestTupleScheme();
    }
  }

  private static class TFetchSplitBatchRequestTupleScheme extends org.apache.thrift.scheme.TupleScheme<TFetchSplitBatchRequest> {

    @Override
    public void write(org.apache.thrift.protocol.TProtocol prot, TFetchSplitBatchRequest struct) throws org.apache.thrift.TException {
      org.apache.thrift.protocol.TTupleProtocol oprot = (org.apache.thrift.protocol.TTupleProtocol) prot;
      java.util.BitSet optionals = new java.util.BitSet();
      if (struct.isSetSplitSourceId()) {
        optionals.set(0);
      }
      if (struct.isSetMaxNumSplits()) {
        optionals.set(1);
      }
      oprot.writeBitSet(optionals, 2);
      if (struct.isSetSplitSourceId()) {
        oprot.writeI64(struct.split_source_id);
      }
      if (struct.isSetMaxNumSplits()) {
        oprot.writeI32(struct.max_num_splits);
      }
    }

    @Override
    public void read(org.apache.thrift.protocol.TProtocol prot, TFetchSplitBatchRequest struct) throws org.apache.thrift.TException {
      org.apache.thrift.protocol.TTupleProtocol iprot = (org.apache.thrift.protocol.TTupleProtocol) prot;
      java.util.BitSet incoming = iprot.readBitSet(2);
      if (incoming.get(0)) {
        struct.split_source_id = iprot.readI64();
        struct.setSplitSourceIdIsSet(true);
      }
      if (incoming.get(1)) {
        struct.max_num_splits = iprot.readI32();
        struct.setMaxNumSplitsIsSet(true);
      }
    }
  }

  private static <S extends org.apache.thrift.scheme.IScheme> S scheme(org.apache.thrift.protocol.TProtocol proto) {
    return (org.apache.thrift.scheme.StandardScheme.class.equals(proto.getScheme()) ? STANDARD_SCHEME_FACTORY : TUPLE_SCHEME_FACTORY).getScheme();
  }
}