From c053d82883ec8e02e12e5fe16a5a82ae09e8acbf Mon Sep 17 00:00:00 2001
From: Logan Hanks <logan@reddit.com>
Date: Mon, 17 Dec 2012 13:00:28 -0800
Subject: [PATCH] Parse dsn param to get key for pg stats.

---
 r2/r2/lib/stats.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/r2/r2/lib/stats.py b/r2/r2/lib/stats.py
index ea6985e71..64d29a532 100644
--- a/r2/r2/lib/stats.py
+++ b/r2/r2/lib/stats.py
@@ -322,7 +322,9 @@ class Stats:
 
     def pg_after_cursor_execute(self, conn, cursor, statement, parameters,
                               context, executemany):
-        self.pg_event(context.engine.url.host, context.engine.url.database,
+        dsn = dict(part.split('=', 1)
+                   for part in context.engine.url.query['dsn'].split())
+        self.pg_event(dsn['host'], dsn['dbname'],
                       time.time() - context._query_start_time)
 
     def pg_event(self, db_server, db_name, service_time):