src/main/java/org/lwes/journaller/DeJournaller.java

   1 package org.lwes.journaller;
   2 /**
   3  * A Class that will "dejournal" files written to by the GZIPEventHandler.  If you want to
   4  * do something other than write the events to stdout, subclass this and override handleEvent.
   5  *
   6  * User: fmaritato
   7  * Date: Apr 16, 2009
   8  */
   9
  10 import org.apache.commons.cli.CommandLine;
  11 import org.apache.commons.cli.CommandLineParser;
  12 import org.apache.commons.cli.HelpFormatter;
  13 import org.apache.commons.cli.Options;
  14 import org.apache.commons.cli.PosixParser;
  15 import org.apache.commons.logging.Log;
  16 import org.apache.commons.logging.LogFactory;
  17 import org.lwes.Event;
  18 import org.lwes.db.EventTemplateDB;
  19 import org.lwes.journaller.util.EventHandlerUtil;
  20 import org.lwes.serializer.DeserializerState;
  21
  22 import java.io.DataInputStream;
  23 import java.io.EOFException;
  24 import java.io.File;
  25 import java.io.FileInputStream;
  26 import java.io.IOException;
  27 import java.util.zip.GZIPInputStream;
  28
  29 public class DeJournaller implements Runnable, JournallerConstants {
  30     private transient Log log = LogFactory.getLog(DeJournaller.class);
  31
  32     private boolean gzipped;
  33     private String fileName;
  34     private String esfFile;
  35     private boolean validate = false;
  36
  37     protected static Options options;
  38
  39     static {
  40         options = new Options();
  41         options.addOption("f", "file", true, "File to read events from.");
  42         options.addOption("e", "esf-file", true, "Event definition file.");
  43         options.addOption("h", "help", false, "Print this message.");
  44         options.addOption("g", "gzipped", false, "Event log is gzipped.");
  45         options.addOption("v", "validate", false, "Validate events");
  46     }
  47
  48     public DeJournaller() {
  49     }
  50
  51     public void run() {
  52
  53         if (fileName == null || "".equals(fileName)) {
  54             log.error("File name was not specified");
  55             return;
  56         }
  57
  58         EventTemplateDB evtTemplate = new EventTemplateDB();
  59         DeserializerState state = new DeserializerState();
  60
  61         if (getEsfFile() != null) {
  62             evtTemplate.setESFFile(new File(getEsfFile()));
  63         }
  64         log.debug("esf: " + evtTemplate.getESFFile());
  65         log.debug("validate: " + validate);
  66         evtTemplate.initialize();
  67
  68         DataInputStream in = null;
  69         try {
  70             log.debug("Opening file: " + fileName);
  71             if (gzipped) {
  72                 in = new DataInputStream(new GZIPInputStream(new FileInputStream(fileName)));
  73             }
  74             else {
  75                 in = new DataInputStream(new FileInputStream(fileName));
  76             }
  77             Event evt;
  78             while ((evt = EventHandlerUtil.readEvent(in, state, evtTemplate, validate)) != null) {
  79                 state.reset();
  80                 handleEvent(evt);
  81             }
  82         }
  83         catch (EOFException e) {
  84             // this is normal. Just catch and ignore.
  85         }
  86         catch (Exception e) {
  87             log.error(e.getMessage(), e);
  88         }
  89         finally {
  90             if (in != null) {
  91                 try {
  92                     in.close();
  93                 }
  94                 catch (IOException e) {
  95                     log.error(e.getMessage(), e);
  96                 }
  97             }
  98             done();
  99         }
 100
 101     }
 102
 103     /**
 104      * This method is called after all events in the file have been processed.
 105      */
 106     public void done() {
 107         if (log.isDebugEnabled()) {
 108             log.debug("done");
 109         }
 110     }
 111
 112     /**
 113      * Subclass and override this method if you want to do something other than print the
 114      * events to the logging system. For example, you may want to write these events to a
 115      * database.
 116      *
 117      * @param event
 118      */
 119     public void handleEvent(Event event) {
 120         System.out.println(event.toOneLineString());
 121     }
 122
 123     public static void main(String[] args) {
 124
 125         DeJournaller dj = new DeJournaller();
 126         try {
 127             CommandLineParser parser = new PosixParser();
 128             CommandLine line = parser.parse(options, args);
 129
 130             if (line.hasOption("h") || line.hasOption("help")) {
 131                 HelpFormatter formatter = new HelpFormatter();
 132                 formatter.printHelp("dejournaller", options);
 133                 Runtime.getRuntime().exit(1);
 134             }
 135             if (line.hasOption("f") || line.hasOption("file")) {
 136                 dj.setFileName(line.getOptionValue("f") == null ?
 137                                line.getOptionValue("file") :
 138                                line.getOptionValue("f"));
 139             }
 140             if (line.hasOption("e") || line.hasOption("esf-file")) {
 141                 dj.setEsfFile(line.getOptionValue("e") == null ?
 142                               line.getOptionValue("esf-file") :
 143                               line.getOptionValue("e"));
 144             }
 145             if (line.hasOption("g") || line.hasOption("gzipped")) {
 146                 dj.setGzipped(true);
 147             }
 148             if (line.hasOption("v") || line.hasOption("validate")) {
 149                 dj.setValidate(true);
 150             }
 151
 152             dj.run();
 153         }
 154         catch (Exception e) {
 155             e.printStackTrace();
 156             Runtime.getRuntime().exit(1);
 157         }
 158     }
 159
 160     public String getFileName() {
 161         return fileName;
 162     }
 163
 164     public void setFileName(String fileName) {
 165         this.fileName = fileName;
 166     }
 167
 168     public String getEsfFile() {
 169         return esfFile;
 170     }
 171
 172     public void setEsfFile(String esfFile) {
 173         this.esfFile = esfFile;
 174     }
 175
 176     public boolean isGzipped() {
 177         return gzipped;
 178     }
 179
 180     public void setGzipped(boolean gzipped) {
 181         this.gzipped = gzipped;
 182     }
 183
 184     public boolean isValidate() {
 185         return validate;
 186     }
 187
 188     public void setValidate(boolean validate) {
 189         this.validate = validate;
 190     }
 191 }