/*
* Copyright 2010 Red Hat, Inc.
* Red Hat licenses this file to you under the Apache License, version
* 2.0 (the "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package org.hornetq.core.protocol.stomp.v10;
import java.io.UnsupportedEncodingException;
import java.util.Map;
import org.hornetq.core.protocol.stomp.FrameEventListener;
import org.hornetq.core.protocol.stomp.HornetQStompException;
import org.hornetq.core.protocol.stomp.Stomp;
import org.hornetq.core.protocol.stomp.StompConnection;
import org.hornetq.core.protocol.stomp.StompDecoder;
import org.hornetq.core.protocol.stomp.StompFrame;
import org.hornetq.core.protocol.stomp.VersionedStompFrameHandler;
import org.hornetq.core.server.HornetQServerLogger;
/**
*
* @author <a href="mailto:hgao@redhat.com">Howard Gao</a>
*/
public class StompFrameHandlerV10 extends VersionedStompFrameHandler implements FrameEventListener
{
public StompFrameHandlerV10(StompConnection connection)
{
super(connection);
decoder = new StompDecoder();
decoder.init();
connection.addStompEventListener(this);
}
@Override
public StompFrame onConnect(StompFrame frame)
{
StompFrame response = null;
Map<String, String> headers = frame.getHeadersMap();
String login = headers.get(Stomp.Headers.Connect.LOGIN);
String passcode = headers.get(Stomp.Headers.Connect.PASSCODE);
String clientID = headers.get(Stomp.Headers.Connect.CLIENT_ID);
String requestID = headers.get(Stomp.Headers.Connect.REQUEST_ID);
if (connection.validateUser(login, passcode))
{
connection.setClientID(clientID);
connection.setValid(true);
response = new StompFrameV10(Stomp.Responses.CONNECTED);
if (frame.hasHeader(Stomp.Headers.ACCEPT_VERSION))
{
response.addHeader(Stomp.Headers.Connected.VERSION, "1.0");
}
response.addHeader(Stomp.Headers.Connected.SESSION, connection.getID().toString());
if (requestID != null)
{
response.addHeader(Stomp.Headers.Connected.RESPONSE_ID, requestID);
}
}
else
{
//not valid
response = new StompFrameV10(Stomp.Responses.ERROR);
response.addHeader(Stomp.Headers.Error.MESSAGE, "Failed to connect");
try
{
response.setBody("The login account is not valid.");
}
catch (UnsupportedEncodingException e)
{
HornetQServerLogger.LOGGER.errorEncodingStompPacket(e);
}
}
return response;
}
@Override
public StompFrame onDisconnect(StompFrame frame)
{
return null;
}
@Override
public StompFrame onUnsubscribe(StompFrame request)
{
StompFrame response = null;
String destination = request.getHeader(Stomp.Headers.Unsubscribe.DESTINATION);
String id = request.getHeader(Stomp.Headers.Unsubscribe.ID);
String subscriptionID = null;
if (id != null)
{
subscriptionID = id;
}
else
{
if (destination == null)
{
response = new HornetQStompException("Must specify the subscription's id or " +
"the destination you are unsubscribing from").getFrame();
return response;
}
subscriptionID = "subscription/" + destination;
}
try
{
connection.unsubscribe(subscriptionID);
}
catch (HornetQStompException e)
{
return e.getFrame();
}
return response;
}
@Override
public StompFrame onAck(StompFrame request)
{
StompFrame response = null;
String messageID = request.getHeader(Stomp.Headers.Ack.MESSAGE_ID);
String txID = request.getHeader(Stomp.Headers.TRANSACTION);
if (txID != null)
{
HornetQServerLogger.LOGGER.stompTXAckNorSupported();
}
try
{
connection.acknowledge(messageID, null);
}
catch (HornetQStompException e)
{
response = e.getFrame();
}
return response;
}
@Override
public StompFrame onStomp(StompFrame request)
{
return onUnknown(request.getCommand());
}
@Override
public StompFrame onNack(StompFrame request)
{
return onUnknown(request.getCommand());
}
@Override
public StompFrame createStompFrame(String command)
{
return new StompFrameV10(command);
}
@Override
public void replySent(StompFrame reply)
{
if (reply.needsDisconnect())
{
connection.destroy();
}
}
@Override
public void requestAccepted(StompFrame request)
{
// TODO Auto-generated method stub
}
}