org.coos.messaging.processor.MessageCounter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of messaging-lib
There is a newer version: 1.3.1
/**
 * COOS - Connected Objects Operating System (www.connectedobjects.org).
 *
 * Copyright (C) 2009 Telenor ASA and Tellu AS. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This library is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 *
 * You may also contact one of the following for additional information:
 * Telenor ASA, Snaroyveien 30, N-1331 Fornebu, Norway (www.telenor.no)
 * Tellu AS, Hagalokkveien 13, N-1383 Asker, Norway (www.tellu.no)
 */
package org.coos.messaging.processor;

import java.lang.management.ManagementFactory;

import java.util.Date;
import java.util.Hashtable;

import javax.management.MBeanServer;
import javax.management.ObjectName;

import org.coos.messaging.Message;
import org.coos.messaging.ProcessorException;
import org.coos.messaging.Service;
import org.coos.messaging.impl.DefaultProcessor;
import org.coos.messaging.routing.Router;
import org.coos.messaging.routing.RouterProcessor;
import org.coos.messaging.util.LogFactory;


/**
 * Counts messages that pass
 * 
 * @author Knut Eilif Husa, Tellu AS
 */
public class MessageCounter extends DefaultProcessor implements RouterProcessor, Service, MessageCounterMBean {

    private static final String PROPERTY_LOGGER_NAME = "counterName";
    private String counterName = "MessageCounter"; // default name

    private long messages = 0;
    private long lastCheck;
    private MBeanServer mbs;
    ObjectName name;

    public MessageCounter() {
        super();

        lastCheck = new Date().getTime();

        // Get the Platform MBean Server
        mbs = ManagementFactory.getPlatformMBeanServer();

        // Construct the ObjectName for the MBean we will register
        try {
            name = new ObjectName("org.coos.messaging.processor:name=" + counterName);

        } catch (Exception e) {
            logger.error("Exception in MessageCounter", e);
        }

    }

    @SuppressWarnings("unchecked")
    public void setProperties(Hashtable properties) {
        super.setProperties(properties);
        counterName = (String) properties.get(PROPERTY_LOGGER_NAME);
    }

    public void processMessage(Message msg) throws ProcessorException {
        messages++;
    }

    public void setRouter(Router router) {
    }

    public void start() throws Exception {
        mbs.registerMBean(this, name);

    }

    public void stop() throws Exception {
        mbs.unregisterMBean(name);
    }

    public double getMessagesPerSecond() {
        long newDate = new Date().getTime();
        double mps = (1000.0 * messages) / (newDate - lastCheck);
        lastCheck = newDate;
        messages = 0;

        return mps;
    }
}