contrib/src/test/java/org/apache/apex/malhar/contrib/helper/SourceModule.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.datatorrent.contrib.helper;

import java.util.HashMap;
import java.util.concurrent.ArrayBlockingQueue;

import org.slf4j.LoggerFactory;

import com.datatorrent.common.util.BaseOperator;
import com.datatorrent.api.Context.OperatorContext;
import com.datatorrent.api.DefaultOutputPort;
import com.datatorrent.api.InputOperator;
import com.datatorrent.api.Operator.ActivationListener;

public class SourceModule extends BaseOperator
implements InputOperator, ActivationListener<OperatorContext>
{
  public final transient DefaultOutputPort<byte[]> outPort = new DefaultOutputPort<byte[]>();
  transient ArrayBlockingQueue<byte[]> holdingBuffer;
  int testNum;

  private static org.slf4j.Logger logger;

  public SourceModule()
  {
    logger =  LoggerFactory.getLogger(SourceModule.class);;
  }

  public SourceModule(org.slf4j.Logger loggerInstance)
  {
    logger = loggerInstance;
  }

  @Override
  public void setup(OperatorContext context)
  {
    holdingBuffer = new ArrayBlockingQueue<byte[]>(1024 * 1024);
  }

  public void emitTuple(byte[] message)
  {
    logger.debug("Emitting message {}", message);
    outPort.emit(message);
  }

  @Override
  public void emitTuples()
  {
    for (int i = holdingBuffer.size(); i-- > 0;) {
      emitTuple(holdingBuffer.poll());
    }
  }

  @Override
  public void activate(OperatorContext ctx)
  {
    for (int i = 0; i < testNum; i++) {
      HashMap<String, Integer> dataMapa = new HashMap<String, Integer>();
      dataMapa.put("a", 2);
      holdingBuffer.add(dataMapa.toString().getBytes());

      HashMap<String, Integer> dataMapb = new HashMap<String, Integer>();
      dataMapb.put("b", 20);
      holdingBuffer.add(dataMapb.toString().getBytes());

      HashMap<String, Integer> dataMapc = new HashMap<String, Integer>();
      dataMapc.put("c", 1000);
      holdingBuffer.add(dataMapc.toString().getBytes());
    }
  }

  public void setTestNum(int testNum)
  {
    this.testNum = testNum;
  }

  @Override
  public void deactivate()
  {
  }

  public void replayTuples(long windowId)
  {
  }
}