integration_test/src/java/com/twitter/heron/integration_test/topology/serialization/CustomCheckBolt.java

// Copyright 2016 Twitter. All rights reserved.
//
// 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.twitter.heron.integration_test.topology.serialization;

import java.util.logging.Logger;

import com.twitter.heron.api.bolt.BaseBasicBolt;
import com.twitter.heron.api.bolt.BasicOutputCollector;
import com.twitter.heron.api.topology.OutputFieldsDeclarer;
import com.twitter.heron.api.tuple.Fields;
import com.twitter.heron.api.tuple.Tuple;

/**
 * A bolt that checks deserialization works fine
 */
public class CustomCheckBolt extends BaseBasicBolt {
  private static final Logger LOG = Logger.getLogger(CustomCheckBolt.class.getName());
  private static final long serialVersionUID = 3404960992657778759L;
  private int nItems;
  private CustomObject[] inputObjects;

  public CustomCheckBolt(CustomObject[] inputObjects) {
    this.nItems = 0;
    this.inputObjects = inputObjects;
  }

  @Override
  public void execute(Tuple input, BasicOutputCollector collector) {
    LOG.info("Received input tuple: " + input.getValueByField("custom").toString());
    if (input.getValueByField("custom")
        .equals(inputObjects[(nItems++) % inputObjects.length])) {
      collector.emit(input.getValues());
    }
  }

  @Override
  public void declareOutputFields(OutputFieldsDeclarer declarer) {
    declarer.declare(new Fields("custom"));
  }
}