Package org.apache.log4j

Source Code of org.apache.log4j.CompositeRollingAppender

package org.apache.log4j;

/*
* Copyright 1999-2005 The Apache Software Foundation.
*
* Licensed 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.
*/

import org.apache.log4j.RollingCalendar;
import org.apache.log4j.helpers.OptionConverter;
import org.apache.log4j.helpers.QuietWriter;
import org.apache.log4j.helpers.CountingQuietWriter;
import org.apache.log4j.helpers.LogLog;
import org.apache.log4j.spi.LoggingEvent;

import java.util.Date;
import java.io.IOException;
import java.io.Writer;
import java.text.SimpleDateFormat;
import java.io.File;
import java.io.FilenameFilter;

/**
<p>CompositeRollingAppender combines RollingFileAppender and DailyRollingFileAppender<br>
*  It can function as either or do both at the same time (making size
*  based rolling files like RollingFileAppender until a data/time boundary
*  is crossed at which time it rolls all of those files as per the DailyRollingFileAppender)
*  based on the setting for <code>rollingStyle</code>.<br>
<br>
*  To use CompositeRollingAppender to roll log files as they reach a certain
*  size (like RollingFileAppender), set rollingStyle=1 (@see config.size)<br>
*  To use CompositeRollingAppender to roll log files at certain time intervals
*  (daily for example), set rollingStyle=2 and a datePattern (@see config.time)<br>
*  To have CompositeRollingAppender roll log files at a certain size AND rename those
*  according to time intervals, set rollingStyle=3 (@see config.composite)<br>
*
<p>A of few additional optional features have been added:<br>
*  -- Attach date pattern for current log file (@see staticLogFileName)<br>
*  -- Backup number increments for newer files (@see countDirection)<br>
*  -- Infinite number of backups by file size (@see maxSizeRollBackups)<br>
<br>
<p>A few notes and warnings:  For large or infinite number of backups
*  countDirection > 0 is highly recommended, with staticLogFileName = false if
*  time based rolling is also used -- this will reduce the number of file renamings
*  to few or none.  Changing staticLogFileName or countDirection without clearing
*  the directory could have nasty side effects.  If Date/Time based rolling
*  is enabled, CompositeRollingAppender will attempt to roll existing files
*  in the directory without a date/time tag based on the last modified date
*  of the base log files last modification.<br>
<br>
<p>A maximum number of backups based on date/time boundries would be nice
*  but is not yet implemented.<br>
*
@author Kevin Steppe
@author Heinz Richter
@author Eirik Lygre
@author Ceki G&uuml;lc&uuml;
*/
public class CompositeRollingAppender extends org.apache.log4j.FileAppender
{
  // The code assumes that the following 'time' constants are in a increasing
  // sequence.
  static final int TOP_OF_TROUBLE=-1;
  static final int TOP_OF_MINUTE = 0;
  static final int TOP_OF_HOUR   = 1;
  static final int HALF_DAY      = 2;
  static final int TOP_OF_DAY    = 3;
  static final int TOP_OF_WEEK   = 4;
  static final int TOP_OF_MONTH  = 5;

  /** Style of rolling to use */
  static final int BY_SIZE = 1;
  static final int BY_DATE = 2;
  static final int BY_COMPOSITE = 3;

  //Not currently used
  static final String S_BY_SIZE = "Size";
  static final String S_BY_DATE = "Date";
  static final String S_BY_COMPOSITE = "Composite";

  /**
   The date pattern. By default, the pattern is set to
   "'.'yyyy-MM-dd" meaning daily rollover.
  */
  private String datePattern = "'.'yyyy-MM-dd";

  /**   The actual formatted filename that is currently being written to
       or will be the file transferred to on roll over
     (based on staticLogFileName). */
  private String scheduledFilename = null;

  /** The timestamp when we shall next recompute the filename. */
  private long nextCheck = System.currentTimeMillis () - 1;

  /** Holds date of last roll over */
  Date now = new Date();

  SimpleDateFormat sdf;

  /** Helper class to determine next rollover time */
  RollingCalendar rc = new RollingCalendar();

  /** Current period for roll overs */
  int checkPeriod = TOP_OF_TROUBLE;

  /**   The default maximum file size is 10MB. */
  protected long maxFileSize = 10*1024*1024;

  /**   There is zero backup files by default. */
  protected int maxSizeRollBackups = 0;
  /** How many sized based backups have been made so far */
  protected int curSizeRollBackups = 0;

  /** not yet implemented */
  protected int maxTimeRollBackups = -1;
  protected int curTimeRollBackups = 0;

  /** By default newer files have lower numbers. (countDirection < 0)
   *  ie. log.1 is most recent, log.5 is the 5th backup, etc...
   *  countDirection > 0 does the opposite ie.
   *  log.1 is the first backup made, log.5 is the 5th backup made, etc.
   *  For infinite backups use countDirection > 0 to reduce rollOver costs.
   */
  protected int countDirection = -1;

  /** Style of rolling to Use.  BY_SIZE (1), BY_DATE(2), BY COMPOSITE(3) */
  protected int rollingStyle = BY_COMPOSITE;
  protected boolean rollDate = true;
  protected boolean rollSize = true;

  /** By default file.log is always the current file.  Optionally
   *  file.log.yyyy-mm-dd for current formated datePattern can by the currently
   *  logging file (or file.log.curSizeRollBackup or even
   *  file.log.yyyy-mm-dd.curSizeRollBackup) This will make time based roll
   *  overs with a large number of backups much faster -- it won't have to
   *  rename all the backups!
   */
  protected boolean staticLogFileName = true;

  /** FileName provided in configuration.  Used for rolling properly */
  protected String baseFileName;

    /** The default constructor does nothing. */
  public CompositeRollingAppender()  {
    }

  /**
   Instantiate a <code>CompositeRollingAppender</code> and open the
   file designated by <code>filename</code>. The opened filename will
   become the ouput destination for this appender.
  */
  public CompositeRollingAppender (Layout layout, String filename,
           String datePattern) throws IOException {
      this(layout, filename, datePattern, true);
  }

  /**
   Instantiate a CompositeRollingAppender and open the file designated by
   <code>filename</code>. The opened filename will become the ouput
   destination for this appender.

   <p>If the <code>append</code> parameter is true, the file will be
   appended to. Otherwise, the file desginated by
   <code>filename</code> will be truncated before being opened.
  */
  public CompositeRollingAppender(Layout layout, String filename, boolean append)
                    throws IOException {
      super(layout, filename, append);
  }

  /**
   Instantiate a CompositeRollingAppender and open the file designated by
   <code>filename</code>. The opened filename will become the ouput
   destination for this appender.
  */
  public CompositeRollingAppender (Layout layout, String filename,
           String datePattern, boolean append) throws IOException {
      super(layout, filename, append);
      this.datePattern = datePattern;
    activateOptions();
  }
  /**
   Instantiate a CompositeRollingAppender and open the file designated by
   <code>filename</code>. The opened filename will become the output
   destination for this appender.

   <p>The file will be appended to.  DatePattern is default.
  */
  public CompositeRollingAppender(Layout layout, String filename) throws IOException {
      super(layout, filename);
  }

  /**
   The <b>DatePattern</b> takes a string in the same format as
   expected by {@link SimpleDateFormat}. This options determines the
   rollover schedule.
  */
  public void setDatePattern(String pattern) {
      datePattern = pattern;
  }

  /** Returns the value of the <b>DatePattern</b> option. */
  public String getDatePattern() {
      return datePattern;
  }

  /**
   Returns the value of the <b>maxSizeRollBackups</b> option.
  */
  public int getMaxSizeRollBackups() {
      return maxSizeRollBackups;
  }

  /**
   Get the maximum size that the output file is allowed to reach
   before being rolled over to backup files.

   @since 1.1
  */
  public long getMaximumFileSize() {
    return maxFileSize;
  }

  /**
   <p>Set the maximum number of backup files to keep around based on file size.

   <p>The <b>MaxSizeRollBackups</b> option determines how many backup
   files are kept before the oldest is erased. This option takes
   an integer value. If set to zero, then there will be no
   backup files and the log file will be truncated when it reaches
   <code>MaxFileSize</code>.  If a negative number is supplied then
   no deletions will be made.  Note that this could result in
   very slow performance as a large number of files are rolled over unless
   {@link #setCountDirection} up is used.

   <p>The maximum applys to -each- time based group of files and -not- the total.
   Using a daily roll the maximum total files would be (#days run) * (maxSizeRollBackups)

  */
  public void setMaxSizeRollBackups(int maxBackups) {
      maxSizeRollBackups = maxBackups;
  }

  /**
   Set the maximum size that the output file is allowed to reach
   before being rolled over to backup files.

   <p>This method is equivalent to {@link #setMaxFileSize} except
   that it is required for differentiating the setter taking a
   <code>long</code> argument from the setter taking a
   <code>String</code> argument by the JavaBeans {@link
   java.beans.Introspector Introspector}.

   @see #setMaxFileSize(String)
  */
  public void setMaxFileSize(long maxFileSize) {
     this.maxFileSize = maxFileSize;
  }

  /**
   Set the maximum size that the output file is allowed to reach
   before being rolled over to backup files.

   <p>This method is equivalent to {@link #setMaxFileSize} except
   that it is required for differentiating the setter taking a
   <code>long</code> argument from the setter taking a
   <code>String</code> argument by the JavaBeans {@link
   java.beans.Introspector Introspector}.

   @see #setMaxFileSize(String)
  */
  public void setMaximumFileSize(long maxFileSize) {
    this.maxFileSize = maxFileSize;
  }

  /**
   Set the maximum size that the output file is allowed to reach
   before being rolled over to backup files.

   <p>In configuration files, the <b>MaxFileSize</b> option takes an
   long integer in the range 0 - 2^63. You can specify the value
   with the suffixes "KB", "MB" or "GB" so that the integer is
   interpreted being expressed respectively in kilobytes, megabytes
   or gigabytes. For example, the value "10KB" will be interpreted
   as 10240.
  */
  public void setMaxFileSize(String value) {
      maxFileSize = OptionConverter.toFileSize(value, maxFileSize + 1);
  }

  protected void setQWForFiles(Writer writer) {
      qw = new CountingQuietWriter(writer, errorHandler);
  }

  //Taken verbatum from DailyRollingFileAppender
  int computeCheckPeriod() {
    RollingCalendar c = new RollingCalendar();
    // set sate to 1970-01-01 00:00:00 GMT
    Date epoch = new Date(0);
    if(datePattern != null) {
      for(int i = TOP_OF_MINUTE; i <= TOP_OF_MONTH; i++) {
        String r0 = sdf.format(epoch);
        c.setType(i);
        Date next = new Date(c.getNextCheckMillis(epoch));
        String r1 = sdf.format(next);
        //LogLog.debug("Type = "+i+", r0 = "+r0+", r1 = "+r1);
        if(r0 != null && r1 != null && !r0.equals(r1)) {
          return i;
        }
      }
    }
    return TOP_OF_TROUBLE; // Deliberately head for trouble...
  }

  //Now for the new stuff
  /**
   * Handles append time behavior for CompositeRollingAppender.  This checks
   * if a roll over either by date (checked first) or time (checked second)
   * is need and then appends to the file last.
  */
  protected void subAppend(LoggingEvent event) {

    if (rollDate) {
      long n = System.currentTimeMillis();
      if (n >= nextCheck) {
        now.setTime(n);
        nextCheck = rc.getNextCheckMillis(now);

        rollOverTime();
      }
    }

    if (rollSize) {
      if ((fileName != null) && ((CountingQuietWriter) qw).getCount() >= maxFileSize) {
          rollOverSize();
      }
    }

    super.subAppend(event);
  }

  public void setFile(String file)
  {
    baseFileName = file.trim();
    fileName = file.trim();
  }

  /**
   * Creates and opens the file for logging.  If <code>staticLogFileName</code>
   * is false then the fully qualified name is determined and used.
   */
  public synchronized void setFile(String fileName, boolean append) throws IOException {
    if (!staticLogFileName) {
        scheduledFilename = fileName = fileName.trim() + sdf.format(now);
      if (countDirection > 0) {
        scheduledFilename = fileName = fileName + '.' + (++curSizeRollBackups);
      }
    }

    super.setFile(fileName, append);
    if(append) {
      File f = new File(fileName);
      ((CountingQuietWriter) qw).setCount(f.length());
    }
  }

  public int getCountDirection() {
    return countDirection;
  }

  public void setCountDirection(int direction) {
    countDirection = direction;
  }

  public int getRollingStyle () {
        return rollingStyle;
  }

  public void setRollingStyle(int style) {
      rollingStyle = style;
    switch (rollingStyle) {
      case BY_SIZE:
         rollDate = false;
         rollSize = true;
         break;
      case BY_DATE:
         rollDate = true;
         rollSize = false;
         break;
      case BY_COMPOSITE:
         rollDate = true;
         rollSize = true;
         break;
      default:
        errorHandler.error("Invalid rolling Style, use 1 (by size only), 2 (by date only) or 3 (both)");
    }
  }

/*
  public void setRollingStyle(String style) {
    if (style == S_BY_SIZE) {
        rollingStyle = BY_SIZE;
    }
    else if (style == S_BY_DATE) {
        rollingStyle = BY_DATE;
    }
    else if (style == S_BY_COMPOSITE) {
      rollingStyle = BY_COMPOSITE;
    }
  }
*/
  public boolean getStaticLogFileName() {
      return staticLogFileName;
  }

  public void setStaticLogFileName(boolean s) {
    staticLogFileName = s;
  }

  public void setStaticLogFileName(String value) {
    setStaticLogFileName(OptionConverter.toBoolean(value, true));
  }

  /**
   *  Initializes based on exisiting conditions at time of <code>
   *  activateOptions</code>.  The following is done:<br>
   <br>
   *  A) determine curSizeRollBackups<br>
   *  B) determine curTimeRollBackups (not implemented)<br>
   *  C) initiates a roll over if needed for crossing a date boundary since
   *  the last run.
   */
  protected void existingInit() {

    curSizeRollBackups = 0;
    curTimeRollBackups = 0;

    //part A starts here
    String filter;
    if (staticLogFileName || !rollDate) {
      filter = baseFileName + ".*";
    }
    else {
      filter = scheduledFilename + ".*";
    }

    File f = new File(baseFileName);
    f = f.getParentFile();
    if (f == null)
       f = new File(".");

    LogLog.debug("Searching for existing files in: " + f);
    String[] files = f.list();

    if (files != null) {
      for (int i = 0; i < files.length; i++) {
        if (!files[i].startsWith(baseFileName))
           continue;

        int index = files[i].lastIndexOf(".");

        if (staticLogFileName) {
           int endLength = files[i].length() - index;
           if (baseFileName.length() + endLength != files[i].length()) {
             //file is probably scheduledFilename + .x so I don't care
             continue;
           }
        }

        try {
          int backup = Integer.parseInt(files[i].substring(index + 1, files[i].length()));
          LogLog.debug("From file: " + files[i] + " -> " + backup);
          if (backup > curSizeRollBackups)
             curSizeRollBackups = backup;
        }
        catch (Exception e) {
          //this happens when file.log -> file.log.yyyy-mm-dd which is normal
          //when staticLogFileName == false
          LogLog.debug("Encountered a backup file not ending in .x " + files[i]);
        }
      }
    }
    LogLog.debug("curSizeRollBackups starts at: " + curSizeRollBackups);
    //part A ends here

    //part B not yet implemented

    //part C
    if (staticLogFileName && rollDate) {
      File old = new File(baseFileName);
      if (old.exists()) {
        Date last = new Date(old.lastModified());
        if (!(sdf.format(last).equals(sdf.format(now)))) {
          scheduledFilename = baseFileName + sdf.format(last);
          LogLog.debug("Initial roll over to: " + scheduledFilename);
          rollOverTime();
        }
      }
    }
    LogLog.debug("curSizeRollBackups after rollOver at: " + curSizeRollBackups);
    //part C ends here

  }

  /**
   * Sets initial conditions including date/time roll over information, first check,
   * scheduledFilename, and calls <code>existingInit</code> to initialize
   * the current # of backups.
   */
  public void activateOptions() {

      //REMOVE removed rollDate from boolean to enable Alex's change
    if(datePattern != null) {
      now.setTime(System.currentTimeMillis());
      sdf = new SimpleDateFormat(datePattern);
      int type = computeCheckPeriod();
      //printPeriodicity(type);
      rc.setType(type);
      //next line added as this removes the name check in rollOver
      nextCheck = rc.getNextCheckMillis(now);
    } else {
      if (rollDate)
          LogLog.error("Either DatePattern or rollingStyle options are not set for ["+
            name+"].");
    }

    existingInit();

    super.activateOptions();

    if (rollDate && fileName != null && scheduledFilename == null)
      scheduledFilename = fileName + sdf.format(now);
  }

  /**
   Rollover the file(s) to date/time tagged file(s).
   Opens the new file (through setFile) and resets curSizeRollBackups.
  */
  protected void rollOverTime() {

      curTimeRollBackups++;

    //delete the old stuff here

    if (staticLogFileName) {
      /* Compute filename, but only if datePattern is specified */
      if (datePattern == null) {
        errorHandler.error("Missing DatePattern option in rollOver().");
        return;
      }

      //is the new file name equivalent to the 'current' one
      //something has gone wrong if we hit this -- we should only
      //roll over if the new file will be different from the old
      String dateFormat = sdf.format(now);
      if (scheduledFilename.equals(fileName + dateFormat)) {
        errorHandler.error("Compare " + scheduledFilename + " : " + fileName + dateFormat);
        return;
      }

      // close current file, and rename it to datedFilename
      this.closeFile();

      //we may have to roll over a large number of backups here
          String from, to;
      for (int i = 1; i <= curSizeRollBackups; i++) {
        from = fileName + '.' + i;
        to = scheduledFilename + '.' + i;
        rollFile(from, to);
          }

      rollFile(fileName, scheduledFilename);
    }

    try {
      // This will also close the file. This is OK since multiple
      // close operations are safe.
      curSizeRollBackups = 0; //We're cleared out the old date and are ready for the new

      //new scheduled name
      scheduledFilename = fileName + sdf.format(now);
      this.setFile(baseFileName, false);
    }
    catch(IOException e) {
      errorHandler.error("setFile("+fileName+", false) call failed.");
    }

  }

  /** Renames file <code>from</code> to file <code>to</code>.  It
   *  also checks for existence of target file and deletes if it does.
   */
  protected static void rollFile(String from, String to) {
    File target = new File(to);
    if (target.exists()) {
      LogLog.debug("deleting existing target file: " + target);
      target.delete();
    }

    File file = new File(from);
    file.renameTo(target);
    LogLog.debug(from +" -> "+ to);
  }

  /** Delete's the specified file if it exists */
  protected static void deleteFile(String fileName) {
    File file = new File(fileName);
    if (file.exists()) {
       file.delete();
    }
  }

  /**
   Implements roll overs base on file size.

   <p>If the maximum number of size based backups is reached
   (<code>curSizeRollBackups == maxSizeRollBackups</code) then the oldest
   file is deleted -- it's index determined by the sign of countDirection.<br>
   If <code>countDirection</code> < 0, then files
   {<code>File.1</code>, ..., <code>File.curSizeRollBackups -1</code>}
   are renamed to {<code>File.2</code>, ...,
   <code>File.curSizeRollBackups</code>}.   Moreover, <code>File</code> is
   renamed <code>File.1</code> and closed.<br>

   A new file is created to receive further log output.

   <p>If <code>maxSizeRollBackups</code> is equal to zero, then the
   <code>File</code> is truncated with no backup files created.

   <p>If <code>maxSizeRollBackups</code> < 0, then <code>File</code> is
   renamed if needed and no files are deleted.
  */

  // synchronization not necessary since doAppend is alreasy synched
  protected void rollOverSize() {
    File file;

    this.closeFile(); // keep windows happy.

    LogLog.debug("rolling over count=" + ((CountingQuietWriter) qw).getCount());
    LogLog.debug("maxSizeRollBackups = " + maxSizeRollBackups);
    LogLog.debug("curSizeRollBackups = " + curSizeRollBackups);
    LogLog.debug("countDirection = " + countDirection);

    // If maxBackups <= 0, then there is no file renaming to be done.
    if (maxSizeRollBackups != 0) {

      if (countDirection < 0) {
        // Delete the oldest file, to keep Windows happy.
        if (curSizeRollBackups == maxSizeRollBackups) {
            deleteFile(fileName + '.' + maxSizeRollBackups);
          curSizeRollBackups--;
        }

        // Map {(maxBackupIndex - 1), ..., 2, 1} to {maxBackupIndex, ..., 3, 2}
        for (int i = curSizeRollBackups; i >= 1; i--) {
          rollFile((fileName + "." + i), (fileName + '.' + (i + 1)));
        }

        curSizeRollBackups++;
        // Rename fileName to fileName.1
        rollFile(fileName, fileName + ".1");

      } //REMOVE This code branching for Alexander Cerna's request
      else if (countDirection == 0) {
        //rollFile based on date pattern
        curSizeRollBackups++;
        now.setTime(System.currentTimeMillis());
        scheduledFilename = fileName + sdf.format(now);
        rollFile(fileName, scheduledFilename);
      }
      else { //countDirection > 0
        if (curSizeRollBackups >= maxSizeRollBackups && maxSizeRollBackups > 0) {
          //delete the first and keep counting up.
          int oldestFileIndex = curSizeRollBackups - maxSizeRollBackups + 1;
          deleteFile(fileName + '.' + oldestFileIndex);
        }

        if (staticLogFileName) {
          curSizeRollBackups++;
          rollFile(fileName, fileName + '.' + curSizeRollBackups);
        }
      }
    }

    try {
      // This will also close the file. This is OK since multiple
      // close operations are safe.
      this.setFile(baseFileName, false);
    }
    catch(IOException e) {
      LogLog.error("setFile("+fileName+", false) call failed.", e);
    }
  }

}
TOP

Related Classes of org.apache.log4j.CompositeRollingAppender

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.