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.util.HashMap;
020    import java.util.Map;
021    
022    import org.apache.activemq.broker.BrokerContext;
023    import org.apache.activemq.broker.BrokerService;
024    import org.apache.activemq.broker.BrokerServiceAware;
025    import org.apache.activemq.transport.MutexTransport;
026    import org.apache.activemq.transport.Transport;
027    import org.apache.activemq.transport.tcp.TcpTransportFactory;
028    import org.apache.activemq.util.IntrospectionSupport;
029    import org.apache.activemq.wireformat.WireFormat;
030    
031    /**
032     * A <a href="http://stomp.codehaus.org/">STOMP</a> transport factory
033     */
034    public class StompTransportFactory extends TcpTransportFactory implements BrokerServiceAware {
035    
036        private BrokerContext brokerContext = null;
037    
038        protected String getDefaultWireFormatType() {
039            return "stomp";
040        }
041    
042        @SuppressWarnings("rawtypes")
043        public Transport compositeConfigure(Transport transport, WireFormat format, Map options) {
044            transport = new StompTransportFilter(transport, format, brokerContext);
045            IntrospectionSupport.setProperties(transport, options);
046            return super.compositeConfigure(transport, format, options);
047        }
048    
049        public void setBrokerService(BrokerService brokerService) {
050            this.brokerContext = brokerService.getBrokerContext();
051        }
052    
053        @SuppressWarnings("rawtypes")
054        @Override
055        public Transport serverConfigure(Transport transport, WireFormat format, HashMap options) throws Exception {
056            transport = super.serverConfigure(transport, format, options);
057    
058            MutexTransport mutex = transport.narrow(MutexTransport.class);
059            if (mutex != null) {
060                mutex.setSyncOnCommand(true);
061            }
062    
063            return transport;
064        }
065    
066        @Override
067        protected Transport createInactivityMonitor(Transport transport, WireFormat format) {
068            StompInactivityMonitor monitor = new StompInactivityMonitor(transport, format);
069    
070            StompTransportFilter filter = (StompTransportFilter) transport.narrow(StompTransportFilter.class);
071            filter.setInactivityMonitor(monitor);
072    
073            return monitor;
074        }
075    }