Examples with TGetResultSetMetadataResp org.apache.hive.service.rpc.thrift.TGetResultSetMetadataResp used on opensource projects

Example 1 with TGetResultSetMetadataResp

use of org.apache.hive.service.rpc.thrift.TGetResultSetMetadataResp in project hive by apache.

the class HiveQueryResultSet method retrieveSchema.

/**
 * Retrieve schema from the server
 */
private void retrieveSchema() throws SQLException {
    try {
        TGetResultSetMetadataReq metadataReq = new TGetResultSetMetadataReq(stmtHandle);
        // TODO need session handle
        TGetResultSetMetadataResp metadataResp;
        metadataResp = client.GetResultSetMetadata(metadataReq);
        Utils.verifySuccess(metadataResp.getStatus());
        TTableSchema schema = metadataResp.getSchema();
        if (schema == null || !schema.isSetColumns()) {
            // TODO: should probably throw an exception here.
            return;
        }
        setSchema(new TableSchema(schema));
        for (final TColumnDesc column : schema.getColumns()) {
            String columnName = column.getColumnName();
            columnNames.add(columnName);
            normalizedColumnNames.add(columnName.toLowerCase());
            TPrimitiveTypeEntry primitiveTypeEntry = column.getTypeDesc().getTypes().get(0).getPrimitiveEntry();
            String columnTypeName = TYPE_NAMES.get(primitiveTypeEntry.getType());
            columnTypes.add(columnTypeName);
            columnAttributes.add(getColumnAttributes(primitiveTypeEntry));
        }
    } catch (SQLException eS) {
        // rethrow the SQLException as is
        throw eS;
    } catch (Exception ex) {
        throw new SQLException("Could not create ResultSet: " + ex.getMessage(), ex);
    }
}

Example 2 with TGetResultSetMetadataResp

use of org.apache.hive.service.rpc.thrift.TGetResultSetMetadataResp in project hive by apache.

the class ThriftCLIService method GetResultSetMetadata.

@Override
public TGetResultSetMetadataResp GetResultSetMetadata(TGetResultSetMetadataReq req) throws TException {
    TGetResultSetMetadataResp resp = new TGetResultSetMetadataResp();
    try {
        TableSchema schema = cliService.getResultSetMetadata(new OperationHandle(req.getOperationHandle()));
        resp.setSchema(schema.toTTableSchema());
        resp.setStatus(OK_STATUS);
    } catch (Exception e) {
        LOG.error("Failed to get result set metadata [request: {}]", req, e);
        resp.setStatus(HiveSQLException.toTStatus(e));
    }
    return resp;
}

Example 3 with TGetResultSetMetadataResp

use of org.apache.hive.service.rpc.thrift.TGetResultSetMetadataResp in project hive by apache.

the class ThriftCLIServiceClient method getResultSetMetadata.

/* (non-Javadoc)
   * @see org.apache.hive.service.cli.ICLIService#getResultSetMetadata(org.apache.hive.service.cli.OperationHandle)
   */
@Override
public TableSchema getResultSetMetadata(OperationHandle opHandle) throws HiveSQLException {
    try {
        TGetResultSetMetadataReq req = new TGetResultSetMetadataReq(opHandle.toTOperationHandle());
        TGetResultSetMetadataResp resp = cliService.GetResultSetMetadata(req);
        checkStatus(resp.getStatus());
        return new TableSchema(resp.getSchema());
    } catch (HiveSQLException e) {
        throw e;
    } catch (Exception e) {
        throw new HiveSQLException(e);
    }
}