com.jkoolcloud.tnt4j.streams.sample.custom.SampleIntegration.java Source code

Introduction

Here is the source code for com.jkoolcloud.tnt4j.streams.sample.custom.SampleIntegration.java

Source

/*
 * Copyright 2014-2017 JKOOL, LLC.
 *
 * 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.jkoolcloud.tnt4j.streams.sample.custom;

import java.util.Collection;

import org.apache.commons.lang3.StringUtils;

import com.jkoolcloud.tnt4j.core.OpLevel;
import com.jkoolcloud.tnt4j.sink.DefaultEventSinkFactory;
import com.jkoolcloud.tnt4j.sink.EventSink;
import com.jkoolcloud.tnt4j.streams.configure.StreamsConfigLoader;
import com.jkoolcloud.tnt4j.streams.inputs.StreamThread;
import com.jkoolcloud.tnt4j.streams.inputs.TNTInputStream;

/**
 * Sample integration of TNT4J-Streams into an application.
 *
 * @version $Revision: 1 $
 */
public final class SampleIntegration {
    private static final EventSink LOGGER = DefaultEventSinkFactory.defaultEventSink(SampleIntegration.class);

    /**
     * Configure streams and parsers, and run each stream in its own thread.
     *
     * @param cfgFileName
     *            configuration file name
     */
    public static void loadConfigAndRun(String cfgFileName) {
        try {
            StreamsConfigLoader cfg = StringUtils.isEmpty(cfgFileName) ? new StreamsConfigLoader()
                    : new StreamsConfigLoader(cfgFileName);
            Collection<TNTInputStream<?, ?>> streams = cfg.getStreams();
            if (streams == null || streams.isEmpty()) {
                throw new IllegalStateException("No Activity Streams found in configuration"); // NON-NLS
            }

            ThreadGroup streamThreads = new ThreadGroup("Streams"); // NON-NLS
            StreamThread ft;
            for (TNTInputStream<?, ?> stream : streams) {
                ft = new StreamThread(streamThreads, stream,
                        String.format("%s:%s", stream.getClass().getSimpleName(), stream.getName())); // NON-NLS
                ft.start();
            }
        } catch (Exception e) {
            LOGGER.log(OpLevel.ERROR, String.valueOf(e.getLocalizedMessage()), e);
        }
    }

    /**
     * The following can be used if using the default configuration file with a single stream.
     *
     * @param cfgFileName
     *            configuration file name
     */
    public static void simpleConfigAndRun(String cfgFileName) {
        try {
            StreamsConfigLoader cfg = new StreamsConfigLoader();
            TNTInputStream<?, ?> stream = cfg.getStream("StreamName"); // NON-NLS
            StreamThread ft = new StreamThread(stream);
            ft.start();
        } catch (Exception e) {
            LOGGER.log(OpLevel.ERROR, String.valueOf(e.getLocalizedMessage()), e);
        }
    }
}