EventCounterListener.java

/*
 * Copyright (c) 2006-2011 Nuxeo SA (http://nuxeo.com/) and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Nuxeo - initial API and implementation
 *
 * $Id$
 */
package org.nuxeo.ecm.core.management.events;

import java.util.Arrays;
import java.util.List;

import org.nuxeo.ecm.core.api.ClientException;
import org.nuxeo.ecm.core.api.LifeCycleConstants;
import org.nuxeo.ecm.core.api.event.DocumentEventTypes;
import org.nuxeo.ecm.core.event.EventBundle;
import org.nuxeo.ecm.core.event.PostCommitEventListener;
import org.nuxeo.ecm.core.event.impl.ReconnectedEventBundleImpl;
import org.nuxeo.runtime.metrics.MetricsService;

import com.codahale.metrics.Counter;
import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.SharedMetricRegistries;

/**
 *
 * AsyncEventListener that collects events to update the Simon counters
 *
 * @author Tiry (tdelprat@nuxeo.com)
 *
 */
public class EventCounterListener implements PostCommitEventListener {

    // @since 5.7.2
    protected final MetricRegistry registry = SharedMetricRegistries.getOrCreate(MetricsService.class.getName());

    // Counters used
    public final Counter createCount = registry.counter(MetricRegistry.name(
            "nuxeo.events", "create"));

    public final Counter updateCount = registry.counter(MetricRegistry.name(
            "nuxeo.events", "update"));

    public final Counter removeCount = registry.counter(MetricRegistry.name(
            "nuxeo.events", "remove"));

    // Event tracked
    protected static final List<String> createEvents = Arrays.asList(new String[] {
            DocumentEventTypes.DOCUMENT_CREATED,
            DocumentEventTypes.DOCUMENT_CREATED_BY_COPY,
            DocumentEventTypes.DOCUMENT_IMPORTED,
            DocumentEventTypes.DOCUMENT_PROXY_PUBLISHED,
            DocumentEventTypes.DOCUMENT_PROXY_UPDATED });

    protected static final List<String> updateEvents = Arrays.asList(new String[] {
            DocumentEventTypes.DOCUMENT_CHECKEDIN,
            DocumentEventTypes.DOCUMENT_CHECKEDOUT,
            DocumentEventTypes.DOCUMENT_CHILDREN_ORDER_CHANGED,
            DocumentEventTypes.DOCUMENT_LOCKED,
            DocumentEventTypes.DOCUMENT_MOVED,
            DocumentEventTypes.DOCUMENT_PUBLISHED,
            DocumentEventTypes.DOCUMENT_SECURITY_UPDATED,
            DocumentEventTypes.DOCUMENT_UNLOCKED,
            DocumentEventTypes.DOCUMENT_UPDATED,
            LifeCycleConstants.TRANSITION_EVENT });

    protected static final List<String> removeEvents = Arrays.asList(new String[] {
            DocumentEventTypes.DOCUMENT_REMOVED,
            DocumentEventTypes.VERSION_REMOVED });

    @Override
    public void handleEvent(EventBundle events) throws ClientException {
        if (events instanceof ReconnectedEventBundleImpl) {
            ReconnectedEventBundleImpl bundle = (ReconnectedEventBundleImpl) events;
            updateCounters(bundle.getEventNames());
        }
    }

    protected void updateCounters(List<String> eventNames) {

        int created = 0;
        int updated = 0;
        int removed = 0;

        for (String eventName : eventNames) {
            if (createEvents.contains(eventName)) {
                created += 1;
            } else if (updateEvents.contains(eventName)) {
                updated += 1;
            } else if (removeEvents.contains(eventName)) {
                removed += 1;
            }
        }

        if (created > 0) {
            createCount.inc(created);
        }
        if (updated > 0) {
            updateCount.inc(updated);
        }
        if (removed > 0) {
            removeCount.inc(removed);
        }
    }

}