external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

wip

pull/5273/head
Nils Dijk 2021-07-15 17:28:30 +02:00
parent 531ad83b8c
commit 5fe4c06284
No known key found for this signature in database
GPG Key ID: CA1177EF9434F241
1 changed files with 55 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/backend/columnar/columnar_tableam.c
View File

@ -83,6 +83,8 @@ typedef struct ColumnarScanDescData
MemoryContext scanContext;
Bitmapset *attr_needed;
List *scanQual;
int tupleindex;
} ColumnarScanDescData;
typedef struct ColumnarScanDescData *ColumnarScanDesc;
@ -2066,6 +2068,57 @@ IsColumnarTableAmTable(Oid relationId)
}
static bool
columnar_bitmap_next_block(TableScanDesc scan, TBMIterateResult *tbmres)
{
ColumnarScanDesc cscan = (ColumnarScanDesc) scan;
cscan->tupleindex = 0;
// ereport(ERROR, (errmsg("bitmap scan not implemented")));
return tbmres->blockno == 0;
}
static bool
columnar_bitmap_next_tuple(TableScanDesc scan, TBMIterateResult *tbmres,
TupleTableSlot *slot)
{
ColumnarScanDesc cscan = (ColumnarScanDesc) scan;
/**/
if (cscan->tupleindex >= tbmres->ntuples || cscan->tupleindex > 0 || tbmres->blockno != 0)
{
return false;
}
ItemPointerData tid = {};
ItemPointerSet(&tid, tbmres->blockno, tbmres->offsets[cscan->tupleindex]);
cscan->tupleindex++;
ExecClearTuple(slot);
/* we need all columns */
Relation rel = cscan->cs_base.rs_rd;
int natts = rel->rd_att->natts;
Bitmapset *attr_needed = bms_add_range(NULL, 0, natts - 1);
TupleDesc relationTupleDesc = RelationGetDescr(rel);
List *relationColumnList = NeededColumnsList(relationTupleDesc, attr_needed);
uint64 rowNumber = tid_to_row_number(tid);
if (!ColumnarReadRowByRowNumber(rel, rowNumber, relationColumnList,
slot->tts_values, slot->tts_isnull,
cscan->cs_base.rs_snapshot))
{
return false;
}
slot->tts_tableOid = RelationGetRelid(rel);
slot->tts_tid = tid;
ExecStoreVirtualTuple(slot);
return true;
}
static const TableAmRoutine columnar_am_methods = {
.type = T_TableAmRoutine,
@ -2119,8 +2172,8 @@ static const TableAmRoutine columnar_am_methods = {
.relation_estimate_size = columnar_estimate_rel_size,
.scan_bitmap_next_block = NULL,
.scan_bitmap_next_tuple = NULL,
.scan_bitmap_next_block = columnar_bitmap_next_block,
.scan_bitmap_next_tuple = columnar_bitmap_next_tuple,
.scan_sample_next_block = columnar_scan_sample_next_block,
.scan_sample_next_tuple = columnar_scan_sample_next_tuple
};