c2cwsgiutils 6.1.8.dev108__tar.gz → 6.1.8.dev112__tar.gz

{c2cwsgiutils-6.1.8.dev108 → c2cwsgiutils-6.1.8.dev112}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: c2cwsgiutils
-Version: 6.1.8.dev108
+Version: 6.1.8.dev112
 Summary: Common utilities for Camptocamp WSGI applications
 Home-page: https://github.com/camptocamp/c2cwsgiutils
 License: BSD-2-Clause

{c2cwsgiutils-6.1.8.dev108 → c2cwsgiutils-6.1.8.dev112}/c2cwsgiutils/db.py

@@ -51,7 +51,7 @@ def setup_session(
     With an accompanying tween that switches between the master and the slave DB
     connection. Uses prefixed entries in the application's settings.
 
-    The slave DB will be used for anything that is GET and OPTIONS queries. The master DB will be used for
+    The slave DB will be used for anything that is GET, HEAD and OPTIONS queries. The master DB will be used for
     all the other queries. You can tweak this behavior with the force_master and force_slave parameters.
     Those parameters are lists of regex that are going to be matched against "{VERB} {PATH}". Warning, the
     path includes the route_prefix.
@@ -107,7 +107,7 @@ def create_session(
     With an accompanying tween that switches between the master and the slave DB
     connection.
 
-    The slave DB will be used for anything that is GET and OPTIONS queries. The master DB will be used for
+    The slave DB will be used for anything that is GET, HEAD and OPTIONS queries. The master DB will be used for
     all the other queries. You can tweak this behavior with the force_master and force_slave parameters.
     Those parameters are lists of regex that are going to be matched against "{VERB} {PATH}". Warning, the
     path includes the route_prefix.
@@ -178,7 +178,10 @@ def _add_tween(
             has_force_master = any(r.match(method_path) for r in master_paths)
             if FORCE_READONLY or (
                 not has_force_master
-                and (request.method in ("GET", "OPTIONS") or any(r.match(method_path) for r in slave_paths))
+                and (
+                    request.method in ("GET", "HEAD", "OPTIONS")
+                    or any(r.match(method_path) for r in slave_paths)
+                )
             ):
                 _LOG.debug(
                     "Using %s database for: %s",
@@ -246,7 +249,7 @@ class SessionFactory(_sessionmaker):
             if FORCE_READONLY or (
                 not has_force_master
                 and (
-                    request.method in ("GET", "OPTIONS")
+                    request.method in ("GET", "OPTIONS", "HEAD")
                     or any(r.match(method_path) for r in self.slave_paths)
                 )
             ):

{c2cwsgiutils-6.1.8.dev108 → c2cwsgiutils-6.1.8.dev112}/c2cwsgiutils/stats_pyramid/_db_spy.py

@@ -15,7 +15,7 @@ _LOG = logging.getLogger(__name__)
 _PROMETHEUS_DB_SUMMARY = prometheus_client.Summary(
     prometheus.build_metric_name("database"),
     "Database requests",
-    ["what"],
+    ["what", "engine"],
     unit="seconds",
 )
 
@@ -66,12 +66,14 @@ def _simplify_sql(sql: str) -> str:
     return re.sub(r"%\(\w+\)\w", "?", sql)
 
 
-def _create_sqlalchemy_timer_cb(what: str) -> Callable[..., Any]:
+def _create_sqlalchemy_timer_cb(what: str, engine_name: str | None = None) -> Callable[..., Any]:
     start = time.perf_counter()
 
     def after(*_args: Any, **_kwargs: Any) -> None:
-        _PROMETHEUS_DB_SUMMARY.labels({"query": what}).observe(time.perf_counter() - start)
-        _LOG.debug("Execute statement '%s' in %d.", what, time.perf_counter() - start)
+        elapsed_time = time.perf_counter() - start
+        _PROMETHEUS_DB_SUMMARY.labels(what=what, engine=engine_name).observe(elapsed_time)
+        engine_suffix = f", on {engine_name}" if engine_name else ""
+        _LOG.debug("Execute statement '%s' in %.3f%ss.", what, elapsed_time, engine_suffix)
 
     return after
 
@@ -79,8 +81,12 @@ def _create_sqlalchemy_timer_cb(what: str) -> Callable[..., Any]:
 def _before_cursor_execute(
     conn: Connection, _cursor: Any, statement: str, _parameters: Any, _context: Any, _executemany: Any
 ) -> None:
+    engine_name = getattr(conn.engine, "c2c_name", None)
     sqlalchemy.event.listen(
-        conn, "after_cursor_execute", _create_sqlalchemy_timer_cb(_simplify_sql(statement)), once=True
+        conn,
+        "after_cursor_execute",
+        _create_sqlalchemy_timer_cb(_simplify_sql(statement), engine_name),
+        once=True,
     )
 
 

{c2cwsgiutils-6.1.8.dev108 → c2cwsgiutils-6.1.8.dev112}/pyproject.toml

@@ -16,7 +16,7 @@ strict = true
 
 [tool.poetry]
 name = "c2cwsgiutils"
-version = "6.1.8.dev108"
+version = "6.1.8.dev112"
 description = "Common utilities for Camptocamp WSGI applications"
 readme = "README.md"
 authors = ["Camptocamp <info@camptocamp.com>"]