appender log4j 扩展_log4j扩展Appender

package com.trunkbow.epay.advice.log4j;

import java.io.File;

import java.io.IOException;

import java.io.InterruptedIOException;

import java.io.Writer;

import java.text.SimpleDateFormat;

import java.util.Calendar;

import java.util.Date;

import java.util.GregorianCalendar;

import java.util.Locale;

import java.util.TimeZone;

import org.apache.log4j.FileAppender;

import org.apache.log4j.Layout;

import org.apache.log4j.helpers.CountingQuietWriter;

import org.apache.log4j.helpers.LogLog;

import org.apache.log4j.helpers.OptionConverter;

import org.apache.log4j.spi.LoggingEvent;/**

* 用于实现文件按照日期和文件大小保存日志

*

* @author wwl

**/

public classMyLog4jFileAppender extends FileAppender {//The code assumes that the following 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;static int fileNum=1;/**

* The default maximum file size is 10MB.*/

protected long maxFileSize = 10 * 1024 * 1024;/**

* There is one backup file by default.*/

protected int maxBackupIndex = 1;/**

* The date pattern. By default, the pattern is set to "'.'yyyy-MM-dd"

* meaning daily rollover.*/

private String datePattern = "'.'yyyy-MM-dd";private String myFilePattern="yyyyMMdd_HH_mm_ss";/**

* The log file will be renamed to the value of the scheduledFilename

* variable when the next interval is entered. For example, if the rollover

* period is one hour, the log file will be renamed to the value of

* "scheduledFilename" at the beginning of the next hour.

*

* The precise time when a rollover occurs depends on logging activity.*/

privateString scheduledFilename;/**

* The next time we estimate a rollover should occur.*/

private long nextCheck = System.currentTimeMillis() - 1;

Date now= newDate();

SimpleDateFormat sdf;

SimpleDateFormat mySdf;

RollingCalendar rc= newRollingCalendar();int checkPeriod =TOP_OF_TROUBLE;//The gmtTimeZone is used only in computeCheckPeriod() method.

static final TimeZone gmtTimeZone = TimeZone.getTimeZone("GMT");/**

* The default constructor does nothing.*/

publicMyLog4jFileAppender() {

}/**

* Instantiate a MyLog4jFileAppender and open the file

* designated by filename. The opened filename will become the

* ouput destination for this appender.*/

publicMyLog4jFileAppender(Layout layout, String filename,

String datePattern) throws IOException {

super(layout, filename,true);this.datePattern =datePattern;

activateOptions();

}/**

* The DatePattern takes a string in the same format as expected by

* {@link SimpleDateFormat}. This options determines the rollover schedule.*/

public voidsetDatePattern(String pattern) {

datePattern=pattern;

}/** Returns the value of the DatePattern option.*/

publicString getDatePattern() {returndatePattern;

}public longgetMaximumFileSize() {returnmaxFileSize;

}public intgetMaxBackupIndex() {returnmaxBackupIndex;

}public voidsetMaxFileSize(String value) {

maxFileSize= OptionConverter.toFileSize(value, maxFileSize + 1);

}public void setMaxBackupIndex(intmaxBackups) {this.maxBackupIndex =maxBackups;

}public voidsetMyFilePattern(String myFilePattern) {this.myFilePattern =myFilePattern;

}public voidactivateOptions() {

super.activateOptions();if (datePattern != null && fileName != null) {

now.setTime(System.currentTimeMillis());

sdf= newSimpleDateFormat(datePattern);int type =computeCheckPeriod();

printPeriodicity(type);

rc.setType(type);

File file= newFile(fileName);

scheduledFilename=fileName+ sdf.format(newDate(file.lastModified()));

}else{

LogLog.error("Either File or DatePattern options are not set for appender ["

+ name + "].");

}

}void printPeriodicity(inttype) {switch(type) {caseTOP_OF_MINUTE:

LogLog.debug("Appender [" + name + "] to be rolled every minute.");break;caseTOP_OF_HOUR:

LogLog.debug("Appender [" +name+ "] to be rolled on top of every hour.");break;caseHALF_DAY:

LogLog.debug("Appender [" +name+ "] to be rolled at midday and midnight.");break;caseTOP_OF_DAY:

LogLog.debug("Appender [" + name + "] to be rolled at midnight.");break;caseTOP_OF_WEEK:

LogLog.debug("Appender [" +name+ "] to be rolled at start of week.");break;caseTOP_OF_MONTH:

LogLog.debug("Appender [" +name+ "] to be rolled at start of every month.");break;default:

LogLog.warn("Unknown periodicity for appender [" + name + "].");

}

}//This method computes the roll over period by looping over the//periods, starting with the shortest, and stopping when the r0 is//different from from r1, where r0 is the epoch formatted according//the datePattern (supplied by the user) and r1 is the//epoch+nextMillis(i) formatted according to datePattern. All date//formatting is done in GMT and not local format because the test//logic is based on comparisons relative to 1970-01-01 00:00:00//GMT (the epoch).

intcomputeCheckPeriod() {

RollingCalendar rollingCalendar= newRollingCalendar(gmtTimeZone,

Locale.getDefault());//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++) {

SimpleDateFormat simpleDateFormat= newSimpleDateFormat(

datePattern);

simpleDateFormat.setTimeZone(gmtTimeZone);//do all date//formatting in GMT

String r0 =simpleDateFormat.format(epoch);

rollingCalendar.setType(i);

Date next= newDate(rollingCalendar.getNextCheckMillis(epoch));

String r1=simpleDateFormat.format(next);//System.out.println("Type = "+i+", r0 = "+r0+", r1 = "+r1);

if (r0 != null && r1 != null && !r0.equals(r1)) {returni;

}

}

}return TOP_OF_TROUBLE; //Deliberately head for trouble...

}/**

* Rollover the current file to a new file.*/

voidrollOver() throws IOException {/*Compute filename, but only if datePattern is specified*/

if (datePattern == null) {

errorHandler.error("Missing DatePattern option in rollOver().");return;

}

String datedFilename= fileName+sdf.format(now);//It is too early to roll over because we are still within the//bounds of the current interval. Rollover will occur once the//next interval is reached.

if(scheduledFilename.equals(datedFilename)) {

myRollOver();return;

}//close current file, and rename it to datedFilename

this.closeFile();

File target= newFile(scheduledFilename);if(target.exists()) {

target.delete();

}

File file= newFile(fileName);

boolean result=file.renameTo(target);if(result) {

LogLog.debug(fileName+"->"+scheduledFilename);

}else{

LogLog.error("Failed to rename ["+fileName+"] to ["+scheduledFilename+"].");

}try{//This will also close the file. This is OK since multiple//close operations are safe.

this.setFile(fileName, true, this.bufferedIO, this.bufferSize);

}catch(IOException e) {

errorHandler.error("setFile("+fileName+", true) call failed.");

}

scheduledFilename=datedFilename;

}/**

* This method differentiates MyLog4jFileAppender from its super class.

*

*

* Before actually logging, this method will check whether it is time to do

* a rollover. If it is, it will schedule the next rollover time and then

* rollover.

**/

protected void subAppend(LoggingEvent event) {long n =System.currentTimeMillis();if (n >=nextCheck) {

now.setTime(n);

nextCheck=rc.getNextCheckMillis(now);try{

rollOver();

}catch(IOException ioe) {if(ioe instanceof InterruptedIOException) {

Thread.currentThread().interrupt();

}

LogLog.error("rollOver() failed.", ioe);

}

}else{

mySubAppend();

}

super.subAppend(event);

}public//synchronization not necessary since doAppend is alreasy synched

voidmyRollOver() {/*Compute filename, but only if datePattern is specified*/

if (datePattern == null) {

errorHandler.error("Missing DatePattern option in rollOver().");return;

}

File file= newFile(fileName);//close current file, and rename it to datedFilename

this.closeFile();

mySdf=newSimpleDateFormat(myFilePattern);

String datedFilename= fileName +"."+ mySdf.format(new Date(file.lastModified()))+"_"+fileNum++;

File target= newFile(datedFilename);if(target.exists()) {

target.delete();

}if(file.exists()) {

LogLog.debug(fileName+ "->" +fileName);

}

boolean result=file.renameTo(target);if(result) {

LogLog.debug(fileName+ "->" +scheduledFilename);

}else{

LogLog.error("Failed to rename [" + fileName + "] to ["

+ scheduledFilename + "].");

}try{//This will also close the file. This is OK since multiple//close operations are safe.

this.setFile(fileName, true, this.bufferedIO, this.bufferSize);

}catch(IOException e) {

errorHandler.error("setFile(" + fileName + ", true) call failed.");

}

}protected voidmySubAppend() {if (fileName != null && qw != null) {long size =((CountingQuietWriter) qw).getCount();if (size >=maxFileSize) {try{

myRollOver();

}catch(Exception e) {

LogLog.error("rollOver() failed.", e);

}

}

}

}protected voidsetQWForFiles(Writer writer) {this.qw = newCountingQuietWriter(writer, errorHandler);

}

}/**

* RollingCalendar is a helper class to MyLog4jFileAppender. Given a periodicity

* type and the current time, it computes the start of the next interval.

**/

classRollingCalendar extends GregorianCalendar {private static final long serialVersionUID = -3560331770601814177L;int type =MyLog4jFileAppender.TOP_OF_TROUBLE;

RollingCalendar() {

super();

}

RollingCalendar(TimeZone tz, Locale locale) {

super(tz, locale);

}void setType(inttype) {this.type =type;

}public longgetNextCheckMillis(Date now) {returngetNextCheckDate(now).getTime();

}publicDate getNextCheckDate(Date now) {this.setTime(now);switch(type) {caseMyLog4jFileAppender.TOP_OF_MINUTE:this.set(Calendar.SECOND, 0);this.set(Calendar.MILLISECOND, 0);this.add(Calendar.MINUTE, 1);break;caseMyLog4jFileAppender.TOP_OF_HOUR:this.set(Calendar.MINUTE, 0);this.set(Calendar.SECOND, 0);this.set(Calendar.MILLISECOND, 0);this.add(Calendar.HOUR_OF_DAY, 1);break;caseMyLog4jFileAppender.HALF_DAY:this.set(Calendar.MINUTE, 0);this.set(Calendar.SECOND, 0);this.set(Calendar.MILLISECOND, 0);int hour = get(Calendar.HOUR_OF_DAY);if (hour < 12) {this.set(Calendar.HOUR_OF_DAY, 12);

}else{this.set(Calendar.HOUR_OF_DAY, 0);this.add(Calendar.DAY_OF_MONTH, 1);

}break;caseMyLog4jFileAppender.TOP_OF_DAY:this.set(Calendar.HOUR_OF_DAY, 0);this.set(Calendar.MINUTE, 0);this.set(Calendar.SECOND, 0);this.set(Calendar.MILLISECOND, 0);this.add(Calendar.DATE, 1);break;caseMyLog4jFileAppender.TOP_OF_WEEK:this.set(Calendar.DAY_OF_WEEK, getFirstDayOfWeek());this.set(Calendar.HOUR_OF_DAY, 0);this.set(Calendar.MINUTE, 0);this.set(Calendar.SECOND, 0);this.set(Calendar.MILLISECOND, 0);this.add(Calendar.WEEK_OF_YEAR, 1);break;caseMyLog4jFileAppender.TOP_OF_MONTH:this.set(Calendar.DATE, 1);this.set(Calendar.HOUR_OF_DAY, 0);this.set(Calendar.MINUTE, 0);this.set(Calendar.SECOND, 0);this.set(Calendar.MILLISECOND, 0);this.add(Calendar.MONTH, 1);break;default:throw new IllegalStateException("Unknown periodicity type.");

}returngetTime();

}

}