org.hsqldb.navigator.RowSetNavigatorDataTable Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqltool Show documentation
Show all versions of sqltool Show documentation
HSQLDB - Lightweight 100% Java SQL Database Engine
/* Copyright (c) 2001-2021, The HSQL Development Group
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* Neither the name of the HSQL Development Group nor the names of its
* contributors may be used to endorse or promote products derived from this
* software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG,
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.hsqldb.navigator;
import org.hsqldb.HsqlException;
import org.hsqldb.QueryExpression;
import org.hsqldb.QuerySpecification;
import org.hsqldb.Row;
import org.hsqldb.Session;
import org.hsqldb.SortAndSlice;
import org.hsqldb.Table;
import org.hsqldb.TableBase;
import org.hsqldb.lib.ArrayUtil;
import org.hsqldb.persist.PersistentStore;
import org.hsqldb.result.ResultMetaData;
import org.hsqldb.rowio.RowInputInterface;
import org.hsqldb.rowio.RowOutputInterface;
/**
* Implementation of RowSetNavigator using a table as the data store.
*
* @author Fred Toussi (fredt@users dot sourceforge.net)
* @version 2.6.1
* @since 1.9.0
*/
public class RowSetNavigatorDataTable extends RowSetNavigatorData {
public TableBase table;
public PersistentStore store;
RowIterator iterator;
Row currentRow;
Object[] tempRowData;
public RowSetNavigatorDataTable(Session session,
QuerySpecification select) {
super(session, select.sortAndSlice);
rangePosition = select.resultRangePosition;
visibleColumnCount = select.indexLimitVisible;
table = select.resultTable.duplicate();
store = session.sessionData.getNewResultRowStore(table,
!select.isAggregated);
table.store = store;
isAggregate = select.isAggregated;
isSimpleAggregate = select.isAggregated && !select.isGrouped;
reindexTable = select.isGrouped;
mainIndex = select.mainIndex;
fullIndex = select.fullIndex;
orderIndex = select.orderIndex;
groupIndex = select.groupIndex;
idIndex = select.idIndex;
tempRowData = new Object[1];
}
public RowSetNavigatorDataTable(Session session,
QuerySpecification select,
RowSetNavigatorData navigator) {
this(session, select);
navigator.reset();
while (navigator.next()) {
add(navigator.getCurrent());
}
}
public RowSetNavigatorDataTable(Session session,
QueryExpression queryExpression) {
super(session, queryExpression.sortAndSlice);
table = queryExpression.resultTable.duplicate();
visibleColumnCount = table.getColumnCount();
store = session.sessionData.getNewResultRowStore(table, true);
table.store = store;
mainIndex = queryExpression.mainIndex;
fullIndex = queryExpression.fullIndex;
}
public RowSetNavigatorDataTable(Session session, Table table) {
super(session, SortAndSlice.noSort);
this.table = table;
visibleColumnCount = table.getColumnCount();
mainIndex = table.getPrimaryIndex();
fullIndex = table.getFullIndex(session);
store = table.getRowStore(session);
size = (int) store.elementCount();
reset();
}
public void sortFull() {
if (reindexTable) {
store.indexRows((Session) session);
}
mainIndex = fullIndex;
reset();
}
public void sortOrder() {
if (orderIndex != null) {
if (reindexTable) {
store.indexRows((Session) session);
}
mainIndex = orderIndex;
if (iterator != null) {
iterator.release();
}
reset();
}
}
public void sortOrderUnion(SortAndSlice sortAndSlice) {
if (sortAndSlice.index != null) {
mainIndex = sortAndSlice.index;
reset();
}
}
public void add(Object[] data) {
try {
Row row = (Row) store.getNewCachedObject((Session) session, data,
false);
store.indexRow((Session) session, row);
size++;
} catch (HsqlException e) {}
}
void addAdjusted(Object[] data, int[] columnMap) {
try {
if (columnMap == null) {
data = (Object[]) ArrayUtil.resizeArrayIfDifferent(data,
visibleColumnCount);
} else {
Object[] newData = new Object[visibleColumnCount];
ArrayUtil.projectRow(data, columnMap, newData);
data = newData;
}
add(data);
} catch (HsqlException e) {}
}
public void update(Object[] oldData, Object[] newData) {
if (isSimpleAggregate) {
return;
}
RowIterator it = groupIndex.findFirstRow((Session) session, store,
oldData);
if (it.next()) {
it.removeCurrent();
it.release();
size--;
add(newData);
}
}
public boolean absolute(int position) {
return super.absolute(position);
}
public Object[] getCurrent() {
return currentRow.getData();
}
public Row getCurrentRow() {
return currentRow;
}
public boolean next() {
boolean result = super.next();
if (!result) {
return false;
}
iterator.next();
currentRow = iterator.getCurrentRow();
return result;
}
public void removeCurrent() {
if (currentRow != null) {
iterator.removeCurrent();
currentRow = null;
currentPos--;
size--;
}
}
public void reset() {
super.reset();
if (iterator != null) {
iterator.release();
}
iterator = mainIndex.firstRow((Session) session, store, null, 0, null);
}
public void release() {
if (isClosed) {
return;
}
iterator.release();
store.release();
isClosed = true;
}
public void clear() {
store.removeAll();
size = 0;
reset();
}
public boolean isMemory() {
return store.isMemory();
}
public void read(RowInputInterface in, ResultMetaData meta) {}
public void write(RowOutputInterface out, ResultMetaData meta) {
reset();
out.writeLong(id);
out.writeInt(size);
out.writeInt(0); // offset
out.writeInt(size);
while (next()) {
Object[] data = getCurrent();
out.writeData(meta.getExtendedColumnCount(), meta.columnTypes,
data, null, null);
}
reset();
}
public Object[] getData(Long rowId) {
tempRowData[0] = rowId;
RowIterator it = idIndex.findFirstRow((Session) session, store,
tempRowData,
idIndex.getDefaultColumnMap());
it.next();
return it.getCurrent();
}
public void copy(RowSetNavigatorData other, int[] rightColumnIndexes) {
while (other.next()) {
Object[] currentData = other.getCurrent();
addAdjusted(currentData, rightColumnIndexes);
}
}
public void union(RowSetNavigatorData other) {
Object[] currentData;
int colCount = table.getColumnTypes().length;
removeDuplicates();
other.reset();
while (other.next()) {
currentData = other.getCurrent();
RowIterator it = findFirstRow(currentData);
if (!it.next()) {
currentData =
(Object[]) ArrayUtil.resizeArrayIfDifferent(currentData,
colCount);
add(currentData);
}
it.release();
}
reset();
}
public void intersect(RowSetNavigatorData other) {
removeDuplicates();
other.sortFull();
while (next()) {
Object[] currentData = getCurrent();
boolean hasRow = other.containsRow(currentData);
if (!hasRow) {
removeCurrent();
}
}
reset();
}
public void intersectAll(RowSetNavigatorData other) {
Object[] compareData = null;
RowIterator it;
Row otherRow = null;
Object[] otherData = null;
sortFull();
other.sortFull();
it = RowIterator.emptyRowIterator;
while (next()) {
Object[] currentData = getCurrent();
boolean newGroup =
compareData == null
|| fullIndex.compareRowNonUnique(
(Session) session, currentData, compareData,
fullIndex.getColumnCount()) != 0;
if (newGroup) {
compareData = currentData;
it = other.findFirstRow(currentData);
}
if (it.next()) {
otherData = it.getCurrent();
if (fullIndex.compareRowNonUnique(
(Session) session, currentData, otherData,
fullIndex.getColumnCount()) == 0) {
continue;
}
}
removeCurrent();
}
reset();
}
public void except(RowSetNavigatorData other) {
removeDuplicates();
other.sortFull();
while (next()) {
Object[] currentData = getCurrent();
boolean hasRow = other.containsRow(currentData);
if (hasRow) {
removeCurrent();
}
}
reset();
}
public void exceptNoDedup(RowSetNavigatorData other) {
other.sortFull();
reset();
while (next()) {
Object[] currentData = getCurrent();
boolean hasRow = other.containsRow(currentData);
if (hasRow) {
removeCurrent();
}
}
reset();
}
public void exceptAll(RowSetNavigatorData other) {
Object[] compareData = null;
RowIterator it;
Object[] otherData = null;
sortFull();
other.sortFull();
it = RowIterator.emptyRowIterator;
while (next()) {
Object[] currentData = getCurrent();
boolean newGroup =
compareData == null
|| fullIndex.compareRowNonUnique(
(Session) session, currentData, compareData,
fullIndex.getColumnCount()) != 0;
if (newGroup) {
compareData = currentData;
it = other.findFirstRow(currentData);
}
if (it.next()) {
otherData = it.getCurrent();
if (fullIndex.compareRowNonUnique(
(Session) session, currentData, otherData,
fullIndex.getColumnCount()) == 0) {
removeCurrent();
}
}
}
reset();
}
public boolean hasUniqueNotNullRows() {
sortFull();
Object[] lastRowData = null;
while (next()) {
Object[] currentData = getCurrent();
if (hasNull(currentData)) {
continue;
}
if (lastRowData != null
&& fullIndex.compareRow(
(Session) session, lastRowData, currentData) == 0) {
return false;
} else {
lastRowData = currentData;
}
}
return true;
}
public void removeDuplicates() {
sortFull();
Object[] lastRowData = null;
while (next()) {
Object[] currentData = getCurrent();
if (lastRowData != null
&& fullIndex.compareRow(
(Session) session, lastRowData, currentData) == 0) {
removeCurrent();
} else {
lastRowData = currentData;
}
}
reset();
}
public void trim(int limitstart, int limitcount) {
if (size == 0) {
return;
}
if (limitstart >= size) {
clear();
return;
}
if (limitstart != 0) {
reset();
for (int i = 0; i < limitstart; i++) {
next();
removeCurrent();
}
}
if (limitcount == 0 || limitcount >= size) {
return;
}
reset();
for (int i = 0; i < limitcount; i++) {
next();
}
while (next()) {
removeCurrent();
}
reset();
}
boolean hasNull(Object[] data) {
for (int i = 0; i < visibleColumnCount; i++) {
if (data[i] == null) {
return true;
}
}
return false;
}
/**
* Special case for isSimpleAggregate cannot use index lookup.
*/
public Object[] getGroupData(Object[] data) {
if (isSimpleAggregate) {
if (simpleAggregateData == null) {
simpleAggregateData = data;
return null;
}
return simpleAggregateData;
}
RowIterator it = groupIndex.findFirstRow((Session) session, store,
data);
if (it.next()) {
Row row = it.getCurrentRow();
if (isAggregate) {
row.setChanged(true);
}
return row.getData();
}
return null;
}
boolean containsRow(Object[] data) {
RowIterator it = mainIndex.findFirstRow((Session) session, store,
data);
boolean result = it.next();
it.release();
return result;
}
RowIterator findFirstRow(Object[] data) {
return mainIndex.findFirstRow((Session) session, store, data);
}
}