patroni.postgresql.mpp.citus module

class _patroni.postgresql.mpp.citus.Citus(_config: Dict[ str, str | int]) View on GitHub

Bases: AbstractMPP +

abc_impl = <_abc._abc_data object>_

+

property _coordinator_group_id: int_

The group id of the Citus coordinator PostgreSQL cluster. +

property _group: int_

The group of this Citus node. +

group_re_: Any = re.compile('^(0|[1-9][0-9]*)$')_

+

static _validate_config(_config: Any | Dict[ str, str | int]) → bool View on GitHub

Check whether provided config is good for a given MPP. +

Parameters

config – configuration of citus MPP section.

Returns

True is config passes validation, otherwise False.

class _patroni.postgresql.mpp.citus.CitusHandler(_postgresql: Postgresql, config: Dict[ str, str | int]) View on GitHub

Bases: Citus, AbstractMPPHandler, Thread + Define the interfaces for handling an underlying Citus cluster. +

__init\\__(postgresql: Postgresql, config: Dict[ str, str | int]) → None View on GitHub

“Initialize a new instance of CitusHandler. +

Parameters

+

abc_impl = <_abc._abc_data object>_

+

add_task(_task: PgDistNode) → bool View on GitHub

+

add_task(event: str, group: int, conn_url: str, timeout: float | None = None, cooldown: float | None = None) → PgDistNode | None View on GitHub

+

adjust_postgres_gucs(parameters: Dict[ str, Any]) → None View on GitHub

Adjust GUCs in the current PostgreSQL configuration. +

Parameters

parameters – dictionary of GUCs, with key as GUC name and the corresponding value as current GUC value. +

bootstrap() → None View on GitHub

Bootstrap handler.
Is called when the new cluster is initialized (through initdb or a custom bootstrap method). +

find_task_by_group(group: int) → int | None View on GitHub

+

handle_event(cluster: Cluster, event: Dict[ str, Any]) → None View on GitHub

Handle an event sent from a worker node. +

Parameters

+

ignore_replication_slot(slot: Dict[ str, str]) → bool View on GitHub

Check whether provided replication slot existing in the database should not be removed. + Note

  MPP database may create replication slots for its own use, for example to migrate data between workers using logical replication, and we don’t want to suddenly drop them.
  +
  Parameters:::
    *slot* – dictionary containing the replication slot settings, like `+name+`, `+database+`, `+type+`, and `+plugin+`.
  Returns:::
    `+True+` if the replication slots should not be removed, otherwise `+False+`.
+
load_pg_dist_node() → bool View on GitHub

Read from the pg_dist_node table and put it into the local cache +

on_demote() → None View on GitHub

On demote handler.
Is called when the primary was demoted. +

pick_task() → Tuple[ int | None, PgDistNode | None] View on GitHub

Returns the tuple(i, task), where i - is the task index in the self._tasks list + Tasks are picked by following priorities: +

  1. If there is already a transaction in progress, pick a task that that will change already affected worker primary.

  2. If the coordinator address should be changed - pick a task with group=0 (coordinators are always in group 0).

  3. Pick a task that is the oldest (first from the self._tasks) +

process_task(task: PgDistNode) → bool View on GitHub

Updates a single row in pg_dist_node table, optionally in a transaction. + The transaction is started if we do a demote of the worker node or before promoting the other worker if there is no transaction in progress. And, the transaction is committed when the switchover/failover completed. +

Parameters

task – reference to a PgDistNode object that represents a row to be updated/created.

Returns

True if the row was succesfully created/updated or transaction in progress was committed as an indicator that the self._pg_dist_node cache should be updated, or, if the new transaction was opened, this method returns False. +

process_tasks() → None View on GitHub

+

query(sql: str, *params: Any) → List[ Tuple[ Any, …​]] View on GitHub

+

run() → None View on GitHub

Method representing the thread’s activity.
You may override this method in a subclass. The standard run() method invokes the callable object passed to the object’s constructor as the target argument, if any, with sequential and keyword arguments taken from the args and kwargs arguments, respectively. +

schedule_cache_rebuild() → None View on GitHub

Cache rebuild handler.
Is called to notify handler that it has to refresh its metadata cache from the database. +

sync_meta_data(cluster: Cluster) → None View on GitHub

Maintain the pg_dist_node from the coordinator leader every heartbeat loop.
We can’t always rely on REST API calls from worker nodes in order to maintain pg_dist_node, therefore at least once per heartbeat loop we make sure that workes registered in self._pg_dist_node cache are matching the cluster view from DCS by creating tasks the same way as it is done from the REST API. +

update_node(task: PgDistNode) → None View on GitHub
class _patroni.postgresql.mpp.citus.PgDistNode(_group: int, host: str, port: int, event: str, nodeid: int | None = None, timeout: float | None = None, cooldown: float | None = None) View on GitHub

Bases: object + Represents a single row in the pg_dist_node table +

__init\\__(group: int, host: str, port: int, event: str, nodeid: int | None = None, timeout: float | None = None, cooldown: float | None = None) → None View on GitHub

+

wait() → None View on GitHub

+

wakeup() → None View on GitHub

Previous Next

© Copyright 2015 Compose, Zalando SE. Revision 9d231aee.

Built with Sphinx using a theme provided by Read the Docs.

Read the Docs v: latest

bugfix-readthedocs

Project Home

+ Builds