Merge pull request #2317 from anvil-works/allow_watch_bookmarks

Add support for allowWatchBookmarks to the dynamic client
This commit is contained in:
Kubernetes Prow Robot 2025-04-24 13:54:31 -07:00 committed by GitHub
commit d010f2efd6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 84 additions and 2 deletions

View File

@ -0,0 +1,78 @@
# Copyright 2025 The Kubernetes Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
Uses watch to print a stream of Pod events from the default namespace.
The allow_watch_bookmarks flag is set to True, so the API server can send
BOOKMARK events.
If the connection to the API server is lost, the script will reconnect and
resume watching from the most recently received resource version.
For more information, see:
- https://kubernetes.io/docs/reference/using-api/api-concepts/#efficient-detection-of-changes
- https://kubernetes.io/docs/reference/using-api/api-concepts/#semantics-for-watch
"""
import urllib3
from kubernetes import config
from kubernetes.client import api_client
from kubernetes.client.exceptions import ApiException
from kubernetes.dynamic.client import DynamicClient
NAMESPACE = "default"
def main():
# Configs can be set in Configuration class directly or using helper
# utility. If no argument provided, the config will be loaded from
# default location.
config.load_kube_config()
client = DynamicClient(api_client.ApiClient())
api = client.resources.get(api_version="v1", kind="Pod")
# Setting resource_version=None means the server will send synthetic
# ADDED events for all resources that exist when the watch starts.
resource_version = None
while True:
try:
for event in api.watch(
namespace=NAMESPACE,
resource_version=resource_version,
allow_watch_bookmarks=True,
):
# Remember the last resourceVersion we saw, so we can resume
# watching from there if the connection is lost.
resource_version = event['object'].metadata.resourceVersion
print("Event: %s %s %s" % (
resource_version,
event['type'],
event['object'].metadata.name,
))
except ApiException as err:
if err.status == 410:
print("ERROR: The requested resource version is no longer available.")
resource_version = None
else:
raise
except urllib3.exceptions.ProtocolError:
print("Lost connection to the k8s API server. Reconnecting...")
if __name__ == "__main__":
main()

View File

@ -163,7 +163,7 @@ class DynamicClient(object):
return self.request('patch', path, body=body, force_conflicts=force_conflicts, **kwargs)
def watch(self, resource, namespace=None, name=None, label_selector=None, field_selector=None, resource_version=None, timeout=None, watcher=None):
def watch(self, resource, namespace=None, name=None, label_selector=None, field_selector=None, resource_version=None, timeout=None, watcher=None, allow_watch_bookmarks=None):
"""
Stream events for a resource from the Kubernetes API
@ -176,6 +176,7 @@ class DynamicClient(object):
a resource_version greater than this value will be returned
:param timeout: The amount of time in seconds to wait before terminating the stream
:param watcher: The Watcher object that will be used to stream the resource
:param allow_watch_bookmarks: Ask the API server to send BOOKMARK events
:return: Event object with these keys:
'type': The type of event such as "ADDED", "DELETED", etc.
@ -206,7 +207,8 @@ class DynamicClient(object):
label_selector=label_selector,
resource_version=resource_version,
serialize=False,
timeout_seconds=timeout
timeout_seconds=timeout,
allow_watch_bookmarks=allow_watch_bookmarks,
):
event['object'] = ResourceInstance(resource, event['object'])
yield event
@ -248,6 +250,8 @@ class DynamicClient(object):
query_params.append(('fieldManager', params['field_manager']))
if params.get('force_conflicts') is not None:
query_params.append(('force', params['force_conflicts']))
if params.get('allow_watch_bookmarks') is not None:
query_params.append(('allowWatchBookmarks', params['allow_watch_bookmarks']))
header_params = params.get('header_params', {})
form_params = []