mirror of https://github.com/citusdata/citus.git
65 lines
2.0 KiB
C
65 lines
2.0 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* test/src/metadata_sync.c
|
|
*
|
|
* This file contains functions to exercise the metadata snapshoy
|
|
* generation functionality within Citus.
|
|
*
|
|
* Copyright (c) 2014-2016, Citus Data, Inc.
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#include "postgres.h"
|
|
#include "c.h"
|
|
#include "fmgr.h"
|
|
|
|
#include "catalog/pg_type.h"
|
|
#include "distributed/metadata_sync.h"
|
|
#include "distributed/test_helper_functions.h" /* IWYU pragma: keep */
|
|
#include "utils/array.h"
|
|
#include "utils/builtins.h"
|
|
|
|
|
|
/* declarations for dynamic loading */
|
|
PG_FUNCTION_INFO_V1(master_metadata_snapshot);
|
|
|
|
|
|
/*
|
|
* master_metadata_snapshot prints all the queries that are required
|
|
* to generate a metadata snapshot.
|
|
*/
|
|
Datum
|
|
master_metadata_snapshot(PG_FUNCTION_ARGS)
|
|
{
|
|
List *dropSnapshotCommands = MetadataDropCommands();
|
|
List *createSnapshotCommands = MetadataCreateCommands();
|
|
List *snapshotCommandList = NIL;
|
|
ListCell *snapshotCommandCell = NULL;
|
|
int snapshotCommandCount = 0;
|
|
Datum *snapshotCommandDatumArray = NULL;
|
|
ArrayType *snapshotCommandArrayType = NULL;
|
|
int snapshotCommandIndex = 0;
|
|
Oid ddlCommandTypeId = TEXTOID;
|
|
|
|
snapshotCommandList = list_concat(snapshotCommandList, dropSnapshotCommands);
|
|
snapshotCommandList = list_concat(snapshotCommandList, createSnapshotCommands);
|
|
|
|
snapshotCommandCount = list_length(snapshotCommandList);
|
|
snapshotCommandDatumArray = palloc0(snapshotCommandCount * sizeof(Datum));
|
|
|
|
foreach(snapshotCommandCell, snapshotCommandList)
|
|
{
|
|
char *metadataSnapshotCommand = (char *) lfirst(snapshotCommandCell);
|
|
Datum metadataSnapshotCommandDatum = CStringGetTextDatum(metadataSnapshotCommand);
|
|
|
|
snapshotCommandDatumArray[snapshotCommandIndex] = metadataSnapshotCommandDatum;
|
|
snapshotCommandIndex++;
|
|
}
|
|
|
|
snapshotCommandArrayType = DatumArrayToArrayType(snapshotCommandDatumArray,
|
|
snapshotCommandCount,
|
|
ddlCommandTypeId);
|
|
|
|
PG_RETURN_ARRAYTYPE_P(snapshotCommandArrayType);
|
|
}
|