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