ActiveMQ example source code file (TransportLogger.java)

This example ActiveMQ source code file (TransportLogger.java) is included in the DevDaily.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" ^TM.

Java - ActiveMQ tags/keywords

futureresponse, io, ioexception, ioexception, logger, logwriter, logwriter, object, object, responsecallback, string, transportfilter, transportlogger, transportlogger, transportloggerview

The ActiveMQ TransportLogger.java source code

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF 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.apache.activemq.transport;

import java.io.IOException;

import org.slf4j.Logger;

/**
 * This TransportFilter implementation writes output to a log
 * as it intercepts commands / events before sending them to the
 * following layer in the Transport stack.
 * 
 * @author David Martin Clavo david(dot)martin(dot)clavo(at)gmail.com
 * 
 */
public class TransportLogger extends TransportFilter {

    private final Logger log;
    private boolean logging;
    private final LogWriter logWriter;
    private TransportLoggerView view;

    public TransportLogger(Transport next, Logger log, boolean startLogging, LogWriter logWriter) {
        // Changed constructor to pass the implementation of the LogWriter interface
        // that will be used to write the messages.
        super(next);
        this.log = log;
        this.logging = startLogging;
        this.logWriter = logWriter;
    }

    /**
     * Returns true if logging is activated for this TransportLogger, false otherwise.
     * @return true if logging is activated for this TransportLogger, false otherwise.
     */
    public boolean isLogging() {
        return logging;
    }

    /**
     * Sets if logging should be activated for this TransportLogger.
     * @param logging true to activate logging, false to deactivate.
     */
    public void setLogging(boolean logging) {
        this.logging = logging;
    } 

    public Object request(Object command) throws IOException {
        // Changed this method to use a LogWriter object to actually 
        // print the messages to the log, and only in case of logging 
        // being active, instead of logging the message directly.
        if (logging)
            logWriter.logRequest(log, command);
        Object rc = super.request(command);
        if (logging)
            logWriter.logResponse(log, command);
        return rc;
    }

    public Object request(Object command, int timeout) throws IOException {
        // Changed this method to use a LogWriter object to actually 
        // print the messages to the log, and only in case of logging 
        // being active, instead of logging the message directly.
        if (logging)
            logWriter.logRequest(log, command);
        Object rc = super.request(command, timeout);
        if (logging)
            logWriter.logResponse(log, command);
        return rc;
    }

    public FutureResponse asyncRequest(Object command, ResponseCallback responseCallback) throws IOException {
        // Changed this method to use a LogWriter object to actually 
        // print the messages to the log, and only in case of logging 
        // being active, instead of logging the message directly.
        if (logging)
            logWriter.logAsyncRequest(log, command);
        FutureResponse rc = next.asyncRequest(command, responseCallback);
        return rc;
    }

    public void oneway(Object command) throws IOException {
        // Changed this method to use a LogWriter object to actually 
        // print the messages to the log, and only in case of logging 
        // being active, instead of logging the message directly.
        if( logging && log.isDebugEnabled() ) {
            logWriter.logOneWay(log, command);
        }
        next.oneway(command);
    }

    public void onCommand(Object command) {
        // Changed this method to use a LogWriter object to actually 
        // print the messages to the log, and only in case of logging 
        // being active, instead of logging the message directly.
        if( logging && log.isDebugEnabled() ) {
            logWriter.logReceivedCommand(log, command);
        }
        getTransportListener().onCommand(command);
    }

    public void onException(IOException error) {
        // Changed this method to use a LogWriter object to actually 
        // print the messages to the log, and only in case of logging 
        // being active, instead of logging the message directly.
        if( logging && log.isDebugEnabled() ) {
            logWriter.logReceivedException(log, error);
        }
        getTransportListener().onException(error);
    }

    /**
     * Gets the associated MBean for this TransportLogger.
     * @return the associated MBean for this TransportLogger.
     */
    public TransportLoggerView getView() {
        return view;
    }

    /**
     * Sets the associated MBean for this TransportLogger.
     * @param view the associated MBean for this TransportLogger.
     */
    public void setView(TransportLoggerView view) {
        this.view = view;
    }


    public String toString() {
        return next.toString();
    }


    /**
     * We need to override this method
     * so that we can unregister the associated
     * MBean to avoid a memory leak.
     */
    public void finalize() throws Throwable {
        if (view != null) {
            view.unregister();    
        }
    }

}