Function UpdateStatistics

contrib/pax_storage/src/cpp/catalog/pax_aux_table.cc:291–331

}

void UpdateStatistics(Oid aux_relid, int block_id,
                      pax::stats::MicroPartitionStatisticsInfo *mp_stats) {
  // NameData pt_visimap_name;
  Datum values[NATTS_PG_PAX_BLOCK_TABLES];
  bool nulls[NATTS_PG_PAX_BLOCK_TABLES];
  bool repls[NATTS_PG_PAX_BLOCK_TABLES];
  ScanAuxContext context;
  HeapTuple newtuple;
  uint32 stats_len;
  void *stats_out;

  Assert(mp_stats);
  stats_len = VARHDRSZ + mp_stats->ByteSizeLong();
  stats_out = palloc(stats_len);
  SET_VARSIZE(stats_out, stats_len);
  mp_stats->SerializeToArray(VARDATA(stats_out), stats_len - VARHDRSZ);

  context.BeginSearchMicroPartition(aux_relid, InvalidOid, NULL,
                                    RowExclusiveLock, block_id);
  auto aux_rel = context.GetRelation();
  auto oldtuple = context.SearchMicroPartitionEntry();
  if (!HeapTupleIsValid(oldtuple))
    elog(ERROR,
         "The micro partition does not exist when we updating statistics.");

  memset(repls, false, sizeof(repls));
  repls[ANUM_PG_PAX_BLOCK_TABLES_PTSTATISITICS - 1] = true;
  nulls[ANUM_PG_PAX_BLOCK_TABLES_PTSTATISITICS - 1] = false;
  values[ANUM_PG_PAX_BLOCK_TABLES_PTSTATISITICS - 1] =
      PointerGetDatum(stats_out);
  newtuple = heap_modify_tuple(oldtuple, RelationGetDescr(aux_rel), values,
                               nulls, repls);

  CatalogTupleUpdate(aux_rel, &oldtuple->t_self, newtuple);
  heap_freetuple(newtuple);
  context.EndSearchMicroPartition(NoLock);

  pfree(stats_out);

  CommandCounterIncrement();
}

// We won't update the visimap in here
// but will update visimap info in the `UpdateVisimap`