Classes in this File | Line Coverage | Branch Coverage | Complexity | ||||
RoundRobinSelectorHandler |
|
| 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 | } |