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