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.generic;
023
024 import java.io.IOException;
025
026 import javax.servlet.ServletException;
027 import javax.servlet.http.HttpServletRequest;
028 import javax.servlet.http.HttpServletResponse;
029
030 import org.granite.gravity.AbstractGravityServlet;
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.AsyncMessage;
037 import flex.messaging.messages.Message;
038
039 /**
040 * @author William DRAI
041 */
042 public class GravityGenericServlet extends AbstractGravityServlet {
043
044 private static final long serialVersionUID = 1L;
045
046 private static final Logger log = Logger.getLogger(GravityGenericServlet.class);
047
048
049 @Override
050 protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
051
052 log.debug("doPost: from %s:%d", request.getRemoteAddr(), request.getRemotePort());
053
054 Gravity gravity = GravityManager.getGravity(getServletContext());
055 GenericChannelFactory channelFactory = new GenericChannelFactory(gravity);
056
057 try {
058 // Setup context (thread local GraniteContext, etc.)
059 initializeRequest(gravity, request, response);
060
061 AsyncMessage connect = getConnectMessage(request);
062
063 // Resumed request (new received messages or timeout).
064 if (connect != null) {
065 try {
066 String channelId = (String)connect.getClientId();
067 GenericChannel channel = gravity.getChannel(channelFactory, channelId);
068 // Reset channel continuation instance and deliver pending messages.
069 synchronized (channel) {
070 channel.close();
071 channel.runReceived(new AsyncHttpContext(request, response, connect));
072 }
073 }
074 finally {
075 removeConnectMessage(request);
076 }
077 return;
078 }
079
080 // New Request.
081 Message[] amf3Requests = deserialize(gravity, request);
082
083 log.debug(">> [AMF3 REQUESTS] %s", (Object)amf3Requests);
084
085 Message[] amf3Responses = null;
086
087 boolean accessed = false;
088 for (int i = 0; i < amf3Requests.length; i++) {
089 Message amf3Request = amf3Requests[i];
090
091 // Ask gravity to create a specific response (will be null for connect request from tunnel).
092 Message amf3Response = gravity.handleMessage(channelFactory, amf3Request);
093 String channelId = (String)amf3Request.getClientId();
094
095 // Mark current channel (if any) as accessed.
096 if (!accessed)
097 accessed = gravity.access(channelId);
098
099 // (Re)Connect message from tunnel.
100 if (amf3Response == null) {
101 if (amf3Requests.length > 1)
102 throw new IllegalArgumentException("Only one request is allowed on tunnel.");
103
104 GenericChannel channel = gravity.getChannel(channelFactory, channelId);
105 if (channel == null)
106 throw new NullPointerException("No channel on tunnel connect");
107
108 // Try to send pending messages if any (using current container thread).
109 if (!channel.runReceived(new AsyncHttpContext(request, response, amf3Request))) {
110 // No pending messages, wait for new ones or timeout.
111 setConnectMessage(request, amf3Request);
112 synchronized (channel) {
113 WaitingContinuation continuation = new WaitingContinuation(channel);
114 channel.setContinuation(continuation);
115 continuation.suspend(getLongPollingTimeout());
116 }
117 }
118
119 return;
120 }
121
122 if (amf3Responses == null)
123 amf3Responses = new Message[amf3Requests.length];
124 amf3Responses[i] = amf3Response;
125 }
126
127 log.debug("<< [AMF3 RESPONSES] %s", (Object)amf3Responses);
128
129 serialize(gravity, response, amf3Responses);
130 }
131 catch (IOException e) {
132 log.error(e, "Gravity message error");
133 throw e;
134 }
135 catch (ClassNotFoundException e) {
136 log.error(e, "Gravity message error");
137 throw new ServletException("Gravity message error", e);
138 }
139 finally {
140 // Cleanup context (thread local GraniteContext, etc.)
141 cleanupRequest(request);
142 }
143
144 removeConnectMessage(request);
145 }
146 }