001    /**
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *      http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    package org.apache.activemq.transport.stomp;
018    
019    import java.io.IOException;
020    import java.net.Socket;
021    import java.net.URI;
022    import java.net.URISyntaxException;
023    import java.net.UnknownHostException;
024    import java.util.Map;
025    
026    import javax.net.ServerSocketFactory;
027    import javax.net.SocketFactory;
028    
029    import org.apache.activemq.broker.BrokerService;
030    import org.apache.activemq.broker.BrokerServiceAware;
031    import org.apache.activemq.transport.Transport;
032    import org.apache.activemq.transport.nio.NIOTransport;
033    import org.apache.activemq.transport.nio.NIOTransportFactory;
034    import org.apache.activemq.transport.tcp.TcpTransport;
035    import org.apache.activemq.transport.tcp.TcpTransportServer;
036    import org.apache.activemq.util.IntrospectionSupport;
037    import org.apache.activemq.wireformat.WireFormat;
038    import org.apache.activemq.xbean.XBeanBrokerService;
039    import org.springframework.context.ApplicationContext;
040    
041    /**
042     * A <a href="http://stomp.codehaus.org/">STOMP</a> over NIO transport factory
043     * 
044     * @version $Revision: 645574 $
045     */
046    public class StompNIOTransportFactory extends NIOTransportFactory implements BrokerServiceAware {
047    
048        private ApplicationContext applicationContext = null;
049    
050        protected String getDefaultWireFormatType() {
051            return "stomp";
052        }
053    
054        protected TcpTransportServer createTcpTransportServer(URI location, ServerSocketFactory serverSocketFactory) throws IOException, URISyntaxException {
055            return new TcpTransportServer(this, location, serverSocketFactory) {
056                protected Transport createTransport(Socket socket, WireFormat format) throws IOException {
057                    return new StompNIOTransport(format, socket);
058                }
059            };
060        }
061    
062        protected TcpTransport createTcpTransport(WireFormat wf, SocketFactory socketFactory, URI location, URI localLocation) throws UnknownHostException, IOException {
063            return new StompNIOTransport(wf, socketFactory, location, localLocation);
064        }  
065    
066        public Transport compositeConfigure(Transport transport, WireFormat format, Map options) {
067            transport = new StompTransportFilter(transport, new LegacyFrameTranslator(), applicationContext);
068            IntrospectionSupport.setProperties(transport, options);
069            return super.compositeConfigure(transport, format, options);
070        }
071    
072        protected boolean isUseInactivityMonitor(Transport transport) {
073            // lets disable the inactivity monitor as stomp does not use keep alive
074            // packets
075            return false;
076        }
077        
078        public void setBrokerService(BrokerService brokerService) {
079            if (brokerService instanceof XBeanBrokerService) {
080                this.applicationContext = ((XBeanBrokerService)brokerService).getApplicationContext();
081            }
082        }
083    
084    }
085