Coverage Report - com.sun.grizzly.RoundRobinSelectorHandler
 
Classes in this File Line Coverage Branch Coverage Complexity
RoundRobinSelectorHandler
65 %
15/23
60 %
6/10
0
 
 1  
 /*
 2  
  * 
 3  
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 4  
  * 
 5  
  * Copyright 2007-2008 Sun Microsystems, Inc. All rights reserved.
 6  
  * 
 7  
  * The contents of this file are subject to the terms of either the GNU
 8  
  * General Public License Version 2 only ("GPL") or the Common Development
 9  
  * and Distribution License("CDDL") (collectively, the "License").  You
 10  
  * may not use this file except in compliance with the License. You can obtain
 11  
  * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
 12  
  * or glassfish/bootstrap/legal/LICENSE.txt.  See the License for the specific
 13  
  * language governing permissions and limitations under the License.
 14  
  * 
 15  
  * When distributing the software, include this License Header Notice in each
 16  
  * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
 17  
  * Sun designates this particular file as subject to the "Classpath" exception
 18  
  * as provided by Sun in the GPL Version 2 section of the License file that
 19  
  * accompanied this code.  If applicable, add the following below the License
 20  
  * Header, with the fields enclosed by brackets [] replaced by your own
 21  
  * identifying information: "Portions Copyrighted [year]
 22  
  * [name of copyright owner]"
 23  
  * 
 24  
  * Contributor(s):
 25  
  * 
 26  
  * If you wish your version of this file to be governed by only the CDDL or
 27  
  * only the GPL Version 2, indicate your decision by adding "[Contributor]
 28  
  * elects to include this software in this distribution under the [CDDL or GPL
 29  
  * Version 2] license."  If you don't indicate a single choice of license, a
 30  
  * recipient has the option to distribute your version of this file under
 31  
  * either the CDDL, the GPL Version 2 or to extend the choice of license to
 32  
  * its licensees as provided above.  However, if you add GPL Version 2 code
 33  
  * and therefore, elected the GPL Version 2 license, then the option applies
 34  
  * only if the new code is made subject to such option by the copyright
 35  
  * holder.
 36  
  *
 37  
  */
 38  
 
 39  
 package com.sun.grizzly;
 40  
 
 41  
 import com.sun.grizzly.Controller.Protocol;
 42  
 import com.sun.grizzly.util.Copyable;
 43  
 import java.io.IOException;
 44  
 import java.nio.channels.SelectableChannel;
 45  
 import java.nio.channels.SelectionKey;
 46  
 import java.util.Set;
 47  
 
 48  
 /**
 49  
  * A SelectorHandler handles all java.nio.channels.Selector operations.
 50  
  * One or more instance of a Selector are handled by SelectorHandler.
 51  
  * The logic for processing of SelectionKey interest (OP_ACCEPT,OP_READ, etc.)
 52  
  * is usually defined using an instance of SelectorHandler.
 53  
  *
 54  
  * This class represents a TCP implementation of a SelectorHandler,
 55  
  * which handles "accept" events by registering newly accepted connections
 56  
  * to auxiliary {@link Controller}s in a round robin fashion.
 57  
  *
 58  
  * @author Alexey Stashok
 59  
  */
 60  
 public class RoundRobinSelectorHandler extends TCPSelectorHandler
 61  
         implements ComplexSelectorHandler {
 62  
     private ReadController[] rrControllers;
 63  
     private int roundRobinCounter;
 64  
     private Set<Protocol> customProtocols;
 65  
     
 66  0
     public RoundRobinSelectorHandler() {}
 67  
     
 68  7
     public RoundRobinSelectorHandler(ReadController[] rrControllers) {
 69  7
         this.rrControllers = rrControllers;
 70  7
     }
 71  
     
 72  
     @Override
 73  
     public void copyTo(Copyable copy) {
 74  0
         super.copyTo(copy);
 75  0
         RoundRobinSelectorHandler copyHandler = (RoundRobinSelectorHandler) copy;
 76  0
         copyHandler.roundRobinCounter = roundRobinCounter;
 77  0
         copyHandler.rrControllers = rrControllers;
 78  0
     }
 79  
     
 80  
     @Override
 81  
     public boolean onAcceptInterest(SelectionKey key, Context context) throws IOException {
 82  30
         ReadController auxController = nextController();
 83  30
         SelectorHandler protocolSelectorHandler = context.getSelectorHandler();
 84  30
         SelectableChannel channel = protocolSelectorHandler.acceptWithoutRegistration(key);
 85  
         
 86  30
         if (channel != null) {
 87  30
             protocolSelectorHandler.configureChannel(channel);
 88  30
             SelectorHandler relativeSelectorHandler = 
 89  
                     auxController.getSelectorHandlerClone(protocolSelectorHandler);
 90  
             
 91  30
             if (relativeSelectorHandler == null) {
 92  
                 // Clone was not found - take correspondent protocol SelectorHandler
 93  2
                 relativeSelectorHandler = 
 94  
                         auxController.getSelectorHandler(protocolSelectorHandler.protocol());
 95  
             }
 96  
             
 97  30
             auxController.addChannel(channel, relativeSelectorHandler);
 98  
         }
 99  30
         return false;
 100  
     }
 101  
     
 102  
     /**
 103  
      * Add custom protocol support
 104  
      * @param customProtocol custom {@link Controller.Protocol}
 105  
      */
 106  
     public void addProtocolSupport(Protocol customProtocol) {
 107  0
         customProtocols.add(customProtocol);
 108  0
     }
 109  
     
 110  
     /**
 111  
      * {@inheritDoc}
 112  
      */
 113  
     public boolean supportsProtocol(Protocol protocol) {
 114  28
         return protocol == Protocol.TCP || protocol == Protocol.TLS ||
 115  
                 customProtocols.contains(protocol);
 116  
     }
 117  
     
 118  
     /**
 119  
      * Return next aux. ReadController to process an accepted connection
 120  
      * @return{@link ReadController}
 121  
      */
 122  
     private ReadController nextController() {
 123  30
         return rrControllers[roundRobinCounter++ % rrControllers.length];
 124  
     }
 125  
 }