Classes in this File | Line Coverage | Branch Coverage | Complexity | ||||
UDPWriteFilter |
|
| 2.5;2,5 |
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.filter; | |
40 | ||
41 | import com.sun.grizzly.Context; | |
42 | import com.sun.grizzly.Controller; | |
43 | import com.sun.grizzly.ProtocolFilter; | |
44 | import com.sun.grizzly.util.OutputWriter; | |
45 | import com.sun.grizzly.util.WorkerThread; | |
46 | import java.io.IOException; | |
47 | import java.net.SocketAddress; | |
48 | import java.nio.ByteBuffer; | |
49 | import java.nio.channels.DatagramChannel; | |
50 | import java.nio.channels.SelectionKey; | |
51 | import java.util.logging.Level; | |
52 | ||
53 | /** | |
54 | * Simple ProtocolFilter implementation which write the available bytes | |
55 | * and delegate the processing to the next ProtocolFilter in the ProtocolChain. | |
56 | * If no bytes are available for write, no new ProtocolHandler will be a invoked and | |
57 | * the connection (SelectionKey) will be cancelled. | |
58 | * | |
59 | * @author Jeanfrancois Arcand | |
60 | */ | |
61 | public class UDPWriteFilter implements ProtocolFilter{ | |
62 | ||
63 | public final static String UDP_SOCKETADDRESS = "socketAddress"; | |
64 | ||
65 | 0 | public UDPWriteFilter(){ |
66 | 0 | } |
67 | ||
68 | ||
69 | /** | |
70 | * Write available bytes and delegate the processing of them to the next | |
71 | * ProtocolFilter in the ProtocolChain. | |
72 | * @return <tt>true</tt> if the next ProtocolFilter on the ProtocolChain | |
73 | * need to bve invoked. | |
74 | */ | |
75 | public boolean execute(Context ctx) throws IOException { | |
76 | 0 | boolean result = true; |
77 | 0 | SocketAddress socketAddress = null; |
78 | 0 | DatagramChannel datagramChannel = null; |
79 | 0 | Exception exception = null; |
80 | 0 | SelectionKey key = ctx.getSelectionKey(); |
81 | 0 | key.attach(null); |
82 | ||
83 | 0 | ByteBuffer byteBuffer = |
84 | ((WorkerThread)Thread.currentThread()).getByteBuffer(); | |
85 | try { | |
86 | 0 | socketAddress = (SocketAddress)ctx.getAttribute(UDP_SOCKETADDRESS); |
87 | ||
88 | 0 | if (socketAddress == null){ |
89 | 0 | throw new IllegalStateException("socketAddress cannot be null"); |
90 | } | |
91 | ||
92 | 0 | datagramChannel = (DatagramChannel)key.channel(); |
93 | 0 | OutputWriter.flushChannel(datagramChannel,socketAddress,byteBuffer); |
94 | 0 | } catch (IOException ex) { |
95 | 0 | exception = ex; |
96 | 0 | log("UDPWriteFilter.execute",ex); |
97 | 0 | } catch (RuntimeException ex) { |
98 | 0 | exception = ex; |
99 | 0 | log("UDPWriteFilter.execute",ex); |
100 | } finally { | |
101 | 0 | if (exception != null){ |
102 | 0 | ctx.setAttribute(Context.THROWABLE,exception); |
103 | 0 | result = false; |
104 | } else { | |
105 | 0 | ctx.setKeyRegistrationState( |
106 | Context.KeyRegistrationState.REGISTER); | |
107 | } | |
108 | 0 | } |
109 | 0 | return result; |
110 | } | |
111 | ||
112 | ||
113 | /** | |
114 | * Re-enable the SelectionKey and make sure the ByteBuffer has been cleared. | |
115 | * @return <tt>true</tt> if the previous ProtocolFilter postExecute method | |
116 | * needs to be invoked. | |
117 | */ | |
118 | public boolean postExecute(Context ctx) throws IOException { | |
119 | 0 | ctx.setKeyRegistrationState(Context.KeyRegistrationState.REGISTER); |
120 | 0 | ((WorkerThread)Thread.currentThread()).getByteBuffer().clear(); |
121 | 0 | return true; |
122 | } | |
123 | ||
124 | ||
125 | /** | |
126 | * Log a message/exception. | |
127 | * @param msg <code>String</code> | |
128 | * @param t <code>Throwable</code> | |
129 | */ | |
130 | protected void log(String msg,Throwable t){ | |
131 | 0 | if (Controller.logger().isLoggable(Level.FINE)){ |
132 | 0 | Controller.logger().log(Level.FINE,"Write,execute()",t); |
133 | } | |
134 | 0 | } |
135 | } |