mirror of https://github.com/citusdata/citus.git
72 lines
2.4 KiB
C
72 lines
2.4 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* worker_manager.h
|
|
* Header and type declarations for managing worker nodes and for placing
|
|
* shards on worker nodes in an intelligent manner.
|
|
*
|
|
* Copyright (c) 2012-2016, Citus Data, Inc.
|
|
*
|
|
* $Id$
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#ifndef WORKER_MANAGER_H
|
|
#define WORKER_MANAGER_H
|
|
|
|
#include "nodes/pg_list.h"
|
|
|
|
|
|
/* Worker node name's maximum length */
|
|
#define WORKER_LENGTH 256
|
|
|
|
/* Maximum length of worker port number (represented as string) */
|
|
#define MAX_PORT_LENGTH 10
|
|
|
|
/* default filename for citus.worker_list_file */
|
|
#define WORKER_LIST_FILENAME "pg_worker_list.conf"
|
|
|
|
/* Implementation specific definitions used in finding worker nodes */
|
|
#define WORKER_RACK_TRIES 5
|
|
#define WORKER_DEFAULT_RACK "default"
|
|
|
|
|
|
/*
|
|
* In memory representation of pg_dist_node table elements. The elements are hold in
|
|
* WorkerNodeHash table.
|
|
*/
|
|
typedef struct WorkerNode
|
|
{
|
|
uint32 nodeId; /* node's unique id, key of the hash table */
|
|
uint32 workerPort; /* node's port */
|
|
char workerName[WORKER_LENGTH]; /* node's name */
|
|
uint32 groupId; /* node's groupId; same for the nodes that are in the same group */
|
|
char workerRack[WORKER_LENGTH]; /* node's network location */
|
|
bool hasMetadata; /* node gets metadata changes */
|
|
} WorkerNode;
|
|
|
|
|
|
/* Config variables managed via guc.c */
|
|
extern int MaxWorkerNodesTracked;
|
|
extern char *WorkerListFileName;
|
|
|
|
|
|
/* Function declarations for finding worker nodes to place shards on */
|
|
extern WorkerNode * WorkerGetRandomCandidateNode(List *currentNodeList);
|
|
extern WorkerNode * WorkerGetRoundRobinCandidateNode(List *workerNodeList,
|
|
uint64 shardId,
|
|
uint32 placementIndex);
|
|
extern WorkerNode * WorkerGetLocalFirstCandidateNode(List *currentNodeList);
|
|
extern WorkerNode * WorkerGetNodeWithName(const char *hostname);
|
|
extern uint32 WorkerGetLiveNodeCount(void);
|
|
extern List * WorkerNodeList(void);
|
|
extern WorkerNode * FindWorkerNode(char *nodeName, int32 nodePort);
|
|
extern List * ReadWorkerNodes(void);
|
|
extern void EnsureCoordinator(void);
|
|
|
|
/* Function declarations for worker node utilities */
|
|
extern int CompareWorkerNodes(const void *leftElement, const void *rightElement);
|
|
extern int WorkerNodeCompare(const void *lhsKey, const void *rhsKey, Size keySize);
|
|
|
|
#endif /* WORKER_MANAGER_H */
|