com.arpnetworking.clusteraggregator.configuration.EmitterConfiguration.java Source code

Introduction

Here is the source code for com.arpnetworking.clusteraggregator.configuration.EmitterConfiguration.java

Source

/**
 * Copyright 2015 Groupon.com
 *
 * Licensed 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 com.arpnetworking.clusteraggregator.configuration;

import com.arpnetworking.commons.builder.OvalBuilder;
import com.arpnetworking.commons.jackson.databind.ObjectMapperFactory;
import com.arpnetworking.tsdcore.sinks.Sink;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.introspect.AnnotationIntrospectorPair;
import com.fasterxml.jackson.module.guice.GuiceAnnotationIntrospector;
import com.fasterxml.jackson.module.guice.GuiceInjectableValues;
import com.google.common.collect.Lists;
import com.google.inject.Injector;
import net.sf.oval.constraint.Min;
import net.sf.oval.constraint.NotEmpty;
import net.sf.oval.constraint.NotNull;

import java.util.Collections;
import java.util.List;

/**
 * Representation of an emitter configuration for cluster aggregator.
 *
 * @author Brandon Arp (brandonarp at gmail dot com)
 */
public final class EmitterConfiguration {
    /**
     * Create an <code>ObjectMapper</code> for Emitter configuration.
     *
     * @param injector The Guice <code>Injector</code> instance.
     * @return An <code>ObjectMapper</code> for Pipeline configuration.
     */
    public static ObjectMapper createObjectMapper(final Injector injector) {
        final ObjectMapper objectMapper = ObjectMapperFactory.createInstance();

        final GuiceAnnotationIntrospector guiceIntrospector = new GuiceAnnotationIntrospector();
        objectMapper.setInjectableValues(new GuiceInjectableValues(injector));
        objectMapper.setAnnotationIntrospectors(
                new AnnotationIntrospectorPair(guiceIntrospector,
                        objectMapper.getSerializationConfig().getAnnotationIntrospector()),
                new AnnotationIntrospectorPair(guiceIntrospector,
                        objectMapper.getDeserializationConfig().getAnnotationIntrospector()));

        return objectMapper;
    }

    public List<Sink> getSinks() {
        return Collections.unmodifiableList(_sinks);
    }

    public int getPoolSize() {
        return _poolSize;
    }

    private EmitterConfiguration(final Builder builder) {
        _sinks = Lists.newArrayList(builder._sinks);
        _poolSize = builder._poolSize;
    }

    private final List<Sink> _sinks;
    private final int _poolSize;

    /**
     * Implementation of builder pattern for {@link EmitterConfiguration}.
     *
     * @author Brandon Arp (brandonarp at gmail dot com)
     */
    public static final class Builder extends OvalBuilder<EmitterConfiguration> {
        /**
         * Public constructor.
         */
        public Builder() {
            super(EmitterConfiguration::new);
        }

        /**
         * The sinks. Required. Cannot be null or empty.
         *
         * @param value The sinks.
         * @return This instance of <code>Builder</code>.
         */
        public Builder setSinks(final List<Sink> value) {
            _sinks = value;
            return this;
        }

        /**
         * The number of emitters to create in a pool. Sinks are not necessarily also duplicated.
         * Cannot be null or empty.
         *
         * @param value The poolSize.
         * @return This instance of <code>Builder</code>.
         */
        public Builder setPoolSize(final Integer value) {
            _poolSize = value;
            return this;
        }

        @NotNull
        @NotEmpty
        private List<Sink> _sinks;

        @NotNull
        @Min(1)
        private Integer _poolSize = 1;
    }
}