com.databricks.jdbc.client.impl.thrift.generated.TSparkArrowBatch Maven / Gradle / Ivy
/**
* Autogenerated by Thrift Compiler (0.19.0)
*
* DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
*
* @generated
*/
package com.databricks.jdbc.client.impl.thrift.generated;
@SuppressWarnings({"cast", "rawtypes", "serial", "unchecked", "unused"})
@javax.annotation.Generated(
value = "Autogenerated by Thrift Compiler (0.19.0)",
date = "2024-05-06")
public class TSparkArrowBatch
implements org.apache.thrift.TBase,
java.io.Serializable,
Cloneable,
Comparable {
private static final org.apache.thrift.protocol.TStruct STRUCT_DESC =
new org.apache.thrift.protocol.TStruct("TSparkArrowBatch");
private static final org.apache.thrift.protocol.TField BATCH_FIELD_DESC =
new org.apache.thrift.protocol.TField(
"batch", org.apache.thrift.protocol.TType.STRING, (short) 1);
private static final org.apache.thrift.protocol.TField ROW_COUNT_FIELD_DESC =
new org.apache.thrift.protocol.TField(
"rowCount", org.apache.thrift.protocol.TType.I64, (short) 2);
private static final org.apache.thrift.scheme.SchemeFactory STANDARD_SCHEME_FACTORY =
new TSparkArrowBatchStandardSchemeFactory();
private static final org.apache.thrift.scheme.SchemeFactory TUPLE_SCHEME_FACTORY =
new TSparkArrowBatchTupleSchemeFactory();
public @org.apache.thrift.annotation.Nullable java.nio.ByteBuffer batch; // required
public long rowCount; // required
/**
* The set of fields this struct contains, along with convenience methods for finding and
* manipulating them.
*/
public enum _Fields implements org.apache.thrift.TFieldIdEnum {
BATCH((short) 1, "batch"),
ROW_COUNT((short) 2, "rowCount");
private static final java.util.Map byName =
new java.util.HashMap();
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: // BATCH
return BATCH;
case 2: // ROW_COUNT
return ROW_COUNT;
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;
}
@Override
public short getThriftFieldId() {
return _thriftId;
}
@Override
public java.lang.String getFieldName() {
return _fieldName;
}
}
// isset id assignments
private static final int __ROWCOUNT_ISSET_ID = 0;
private byte __isset_bitfield = 0;
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.BATCH,
new org.apache.thrift.meta_data.FieldMetaData(
"batch",
org.apache.thrift.TFieldRequirementType.REQUIRED,
new org.apache.thrift.meta_data.FieldValueMetaData(
org.apache.thrift.protocol.TType.STRING, true)));
tmpMap.put(
_Fields.ROW_COUNT,
new org.apache.thrift.meta_data.FieldMetaData(
"rowCount",
org.apache.thrift.TFieldRequirementType.REQUIRED,
new org.apache.thrift.meta_data.FieldValueMetaData(
org.apache.thrift.protocol.TType.I64)));
metaDataMap = java.util.Collections.unmodifiableMap(tmpMap);
org.apache.thrift.meta_data.FieldMetaData.addStructMetaDataMap(
TSparkArrowBatch.class, metaDataMap);
}
public TSparkArrowBatch() {}
public TSparkArrowBatch(java.nio.ByteBuffer batch, long rowCount) {
this();
this.batch = org.apache.thrift.TBaseHelper.copyBinary(batch);
this.rowCount = rowCount;
setRowCountIsSet(true);
}
/** Performs a deep copy on other. */
public TSparkArrowBatch(TSparkArrowBatch other) {
__isset_bitfield = other.__isset_bitfield;
if (other.isSetBatch()) {
this.batch = org.apache.thrift.TBaseHelper.copyBinary(other.batch);
}
this.rowCount = other.rowCount;
}
@Override
public TSparkArrowBatch deepCopy() {
return new TSparkArrowBatch(this);
}
@Override
public void clear() {
this.batch = null;
setRowCountIsSet(false);
this.rowCount = 0;
}
public byte[] getBatch() {
setBatch(org.apache.thrift.TBaseHelper.rightSize(batch));
return batch == null ? null : batch.array();
}
public java.nio.ByteBuffer bufferForBatch() {
return org.apache.thrift.TBaseHelper.copyBinary(batch);
}
public TSparkArrowBatch setBatch(byte[] batch) {
this.batch =
batch == null ? (java.nio.ByteBuffer) null : java.nio.ByteBuffer.wrap(batch.clone());
return this;
}
public TSparkArrowBatch setBatch(
@org.apache.thrift.annotation.Nullable java.nio.ByteBuffer batch) {
this.batch = org.apache.thrift.TBaseHelper.copyBinary(batch);
return this;
}
public void unsetBatch() {
this.batch = null;
}
/** Returns true if field batch is set (has been assigned a value) and false otherwise */
public boolean isSetBatch() {
return this.batch != null;
}
public void setBatchIsSet(boolean value) {
if (!value) {
this.batch = null;
}
}
public long getRowCount() {
return this.rowCount;
}
public TSparkArrowBatch setRowCount(long rowCount) {
this.rowCount = rowCount;
setRowCountIsSet(true);
return this;
}
public void unsetRowCount() {
__isset_bitfield =
org.apache.thrift.EncodingUtils.clearBit(__isset_bitfield, __ROWCOUNT_ISSET_ID);
}
/** Returns true if field rowCount is set (has been assigned a value) and false otherwise */
public boolean isSetRowCount() {
return org.apache.thrift.EncodingUtils.testBit(__isset_bitfield, __ROWCOUNT_ISSET_ID);
}
public void setRowCountIsSet(boolean value) {
__isset_bitfield =
org.apache.thrift.EncodingUtils.setBit(__isset_bitfield, __ROWCOUNT_ISSET_ID, value);
}
@Override
public void setFieldValue(
_Fields field, @org.apache.thrift.annotation.Nullable java.lang.Object value) {
switch (field) {
case BATCH:
if (value == null) {
unsetBatch();
} else {
if (value instanceof byte[]) {
setBatch((byte[]) value);
} else {
setBatch((java.nio.ByteBuffer) value);
}
}
break;
case ROW_COUNT:
if (value == null) {
unsetRowCount();
} else {
setRowCount((java.lang.Long) value);
}
break;
}
}
@org.apache.thrift.annotation.Nullable
@Override
public java.lang.Object getFieldValue(_Fields field) {
switch (field) {
case BATCH:
return getBatch();
case ROW_COUNT:
return getRowCount();
}
throw new java.lang.IllegalStateException();
}
/**
* Returns true if field corresponding to fieldID is set (has been assigned a value) and false
* otherwise
*/
@Override
public boolean isSet(_Fields field) {
if (field == null) {
throw new java.lang.IllegalArgumentException();
}
switch (field) {
case BATCH:
return isSetBatch();
case ROW_COUNT:
return isSetRowCount();
}
throw new java.lang.IllegalStateException();
}
@Override
public boolean equals(java.lang.Object that) {
if (that instanceof TSparkArrowBatch) return this.equals((TSparkArrowBatch) that);
return false;
}
public boolean equals(TSparkArrowBatch that) {
if (that == null) return false;
if (this == that) return true;
boolean this_present_batch = true && this.isSetBatch();
boolean that_present_batch = true && that.isSetBatch();
if (this_present_batch || that_present_batch) {
if (!(this_present_batch && that_present_batch)) return false;
if (!this.batch.equals(that.batch)) return false;
}
boolean this_present_rowCount = true;
boolean that_present_rowCount = true;
if (this_present_rowCount || that_present_rowCount) {
if (!(this_present_rowCount && that_present_rowCount)) return false;
if (this.rowCount != that.rowCount) return false;
}
return true;
}
@Override
public int hashCode() {
int hashCode = 1;
hashCode = hashCode * 8191 + ((isSetBatch()) ? 131071 : 524287);
if (isSetBatch()) hashCode = hashCode * 8191 + batch.hashCode();
hashCode = hashCode * 8191 + org.apache.thrift.TBaseHelper.hashCode(rowCount);
return hashCode;
}
@Override
public int compareTo(TSparkArrowBatch other) {
if (!getClass().equals(other.getClass())) {
return getClass().getName().compareTo(other.getClass().getName());
}
int lastComparison = 0;
lastComparison = java.lang.Boolean.compare(isSetBatch(), other.isSetBatch());
if (lastComparison != 0) {
return lastComparison;
}
if (isSetBatch()) {
lastComparison = org.apache.thrift.TBaseHelper.compareTo(this.batch, other.batch);
if (lastComparison != 0) {
return lastComparison;
}
}
lastComparison = java.lang.Boolean.compare(isSetRowCount(), other.isSetRowCount());
if (lastComparison != 0) {
return lastComparison;
}
if (isSetRowCount()) {
lastComparison = org.apache.thrift.TBaseHelper.compareTo(this.rowCount, other.rowCount);
if (lastComparison != 0) {
return lastComparison;
}
}
return 0;
}
@org.apache.thrift.annotation.Nullable
@Override
public _Fields fieldForId(int fieldId) {
return _Fields.findByThriftId(fieldId);
}
@Override
public void read(org.apache.thrift.protocol.TProtocol iprot) throws org.apache.thrift.TException {
scheme(iprot).read(iprot, this);
}
@Override
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("TSparkArrowBatch(");
boolean first = true;
sb.append("batch:");
if (this.batch == null) {
sb.append("null");
} else {
org.apache.thrift.TBaseHelper.toString(this.batch, sb);
}
first = false;
if (!first) sb.append(", ");
sb.append("rowCount:");
sb.append(this.rowCount);
first = false;
sb.append(")");
return sb.toString();
}
public void validate() throws org.apache.thrift.TException {
// check for required fields
if (batch == null) {
throw new org.apache.thrift.protocol.TProtocolException(
"Required field 'batch' was not present! Struct: " + toString());
}
// alas, we cannot check 'rowCount' because it's a primitive and you chose the non-beans
// generator.
// 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 TSparkArrowBatchStandardSchemeFactory
implements org.apache.thrift.scheme.SchemeFactory {
@Override
public TSparkArrowBatchStandardScheme getScheme() {
return new TSparkArrowBatchStandardScheme();
}
}
private static class TSparkArrowBatchStandardScheme
extends org.apache.thrift.scheme.StandardScheme {
@Override
public void read(org.apache.thrift.protocol.TProtocol iprot, TSparkArrowBatch 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: // BATCH
if (schemeField.type == org.apache.thrift.protocol.TType.STRING) {
struct.batch = iprot.readBinary();
struct.setBatchIsSet(true);
} else {
org.apache.thrift.protocol.TProtocolUtil.skip(iprot, schemeField.type);
}
break;
case 2: // ROW_COUNT
if (schemeField.type == org.apache.thrift.protocol.TType.I64) {
struct.rowCount = iprot.readI64();
struct.setRowCountIsSet(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
if (!struct.isSetRowCount()) {
throw new org.apache.thrift.protocol.TProtocolException(
"Required field 'rowCount' was not found in serialized data! Struct: " + toString());
}
struct.validate();
}
@Override
public void write(org.apache.thrift.protocol.TProtocol oprot, TSparkArrowBatch struct)
throws org.apache.thrift.TException {
struct.validate();
oprot.writeStructBegin(STRUCT_DESC);
if (struct.batch != null) {
oprot.writeFieldBegin(BATCH_FIELD_DESC);
oprot.writeBinary(struct.batch);
oprot.writeFieldEnd();
}
oprot.writeFieldBegin(ROW_COUNT_FIELD_DESC);
oprot.writeI64(struct.rowCount);
oprot.writeFieldEnd();
oprot.writeFieldStop();
oprot.writeStructEnd();
}
}
private static class TSparkArrowBatchTupleSchemeFactory
implements org.apache.thrift.scheme.SchemeFactory {
@Override
public TSparkArrowBatchTupleScheme getScheme() {
return new TSparkArrowBatchTupleScheme();
}
}
private static class TSparkArrowBatchTupleScheme
extends org.apache.thrift.scheme.TupleScheme {
@Override
public void write(org.apache.thrift.protocol.TProtocol prot, TSparkArrowBatch struct)
throws org.apache.thrift.TException {
org.apache.thrift.protocol.TTupleProtocol oprot =
(org.apache.thrift.protocol.TTupleProtocol) prot;
oprot.writeBinary(struct.batch);
oprot.writeI64(struct.rowCount);
}
@Override
public void read(org.apache.thrift.protocol.TProtocol prot, TSparkArrowBatch struct)
throws org.apache.thrift.TException {
org.apache.thrift.protocol.TTupleProtocol iprot =
(org.apache.thrift.protocol.TTupleProtocol) prot;
struct.batch = iprot.readBinary();
struct.setBatchIsSet(true);
struct.rowCount = iprot.readI64();
struct.setRowCountIsSet(true);
}
}
private static 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();
}
}