com.mantledillusion.metrics.trail.MetricsReceiver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of trail-metrics-adaptor-web Show documentation
XML based (REST/SOAP) JAXB web adaptor for the trail-metrics-api.
There is a newer version: 2.2.2
Show newest version
package com.mantledillusion.metrics.trail;

import com.mantledillusion.metrics.trail.api.Metric;
import com.mantledillusion.metrics.trail.api.MetricAttribute;
import com.mantledillusion.metrics.trail.api.MetricType;
import com.mantledillusion.metrics.trail.api.web.WebMetric;
import com.mantledillusion.metrics.trail.api.web.WebMetricRequest;

import java.util.*;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

/**
 * A receiver for {@link WebMetricRequest}s that is able to deliver them to registered {@link MetricsConsumer}s.
 */
public class MetricsReceiver {

    private class MatchedMetricsTrailConsumer {

        private final String matcher;
        private final MetricsTrailConsumer consumer;

        private MatchedMetricsTrailConsumer(String matcher, MetricsTrailConsumer consumer) {
            this.matcher = matcher;
            this.consumer = consumer;
        }

        private MetricsTrailConsumer.MetricsTrailConsumerQueue queueFor(UUID trailId) {
            return this.consumer.queueFor(trailId);
        }
    }

    private final Set consumers = Collections.newSetFromMap(new IdentityHashMap<>());

    /**
     * Adds a {@link MetricsConsumer} to this {@link MetricsReceiver} that will
     * receive all events the receiver gets aware of.
     *
     * @param senderConsumerMatcher An as {@link Pattern} parsable {@link String}
     *                              matcher for the consumer ID the
     *                              {@link MetricsSender} is registered under on the
     *                              client side; might not be null.
     * @param consumerId            The id to add the consumer under, which will be
     *                              delivered to the consumer on each
     *                              {@link MetricsConsumer#consume(String, UUID, Metric)}
     *                              invocation. Allows the same consumer to be
     *                              registered multiple times with differing
     *                              configurations; might not be null.
     * @param consumer              The consumer to add; might not be null.
     * @return The created {@link MetricsTrailConsumer} that can be used to configure the consumer, never null
     */
    public MetricsTrailConsumer addConsumer(String senderConsumerMatcher, String consumerId, MetricsConsumer consumer) {
        return addConsumer(senderConsumerMatcher, consumerId, consumer, null, null);
    }

    /**
     * Adds a {@link MetricsConsumer} to this {@link MetricsReceiver} that will
     * receive all events the receiver gets aware of.
     *
     * @param senderConsumerMatcher An as {@link Pattern} parsable {@link String}
     *                              matcher for the consumer ID the
     *                              {@link MetricsSender} is registered under on the
     *                              client side; might not be null.
     * @param consumerId            The unique id to add the consumer under, which
     *                              will be delivered to the consumer on each
     *                              {@link MetricsConsumer#consume(String, UUID, Metric)}
     *                              invocation. Allows the same consumer to be
     *                              registered multiple times with differing
     *                              configurations; might not be null.
     * @param consumer              The consumer to add; might not be null.
     * @param gate                  The predicate that needs to
     *                              {@link MetricsPredicate#test(Metric)} true
     *                              to trigger flushing all of a session's
     *                              accumulated {@link Metric}s; might be null.
     * @param filter                The predicate that needs to
     *                              {@link MetricsPredicate#test(Metric)} true
     *                              to allow an about-to-be-flushed event to be
     *                              delivered to the consumer; might be null.
     * @return The created {@link MetricsTrailConsumer} that can be used to configure the consumer, never null
     */
    public synchronized MetricsTrailConsumer addConsumer(String senderConsumerMatcher, String consumerId,
                                                  MetricsConsumer consumer, MetricsPredicate gate, MetricsPredicate filter) {
        if (senderConsumerMatcher == null) {
            throw new IllegalArgumentException("Cannot add a consumer using a null sender consumer matcher");
        }
        MetricsTrailConsumer trailConsumer = MetricsTrailConsumer.from(consumerId, consumer, gate, filter);
        this.consumers.add(new MatchedMetricsTrailConsumer(senderConsumerMatcher, trailConsumer));
        return trailConsumer;
    }

    /**
     * Unpacks the given packaged {@link Metric}s and dispatches them to this
     * receiver's {@link MetricsConsumer}s.
     *
     * @param request The request to unpack and disptach; might not be null.
     */
    public synchronized void receive(WebMetricRequest request) {
        // For each consumer...
        this.consumers.parallelStream().forEach(consumer -> {
            // ...go over the request's consumers...
            request.getConsumers().parallelStream().forEach(requestConsumer -> {
                // ...and if the request consumer's ID matches the consumer's one...
                if (requestConsumer.getConsumerId().matches(consumer.matcher)) {
                    // ...go over the request consumer's trails...
                    requestConsumer.getTrails().parallelStream().forEach(trail -> {
                        UUID trailId = UUID.fromString(trail.getTrailId());
                        MetricsTrailConsumer.MetricsTrailConsumerQueue queue = consumer.queueFor(trailId);
                        // ...and that trail's metrics...
                        trail.getMetrics().stream().forEach(metric -> {
                            // ...enqueue them to the queue...
                            queue.enqueue(metric.to());
                        });
                        // ...and then end the trail
                        queue.onTrailEnd();
                    });
                }
            });
        });
    }
}