java/pig-galago/src/com/yahoo/pig/data/DataBagFileWriter.java

   1 /*
   2  * Copyright (c) 2007 Yahoo! Inc.  All rights reserved.
   3  * See accompanying LICENSE file.
   4  */
   5 package com.yahoo.pig.data;
   6
   7 import java.io.BufferedOutputStream;
   8 import java.io.DataOutputStream;
   9 import java.io.File;
  10 import java.io.FileOutputStream;
  11 import java.io.IOException;
  12 import java.util.Iterator;
  13
  14
  15 public class DataBagFileWriter {
  16         File store;
  17         DataOutputStream out;
  18
  19         public DataBagFileWriter(File store) throws IOException{
  20                 this.store = store;
  21                 out = new DataOutputStream(new BufferedOutputStream(new FileOutputStream(store)));
  22         }
  23
  24         public void write(Tuple t) throws IOException{
  25                 t.write(out);
  26         }
  27
  28         public void write(Iterator<Tuple> iter) throws IOException{
  29                 while (iter.hasNext())
  30                         iter.next().write(out);
  31         }
  32
  33         public void close() throws IOException{
  34                 flush();
  35                 out.close();
  36         }
  37
  38         public void flush() throws IOException{
  39                 out.flush();
  40         }
  41
  42 }