From f500a01511e7d8a92d88cbcbfe06835b60fcd2fe Mon Sep 17 00:00:00 2001
From: Alex Huang <alex@cloud.com>
Date: Tue, 19 Jul 2011 09:26:05 -0700
Subject: [PATCH] more db connection controls through jmx

---
 .../com/cloud/utils/db/ConnectionConcierge.java    | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/utils/src/com/cloud/utils/db/ConnectionConcierge.java b/utils/src/com/cloud/utils/db/ConnectionConcierge.java
index 05755321750..08b8336b76f 100644
--- a/utils/src/com/cloud/utils/db/ConnectionConcierge.java
+++ b/utils/src/com/cloud/utils/db/ConnectionConcierge.java
@@ -57,11 +57,19 @@ public class ConnectionConcierge {
     String _name;
     boolean _keepAlive;
     boolean _autoCommit;
+    int _isolationLevel;
+    int _holdability;
     
     public ConnectionConcierge(String name, Connection conn, boolean autoCommit, boolean keepAlive) {
         _name = name + s_mgr.getNextId();
         _keepAlive = keepAlive;
-        _autoCommit = autoCommit;
+        try {
+            _autoCommit = conn.getAutoCommit();
+            _isolationLevel = conn.getTransactionIsolation();
+            _holdability = conn.getHoldability();
+        } catch (SQLException e) {
+            throw new CloudRuntimeException("Unable to get information from the connection object", e);
+        }
         reset(conn);
     }
     
@@ -74,6 +82,8 @@ public class ConnectionConcierge {
         _conn = conn;
         try {
             _conn.setAutoCommit(_autoCommit);
+            _conn.setHoldability(_holdability);
+            _conn.setTransactionIsolation(_isolationLevel);
         } catch (SQLException e) {
             s_logger.error("Unable to release a connection", e);
         }
@@ -117,7 +127,7 @@ public class ConnectionConcierge {
             super(ConnectionConciergeMBean.class, false);
             resetKeepAliveTask(20);
             try {
-                JmxUtil.registerMBean("ConnectionConciergeManager", "ConnectionConciergeManager", this);
+                JmxUtil.registerMBean("DB Connections", "DB Connections", this);
             } catch (Exception e) {
                 s_logger.error("Unable to register mbean", e);
             }