patroni.dcs.kubernetes module¶
-
class
patroni.dcs.kubernetes.
CoreV1ApiProxy
(use_endpoints: Optional[bool] = False, bypass_api_service: Optional[bool] = False)¶ Bases:
object
Proxy class to work with k8s_client.CoreV1Api() object
-
_DEFAULT_RETRIABLE_HTTP_CODES
= frozenset({504, 500, 503})¶
-
__init__
(use_endpoints: Optional[bool] = False, bypass_api_service: Optional[bool] = False) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
configure_retriable_http_codes
(retriable_http_codes: List[int]) → None¶
-
configure_timeouts
(loop_wait: int, retry_timeout: Union[int, float], ttl: int) → None¶
-
refresh_api_servers_cache
() → None¶
-
use_endpoints
¶
-
-
class
patroni.dcs.kubernetes.
K8sClient
¶ Bases:
object
-
class
ApiClient
(bypass_api_service: Optional[bool] = False)¶ Bases:
object
-
_API_URL_PREFIX
= '/api/v1/namespaces/'¶
-
__init__
(bypass_api_service: Optional[bool] = False) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
_calculate_timeouts
(api_servers: int, timeout: Optional[float] = None) → Tuple[int, float, int]¶ 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: Optional[patroni.utils.Retry], api_servers_cache: List[str], method: str, path: str, **kwargs) → urllib3.response.HTTPResponse¶
-
_get_api_servers
(api_servers_cache: List[str]) → List[str]¶
-
static
_handle_server_response
(response: urllib3.response.HTTPResponse, _preload_content: bool) → Union[urllib3.response.HTTPResponse, patroni.dcs.kubernetes.K8sObject]¶
-
_load_api_servers_cache
() → None¶
-
static
_make_headers
(headers: Optional[Dict[str, str]]) → Dict[str, str]¶
-
_refresh_api_servers_cache
(updating_cache: Optional[bool] = False) → None¶
-
api_servers_cache
¶
-
call_api
(method: str, path: str, headers: Optional[Dict[str, str]] = None, body: Optional[Any] = None, _retry: Optional[patroni.utils.Retry] = None, _preload_content: bool = True, _request_timeout: Optional[float] = None, **kwargs) → Union[urllib3.response.HTTPResponse, patroni.dcs.kubernetes.K8sObject]¶
-
refresh_api_servers_cache
() → None¶
-
request
(retry: Optional[patroni.utils.Retry], method: str, path: str, timeout: Union[int, float, Tuple[Union[int, float], Union[int, float]], urllib3.util.timeout.Timeout, None] = None, **kwargs) → urllib3.response.HTTPResponse¶
-
set_api_servers_cache_ttl
(ttl: int) → None¶
-
set_base_uri
(value: str) → None¶
-
set_read_timeout
(timeout: Union[int, float]) → None¶
-
-
class
CoreV1Api
(api_client: Optional[K8sClient.ApiClient] = None)¶ Bases:
object
-
__init__
(api_client: Optional[K8sClient.ApiClient] = None) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
-
class
_K8sObjectTemplate
(**kwargs)¶ Bases:
patroni.dcs.kubernetes.K8sObject
The template for objects which we create locally, e.g. k8s_client.V1ObjectMeta & co
-
__init__
(**kwargs) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
-
__init__
() → None¶ Initialize self. See help(type(self)) for accurate signature.
-
class
rest
¶ Bases:
object
-
exception
ApiException
(status: Optional[int] = None, reason: Optional[str] = None, http_resp: Optional[urllib3.response.HTTPResponse] = None)¶ Bases:
Exception
-
__init__
(status: Optional[int] = None, reason: Optional[str] = None, http_resp: Optional[urllib3.response.HTTPResponse] = None) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
-
exception
-
class
-
class
patroni.dcs.kubernetes.
K8sConfig
¶ Bases:
object
-
__init__
() → None¶ Initialize self. See help(type(self)) for accurate signature.
-
static
_get_by_name
(config: Dict[str, List[Dict[str, Any]]], section: str, name: str) → Optional[Dict[str, Any]]¶
-
_make_headers
(token: Optional[str] = None, **kwargs) → None¶
-
_pool_config_from_file_or_data
(config: Dict[str, str], file_key_name: str, pool_key_name: str) → None¶
-
_read_token_file
() → str¶
-
_set_token
(token: str) → None¶
-
headers
¶
-
load_incluster_config
(ca_certs: str = '/var/run/secrets/kubernetes.io/serviceaccount/ca.crt', token_refresh_interval: datetime.timedelta = datetime.timedelta(seconds=60)) → None¶
-
load_kube_config
(context: Optional[str] = None) → None¶
-
server
¶
-
-
exception
patroni.dcs.kubernetes.
K8sConnectionFailed
¶
-
class
patroni.dcs.kubernetes.
K8sObject
(kwargs: Dict[str, Any])¶ Bases:
object
-
__init__
(kwargs: Dict[str, Any]) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
classmethod
_wrap
(parent: Optional[str], value: Any) → Any¶
-
get
(name: str, default: Optional[Any] = None) → Optional[Any]¶
-
to_dict
() → Dict[str, Any]¶
-
-
class
patroni.dcs.kubernetes.
Kubernetes
(config: Dict[str, Any], mpp: patroni.postgresql.mpp.AbstractMPP)¶ Bases:
patroni.dcs.AbstractDCS
-
_Kubernetes__load_cluster
(group: Optional[str], loader: Callable[[Dict[str, Any]], Union[patroni.dcs.Cluster, Dict[int, patroni.dcs.Cluster]]]) → Union[patroni.dcs.Cluster, Dict[int, patroni.dcs.Cluster]]¶
-
_Kubernetes__target_ref
(leader_ip: str, latest_subsets: List[patroni.dcs.kubernetes.K8sObject], pod: patroni.dcs.kubernetes.K8sObject) → patroni.dcs.kubernetes.K8sObject¶
-
__init__
(config: Dict[str, Any], mpp: patroni.postgresql.mpp.AbstractMPP) → None¶ Prepare DCS paths, MPP object, initial values for state information and processing dependencies.
Parameters: - config –
dict
, reference to config section of selected DCS. i.e.:zookeeper
for zookeeper,etcd
for etcd, etc… - mpp – an object implementing
AbstractMPP
interface.
- config –
-
_abc_impl
= <_abc_data object>¶
-
_cluster_from_nodes
(group: str, nodes: Dict[str, patroni.dcs.kubernetes.K8sObject], pods: Collection[patroni.dcs.kubernetes.K8sObject]) → patroni.dcs.Cluster¶
-
_config_resource_version
¶
-
_create_config_service
() → None¶
-
_delete_leader
(leader: patroni.dcs.Leader) → bool¶ Unused
-
_load_cluster
(path: str, loader: Callable[[Any], Union[patroni.dcs.Cluster, Dict[int, patroni.dcs.Cluster]]]) → Union[patroni.dcs.Cluster, Dict[int, patroni.dcs.Cluster]]¶ 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 theget_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]) → None¶
-
_mpp_cluster_loader
(path: Dict[str, Any]) → Dict[int, patroni.dcs.Cluster]¶ 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 andCluster
objects as values.
-
_patch_or_create
(name: str, annotations: Dict[str, Any], resource_version: Optional[str] = None, patch: bool = False, retry: Optional[Callable[[...], Any]] = None, ips: Optional[List[str]] = None) → patroni.dcs.kubernetes.K8sObject¶ 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, whendelete_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 thekubernetes.pod_ip
;['']
- when we want to make sure that the subsets of the leader endpoint contains the IP address of the leader, butkubernetes.pod_ip
configuration is missing. In this case we will try to take the IP address of the Pod which name matchesname
from the config file.
Returns: the new
V1Endpoints
orV1ConfigMap
object, that was created or updated.
-
_postgresql_cluster_loader
(path: Dict[str, Any]) → patroni.dcs.Cluster¶ 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: patroni.dcs.Leader) → bool¶ Unused
-
_update_leader_with_retry
(annotations: Dict[str, Any], resource_version: Optional[str], ips: List[str]) → bool¶
-
_wait_caches
(stop_time: float) → None¶
-
_write_failsafe
(value: str) → bool¶ Unused
-
_write_leader_optime
(last_lsn: str) → bool¶ Unused
-
_write_status
(value: str) → bool¶ Unused
-
attempt_to_acquire_leader
() → bool¶ 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
() → bool¶ Removes the
initialize
key for a cluster.Returns: True
if successfully committed to DCS.
-
client_path
(path: str) → str¶ 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: patroni.dcs.kubernetes.K8sObject, p2: patroni.dcs.kubernetes.K8sObject) → bool¶
-
delete_cluster
(*args, **kwargs) → Any¶
-
delete_leader
(leader: Optional[patroni.dcs.Leader], last_lsn: Optional[int] = None) → bool¶ 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()
.- leader –
-
delete_sync_state
(version: Optional[str] = None) → bool¶ 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
() → Optional[patroni.dcs.Cluster]¶ 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 = '') → bool¶ Race for cluster initialization.
This method should atomically create
initialize
key and returnTrue
, otherwise it should returnFalse
.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.- create_new –
-
leader_path
¶ Get the client path for
leader
.
-
manual_failover
(leader: Optional[str], candidate: Optional[str], scheduled_at: Optional[datetime.datetime] = None, version: Optional[str] = None) → bool¶ 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.- leader – value to set for
-
static
member
(pod: patroni.dcs.kubernetes.K8sObject) → patroni.dcs.Member¶
-
patch_or_create
(*args, **kwargs) → Any¶
-
patch_or_create_config
(annotations: Dict[str, Any], resource_version: Optional[str] = None, patch: bool = False, retry: bool = True) → bool¶
-
reload_config
(config: Union[Config, Dict[str, Any]]) → None¶ 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, **kwargs) → Any¶
-
set_config_value
(value: str, version: Optional[str] = None) → bool¶ 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.- value – new value to set in the
-
set_failover_value
(value: str, version: Optional[str] = None) → bool¶ Unused
-
set_history_value
(value: str) → bool¶ 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) → None¶ Set the new value for retry_timeout.
-
set_sync_state_value
(value: str, version: Optional[str] = None) → bool¶ Unused
-
set_ttl
(ttl: int) → Optional[bool]¶ Set the new ttl value for DCS keys.
-
static
subsets_changed
(last_observed_subsets: List[patroni.dcs.kubernetes.K8sObject], ip: str, ports: List[patroni.dcs.kubernetes.K8sObject]) → bool¶ >>> 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
() → bool¶ Establish a new leader in DCS.
Note
This method should create leader key with value of
_name
andttl
ofttl
.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, **kwargs) → Any¶
-
ttl
¶ Get current
ttl
value.
-
update_leader
(cluster: patroni.dcs.Cluster, last_lsn: Optional[int], slots: Optional[Dict[str, int]] = None, failsafe: Optional[Dict[str, str]] = None) → bool¶ Update
leader
key (or session) ttl,/status
, and/failsafe
keys.Parameters: - cluster –
Cluster
object with information about the current cluster state. - 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
ifleader
key (or session) has been updated successfully.- cluster –
-
watch
(leader_version: Optional[str], timeout: float) → bool¶ 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) → None¶ Write value for WAL LSN to
optime
annotation of the leader object.Parameters: last_lsn – absolute WAL LSN in bytes.
-
write_sync_state
(leader: Optional[str], sync_standby: Optional[Collection[str]], quorum: Optional[int], version: Optional[str] = None) → Optional[patroni.dcs.SyncState]¶ 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
- quorum – if the node from sync_standby list is doing a leader race it should see at least quorum other nodes from the sync_standby + leader list
- 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)¶ Bases:
patroni.exceptions.DCSError
-
exception
patroni.dcs.kubernetes.
KubernetesRetriableException
(orig: patroni.dcs.kubernetes.K8sClient.rest.ApiException)¶ Bases:
patroni.dcs.kubernetes.ApiException
-
__init__
(orig: patroni.dcs.kubernetes.K8sClient.rest.ApiException) → None¶ Initialize self. See help(type(self)) for accurate signature.
-
sleeptime
¶
-
-
class
patroni.dcs.kubernetes.
ObjectCache
(dcs: patroni.dcs.kubernetes.Kubernetes, func: Callable[[...], Any], retry: patroni.utils.Retry, condition: threading.Condition, name: Optional[str] = None)¶ Bases:
threading.Thread
-
__init__
(dcs: patroni.dcs.kubernetes.Kubernetes, func: Callable[[...], Any], retry: patroni.utils.Retry, condition: threading.Condition, name: Optional[str] = None) → None¶ 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 the argument tuple 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
() → None¶
-
_do_watch
(resource_version: str) → None¶
-
static
_finish_response
(response: urllib3.response.HTTPResponse) → None¶
-
_list
() → patroni.dcs.kubernetes.K8sObject¶
-
_process_event
(event: Dict[str, Any]) → None¶
-
_watch
(resource_version: str) → urllib3.response.HTTPResponse¶
-
copy
() → Dict[str, patroni.dcs.kubernetes.K8sObject]¶
-
delete
(name: str, resource_version: str) → Tuple[bool, Optional[patroni.dcs.kubernetes.K8sObject]]¶
-
get
(name: str) → Optional[patroni.dcs.kubernetes.K8sObject]¶
-
is_ready
() → bool¶ Must be called only when holding the lock on _condition
-
kill_stream
() → None¶
-
run
() → None¶ 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: patroni.dcs.kubernetes.K8sObject) → Tuple[bool, Optional[patroni.dcs.kubernetes.K8sObject]]¶
-
-
patroni.dcs.kubernetes.
_cleanup_temp_files
() → None¶
-
patroni.dcs.kubernetes.
_create_temp_file
(content: bytes) → str¶
-
patroni.dcs.kubernetes.
_run_and_handle_exceptions
(method: Callable[[...], Any], *args, **kwargs) → Any¶
-
patroni.dcs.kubernetes.
catch_kubernetes_errors
(func: Callable[[...], Any]) → Callable[[...], Any]¶
-
patroni.dcs.kubernetes.
k8s_client
¶
-
patroni.dcs.kubernetes.
to_camel_case
(value: str) → str¶