[jboss-cvs] JBossAS SVN: r112558 - projects/jboss-jca/branches/Branch_1_0/core/src/main/java/org/jboss/jca/core/recovery.

[jboss-cvs-commits at lists.jboss.org]

Author: jesper.pedersen
Date: 2012-01-06 13:26:34 -0500 (Fri, 06 Jan 2012)
New Revision: 112558

Added:
   projects/jboss-jca/branches/Branch_1_0/core/src/main/java/org/jboss/jca/core/recovery/ConfigurableRecoveryPlugin.java
Log:
[JBJCA-724] Configurable recovery plugin

Added: projects/jboss-jca/branches/Branch_1_0/core/src/main/java/org/jboss/jca/core/recovery/ConfigurableRecoveryPlugin.java
===================================================================
--- projects/jboss-jca/branches/Branch_1_0/core/src/main/java/org/jboss/jca/core/recovery/ConfigurableRecoveryPlugin.java	                        (rev 0)
+++ projects/jboss-jca/branches/Branch_1_0/core/src/main/java/org/jboss/jca/core/recovery/ConfigurableRecoveryPlugin.java	2012-01-06 18:26:34 UTC (rev 112558)
@@ -0,0 +1,191 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2011, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.jca.core.recovery;
+
+import org.jboss.jca.core.CoreBundle;
+import org.jboss.jca.core.CoreLogger;
+import org.jboss.jca.core.spi.recovery.RecoveryPlugin;
+
+import java.lang.reflect.Method;
+
+import javax.resource.ResourceException;
+
+import org.jboss.logging.Logger;
+import org.jboss.logging.Messages;
+
+/**
+ * Configurable implementation of a recovery plugin.
+ *
+ * @author <a href="jesper.pedersen at jboss.org">Jesper Pedersen</a>
+ */
+public class ConfigurableRecoveryPlugin implements RecoveryPlugin
+{
+   /** Log instance */
+   private static CoreLogger log = Logger.getMessageLogger(CoreLogger.class, DefaultRecoveryPlugin.class.getName());
+
+   /** The bundle */
+   private static CoreBundle bundle = Messages.getBundle(CoreBundle.class);
+   
+   /** Enable isValid */
+   private boolean enableIsValid;
+
+   /** isValid override */
+   private boolean isValidOverride;
+
+   /* isValid value */
+   private int isValidValue;
+
+   /** Enable close */
+   private boolean enableClose;
+
+   /* Close method */
+   private String closeMethod;
+
+   /**
+    * Constructor
+    */
+   public ConfigurableRecoveryPlugin()
+   {
+      this.enableIsValid = true;
+      this.isValidValue = 5;
+      this.isValidOverride = false;
+      this.enableClose = true;
+      this.closeMethod = "close";
+   }
+
+   /**
+    * Enable isValid method call
+    * @param v The value
+    */
+   public void setEnableIsValid(boolean v)
+   {
+      enableIsValid = v;
+   }
+
+   /**
+    * Set isValid method parameter
+    * @param v The value
+    */
+   public void setIsValidValue(int v)
+   {
+      isValidValue = v;
+   }
+
+   /**
+    * Override value for isValid method call if not enabled
+    * @param v The value
+    */
+   public void setIsValidOverride(boolean v)
+   {
+      isValidOverride = v;
+   }
+
+   /**
+    * Enable close method call
+    * @param v The value
+    */
+   public void setEnableClose(boolean v)
+   {
+      enableClose = v;
+   }
+
+   /**
+    * Set close method name
+    * @param v The value
+    */
+   public void setCloseMethod(String v)
+   {
+      closeMethod = v;
+   }
+
+   /**
+    * {@inheritDoc}
+    */
+   @Override
+   public boolean isValid(Object c) throws ResourceException
+   {
+      if (enableIsValid)
+      {
+         if (c != null)
+         {
+            try
+            {
+               Method method = c.getClass().getMethod("isValid", new Class[] {int.class});
+               SecurityActions.setAccessible(method, true);
+               Boolean b = (Boolean)method.invoke(c, new Object[] {Integer.valueOf(isValidValue)});
+               return b.booleanValue();
+            }
+            catch (Throwable t)
+            {
+               log.debugf("No isValid(int) method defined on connection interface (%s)", c.getClass().getName());
+            }
+         }
+      }
+      else
+      {
+         return isValidOverride;
+      }
+
+      return false;
+   }
+
+   /**
+    * {@inheritDoc}
+    */
+   @Override
+   public void close(Object c) throws ResourceException
+   {
+      if (enableClose)
+      {
+         if (c != null)
+         {
+            if (c instanceof javax.resource.cci.Connection)
+            {
+               try
+               {
+                  javax.resource.cci.Connection cci = (javax.resource.cci.Connection)c;
+                  cci.close();
+               }
+               catch (ResourceException re)
+               {
+                  log.exceptionDuringConnectionClose(re);
+                  throw new ResourceException(bundle.errorDuringConnectionClose(), re);
+               }
+            }
+            else
+            {
+               try
+               {
+                  Method method = c.getClass().getMethod(closeMethod, (Class<?>)null);
+                  SecurityActions.setAccessible(method, true);
+                  method.invoke(c, (Object)null);
+               }
+               catch (Throwable t)
+               {
+                  log.debug("Error during connection " + closeMethod + "()", t);
+                  throw new ResourceException(bundle.errorDuringConnectionClose(), t);
+               }
+            }
+         }
+      }
+   }
+}