patroni.dcs.kubernetes module

class patroni.dcs.kubernetes.CoreV1ApiProxy(use_endpoints: bool | None = False, bypass_api_service: bool | None = False)View on GitHub 

Bases: object

Proxy class to work with k8s_client.CoreV1Api() object

_DEFAULT_RETRIABLE_HTTP_CODES = frozenset({500, 503, 504})

__init__(use_endpoints: bool | None = False, bypass_api_service: bool | None = False) → NoneView on GitHub 

configure_retriable_http_codes(retriable_http_codes: List[int]) → NoneView on GitHub 

configure_timeouts(loop_wait: int, retry_timeout: int | float, ttl: int) → NoneView on GitHub 

refresh_api_servers_cache() → NoneView on GitHub 

property use_endpoints: bool

class patroni.dcs.kubernetes.K8sClientView on GitHub 

Bases: object

class ApiClient(bypass_api_service: bool | None = False)View on GitHub 

Bases: object

_API_URL_PREFIX = '/api/v1/namespaces/'

__init__(bypass_api_service: bool | None = False) → NoneView on GitHub 

_calculate_timeouts(api_servers: int, timeout: float | None = None) → Tuple[int, float, int]View on GitHub : Calculate a request timeout and number of retries per single K8s API server node. In case if the timeout per node is too small (less than one second) we will reduce the number of nodes. For the cluster with only one API server node we will try to do 1 retry. No retries for clusters with 2 or more API server nodes. We better rely on switching to a different node.

_do_http_request(retry: Retry | None, api_servers_cache: List[str], method: str, path: str, **kwargs: Any) → HTTPResponseView on GitHub 

_get_api_servers(api_servers_cache: List[str]) → List[str]View on GitHub 

static _handle_server_response(response: HTTPResponse, _preload_content: bool) → HTTPResponse | K8sObjectView on GitHub 

_load_api_servers_cache() → NoneView on GitHub 

static _make_headers(headers: Dict[str, str] | None) → Dict[str, str]View on GitHub 

_refresh_api_servers_cache(updating_cache: bool | None = False) → NoneView on GitHub 

property api_servers_cache: List[str]

call_api(method: str, path: str, headers: Dict[str, str] | None = None, body: Any | None = None, _retry: Retry | None = None, _preload_content: bool = True, _request_timeout: float | None = None, **kwargs: Any) → HTTPResponse | K8sObjectView on GitHub 

refresh_api_servers_cache() → NoneView on GitHub 

request(retry: Retry | None, method: str, path: str, timeout: int | float | Tuple[int | float, int | float] | Timeout | None = None, **kwargs: Any) → HTTPResponseView on GitHub 

set_api_servers_cache_ttl(ttl: int) → NoneView on GitHub 

set_base_uri(value: str) → NoneView on GitHub 

set_read_timeout(timeout: int | float) → NoneView on GitHub 

class CoreV1Api(api_client: ApiClient | None = None)View on GitHub 

Bases: object

__init__(api_client: ApiClient | None = None) → NoneView on GitHub 

class _K8sObjectTemplate(**kwargs: Any)View on GitHub 

Bases: K8sObject

The template for objects which we create locally, e.g. k8s_client.V1ObjectMeta & co

__init__(**kwargs: Any) → NoneView on GitHub 

__init__() → NoneView on GitHub 

class restView on GitHub 

Bases: object

exception ApiException(status: int | None = None, reason: str | None = None, http_resp: HTTPResponse | None = None)View on GitHub 

Bases: Exception

__init__(status: int | None = None, reason: str | None = None, http_resp: HTTPResponse | None = None) → NoneView on GitHub 

class patroni.dcs.kubernetes.K8sConfigView on GitHub 

Bases: object

exception ConfigExceptionView on GitHub : Bases: Exception

__init__() → NoneView on GitHub 

static _get_by_name(config: Dict[str, List[Dict[str, Any]]], section: str, name: str) → Dict[str, Any] | NoneView on GitHub 

_make_headers(token: str | None = None, **kwargs: Any) → NoneView on GitHub 

_pool_config_from_file_or_data(config: Dict[str, str], file_key_name: str, pool_key_name: str) → NoneView on GitHub 

_read_token_file() → strView on GitHub 

_set_token(token: str) → NoneView on GitHub 

property headers: Dict[str, str]

load_incluster_config(ca_certs: str = '/var/run/secrets/kubernetes.io/serviceaccount/ca.crt', token_refresh_interval: timedelta = datetime.timedelta(seconds=60)) → NoneView on GitHub 

load_kube_config(context: str | None = None) → NoneView on GitHub 

property server: str

exception patroni.dcs.kubernetes.K8sConnectionFailedView on GitHub : Bases: K8sException

exception patroni.dcs.kubernetes.K8sExceptionView on GitHub : Bases: Exception

class patroni.dcs.kubernetes.K8sObject(kwargs: Dict[str, Any])View on GitHub 

Bases: object

__init__(kwargs: Dict[str, Any]) → NoneView on GitHub 

classmethod _wrap(parent: str | None, value: Any) → AnyView on GitHub 

get(name: str, default: Any | None = None) → Any | NoneView on GitHub 

to_dict() → Dict[str, Any]View on GitHub 

class patroni.dcs.kubernetes.Kubernetes(config: Dict[str, Any], mpp: AbstractMPP)View on GitHub 

Bases: AbstractDCS

__init__(config: Dict[str, Any], mpp: AbstractMPP) → NoneView on GitHub 

Prepare DCS paths, MPP object, initial values for state information and processing dependencies.

Variables:: config – dict, reference to config section of selected DCS. i.e.: zookeeper for zookeeper, etcd for etcd, etc…

__load_cluster(group: str | None, loader: Callable[[Dict[str, Any]], Cluster | Dict[int, Cluster]]) → Cluster | Dict[int, Cluster]

__target_ref(leader_ip: str, latest_subsets: List[K8sObject], pod: K8sObject) → K8sObject

_abc_impl = <_abc._abc_data object>

_cluster_from_nodes(group: str, nodes: Dict[str, K8sObject], pods: Collection[K8sObject]) → ClusterView on GitHub 

property _config_resource_version: str | None

_create_config_service() → NoneView on GitHub 

_delete_leader(leader: Leader) → boolView on GitHub : Unused

_load_cluster(path: str, loader: Callable[[Any], Cluster | Dict[int, Cluster]]) → Cluster | Dict[int, Cluster]View on GitHub 

Main abstract method that implements the loading of Cluster instance.

Note

Internally this method should call the loader method that will build Cluster object which represents current state and topology of the cluster in DCS. This method supposed to be called only by the get_cluster() method.

Parameters:

path – the path in DCS where to load Cluster(s) from.
loader – one of _postgresql_cluster_loader() or _mpp_cluster_loader().

Raise:

DCSError in case of communication problems with DCS. If the current node was running as a primary and exception raised, instance would be demoted.

_map_subsets(endpoints: Dict[str, Any], ips: List[str]) → NoneView on GitHub 

_mpp_cluster_loader(path: Dict[str, Any]) → Dict[int, Cluster]View on GitHub 

Load and build all PostgreSQL clusters from a single MPP cluster.

Parameters:: path – the path in DCS where to load Cluster(s) from.
Returns:: all MPP groups as dict, with group IDs as keys and Cluster objects as values.

_patch_or_create(name: str, annotations: Dict[str, Any], resource_version: str | None = None, patch: bool = False, retry: Callable[[...], Any] | None = None, ips: List[str] | None = None) → K8sObjectView on GitHub 

Patch or create K8s object, Endpoint or ConfigMap.

Parameters:

name – the name of the object.
annotations – mapping of annotations that we want to create/update.
resource_version – object should be updated only if the resource_version matches provided value.
patch – True if we know in advance that the object already exists and we should patch it.
retry – a callable that will take care of retries
ips –
IP address that we want to put to the subsets of the endpoint. Could have following values:
- None - when we don’t need to touch subset;
- [] - to set subsets to the empty list, when delete_leader() method is called;
- ['ip.add.re.ss'] - when we want to make sure that the subsets of the leader endpoint contains the IP address of the leader, that we get from the kubernetes.pod_ip;
- [''] - when we want to make sure that the subsets of the leader endpoint contains the IP address of the leader, but kubernetes.pod_ip configuration is missing. In this case we will try to take the IP address of the Pod which name matches name from the config file.

Returns:

the new V1Endpoints or V1ConfigMap object, that was created or updated.

_postgresql_cluster_loader(path: Dict[str, Any]) → ClusterView on GitHub 

Load and build the Cluster object from DCS, which represents a single PostgreSQL cluster.

Parameters:: path – the path in DCS where to load Cluster from.
Returns:: Cluster instance.

_update_leader(leader: Leader) → boolView on GitHub : Unused

_update_leader_with_retry(annotations: Dict[str, Any], resource_version: str | None, ips: List[str]) → boolView on GitHub 

_wait_caches(stop_time: float) → NoneView on GitHub 

_write_failsafe(value: str) → boolView on GitHub : Unused

_write_leader_optime(last_lsn: str) → boolView on GitHub : Unused

_write_status(value: str) → boolView on GitHub : Unused

attempt_to_acquire_leader() → boolView on GitHub 

Attempt to acquire leader lock.

Note

This method should create /leader key with the value _name.

The key must be created atomically. In case the key already exists it should not be overwritten and False must be returned.

If key creation fails due to DCS not being accessible or because it is not able to process requests (hopefully temporary), the DCSError exception should be raised.

Returns:: True if key has been created successfully.

cancel_initialization() → boolView on GitHub 

Removes the initialize key for a cluster.

Returns:: True if successfully committed to DCS.

client_path(path: str) → strView on GitHub 

Construct the absolute key name from appropriate parts for the DCS type.

Parameters:: path – The key name within the current Patroni cluster.
Returns:: absolute key name for the current Patroni cluster.

static compare_ports(p1: K8sObject, p2: K8sObject) → boolView on GitHub 

delete_cluster(*args: Any, **kwargs: Any) → AnyView on GitHub 

Delete cluster from DCS.

Returns:: True if successfully committed to DCS.

delete_leader(leader: Leader | None, last_lsn: int | None = None) → boolView on GitHub 

Update optime/leader and voluntarily remove leader key from DCS.

This method should remove leader key if current instance is the leader.

Parameters:

leader – Leader object with information about the leader.
last_lsn – latest checkpoint location in bytes.

Returns:

boolean result of called abstract _delete_leader().

delete_sync_state(version: str | None = None) → boolView on GitHub 

Patch annotations of $SCOPE-sync Endpoint or ConfigMap with empty values.

Effectively it removes “leader” and “sync_standby” annotations from the object. :param version: last known resource_version for conditional update of the object :returns: True if “delete” was successful

get_mpp_coordinator() → Cluster | NoneView on GitHub 

Load the PostgreSQL cluster for the MPP Coordinator.

Note

This method is only executed on the worker nodes to find the coordinator.

Returns:: Select Cluster instance associated with the MPP Coordinator group ID.

initialize(create_new: bool = True, sysid: str = '') → boolView on GitHub 

Race for cluster initialization.

This method should atomically create initialize key and return True, otherwise it should return False.

Parameters:

create_new – False if the key should already exist (in the case we are setting the system_id).
sysid – PostgreSQL cluster system identifier, if specified, is written to the key.

Returns:

True if key has been created successfully.

property leader_path: str: Get the client path for leader.

manual_failover(leader: str | None, candidate: str | None, scheduled_at: datetime | None = None, version: str | None = None) → boolView on GitHub 

Prepare dictionary with given values and set /failover key in DCS.

Parameters:

leader – value to set for leader.
candidate – value to set for member.
scheduled_at – value converted to ISO date format for scheduled_at.
version – for conditional update of the key/object.

Returns:

True if successfully committed to DCS.

static member(pod: K8sObject) → MemberView on GitHub 

patch_or_create(*args: Any, **kwargs: Any) → AnyView on GitHub 

patch_or_create_config(annotations: Dict[str, Any], resource_version: str | None = None, patch: bool = False, retry: bool = True) → boolView on GitHub 

reload_config(config: Config | Dict[str, Any]) → NoneView on GitHub 

Handles dynamic config changes.

Either cause by changes in the local configuration file + SIGHUP or by changes of dynamic configuration

retry(method: Callable[[...], Any], *args: Any, **kwargs: Any) → AnyView on GitHub 

set_config_value(value: str, version: str | None = None) → boolView on GitHub 

Create or update /config key in DCS.

Parameters:

value – new value to set in the config key.
version – for conditional update of the key/object.

Returns:

True if successfully committed to DCS.

set_failover_value(value: str, version: str | None = None) → boolView on GitHub : Unused

set_history_value(value: str) → boolView on GitHub 

Set value for history in DCS.

Parameters:: value – new value of history key/object.
Returns:: True if successfully committed to DCS.

set_retry_timeout(retry_timeout: int) → NoneView on GitHub : Set the new value for retry_timeout.

set_sync_state_value(value: str, version: str | None = None) → boolView on GitHub : Unused

set_ttl(ttl: int) → bool | NoneView on GitHub : Set the new ttl value for DCS keys.

static subsets_changed(last_observed_subsets: List[K8sObject], ip: str, ports: List[K8sObject]) → boolView on GitHub 

>>> ip = '1.2.3.4'
>>> a = [k8s_client.V1EndpointAddress(ip=ip)]
>>> s = [k8s_client.V1EndpointSubset(addresses=a)]
>>> Kubernetes.subsets_changed(s, '1.2.3.5', [])
True
>>> s = [k8s_client.V1EndpointSubset(addresses=a, ports=[k8s_client.V1EndpointPort(protocol='TCP', port=1)])]
>>> Kubernetes.subsets_changed(s, '1.2.3.4', [k8s_client.V1EndpointPort(port=5432)])
True
>>> p1 = k8s_client.V1EndpointPort(name='port1', port=1)
>>> p2 = k8s_client.V1EndpointPort(name='port2', port=2)
>>> p3 = k8s_client.V1EndpointPort(name='port3', port=3)
>>> s = [k8s_client.V1EndpointSubset(addresses=a, ports=[p1, p2])]
>>> Kubernetes.subsets_changed(s, ip, [p2, p3])
True
>>> s2 = [k8s_client.V1EndpointSubset(addresses=a, ports=[p2, p1])]
>>> Kubernetes.subsets_changed(s, ip, [p2, p1])
False

take_leader() → boolView on GitHub 

Establish a new leader in DCS.

Note

This method should create leader key with value of _name and ttl of ttl.

Since it could be called only on initial cluster bootstrap it could create this key regardless, overwriting the key if necessary.

Returns:: True if successfully committed to DCS.

touch_member(*args: Any, **kwargs: Any) → AnyView on GitHub 

Update member key in DCS.

Note

This method should create or update key with the name with /members/ + _name and the value of data in a given DCS.

Parameters:: data – information about an instance (including connection strings).
Returns:: True if successfully committed to DCS.

property ttl: int: Get current ttl value.

update_leader(leader: Leader, last_lsn: int | None, slots: Dict[str, int] | None = None, failsafe: Dict[str, str] | None = None) → boolView on GitHub 

Update leader key (or session) ttl and optime/leader.

Parameters:

leader – Leader object with information about the leader.
last_lsn – absolute WAL LSN in bytes.
slots – dictionary with permanent slots confirmed_flush_lsn.
failsafe – if defined dictionary passed to write_failsafe().

Returns:

True if leader key (or session) has been updated successfully.

watch(leader_version: str | None, timeout: float) → boolView on GitHub 

Sleep if the current node is a leader, otherwise, watch for changes of leader key with a given timeout.

Parameters:

leader_version – version of a leader key.
timeout – timeout in seconds.

Returns:

if True this will reschedule the next run of the HA cycle.

write_leader_optime(last_lsn: int) → NoneView on GitHub 

Write value for WAL LSN to optime annotation of the leader object.

Parameters:: last_lsn – absolute WAL LSN in bytes.

write_sync_state(leader: str | None, sync_standby: Collection[str] | None, version: str | None = None) → SyncState | NoneView on GitHub 

Prepare and write annotations to $SCOPE-sync Endpoint or ConfigMap.

Parameters:

leader – name of the leader node that manages /sync key
sync_standby – collection of currently known synchronous standby node names
version – last known resource_version for conditional update of the object

Returns:

the new SyncState object or None

exception patroni.dcs.kubernetes.KubernetesError(value: Any)View on GitHub : Bases: DCSError

exception patroni.dcs.kubernetes.KubernetesRetriableException(orig: ApiException)View on GitHub 

Bases: ApiException

__init__(orig: ApiException) → NoneView on GitHub 

property sleeptime: int | None

class patroni.dcs.kubernetes.ObjectCache(dcs: Kubernetes, func: Callable[[...], Any], retry: Retry, condition: Condition, name: str | None = None)View on GitHub 

Bases: Thread

__init__(dcs: Kubernetes, func: Callable[[...], Any], retry: Retry, condition: Condition, name: str | None = None) → NoneView on GitHub 

This constructor should always be called with keyword arguments. Arguments are:

group should be None; reserved for future extension when a ThreadGroup class is implemented.

target is the callable object to be invoked by the run() method. Defaults to None, meaning nothing is called.

name is the thread name. By default, a unique name is constructed of the form “Thread-N” where N is a small decimal number.

args is a list or tuple of arguments for the target invocation. Defaults to ().

kwargs is a dictionary of keyword arguments for the target invocation. Defaults to {}.

If a subclass overrides the constructor, it must make sure to invoke the base class constructor (Thread.__init__()) before doing anything else to the thread.

_build_cache() → NoneView on GitHub 

_do_watch(resource_version: str) → NoneView on GitHub 

static _finish_response(response: HTTPResponse) → NoneView on GitHub 

_list() → K8sObjectView on GitHub 

_process_event(event: Dict[str, Any | Dict[str, Dict[str, Any] | Any]]) → NoneView on GitHub 

_watch(resource_version: str) → HTTPResponseView on GitHub 

copy() → Dict[str, K8sObject]View on GitHub 

delete(name: str, resource_version: str) → Tuple[bool, K8sObject | None]View on GitHub 

get(name: str) → K8sObject | NoneView on GitHub 

is_ready() → boolView on GitHub : Must be called only when holding the lock on _condition

kill_stream() → NoneView on GitHub 

run() → NoneView 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.

set(name: str, value: K8sObject) → Tuple[bool, K8sObject | None]View on GitHub 

patroni.dcs.kubernetes._cleanup_temp_files() → NoneView on GitHub 

patroni.dcs.kubernetes._create_temp_file(content: bytes) → strView on GitHub 

patroni.dcs.kubernetes._run_and_handle_exceptions(method: Callable[[...], Any], *args: Any, **kwargs: Any) → AnyView on GitHub 

patroni.dcs.kubernetes.catch_kubernetes_errors(func: Callable[[...], Any]) → Callable[[...], Any]View on GitHub 

patroni.dcs.kubernetes.to_camel_case(value: str) → strView on GitHub 