LoggerBolt.java

Go to the documentation of this file.

/* Copyright 2017 Telstra Open Source
 *
 *   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 org.openkilda.wfm.topology.utils;

import org.apache.storm.task.OutputCollector;
import org.apache.storm.task.TopologyContext;
import org.apache.storm.topology.OutputFieldsDeclarer;
import org.apache.storm.topology.base.BaseRichBolt;
import org.apache.storm.tuple.Tuple;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.event.Level;

import java.util.Map;

public class LoggerBolt extends BaseRichBolt {

    private static Logger logger = LoggerFactory.getLogger(LoggerBolt.class);
    public Level level = Level.DEBUG;
    public String watermark = "";
    private OutputCollector _collector;

    private void log(Level level, String format, Object[] argArray) {
        switch (level) {
            case TRACE:
                logger.trace(format, argArray);
                break;
            case DEBUG:
                logger.debug(format, argArray);
                break;
            case INFO:
                logger.info(format, argArray);
                break;
            case WARN:
                logger.warn(format, argArray);
                break;
            case ERROR:
                logger.error(format, argArray);
                break;
        }

    }

    public LoggerBolt withLevel(Level level) {
        this.level = level;
        return this;
    }

    public LoggerBolt withWatermark(String watermark) {
        this.watermark = watermark;
        return this;
    }

    @Override
    public void prepare(Map conf, TopologyContext context, OutputCollector collector) {
        _collector = collector;
    }

    @Override
    public void execute(Tuple tuple) {
        System.out.println("this = " + this);

        // No way to do dynamic log level with SLF4J
        // https://stackoverflow.com/questions/2621701/setting-log-level-of-message-at-runtime-in-slf4j
        // https://jira.qos.ch/browse/SLF4J-124

        log(level, "\n{}: fields: {} :: values: {}",
                new Object[] {watermark, tuple.getFields(), tuple.getValues()});

        _collector.ack(tuple);
    }

    @Override
    public void declareOutputFields(OutputFieldsDeclarer declarer) {
    }
}