/**
* Copyright (C) 2009-2013 FoundationDB, LLC
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.foundationdb.server.test.it.qp;
import com.foundationdb.ais.model.GroupIndex;
import com.foundationdb.ais.model.Table;
import com.foundationdb.ais.model.TableName;
import com.foundationdb.qp.expression.IndexKeyRange;
import com.foundationdb.qp.operator.*;
import com.foundationdb.qp.row.Row;
import com.foundationdb.qp.rowtype.IndexRowType;
import com.foundationdb.qp.rowtype.RowType;
import com.foundationdb.qp.rowtype.Schema;
import com.foundationdb.qp.rowtype.TableRowType;
import com.foundationdb.server.test.it.ITBase;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public final class GroupIndexScanIT extends ITBase {
@Test
public void scanAtLeastO () {
Operator plan = API.indexScan_Default(giRowType, false, unboundedRange(giRowType), tableRowType(o));
compareResults(plan,
array("01-01-2001", null),
array("02-02-2002", "1111"),
array("03-03-2003", null),
array("03-03-2003", "3333")
);
}
@Test
public void scanAtLeastI () {
Operator plan = API.indexScan_Default(giRowType, false, unboundedRange(giRowType), tableRowType(i));
compareResults(plan,
array("02-02-2002", "1111"),
array("03-03-2003", null),
array("03-03-2003", "3333")
);
}
@Test
public void defaultDepth() {
Operator explicit = API.indexScan_Default(giRowType, false, unboundedRange(giRowType), tableRowType(i));
Operator defaulted = API.indexScan_Default(giRowType, false, unboundedRange(giRowType));
List<List<?>> explicitList = planToList(explicit);
List<List<?>> defaultedList = planToList(defaulted);
assertEqualLists("results", explicitList, defaultedList);
}
@Test(expected = IllegalArgumentException.class)
public void scanAtLeastC () {
API.indexScan_Default(giRowType, false, null, tableRowType(c));
}
@Test(expected = IllegalArgumentException.class)
public void scanAtLeastH () {
API.indexScan_Default(giRowType, false, null, tableRowType(h));
}
@Before
public void setUp() {
c = createTable(SCHEMA, "c", "cid int not null primary key, name varchar(32)");
o = createTable(SCHEMA, "o", "oid int not null primary key, c_id int", "when varchar(32)", akibanFK("c_id", "c", "cid"));
i = createTable(SCHEMA, "i", "iid int not null primary key, o_id int", "sku varchar(6)", akibanFK("o_id", "o", "oid"));
h = createTable(SCHEMA, "h", "hid int not null primary key, i_id int", akibanFK("i_id", "i", "iid"));
TableName groupName = getTable(c).getGroup().getName();
GroupIndex gi = createLeftGroupIndex(groupName, GI_NAME, "o.when", "i.sku");
schema = new Schema(ddl().getAIS(session()));
adapter = newStoreAdapter(schema);
queryContext = queryContext(adapter);
queryBindings = queryContext.createBindings();
giRowType = schema.indexRowType(gi);
writeRows(
row(c, 1L, "One"),
row(o, 10L, 1L, "01-01-2001"), // no children
row(o, 11L, 1L, "02-02-2002"), // one child
row(i, 100L, 11L, "1111"),
row(o, 12L, 2L, "03-03-2003"), // orphaned, two children
row(i, 101L, 12L, null),
row(i, 102L, 12L, "3333")
);
txnService().beginTransaction(session());
}
@After
public void tearDown() {
txnService().rollbackTransaction(session());
c = o = i = h = null;
schema = null;
giRowType = null;
adapter = null;
}
private TableRowType tableRowType(int tableId) {
Table table = ddl().getAIS(session()).getTable(tableId);
TableRowType rowType = schema.tableRowType(table);
if (rowType == null) {
throw new NullPointerException(table.toString());
}
return rowType;
}
private void compareResults(Operator plan, Object[]... expectedResults) {
assertEqualLists("rows scanned", nestedList(expectedResults), planToList(plan));
}
private List<List<?>> planToList(Operator plan) {
List<List<?>> actualResults = new ArrayList<>();
Cursor cursor = API.cursor(plan, queryContext, queryBindings);
cursor.openTopLevel();
try {
for (Row row = cursor.next(); row != null; row = cursor.next()) {
RowType rowType = row.rowType();
int fields =
rowType instanceof IndexRowType
? ((IndexRowType)rowType).index().getKeyColumns().size()
: rowType.nFields();
Object[] rowArray = new Object[fields];
for (int i=0; i < rowArray.length; ++i) {
Object fromRow;
fromRow = getObject(row.value(i));
rowArray[i] = fromRow;
}
actualResults.add(Arrays.asList(rowArray));
}
} finally {
cursor.closeTopLevel();
}
return actualResults;
}
private List<List<?>> nestedList(Object[][] input) {
List<List<?>> listList = new ArrayList<>();
for (Object[] array : input) {
listList.add(Arrays.asList(array));
}
return listList;
}
private IndexKeyRange unboundedRange(IndexRowType indexRowType)
{
return IndexKeyRange.unbounded(indexRowType);
}
private Integer c, o, i, h;
private Schema schema;
private StoreAdapter adapter;
private QueryContext queryContext;
private QueryBindings queryBindings;
private IndexRowType giRowType;
private final static String SCHEMA = "schema";
private final static String GI_NAME = "when_sku";
}