Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import com.databricks.jdbc.common.util.WrapperUtil;
import com.databricks.jdbc.log.JdbcLogger;
import com.databricks.jdbc.log.JdbcLoggerFactory;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import java.sql.ParameterMetaData;
import java.sql.SQLException;
import java.util.HashMap;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,10 @@
import com.databricks.jdbc.log.JdbcLogger;
import com.databricks.jdbc.log.JdbcLoggerFactory;
import com.databricks.jdbc.model.client.thrift.generated.*;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnMetadata;
import com.databricks.jdbc.model.core.ResultManifest;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.google.common.collect.ImmutableList;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
package com.databricks.jdbc.api.impl;

import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import javax.annotation.Nullable;
import org.immutables.value.Value;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@

import com.databricks.jdbc.api.impl.converters.ArrowToJavaObjectConverter;
import com.databricks.jdbc.exception.DatabricksSQLException;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import org.apache.arrow.vector.ValueVector;

public class ArrowResultChunkIterator {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,10 +16,10 @@
import com.databricks.jdbc.model.client.thrift.generated.TColumnDesc;
import com.databricks.jdbc.model.client.thrift.generated.TFetchResultsResp;
import com.databricks.jdbc.model.client.thrift.generated.TGetResultSetMetadataResp;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ResultData;
import com.databricks.jdbc.model.core.ResultManifest;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.google.common.annotations.VisibleForTesting;
import java.util.ArrayList;
import java.util.List;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@
import com.databricks.jdbc.exception.DatabricksValidationException;
import com.databricks.jdbc.log.JdbcLogger;
import com.databricks.jdbc.log.JdbcLoggerFactory;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.sql.Date;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,11 @@
import com.databricks.jdbc.log.JdbcLogger;
import com.databricks.jdbc.log.JdbcLoggerFactory;
import com.databricks.jdbc.model.client.thrift.generated.*;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ExternalLink;
import com.databricks.jdbc.model.core.StatementStatus;
import com.databricks.jdbc.model.telemetry.enums.DatabricksDriverErrorCode;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.sdk.service.sql.StatementState;
import java.nio.ByteBuffer;
import java.util.*;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import com.databricks.jdbc.model.client.thrift.generated.TTypeDesc;
import com.databricks.jdbc.model.client.thrift.generated.TTypeEntry;
import com.databricks.jdbc.model.client.thrift.generated.TTypeId;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import java.math.BigDecimal;
import java.sql.Date;
import java.sql.SQLException;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

import com.databricks.jdbc.api.impl.ImmutableSqlParameter;
import com.databricks.jdbc.exception.DatabricksValidationException;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
Expand Down
138 changes: 138 additions & 0 deletions src/main/java/com/databricks/jdbc/model/core/ColumnInfo.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,138 @@
package com.databricks.jdbc.model.core;

import com.databricks.sdk.support.ToStringer;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.Objects;

Comment thread
sreekanth-db marked this conversation as resolved.
/**
* Column info POJO
*
* <p>TODO: Replace this class with the corresponding SDK implementation once it becomes available
*/
public class ColumnInfo {
@JsonProperty("name")
private String name;

@JsonProperty("position")
private Long position;

@JsonProperty("type_interval_type")
private String typeIntervalType;

@JsonProperty("type_name")
private ColumnInfoTypeName typeName;

@JsonProperty("type_precision")
private Long typePrecision;

@JsonProperty("type_scale")
private Long typeScale;

@JsonProperty("type_text")
private String typeText;

public ColumnInfo setName(String name) {
this.name = name;
return this;
}

public String getName() {
return this.name;
}

public ColumnInfo setPosition(Long position) {
this.position = position;
return this;
}

public Long getPosition() {
return this.position;
}

public ColumnInfo setTypeIntervalType(String typeIntervalType) {
this.typeIntervalType = typeIntervalType;
return this;
}

public String getTypeIntervalType() {
return this.typeIntervalType;
}

public ColumnInfo setTypeName(ColumnInfoTypeName typeName) {
this.typeName = typeName;
return this;
}

public ColumnInfoTypeName getTypeName() {
return this.typeName;
}

public ColumnInfo setTypePrecision(Long typePrecision) {
this.typePrecision = typePrecision;
return this;
}

public Long getTypePrecision() {
return this.typePrecision;
}

public ColumnInfo setTypeScale(Long typeScale) {
this.typeScale = typeScale;
return this;
}

public Long getTypeScale() {
return this.typeScale;
}

public ColumnInfo setTypeText(String typeText) {
this.typeText = typeText;
return this;
}

public String getTypeText() {
return this.typeText;
}

public boolean equals(Object o) {
if (this == o) {
return true;
} else if (o != null && this.getClass() == o.getClass()) {
ColumnInfo that = (ColumnInfo) o;
return Objects.equals(this.name, that.name)
&& Objects.equals(this.position, that.position)
&& Objects.equals(this.typeIntervalType, that.typeIntervalType)
&& Objects.equals(this.typeName, that.typeName)
&& Objects.equals(this.typePrecision, that.typePrecision)
&& Objects.equals(this.typeScale, that.typeScale)
&& Objects.equals(this.typeText, that.typeText);
} else {
return false;
}
}

public int hashCode() {
return Objects.hash(
new Object[] {
this.name,
this.position,
this.typeIntervalType,
this.typeName,
this.typePrecision,
this.typeScale,
this.typeText
});
}

public String toString() {
return (new ToStringer(ColumnInfo.class))
.add("name", this.name)
.add("position", this.position)
.add("typeIntervalType", this.typeIntervalType)
.add("typeName", this.typeName)
.add("typePrecision", this.typePrecision)
.add("typeScale", this.typeScale)
.add("typeText", this.typeText)
.toString();
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
package com.databricks.jdbc.model.core;

Comment thread
sreekanth-db marked this conversation as resolved.
/**
* Column info type name POJO
*
* <p>TODO: Replace this class with the corresponding SDK implementation once it becomes available
*/
public enum ColumnInfoTypeName {
ARRAY,
BINARY,
BOOLEAN,
BYTE,
CHAR,
DATE,
DECIMAL,
DOUBLE,
FLOAT,
GEOGRAPHY,
GEOMETRY,
INT,
INTERVAL,
LONG,
MAP,
NULL,
SHORT,
STRING,
STRUCT,
TIMESTAMP,
USER_DEFINED_TYPE;
}
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import com.databricks.jdbc.common.CompressionCodec;
import com.databricks.sdk.service.sql.BaseChunkInfo;
import com.databricks.sdk.service.sql.Format;
import com.databricks.sdk.service.sql.ResultSchema;
import com.databricks.sdk.support.ToStringer;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.Collection;
Expand Down
60 changes: 60 additions & 0 deletions src/main/java/com/databricks/jdbc/model/core/ResultSchema.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
package com.databricks.jdbc.model.core;

import com.databricks.sdk.support.ToStringer;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.Collection;
import java.util.Objects;

/**
* Result schema POJO
*
* <p>TODO: Replace this class with the corresponding SDK implementation once it becomes available
*/
public class ResultSchema {
@JsonProperty("column_count")
private Long columnCount;

@JsonProperty("columns")
private Collection<ColumnInfo> columns;

public ResultSchema setColumnCount(Long columnCount) {
this.columnCount = columnCount;
return this;
}

public Long getColumnCount() {
return this.columnCount;
}

public ResultSchema setColumns(Collection<ColumnInfo> columns) {
this.columns = columns;
return this;
}

public Collection<ColumnInfo> getColumns() {
return this.columns;
}

public boolean equals(Object o) {
if (this == o) {
return true;
} else if (o != null && this.getClass() == o.getClass()) {
ResultSchema that = (ResultSchema) o;
return Objects.equals(this.columnCount, that.columnCount)
&& Objects.equals(this.columns, that.columns);
} else {
return false;
}
}

public int hashCode() {
return Objects.hash(new Object[] {this.columnCount, this.columns});
}

public String toString() {
return (new ToStringer(ResultSchema.class))
.add("columnCount", this.columnCount)
.add("columns", this.columns)
.toString();
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import static java.sql.ParameterMetaData.parameterModeIn;
import static org.junit.jupiter.api.Assertions.*;

import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import java.sql.SQLException;
import java.sql.Types;
import org.junit.jupiter.api.BeforeEach;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,10 @@
import com.databricks.jdbc.common.util.DatabricksTypeUtil;
import com.databricks.jdbc.dbclient.impl.common.StatementId;
import com.databricks.jdbc.model.client.thrift.generated.*;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ResultManifest;
import com.databricks.sdk.service.sql.*;
import com.databricks.jdbc.model.core.ResultSchema;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Types;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@
import com.databricks.jdbc.model.client.thrift.generated.*;
import com.databricks.jdbc.model.core.ResultData;
import com.databricks.jdbc.model.core.ResultManifest;
import com.databricks.jdbc.model.core.ResultSchema;
import com.databricks.sdk.service.sql.Format;
import com.databricks.sdk.service.sql.ResultSchema;
import java.sql.SQLException;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
import com.databricks.jdbc.exception.DatabricksParsingException;
import com.databricks.jdbc.exception.DatabricksSQLException;
import com.databricks.jdbc.model.client.thrift.generated.TSparkArrowResultLink;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import com.databricks.sdk.service.sql.BaseChunkInfo;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,13 +21,13 @@
import com.databricks.jdbc.model.client.thrift.generated.TGetResultSetMetadataResp;
import com.databricks.jdbc.model.client.thrift.generated.TRowSet;
import com.databricks.jdbc.model.client.thrift.generated.TSparkArrowResultLink;
import com.databricks.jdbc.model.core.ColumnInfo;
import com.databricks.jdbc.model.core.ColumnInfoTypeName;
import com.databricks.jdbc.model.core.ExternalLink;
import com.databricks.jdbc.model.core.ResultData;
import com.databricks.jdbc.model.core.ResultManifest;
import com.databricks.jdbc.model.core.ResultSchema;
import com.databricks.sdk.service.sql.BaseChunkInfo;
import com.databricks.sdk.service.sql.ColumnInfo;
import com.databricks.sdk.service.sql.ColumnInfoTypeName;
import com.databricks.sdk.service.sql.ResultSchema;
import com.google.common.collect.ImmutableList;
import java.io.*;
import java.time.Instant;
Expand Down
Loading
Loading