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 */
017package org.apache.activemq.transport.mqtt;
018
019import java.util.HashMap;
020import java.util.Map;
021
022import org.apache.activemq.broker.BrokerContext;
023import org.apache.activemq.broker.BrokerService;
024import org.apache.activemq.broker.BrokerServiceAware;
025import org.apache.activemq.transport.MutexTransport;
026import org.apache.activemq.transport.Transport;
027import org.apache.activemq.transport.tcp.TcpTransportFactory;
028import org.apache.activemq.util.IntrospectionSupport;
029import org.apache.activemq.wireformat.WireFormat;
030
031/**
032 * A <a href="http://mqtt.org/">MQTT</a> transport factory
033 */
034public class MQTTTransportFactory extends TcpTransportFactory implements BrokerServiceAware {
035
036    private BrokerContext brokerContext = null;
037
038    protected String getDefaultWireFormatType() {
039        return "mqtt";
040    }
041
042    @SuppressWarnings("rawtypes")
043    public Transport compositeConfigure(Transport transport, WireFormat format, Map options) {
044        transport = new MQTTTransportFilter(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        MQTTInactivityMonitor monitor = new MQTTInactivityMonitor(transport, format);
069
070        MQTTTransportFilter filter = transport.narrow(MQTTTransportFilter.class);
071        filter.setInactivityMonitor(monitor);
072
073        return monitor;
074    }
075}