Fix protocol AttributeError and KeyError (v1.1.7)

Author: tom-dyar

CHANGELOG.md

@@ -5,6 +5,14 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.1.7] - 2026-01-17
+
+### Fixed
+- **AttributeError in Protocol**: Fixed `PGWireProtocol` to use dictionary access instead of dot notation for `performance_stats` returned by the new `SQLPipeline`.
+- **KeyError in Protocol**: Added missing `cache_hit` key to the `performance_stats` dictionary in `SQLTranslator` to prevent crashes during statement parsing.
+- **Embedded Namespace Context**: Strengthened the `SetNamespace` logic in background execution threads to ensure consistent IRIS namespace context and prevent intermittent "Class not found" errors.
+- **Redundant SQL Mapping**: Removed legacy schema translation code in `iris_executor.py` that was conflicting with the centralized `SQLPipeline`.
+
 ## [1.1.6] - 2026-01-17
 
 ### Changed

src/iris_pgwire/__init__.py

@@ -6,7 +6,7 @@
 caretdev/sqlalchemy-iris.
 """
 
-__version__ = "1.1.6"
+__version__ = "1.1.7"
 __author__ = "Thomas Dyar <thomas.dyar@intersystems.com>"
 
 # Don't import server/protocol in __init__ to avoid sys.modules conflicts

src/iris_pgwire/iris_executor.py

@@ -1049,23 +1049,17 @@ def fetchone(self):
     async def _execute_embedded_async(
         self, sql: str, params: list | None = None, session_id: str | None = None
     ) -> dict[str, Any]:
-        """
-        Execute SQL using IRIS embedded Python with proper async threading
-
-        This method runs the blocking IRIS operations in a thread pool to avoid
-        blocking the event loop, following constitutional async requirements.
-        """
+        """Execute query in IRIS embedded Python environment (async wrapper)"""
 
         def _sync_execute():
             """Synchronous IRIS execution in thread pool"""
             import iris
 
-            # CRITICAL: Ensure correct namespace context in background thread (Feature 036 Fix)
-            # Embedded Python process context is thread-local or needs explicit refresh
             if hasattr(iris, "system") and hasattr(iris.system, "Process"):
                 iris.system.Process.SetNamespace(self.iris_config.get("namespace", "USER"))
 
             # Log entry to embedded execution path
+
             logger.info(
                 "🔍 EXECUTING IN EMBEDDED MODE",
                 sql_preview=sql[:100],

src/iris_pgwire/protocol.py

@@ -229,10 +229,10 @@ async def translate_sql(
                     "skip_reason": result.skip_reason,
                     "command_tag": result.command_tag,
                     "translation_used": True,
-                    "performance_stats": {
-                        "translation_time_ms": tracker.start_time
-                        and (time.perf_counter() - tracker.start_time) * 1000
-                    },
+                    "performance_stats": result.performance_stats,
+                    "construct_mappings": [],
+                    "warnings": [],
+                    "validation_result": None,
                 }
 
         except Exception as e:
@@ -1361,26 +1361,30 @@ async def _handle_single_statement(self, query: str, send_ready: bool = True):
             if translation_result.get("translation_used") and translation_result.get(
                 "construct_mappings"
             ):
-                perf_stats = translation_result["performance_stats"]
+                perf_stats = translation_result.get("performance_stats", {})
                 logger.info(
                     "IRIS constructs translated",
                     connection_id=self.connection_id,
                     constructs_count=len(translation_result["construct_mappings"]),
-                    translation_time_ms=perf_stats.translation_time_ms if perf_stats else 0,
-                    cache_hit=perf_stats.cache_hit if perf_stats else False,
+                    translation_time_ms=perf_stats.get("normalization_time_ms", 0)
+                    if perf_stats
+                    else 0,
+                    cache_hit=perf_stats.get("cache_hit", False) if perf_stats else False,
                 )
 
             # Execute translated SQL against IRIS
             result = await self.iris_executor.execute_query(final_sql)
 
             # Add translation metadata to result for debugging/monitoring
             if translation_result.get("translation_used"):
-                perf_stats = translation_result["performance_stats"]
+                perf_stats = translation_result.get("performance_stats", {})
                 result["translation_metadata"] = {
                     "original_sql": translation_result["original_sql"],
                     "constructs_translated": len(translation_result.get("construct_mappings", [])),
-                    "translation_time_ms": perf_stats.translation_time_ms if perf_stats else 0,
-                    "cache_hit": perf_stats.cache_hit if perf_stats else False,
+                    "translation_time_ms": perf_stats.get("normalization_time_ms", 0)
+                    if perf_stats
+                    else 0,
+                    "cache_hit": perf_stats.get("cache_hit", False) if perf_stats else False,
                     "warnings": translation_result.get("warnings", []),
                 }
 
@@ -2626,10 +2630,12 @@ async def handle_parse_message(self, body: bytes):
                 "param_types": param_types,
                 "translation_metadata": {
                     "constructs_translated": len(translation_result.get("construct_mappings", [])),
-                    "translation_time_ms": translation_result[
-                        "performance_stats"
-                    ].translation_time_ms,
-                    "cache_hit": translation_result["performance_stats"].cache_hit,
+                    "translation_time_ms": translation_result.get("performance_stats", {}).get(
+                        "normalization_time_ms", 0
+                    ),
+                    "cache_hit": translation_result.get("performance_stats", {}).get(
+                        "cache_hit", False
+                    ),
                     "warnings": translation_result.get("warnings", []),
                 },
             }
@@ -2652,13 +2658,15 @@ async def handle_parse_message(self, body: bytes):
             if translation_result.get("translation_used") and translation_result.get(
                 "construct_mappings"
             ):
-                perf_stats = translation_result["performance_stats"]
+                perf_stats = translation_result.get("performance_stats", {})
                 logger.info(
                     "IRIS constructs translated in prepared statement",
                     connection_id=self.connection_id,
                     statement_name=statement_name,
                     constructs_count=len(translation_result["construct_mappings"]),
-                    translation_time_ms=perf_stats.translation_time_ms if perf_stats else 0,
+                    translation_time_ms=perf_stats.get("normalization_time_ms", 0)
+                    if perf_stats
+                    else 0,
                 )
 
             # Send ParseComplete response

src/iris_pgwire/sql_translator/normalizer.py

@@ -19,7 +19,8 @@
 
 import re
 import time
-from dataclasses import dataclass
+from dataclasses import dataclass, field
+from typing import Any
 
 from ..conversions.json_path import JsonPathBuilder
 from ..schema_mapper import translate_input_schema
@@ -42,6 +43,7 @@ class TranslationResult:
     was_skipped: bool = False
     skip_reason: SkipReason | None = None
     command_tag: str = ""
+    performance_stats: dict[str, Any] = field(default_factory=dict)
 
 
 class SQLTranslator:
@@ -192,6 +194,7 @@ def normalize_sql_with_result(
                 was_skipped=True,
                 skip_reason=filter_result.reason,
                 command_tag=filter_result.command_tag,
+                performance_stats=self._last_metrics.copy(),
             )
 
         normalized_sql, enum_count = self.enum_translator.translate(normalized_sql)
@@ -216,9 +219,10 @@ def normalize_sql_with_result(
             "boolean_translation_count": bool_count,
             "enum_translation_count": enum_count,
             "sla_violated": sla_violated,
+            "cache_hit": False,
         }
 
-        return TranslationResult(sql=normalized_sql)
+        return TranslationResult(sql=normalized_sql, performance_stats=self._last_metrics.copy())
 
     def _translate_json_operators(self, sql: str) -> tuple[str, int]:
         """Translate PostgreSQL JSON operators to IRIS JSON_VALUE/JSON_QUERY"""