com.tigergraph.jdbc.common.ResultSetMetaData Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tigergraph-jdbc-driver Show documentation
This project is a type 4 jdbc driver which implemented the standard jdbc interface. It supports connectivity to tigergraph server and varieties of query types.
There is a newer version: 1.3.20
Show newest version
package com.tigergraph.jdbc.common;

import java.sql.SQLException;
import java.sql.Types;
import java.util.List;

public class ResultSetMetaData implements java.sql.ResultSetMetaData {

  private String table_name;
  private List attributeList;
  private int column_size;

  /** Default constructor. */
  public ResultSetMetaData(String table_name, List attributeList) {
    this.table_name = table_name;
    this.attributeList = attributeList;
    this.column_size = attributeList.size();
  }

  private int getValidColumnIndex(int column) throws SQLException {
    int index = column - 1;
    if ((index < 0) || (index >= this.column_size)) {
      throw new SQLException("column number exceeds the boundary.");
    }

    return index;
  }

  @Override
  public int getColumnCount() throws SQLException {
    return this.column_size;
  }

  @Override
  public String getColumnLabel(int column) throws SQLException {
    int index = getValidColumnIndex(column);

    return this.attributeList.get(index).getName();
  }

  @Override
  public String getColumnName(int column) throws SQLException {
    int index = getValidColumnIndex(column);

    return this.attributeList.get(index).getName();
  }

  @Override
  public String getCatalogName(int column) throws SQLException {
    int index = getValidColumnIndex(column);

    return this.table_name;
  }

  @Override
  public boolean isAutoIncrement(int column) throws SQLException {
    return Boolean.FALSE;
  }

  @Override
  public boolean isSearchable(int column) throws SQLException {
    return Boolean.FALSE;
  }

  @Override
  public boolean isCurrency(int column) throws SQLException {
    return Boolean.FALSE;
  }

  @Override
  public int isNullable(int column) throws SQLException {
    return java.sql.ResultSetMetaData.columnNoNulls;
  }

  // Spark determines to convert Types.BIGINT to LongType or DecimalType(20,0) base on 'isSigned'
  @Override
  public boolean isSigned(int column) throws SQLException {
    if (this.getColumnTypeName(column).toLowerCase().equals("uint")) {
      return Boolean.FALSE;
    }
    return Boolean.TRUE;
  }

  @Override
  public int getPrecision(int column) throws SQLException {
    int index = getValidColumnIndex(column);
    return this.attributeList.get(index).getPrecision();
  }

  @Override
  public int getScale(int column) throws SQLException {
    int index = getValidColumnIndex(column);
    return this.attributeList.get(index).getScale();
  }

  @Override
  public int getColumnType(int column) throws SQLException {
    int index = getValidColumnIndex(column);
    String type = this.attributeList.get(index).getType().toLowerCase();
    if (type.equals("string") || type.equals("string compress")) {
      return Types.VARCHAR;
    }
    if (type.equals("bool")) {
      return Types.BOOLEAN;
    }
    if (type.equals("int")) {
      return Types.BIGINT;
    }
    if (type.equals("uint")) {
      return Types.BIGINT;
    }
    if (type.equals("double")) {
      return Types.DOUBLE;
    }
    if (type.equals("float")) {
      return Types.FLOAT;
    }
    if (type.startsWith("list") || type.startsWith("set")) {
      return Types.ARRAY;
    }
    /*
     * For map/udt and other types, they will be treated as "Types.VARCHAR".
     * Spark will panic if it returns "Types.OTHER".
     */
    return Types.VARCHAR;
  }

  @Override
  public String getColumnTypeName(int column) throws SQLException {
    int index = getValidColumnIndex(column);

    return this.attributeList.get(index).getType();
  }

  /** Methods not implemented yet. */
  @Override
  public boolean isWritable(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public boolean isDefinitelyWritable(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public  T unwrap(Class iface) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public boolean isWrapperFor(Class iface) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public String getColumnClassName(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public String getTableName(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public String getSchemaName(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public boolean isCaseSensitive(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public boolean isReadOnly(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }

  @Override
  public int getColumnDisplaySize(int column) throws SQLException {
    throw new UnsupportedOperationException("Not implemented yet.");
  }
}