mirror of https://github.com/citusdata/citus.git
97 lines
3.3 KiB
C
97 lines
3.3 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* listutils.h
|
|
*
|
|
* Declarations for public utility functions related to lists.
|
|
*
|
|
* Copyright (c) Citus Data, Inc.
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#ifndef CITUS_LISTUTILS_H
|
|
#define CITUS_LISTUTILS_H
|
|
|
|
#include "postgres.h"
|
|
#include "c.h"
|
|
|
|
#include "nodes/pg_list.h"
|
|
#include "utils/array.h"
|
|
#include "utils/hsearch.h"
|
|
#include "distributed/version_compat.h"
|
|
|
|
|
|
/*
|
|
* ListCellAndListWrapper stores a list and list cell.
|
|
* This struct is used for functionContext. When iterating a list
|
|
* in separate function calls, we need both the list and the current cell.
|
|
* Therefore this wrapper stores both of them.
|
|
*/
|
|
typedef struct ListCellAndListWrapper
|
|
{
|
|
List *list;
|
|
ListCell *listCell;
|
|
} ListCellAndListWrapper;
|
|
|
|
/*
|
|
* foreach_ptr -
|
|
* a convenience macro which loops through a pointer list without needing a
|
|
* ListCell, just a declared pointer variable to store the pointer of the
|
|
* cell in.
|
|
*
|
|
* How it works:
|
|
* - A ListCell is declared with the name {var}CellDoNotUse and used
|
|
* throughout the for loop using ## to concat.
|
|
* - To assign to var it needs to be done in the condition of the for loop,
|
|
* because we cannot use the initializer since a ListCell* variable is
|
|
* declared there.
|
|
* - || true is used to always enter the loop when cell is not null even if
|
|
* var is NULL.
|
|
*/
|
|
#define foreach_ptr(var, l) \
|
|
for (ListCell *(var ## CellDoNotUse) = list_head(l); \
|
|
(var ## CellDoNotUse) != NULL && \
|
|
(((var) = lfirst(var ## CellDoNotUse)) || true); \
|
|
var ## CellDoNotUse = lnext_compat(l, var ## CellDoNotUse))
|
|
|
|
|
|
/*
|
|
* foreach_int -
|
|
* a convenience macro which loops through an int list without needing a
|
|
* ListCell, just a declared int variable to store the int of the cell in.
|
|
* For explanation of how it works see foreach_ptr.
|
|
*/
|
|
#define foreach_int(var, l) \
|
|
for (ListCell *(var ## CellDoNotUse) = list_head(l); \
|
|
(var ## CellDoNotUse) != NULL && \
|
|
(((var) = lfirst_int(var ## CellDoNotUse)) || true); \
|
|
var ## CellDoNotUse = lnext_compat(l, var ## CellDoNotUse))
|
|
|
|
|
|
/*
|
|
* foreach_oid -
|
|
* a convenience macro which loops through an oid list without needing a
|
|
* ListCell, just a declared Oid variable to store the oid of the cell in.
|
|
* For explanation of how it works see foreach_ptr.
|
|
*/
|
|
#define foreach_oid(var, l) \
|
|
for (ListCell *(var ## CellDoNotUse) = list_head(l); \
|
|
(var ## CellDoNotUse) != NULL && \
|
|
(((var) = lfirst_oid(var ## CellDoNotUse)) || true); \
|
|
var ## CellDoNotUse = lnext_compat(l, var ## CellDoNotUse))
|
|
|
|
|
|
/* utility functions declaration shared within this module */
|
|
extern List * SortList(List *pointerList,
|
|
int (*ComparisonFunction)(const void *, const void *));
|
|
extern void ** PointerArrayFromList(List *pointerList);
|
|
extern ArrayType * DatumArrayToArrayType(Datum *datumArray, int datumCount,
|
|
Oid datumTypeId);
|
|
extern HTAB * ListToHashSet(List *pointerList, Size keySize, bool isStringList);
|
|
extern char * StringJoin(List *stringList, char delimiter);
|
|
extern List * ListTake(List *pointerList, int size);
|
|
extern void * safe_list_nth(const List *list, int index);
|
|
extern List * GenerateListFromElement(void *listElement, int listLength);
|
|
|
|
#endif /* CITUS_LISTUTILS_H */
|