[jboss-cvs] JBossAS SVN: r111214 - projects/jboss-jca/trunk/codegenerator/src/main/java/org/jboss/jca/codegenerator/code.

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

Author: jeff.zhang
Date: 2011-04-20 02:28:23 -0400 (Wed, 20 Apr 2011)
New Revision: 111214

Modified:
   projects/jboss-jca/trunk/codegenerator/src/main/java/org/jboss/jca/codegenerator/code/McfCodeGen.java
Log:
[JBJCA-549] Support matchConnections

Modified: projects/jboss-jca/trunk/codegenerator/src/main/java/org/jboss/jca/codegenerator/code/McfCodeGen.java
===================================================================
--- projects/jboss-jca/trunk/codegenerator/src/main/java/org/jboss/jca/codegenerator/code/McfCodeGen.java	2011-04-20 03:11:06 UTC (rev 111213)
+++ projects/jboss-jca/trunk/codegenerator/src/main/java/org/jboss/jca/codegenerator/code/McfCodeGen.java	2011-04-20 06:28:23 UTC (rev 111214)
@@ -155,6 +155,8 @@
       writeEol(out);
       out.write("import java.io.PrintWriter;");
       writeEol(out);
+      out.write("import java.util.Iterator;");
+      writeEol(out);
       out.write("import java.util.Set;");
       writeEol(out);
       writeEol(out);
@@ -386,7 +388,27 @@
       out.write("log.finest(\"matchManagedConnections()\");");
       writeEol(out);
       writeIndent(out, indent + 1);
-      out.write("return null;");
+      out.write("ManagedConnection result = null;");
+      writeEol(out);
+      writeIndent(out, indent + 1);
+      out.write("Iterator it = connectionSet.iterator();");
+      writeEol(out);
+      writeIndent(out, indent + 1);
+      out.write("while (result == null && it.hasNext())");
+      writeLeftCurlyBracket(out, indent + 1);
+      writeIndent(out, indent + 2);
+      out.write("ManagedConnection mc = (ManagedConnection)it.next();");
+      writeEol(out);
+      writeIndent(out, indent + 2);
+      out.write("if (mc instanceof " + def.getMcClass() + ")");
+      writeLeftCurlyBracket(out, indent + 2);
+      writeIndent(out, indent + 3);
+      out.write("result = mc;");
+      writeRightCurlyBracket(out, indent + 2);
+      writeRightCurlyBracket(out, indent + 1);
+      writeIndent(out, indent + 1);
+      out.write("return result;");
+
       writeRightCurlyBracket(out, indent);
       writeEol(out);
    }