com.hazelcast.jet.sql.impl.SqlResultImpl Maven / Gradle / Ivy
/*
* Copyright 2024 Hazelcast Inc.
*
* Licensed under the Hazelcast Community License (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://hazelcast.com/hazelcast-community-license
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.jet.sql.impl;
import com.hazelcast.sql.SqlRow;
import com.hazelcast.sql.SqlRowMetadata;
import com.hazelcast.sql.impl.AbstractSqlResult;
import com.hazelcast.sql.impl.CoreQueryUtils;
import com.hazelcast.sql.impl.QueryException;
import com.hazelcast.sql.impl.QueryId;
import com.hazelcast.sql.impl.QueryResultProducer;
import com.hazelcast.sql.impl.ResultIterator;
import com.hazelcast.sql.impl.SqlRowImpl;
import com.hazelcast.sql.impl.row.JetSqlRow;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.NoSuchElementException;
import java.util.concurrent.TimeUnit;
class SqlResultImpl extends AbstractSqlResult {
private final QueryId queryId;
private final QueryResultProducer rootResultConsumer;
private final SqlRowMetadata rowMetadata;
private final boolean isInfiniteRows;
private final int partitionArgumentIndex;
private ResultIterator iterator;
SqlResultImpl(
QueryId queryId,
QueryResultProducer rootResultConsumer,
SqlRowMetadata rowMetadata,
boolean isInfiniteRows
) {
this.queryId = queryId;
this.rootResultConsumer = rootResultConsumer;
this.rowMetadata = rowMetadata;
this.isInfiniteRows = isInfiniteRows;
this.partitionArgumentIndex = -1;
}
SqlResultImpl(
QueryId queryId,
QueryResultProducer rootResultConsumer,
SqlRowMetadata rowMetadata,
boolean isInfiniteRows,
int partitionArgumentIndex
) {
this.queryId = queryId;
this.rootResultConsumer = rootResultConsumer;
this.rowMetadata = rowMetadata;
this.isInfiniteRows = isInfiniteRows;
this.partitionArgumentIndex = partitionArgumentIndex;
}
@Override
public QueryId getQueryId() {
return queryId;
}
@Override
public boolean isInfiniteRows() {
return isInfiniteRows;
}
@Override
public int getPartitionArgumentIndex() {
return partitionArgumentIndex;
}
@Nonnull @Override
public SqlRowMetadata getRowMetadata() {
return rowMetadata;
}
@Nonnull @Override
public ResultIterator iterator() {
if (iterator == null) {
iterator = new RowToSqlRowIterator(rootResultConsumer.iterator());
return iterator;
} else {
throw new IllegalStateException("Iterator can be requested only once.");
}
}
@Override
public long updateCount() {
return -1;
}
@Override
public void close(@Nullable QueryException exception) {
// TODO: [sasha] use Job.cancel() here instead of injecting exception
if (exception == null) {
exception = QueryException.cancelledByUser();
}
rootResultConsumer.onError(exception);
}
private final class RowToSqlRowIterator implements ResultIterator {
private final ResultIterator delegate;
private RowToSqlRowIterator(ResultIterator delegate) {
this.delegate = delegate;
}
@Override
public boolean hasNext() {
try {
return delegate.hasNext();
} catch (Exception e) {
throw CoreQueryUtils.toPublicException(e, queryId.getMemberId());
}
}
@Override
public HasNextResult hasNext(long timeout, TimeUnit timeUnit) {
try {
return delegate.hasNext(timeout, timeUnit);
} catch (Exception e) {
throw CoreQueryUtils.toPublicException(e, queryId.getMemberId());
}
}
@Override
public SqlRow next() {
try {
return new SqlRowImpl(getRowMetadata(), delegate.next());
} catch (NoSuchElementException e) {
throw e;
} catch (Exception e) {
throw CoreQueryUtils.toPublicException(e, queryId.getMemberId());
}
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy