##### Coverage Report - com.sun.grizzly.RoundRobinSelectorHandler

Classes in this File | Line Coverage | Branch Coverage | Complexity
RoundRobinSelectorHandler |
65 % | 15/23
60 % | 6/10
| 0.0;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 | 21 |
             ReadController auxController = nextController();
83 | 21 |
             SelectorHandler protocolSelectorHandler = context.getSelectorHandler();
84 | 21 |
             SelectableChannel channel = protocolSelectorHandler.acceptWithoutRegistration(key);
85 |  |
86 | 21 |
             if (channel != null) {
87 | 21 |
                 protocolSelectorHandler.configureChannel(channel);
88 | 21 |
                 SelectorHandler relativeSelectorHandler =
89 |  |
                         auxController.getSelectorHandlerClone(protocolSelectorHandler);
90 |  |
91 | 21 |
                 if (relativeSelectorHandler == null) {
92 |  |
                     // Clone was not found - take correspondent protocol SelectorHandler
93 | 2 |
                     relativeSelectorHandler =
94 |  |
                             auxController.getSelectorHandler(protocolSelectorHandler.protocol());
95 |  |
                 }
96 |  |
97 | 21 |
                 auxController.addChannel(channel, relativeSelectorHandler);
98 |  |
             }
99 | 21 |
             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 | 19 |
             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 | 21 |
             return rrControllers[roundRobinCounter++ % rrControllers.length];
124 |  |
         }
125 |  |
     }