TupleToBag.java

package com.hipmunk.pig.udfs;

import org.apache.pig.EvalFunc;
import org.apache.pig.backend.executionengine.ExecException;
import org.apache.pig.data.BagFactory;
import org.apache.pig.data.DataBag;
import org.apache.pig.data.DefaultBagFactory;
import org.apache.pig.data.Tuple;
import org.apache.pig.data.TupleFactory;

public class TupleToBag extends EvalFunc<DataBag> {
	private static final BagFactory bagFactory_ = DefaultBagFactory.getInstance();
	private static final TupleFactory tupleFactory_ = TupleFactory.getInstance();
	
	public DataBag exec(Tuple args) throws ExecException {
		assert args.size() == 1;
		DataBag db = bagFactory_.newDefaultBag();
		Tuple t = (Tuple)args.get(0);
		for(Object i: t.getAll()) {
			db.add(tupleFactory_.newTuple(i));
		}
		return db;
	}
}