Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.session;
import org.apache.iotdb.isession.Config;
import org.apache.iotdb.isession.IDataIterator;
import org.apache.iotdb.isession.ISessionDataSet;
import org.apache.iotdb.rpc.IoTDBConnectionException;
import org.apache.iotdb.rpc.IoTDBRpcDataSet;
import org.apache.iotdb.rpc.StatementExecutionException;
import org.apache.iotdb.service.rpc.thrift.TSIService;
import org.apache.iotdb.service.rpc.thrift.TSQueryDataSet;
import org.apache.iotdb.tsfile.exception.write.UnSupportedDataTypeException;
import org.apache.iotdb.tsfile.file.metadata.enums.TSDataType;
import org.apache.iotdb.tsfile.read.common.Field;
import org.apache.iotdb.tsfile.read.common.RowRecord;
import org.apache.iotdb.tsfile.utils.Binary;
import org.apache.iotdb.tsfile.utils.BytesUtils;
import org.apache.thrift.TException;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import static org.apache.iotdb.rpc.IoTDBRpcDataSet.START_INDEX;
public class SessionDataSet implements ISessionDataSet {
private final IoTDBRpcDataSet ioTDBRpcDataSet;
public SessionDataSet(
String sql,
List columnNameList,
List columnTypeList,
Map columnNameIndex,
long queryId,
long statementId,
TSIService.Iface client,
long sessionId,
TSQueryDataSet queryDataSet,
boolean ignoreTimeStamp) {
this.ioTDBRpcDataSet =
new IoTDBRpcDataSet(
sql,
columnNameList,
columnTypeList,
columnNameIndex,
ignoreTimeStamp,
queryId,
statementId,
client,
sessionId,
queryDataSet,
Config.DEFAULT_FETCH_SIZE,
0);
}
public SessionDataSet(
String sql,
List columnNameList,
List columnTypeList,
Map columnNameIndex,
long queryId,
long statementId,
TSIService.Iface client,
long sessionId,
TSQueryDataSet queryDataSet,
boolean ignoreTimeStamp,
long timeout) {
this.ioTDBRpcDataSet =
new IoTDBRpcDataSet(
sql,
columnNameList,
columnTypeList,
columnNameIndex,
ignoreTimeStamp,
queryId,
statementId,
client,
sessionId,
queryDataSet,
Config.DEFAULT_FETCH_SIZE,
timeout);
}
@Override
public int getFetchSize() {
return ioTDBRpcDataSet.fetchSize;
}
@Override
public void setFetchSize(int fetchSize) {
ioTDBRpcDataSet.fetchSize = fetchSize;
}
@Override
public List getColumnNames() {
return new ArrayList<>(ioTDBRpcDataSet.columnNameList);
}
@Override
public List getColumnTypes() {
return new ArrayList<>(ioTDBRpcDataSet.columnTypeList);
}
@Override
public boolean hasNext() throws StatementExecutionException, IoTDBConnectionException {
return ioTDBRpcDataSet.next();
}
private RowRecord constructRowRecordFromValueArray() throws StatementExecutionException {
List outFields = new ArrayList<>();
for (int i = 0; i < ioTDBRpcDataSet.columnSize; i++) {
Field field;
int index = i + 1;
int datasetColumnIndex = i + START_INDEX;
if (ioTDBRpcDataSet.ignoreTimeStamp) {
index--;
datasetColumnIndex--;
}
int loc =
ioTDBRpcDataSet.columnOrdinalMap.get(ioTDBRpcDataSet.columnNameList.get(index))
- START_INDEX;
if (!ioTDBRpcDataSet.isNull(datasetColumnIndex)) {
byte[] valueBytes = ioTDBRpcDataSet.values[loc];
TSDataType dataType = ioTDBRpcDataSet.columnTypeDeduplicatedList.get(loc);
field = new Field(dataType);
switch (dataType) {
case BOOLEAN:
boolean booleanValue = BytesUtils.bytesToBool(valueBytes);
field.setBoolV(booleanValue);
break;
case INT32:
int intValue = BytesUtils.bytesToInt(valueBytes);
field.setIntV(intValue);
break;
case INT64:
long longValue = BytesUtils.bytesToLong(valueBytes);
field.setLongV(longValue);
break;
case FLOAT:
float floatValue = BytesUtils.bytesToFloat(valueBytes);
field.setFloatV(floatValue);
break;
case DOUBLE:
double doubleValue = BytesUtils.bytesToDouble(valueBytes);
field.setDoubleV(doubleValue);
break;
case TEXT:
field.setBinaryV(new Binary(valueBytes));
break;
default:
throw new UnSupportedDataTypeException(
String.format(
"Data type %s is not supported.",
ioTDBRpcDataSet.columnTypeDeduplicatedList.get(i)));
}
} else {
field = new Field(null);
}
outFields.add(field);
}
return new RowRecord(BytesUtils.bytesToLong(ioTDBRpcDataSet.time), outFields);
}
@Override
public RowRecord next() throws StatementExecutionException, IoTDBConnectionException {
if (!ioTDBRpcDataSet.hasCachedRecord && !hasNext()) {
return null;
}
ioTDBRpcDataSet.hasCachedRecord = false;
return constructRowRecordFromValueArray();
}
@Override
public void closeOperationHandle() throws StatementExecutionException, IoTDBConnectionException {
try {
ioTDBRpcDataSet.close();
} catch (TException e) {
throw new IoTDBConnectionException(e.getMessage());
}
}
@Override
public DataIterator iterator() {
return new DataIterator();
}
@Override
public void close() throws IoTDBConnectionException, StatementExecutionException {
closeOperationHandle();
}
public class DataIterator implements IDataIterator {
@Override
public boolean next() throws StatementExecutionException, IoTDBConnectionException {
return ioTDBRpcDataSet.next();
}
@Override
public boolean isNull(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.isNull(columnIndex);
}
@Override
public boolean isNull(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.isNull(columnName);
}
@Override
public boolean getBoolean(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getBoolean(columnIndex);
}
@Override
public boolean getBoolean(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getBoolean(columnName);
}
@Override
public double getDouble(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getDouble(columnIndex);
}
@Override
public double getDouble(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getDouble(columnName);
}
@Override
public float getFloat(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getFloat(columnIndex);
}
@Override
public float getFloat(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getFloat(columnName);
}
@Override
public int getInt(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getInt(columnIndex);
}
@Override
public int getInt(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getInt(columnName);
}
@Override
public long getLong(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getLong(columnIndex);
}
@Override
public long getLong(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getLong(columnName);
}
@Override
public Object getObject(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getObject(columnIndex);
}
@Override
public Object getObject(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getObject(columnName);
}
@Override
public String getString(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getString(columnIndex);
}
@Override
public String getString(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getString(columnName);
}
@Override
public Timestamp getTimestamp(int columnIndex) throws StatementExecutionException {
return ioTDBRpcDataSet.getTimestamp(columnIndex);
}
@Override
public Timestamp getTimestamp(String columnName) throws StatementExecutionException {
return ioTDBRpcDataSet.getTimestamp(columnName);
}
@Override
public int findColumn(String columnName) {
return ioTDBRpcDataSet.findColumn(columnName);
}
}
}
