From 8903d1be4b3d97d66f92074365e781efc7721d42 Mon Sep 17 00:00:00 2001 From: Robert Yokota Date: Mon, 24 Feb 2025 11:24:12 -0800 Subject: [PATCH] Ensure cached schema matches input --- .../schema_registry/schema_registry_client.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/confluent_kafka/schema_registry/schema_registry_client.py b/src/confluent_kafka/schema_registry/schema_registry_client.py index 93af2f028..0e301e73d 100644 --- a/src/confluent_kafka/schema_registry/schema_registry_client.py +++ b/src/confluent_kafka/schema_registry/schema_registry_client.py @@ -473,7 +473,7 @@ def set_schema(self, subject: str, schema_id: int, schema: 'Schema'): self.schema_id_index[subject][schema_id] = schema self.schema_index[subject][schema] = schema_id - def set_registered_schema(self, registered_schema: 'RegisteredSchema'): + def set_registered_schema(self, schema: 'Schema', registered_schema: 'RegisteredSchema'): """ Add a RegisteredSchema to the cache. @@ -484,7 +484,6 @@ def set_registered_schema(self, registered_schema: 'RegisteredSchema'): subject = registered_schema.subject schema_id = registered_schema.schema_id version = registered_schema.version - schema = registered_schema.schema with self.lock: self.schema_id_index[subject][schema_id] = schema self.schema_index[subject][schema] = schema_id @@ -786,7 +785,7 @@ def register_schema_full_response( registered_schema = RegisteredSchema.from_dict(response) - self._cache.set_schema(subject_name, registered_schema.schema_id, registered_schema.schema) + self._cache.set_schema(subject_name, registered_schema.schema_id, schema) return registered_schema @@ -866,7 +865,7 @@ def lookup_schema( registered_schema = RegisteredSchema.from_dict(response) - self._cache.set_registered_schema(registered_schema) + self._cache.set_registered_schema(schema, registered_schema) return registered_schema @@ -1025,7 +1024,7 @@ def get_version( registered_schema = RegisteredSchema.from_dict(response) - self._cache.set_registered_schema(registered_schema) + self._cache.set_registered_schema(registered_schema.schema, registered_schema) return registered_schema