1 /* 2 * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* 25 * @test 26 * @bug 7009998 27 * @summary Tests the correct processing of concurrent ClientComunicatorAdmin reconnect requests. 28 * @author Jaroslav Bachorik 29 * @run clean CCAdminReconnectTest 30 * @run build CCAdminReconnectTest 31 * @run main CCAdminReconnectTest 32 */ 33 34 import com.sun.jmx.remote.internal.ClientCommunicatorAdmin; 35 import java.io.*; 36 import java.util.Collection; 37 import java.util.LinkedList; 38 import java.util.concurrent.ExecutorService; 39 import java.util.concurrent.Executors; 40 import java.util.concurrent.TimeUnit; 41 import java.util.concurrent.atomic.AtomicBoolean; 42 43 public class CCAdminReconnectTest { 44 final private static int THREAD_COUNT = 3; 45 46 public static void main(String ... args) throws Exception { 47 final ExecutorService e = Executors.newFixedThreadPool(THREAD_COUNT); 48 final AtomicBoolean beingReconnected = new AtomicBoolean(); 49 final Collection<Exception> thrownExceptions = new LinkedList<>(); 50 51 System.out.println(": Testing concurrent restart of ClientCommunicatorAdmin"); 52 53 final ClientCommunicatorAdmin cca = new ClientCommunicatorAdmin(50) { 54 55 @Override 56 protected void checkConnection() throws IOException { 57 // empty 58 } 59 60 @Override 61 protected void doStart() throws IOException { 62 if (!beingReconnected.compareAndSet(false, true)) { 63 IOException e = new IOException("Detected overlayed reconnect requests"); 64 thrownExceptions.add(e); 65 throw e; 66 } 67 try { 68 Thread.sleep(800); // simulating a workload 69 beingReconnected.set(false); 70 } catch (InterruptedException e) { 71 } 72 } 73 74 @Override 75 protected void doStop() { 76 // empty 77 } 78 }; 79 80 Runnable r = new Runnable() { 81 final private IOException e = new IOException("Forced reconnect"); 82 @Override 83 public void run() { 84 try { 85 // forcing the reconnect request 86 cca.gotIOException(e); 87 } catch (Exception ex) { 88 ex.printStackTrace(); 89 } 90 } 91 }; 92 93 System.out.println(": Spawning " + THREAD_COUNT + " concurrent reconnect requests"); 94 95 for(int i=0;i<THREAD_COUNT;i++) { 96 e.execute(r); 97 } 98 99 Thread.sleep(THREAD_COUNT * 1000); 100 e.shutdown(); 101 e.awaitTermination(10, TimeUnit.SECONDS); 102 103 cca.terminate(); 104 105 for(Exception thrown : thrownExceptions) { 106 throw thrown; 107 } 108 System.out.println(": Requests processed successfully"); 109 } 110 } 111