[Apache Pig] Extending CSVExcelLoader to append file name of the split

CSVExcelLoader doesn’t have an option to append the filename of the split it is processing. It comes in handy in certain situations. Here is a quick way to add the support

public class CSVExcelLoaderWithFileName extends CSVLoader {

  Path path;

  @Override
  public void prepareToRead(RecordReader reader, PigSplit split) throws IOException {
    super.prepareToRead(reader, split);
    path = ((FileSplit)split.getWrappedSplit()).getPath();
  }

  @Override
  public Tuple getNext() throws IOException {
    Tuple superTuple =  super.getNext();
    if(superTuple != null) {
      superTuple.append(path.getName());
    }
    return superTuple;
  }
}

The code is simple. When the Load function is ready to read, we get the path of the split it shall be processing.

Caution: Watch out forpig.splitCombination property. More info at http://pig.apache.org/docs/r0.12.1/perf.html#combine-files