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.mqtt;
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.HashMap;
025    import java.util.Map;
026    
027    import javax.net.ServerSocketFactory;
028    import javax.net.SocketFactory;
029    import org.apache.activemq.broker.BrokerContext;
030    import org.apache.activemq.broker.BrokerService;
031    import org.apache.activemq.broker.BrokerServiceAware;
032    import org.apache.activemq.transport.MutexTransport;
033    import org.apache.activemq.transport.Transport;
034    import org.apache.activemq.transport.nio.NIOTransportFactory;
035    import org.apache.activemq.transport.tcp.TcpTransport;
036    import org.apache.activemq.transport.tcp.TcpTransportServer;
037    import org.apache.activemq.util.IntrospectionSupport;
038    import org.apache.activemq.wireformat.WireFormat;
039    
040    /**
041     * A <a href="http://mqtt.org/">MQTT</a> over NIO transport factory
042     */
043    public class MQTTNIOTransportFactory extends NIOTransportFactory implements BrokerServiceAware {
044    
045        private BrokerContext brokerContext = null;
046    
047        protected String getDefaultWireFormatType() {
048            return "mqtt";
049        }
050    
051        protected TcpTransportServer createTcpTransportServer(URI location, ServerSocketFactory serverSocketFactory) throws IOException, URISyntaxException {
052            return new TcpTransportServer(this, location, serverSocketFactory) {
053                protected Transport createTransport(Socket socket, WireFormat format) throws IOException {
054                    return new MQTTNIOTransport(format, socket);
055                }
056            };
057        }
058    
059        protected TcpTransport createTcpTransport(WireFormat wf, SocketFactory socketFactory, URI location, URI localLocation) throws UnknownHostException, IOException {
060            return new MQTTNIOTransport(wf, socketFactory, location, localLocation);
061        }
062    
063        @SuppressWarnings("rawtypes")
064        @Override
065        public Transport serverConfigure(Transport transport, WireFormat format, HashMap options) throws Exception {
066            transport = super.serverConfigure(transport, format, options);
067    
068            MutexTransport mutex = transport.narrow(MutexTransport.class);
069            if (mutex != null) {
070                mutex.setSyncOnCommand(true);
071            }
072    
073            return transport;
074        }
075    
076        @SuppressWarnings("rawtypes")
077        public Transport compositeConfigure(Transport transport, WireFormat format, Map options) {
078            transport = new MQTTTransportFilter(transport, format, brokerContext);
079            IntrospectionSupport.setProperties(transport, options);
080            return super.compositeConfigure(transport, format, options);
081        }
082    
083        public void setBrokerService(BrokerService brokerService) {
084            this.brokerContext = brokerService.getBrokerContext();
085        }
086    
087        protected Transport createInactivityMonitor(Transport transport, WireFormat format) {
088            MQTTInactivityMonitor monitor = new MQTTInactivityMonitor(transport, format);
089            MQTTTransportFilter filter = transport.narrow(MQTTTransportFilter.class);
090            filter.setInactivityMonitor(monitor);
091            return monitor;
092        }
093    
094    }
095