001 /**
002 * GRANITE DATA SERVICES
003 * Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004 *
005 * This file is part of the Granite Data Services Platform.
006 *
007 * Granite Data Services is free software; you can redistribute it and/or
008 * modify it under the terms of the GNU Lesser General Public
009 * License as published by the Free Software Foundation; either
010 * version 2.1 of the License, or (at your option) any later version.
011 *
012 * Granite Data Services is distributed in the hope that it will be useful,
013 * but WITHOUT ANY WARRANTY; without even the implied warranty of
014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
015 * General Public License for more details.
016 *
017 * You should have received a copy of the GNU Lesser General Public
018 * License along with this library; if not, write to the Free Software
019 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020 * USA, or see <http://www.gnu.org/licenses/>.
021 */
022 package org.granite.gravity.jbossweb;
023
024 import java.io.IOException;
025 import java.io.InputStream;
026
027 import javax.servlet.ServletException;
028 import javax.servlet.http.HttpServletRequest;
029 import javax.servlet.http.HttpServletResponse;
030
031 import org.granite.gravity.AsyncHttpContext;
032 import org.granite.gravity.Gravity;
033 import org.granite.gravity.GravityManager;
034 import org.granite.logging.Logger;
035 import org.jboss.servlet.http.HttpEvent;
036
037 import flex.messaging.messages.Message;
038
039 /**
040 * @author Franck WOLFF
041 */
042 public class GravityJBossWebServlet extends AbstractHttpEventServlet {
043
044 private static final long serialVersionUID = 1L;
045
046 private static final Logger log = Logger.getLogger(GravityJBossWebServlet.class);
047
048
049 @Override
050 public CometIO createCometIO() {
051 return new ByteArrayCometIO();
052 }
053
054 @Override
055 public boolean handleRequest(HttpEvent event, InputStream content) throws IOException, ServletException {
056
057 HttpServletRequest request = event.getHttpServletRequest();
058 HttpServletResponse response = event.getHttpServletResponse();
059
060 rejectJMFContentType(request);
061
062 Gravity gravity = GravityManager.getGravity(getServletContext());
063 JBossWebChannelFactory channelFactory = new JBossWebChannelFactory(gravity);
064
065 try {
066 initializeRequest(gravity, request, response);
067
068 Message[] amf3Requests = deserialize(gravity, request, content);
069
070 log.debug(">> [AMF3 REQUESTS] %s", (Object)amf3Requests);
071
072 Message[] amf3Responses = null;
073
074 boolean accessed = false;
075 for (int i = 0; i < amf3Requests.length; i++) {
076 Message amf3Request = amf3Requests[i];
077
078 // Ask gravity to create a specific response (will be null for connect request from tunnel).
079 Message amf3Response = gravity.handleMessage(channelFactory, amf3Request);
080 String channelId = (String)amf3Request.getClientId();
081
082 // Mark current channel (if any) as accessed.
083 if (!accessed)
084 accessed = gravity.access(channelId);
085
086 // (Re)Connect message from tunnel...
087 if (amf3Response == null) {
088 if (amf3Requests.length > 1)
089 throw new IllegalArgumentException("Only one request is allowed on tunnel.");
090
091 JBossWebChannel channel = gravity.getChannel(channelFactory, channelId);
092 if (channel == null)
093 throw new NullPointerException("No channel on tunnel connect");
094
095 // Try to send pending messages if any (using current container thread).
096 if (channel.runReceived(new AsyncHttpContext(request, response, amf3Request)))
097 return true; // Close http event.
098
099 // No pending messages, wait for new ones or timeout.
100 setConnectMessage(request, amf3Request);
101 channel.setHttpEvent(event);
102 return false; // Do not close http event.
103 }
104
105 if (amf3Responses == null)
106 amf3Responses = new Message[amf3Requests.length];
107 amf3Responses[i] = amf3Response;
108 }
109
110 log.debug("<< [AMF3 RESPONSES] %s", (Object)amf3Responses);
111
112 serialize(gravity, response, amf3Responses);
113 }
114 catch (IOException e) {
115 log.error(e, "Gravity message error");
116 throw e;
117 }
118 catch (Exception e) {
119 log.error(e, "Gravity message error");
120 throw new ServletException(e);
121 }
122 finally {
123 try {
124 if (content != null)
125 content.close();
126 }
127 finally {
128 cleanupRequest(request);
129 }
130 }
131
132 return true; // Close http event.
133 }
134
135 @Override
136 public boolean handleEnd(HttpEvent event) throws IOException, ServletException {
137 return true; // Close http event.
138 }
139
140 @Override
141 public boolean handleError(HttpEvent event) throws IOException {
142 if (EventUtil.isErrorButNotTimeout(event))
143 log.error("Got an error event: %s", EventUtil.toString(event));
144
145 try {
146 HttpServletRequest request = event.getHttpServletRequest();
147 Message connect = getConnectMessage(request);
148 if (connect != null) { // This should be a timeout.
149 Gravity gravity = GravityManager.getGravity(getServletContext());
150 String channelId = (String)connect.getClientId();
151 JBossWebChannelFactory channelFactory = new JBossWebChannelFactory(gravity);
152 JBossWebChannel channel = gravity.getChannel(channelFactory, channelId);
153
154 // Cancel channel's execution (timeout or other errors).
155 if (channel != null)
156 channel.setHttpEvent(null);
157 }
158 }
159 catch (Exception e) {
160 log.error(e, "Error while processing event: %s", EventUtil.toString(event));
161 }
162
163 return true; // Close http event.
164 }
165 }