com.datatorrent.lib.logs.LineToTokenArrayList Maven / Gradle / Ivy

Go to download
/**
 * 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.lib.logs;

import java.util.ArrayList;
import java.util.HashMap;

import com.datatorrent.api.DefaultOutputPort;
import com.datatorrent.api.annotation.OperatorAnnotation;
import com.datatorrent.api.annotation.OutputPortFieldAnnotation;
import com.datatorrent.api.annotation.Stateless;

import com.datatorrent.lib.util.BaseLineTokenizer;
import com.datatorrent.lib.util.UnifierArrayList;

/**
 * This operator splits string objects into tokens and sub tokens. 
 * This operator emits a list of tokens, as well as a map from tokens to sub tokens.
 * 
 * An ArrayList of all tkns are emitted on output port "tokens".
 * An ArrayList of all subtokens are emitted on port splittokens

 * This module is a pass through. Ideal for applications like log processing

 * 

 * StateFull : No,  tokens are processed in current window. 

 * Partitions : Yes,  output port unifier operator. 

 * 

 * Ports:

 * data: expects String

 * tokens: emits ArrayList<String>

 * splittokens: emits ArrayList<HashMap<String,ArrayList<String>>>

 * 

 * Properties:

 * splitby: The characters used to split the line. Default is ";\t "

 * splittokenby: The characters used to split a token into key,val1,val2,.... If not specified the value is set to null. Default is "", i.e. tokens are not split

 * 
 * @displayName Line To Token (ArrayList)
 * @category Tuple Converters
 * @tags string, arraylist
 *
 * @since 0.3.2
 */
@Stateless
@OperatorAnnotation(partitionable = true)
public class LineToTokenArrayList extends BaseLineTokenizer
{
  protected transient ArrayList tokentuple = null;
  protected transient ArrayList>> stokentuple = null;
  protected transient HashMap> smap = null;
  protected transient ArrayList vals = null;


  /**
   * This emits the tokens a string is broken up into.
   */
  @OutputPortFieldAnnotation(optional = true)
  public final transient DefaultOutputPort> tokens = new DefaultOutputPort>()
  {
    @Override
    public Unifier> getUnifier()
    {
      return new UnifierArrayList();
    }
  };

  /**
   * This output port emits a map from tokens to sub tokens.
   */
  @OutputPortFieldAnnotation(optional = true)
  public final transient DefaultOutputPort>>> splittokens =
      new DefaultOutputPort>>>()
  {
    @Override
    public Unifier>>> getUnifier()
    {
      return new UnifierArrayList<>();
    }
  };

  /**
   * sets up output tuples
   */
  @Override
  public void beginProcessTokens()
  {
    tokentuple = new ArrayList();
    stokentuple = new ArrayList>>();
  }

  /**
   * adds tok to output token tuple
   * if splittoken is set starts subtoken processing
   *
   * @param tok
   */
  @Override
  public void processToken(String tok)
  {
    if (tokens.isConnected()) {
      tokentuple.add(tok);
    }
    if (splittokens.isConnected() && hasSplitTokenBy()) {
      super.processToken(tok);
    }
  }

  /**
   * sets up data for sub token processing
   */
  @Override
  public void beginProcessSubTokens()
  {
    smap = new HashMap>(1);
    vals = new ArrayList(4);
  }

  /**
   * Added data to subtoken tuple
   */
  @Override
  public void endProcessSubTokens()
  {
    if (!smap.isEmpty()) {
      stokentuple.add(smap);
    }

    smap = null;
    vals = null;
  }

  /**
   * Addd first subtoken to key, and rest to value ArrayList
   *
   * @param subtok
   */
  @Override
  public void processSubToken(String subtok)
  {
    if (smap.isEmpty()) {
      smap.put(subtok, vals);
    } else {
      vals.add(subtok);
    }
  }

  /**
   * emits token tuple and subtoken tuple if respective ports are connected
   */
  @Override
  public void endProcessTokens()
  {
    if (tokens.isConnected()) {
      tokens.emit(tokentuple);
      tokentuple = null;
    }

    if (splittokens.isConnected()) {
      splittokens.emit(stokentuple);
      stokentuple = null;
      smap = null;
      vals = null;
    }
  }
}