/* * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved. * * Licensed under the Apache License, Version 2.0 (the "License"); you * may not use this file except in compliance with the License. You * may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or * implied. See the License for the specific language governing * permissions and limitations under the License. See accompanying * LICENSE file. */ package com.gemstone.gemfire.internal.jta.dunit; import dunit.*; import java.io.*; import java.util.*; //import java.net.*; import com.gemstone.gemfire.cache.*; import com.gemstone.gemfire.distributed.*; //import java.util.Hashtable; //import javax.naming.InitialContext; import javax.naming.Context; import javax.sql.*; //import javax.transaction.*; import java.sql.*; //import java.lang.Exception.*; //import java.lang.RuntimeException; //import java.sql.SQLException.*; import javax.naming.NamingException; //import javax.naming.NoInitialContextException; //import javax.transaction.SystemException; public class MaxPoolSizeDUnitTest extends DistributedTestCase { static DistributedSystem ds; static Cache cache; private static String tblName; private static String readFile(String filename) throws IOException { // String lineSep = System.getProperty("\n"); BufferedReader br = new BufferedReader(new FileReader(filename)); String nextLine = ""; StringBuffer sb = new StringBuffer(); while ((nextLine = br.readLine()) != null) { sb.append(nextLine); // // note: // BufferedReader strips the EOL character. // // sb.append(lineSep); } getLogWriter().fine("***********\n " + sb); return sb.toString(); } public MaxPoolSizeDUnitTest(String name) { super(name); } private static String modifyFile(String str) throws IOException { String search = "<jndi-binding type=\"XAPooledDataSource\""; String last_search = "</jndi-binding>"; String newDB = "newDB_" + hydra.ProcessMgr.getProcessId(); String jndi_str = "<jndi-binding type=\"XAPooledDataSource\" jndi-name=\"XAPooledDataSource\" jdbc-driver-class=\"org.apache.derby.jdbc.EmbeddedDriver\" init-pool-size=\"1\" max-pool-size=\"2\" idle-timeout-seconds=\"600\" blocking-timeout-seconds=\"2\" login-timeout-seconds=\"1\" conn-pooled-datasource-class=\"org.apache.derby.jdbc.EmbeddedConnectionPoolDataSource\" xa-datasource-class=\"org.apache.derby.jdbc.EmbeddedXADataSource\" user-name=\"mitul\" password=\"83f0069202c571faf1ae6c42b4ad46030e4e31c17409e19a\" connection-url=\"jdbc:derby:"+newDB+";create=true\" >"; String config_prop = "<config-property>" + "<config-property-name>description</config-property-name>" + "<config-property-type>java.lang.String</config-property-type>" + "<config-property-value>hi</config-property-value>" + "</config-property>" + "<config-property>" + "<config-property-name>user</config-property-name>" + "<config-property-type>java.lang.String</config-property-type>" + "<config-property-value>mitul</config-property-value>" + "</config-property>" + "<config-property>" + "<config-property-name>password</config-property-name>" + "<config-property-type>java.lang.String</config-property-type>" + "<config-property-value>83f0069202c571faf1ae6c42b4ad46030e4e31c17409e19a</config-property-value> " + "</config-property>" + "<config-property>" + "<config-property-name>databaseName</config-property-name>" + "<config-property-type>java.lang.String</config-property-type>" + "<config-property-value>"+newDB+"</config-property-value>" + "</config-property>\n"; String new_str = jndi_str + config_prop; /* * String new_str = " <jndi-binding type=\"XAPooledDataSource\" * jndi-name=\"XAPooledDataSource\" * jdbc-driver-class=\"org.apache.derby.jdbc.EmbeddedDriver\" * init-pool-size=\"5\" max-pool-size=\"30\" idle-timeout-seconds=\"600\" * blocking-timeout-seconds=\"60\" login-timeout-seconds=\"25\" * conn-pooled-datasource-class=\"org.apache.derby.jdbc.EmbeddedConnectionPoolDataSource\" * xa-datasource-class=\"org.apache.derby.jdbc.EmbeddedXADataSource\" * user-name=\"mitul\" * password=\"83f0069202c571faf1ae6c42b4ad46030e4e31c17409e19a\" * connection-url=\"jdbc:derby:"+newDB+";create=true\" > <property * key=\"description\" value=\"hi\"/> <property key=\"databaseName\" * value=\""+newDB+"\"/> <property key=\"user\" value=\"mitul\"/> <property * key=\"password\" * value=\"83f0069202c571faf1ae6c42b4ad46030e4e31c17409e19a\"/>"; */ int n1 = str.indexOf(search); getLogWriter().fine("Start Index = " + n1); int n2 = str.indexOf(last_search, n1); StringBuffer sbuff = new StringBuffer(str); getLogWriter().fine("END Index = " + n2); String modified_str = sbuff.replace(n1, n2, new_str).toString(); return modified_str; } public static String init(String className) throws Exception { getLogWriter().fine("PATH11 "); Properties props = new Properties(); String jtest = System.getProperty("JTESTS"); int pid = hydra.ProcessMgr.getProcessId(); String path = System.getProperty("JTESTS") + "/lib/dunit-cachejta_" + pid + ".xml"; getLogWriter().fine("PATH " + path); /** * Return file as string and then modify the string accordingly ** */ String file_as_str = readFile(jtest + "/lib/cachejta.xml"); file_as_str = file_as_str.replaceAll("newDB", "newDB_" + pid); String modified_file_str = modifyFile(file_as_str); FileOutputStream fos = new FileOutputStream(path); BufferedWriter wr = new BufferedWriter(new OutputStreamWriter(fos)); wr.write(modified_file_str); wr.flush(); wr.close(); props.setProperty("cache-xml-file", path); String tableName = ""; // props.setProperty("mcast-port", "10339"); try { // ds = DistributedSystem.connect(props); ds = (new MaxPoolSizeDUnitTest("temp")).getSystem(props); cache = CacheFactory.create(ds); if (className != null && !className.equals("")) { String time = new Long(System.currentTimeMillis()).toString(); tableName = className + time; createTable(tableName); } } catch (Exception e) { e.printStackTrace(System.err); throw new Exception("" + e); } tblName = tableName; return tableName; } public static void createTable(String tableName) throws NamingException, SQLException { Context ctx = cache.getJNDIContext(); DataSource ds = (DataSource) ctx.lookup("java:/SimpleDataSource"); //String sql = "create table " + tableName + " (id number primary key, name // varchar2(50))"; //String sql = "create table " + tableName + " (id integer primary key, // name varchar(50))"; String sql = "create table " + tableName + " (id integer NOT NULL, name varchar(50), CONSTRAINT the_key PRIMARY KEY(id))"; getLogWriter().fine(sql); Connection conn = ds.getConnection(); Statement sm = conn.createStatement(); sm.execute(sql); sm.close(); sm = conn.createStatement(); for (int i = 1; i <= 10; i++) { sql = "insert into " + tableName + " values (" + i + ",'name" + i + "')"; sm.addBatch(sql); getLogWriter().fine(sql); } sm.executeBatch(); conn.close(); } public static void destroyTable() throws NamingException, SQLException { try { String tableName = tblName; Context ctx = cache.getJNDIContext(); DataSource ds = (DataSource) ctx.lookup("java:/SimpleDataSource"); Connection conn = ds.getConnection(); getLogWriter().fine(" trying to drop table: " + tableName); String sql = "drop table " + tableName; Statement sm = conn.createStatement(); sm.execute(sql); conn.close(); } catch (NamingException ne) { getLogWriter().fine("destroy table naming exception: " + ne); throw ne; } catch (SQLException se) { getLogWriter().fine("destroy table sql exception: " + se); throw se; } closeCache(); } public static Cache getCache() { return cache; } public static void startCache() { try { if (cache.isClosed()) { cache = CacheFactory.create(ds); } } catch (Exception e) { e.printStackTrace(); } } public static void closeCache() { try { if (!cache.isClosed()) { cache.close(); } } catch (Exception e) { e.printStackTrace(); } try { ds.disconnect(); } catch (Exception e) { getLogWriter().fine("Error in disconnecting from Distributed System"); } } @Override public void setUp() throws Exception { Host host = Host.getHost(0); VM vm0 = host.getVM(0); Object o[] = new Object[1]; o[0] = "MaxPoolSizeDUnitTest"; vm0.invoke(MaxPoolSizeDUnitTest.class, "init", o); } @Override public void tearDown2() throws NamingException, SQLException { Host host = Host.getHost(0); VM vm0 = host.getVM(0); vm0.invoke(MaxPoolSizeDUnitTest.class, "destroyTable"); } public static void testMaxPoolSize() throws Throwable{ Host host = Host.getHost(0); VM vm0 = host.getVM(0); AsyncInvocation asyncObj = vm0.invokeAsync(MaxPoolSizeDUnitTest.class, "runTest1"); DistributedTestCase.join(asyncObj, 30 * 1000, getLogWriter()); if(asyncObj.exceptionOccurred()){ fail("asyncObj failed", asyncObj.getException()); } } public static void runTest1() throws Exception { final int MAX_CONNECTIONS = 3; int count = 0; DataSource ds = null; try { Context ctx = cache.getJNDIContext(); ds = (DataSource) ctx.lookup("java:/XAPooledDataSource"); } catch (NamingException e) { getLogWriter().fine("Naming Exception caught in lookup: " + e); fail("failed in naming lookup: " + e); return; } catch (Exception e) { getLogWriter().fine("Exception caught during naming lookup: " + e); fail("failed in naming lookup: " + e); return; } try { for (count = 0; count < MAX_CONNECTIONS; count++) { ds.getConnection(); getLogWriter().fine("Thread 1 acquired connection #" + count); } fail("expected max connect exception"); } catch (SQLException e) { if (count < (MAX_CONNECTIONS-1)) { fail("runTest1 SQL Exception caught: " + e); } else { getLogWriter().fine("Success SQLException caught at connection #" + count); } } catch (Exception e) { getLogWriter().fine("Exception caught in runTest1: " + e); fail("Exception caught in runTest1: " + e); e.printStackTrace(); } } }