added archive job

Author: janithb

src/main/java/org/audit4j/core/handler/file/ArchiveJob.java

@@ -0,0 +1,17 @@
+package org.audit4j.core.handler.file;
+
+public abstract class ArchiveJob {
+	protected Integer archiveDateDiff;
+
+	protected String path;
+
+	abstract void archive();
+
+	public void setArchiveDateDiff(Integer archiveDateDiff) {
+		this.archiveDateDiff = archiveDateDiff;
+	}
+
+	public void setPath(String path) {
+		this.path = path;
+	}
+}

src/main/java/org/audit4j/core/handler/file/ArchiveManager.java

@@ -0,0 +1,66 @@
+package org.audit4j.core.handler.file;
+
+import it.sauronsoftware.cron4j.Scheduler;
+
+public class ArchiveManager {
+
+	private String datePattern;
+
+	private String path;
+
+	private String cronPattern;
+
+	private ArchiveJob job;
+
+	public void init() {
+		Scheduler s = new Scheduler();
+		System.out.println("Start Archive Manager");
+		s.schedule(cronPattern, new Runnable() {
+			@Override
+			public void run() {
+				System.out.println("Start Archive Manager");
+				ArchiveJob archiveJob = job;
+				archiveJob.setArchiveDateDiff(extractArchiveDateCount(datePattern));
+				archiveJob.setPath(path);
+				archiveJob.archive();
+			}
+		});
+	}
+
+	public Integer extractArchiveDateCount(String datePattern) {
+		int dateCount = 0;
+		String[] splits = datePattern.split("d|M|y");
+		if (splits.length>1) {
+			dateCount = dateCount + Integer.valueOf(splits[0]);
+		}
+		
+		System.out.println(dateCount);
+		
+		if (splits.length>2) {
+			dateCount = dateCount + (Integer.valueOf(splits[1]) * 30);
+		}
+		System.out.println(dateCount);
+		
+		if (splits.length>3) {
+			dateCount = dateCount + (Integer.valueOf(splits[2]) * 365);
+		}
+		System.out.println(dateCount);
+		return dateCount;
+	}
+
+	public void setArchiveDate(String datePattern) {
+		this.datePattern = datePattern;
+	}
+
+	public void setPath(String path) {
+		this.path = path;
+	}
+
+	public void setCronPattern(String cronPattern) {
+		this.cronPattern = cronPattern;
+	}
+
+	public void setJob(ArchiveJob job) {
+		this.job = job;
+	}
+}

src/main/java/org/audit4j/core/handler/file/FileArchiveJob.java

@@ -0,0 +1,60 @@
+package org.audit4j.core.handler.file;
+
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.Date;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+import org.audit4j.core.Log;
+
+public class FileArchiveJob extends ArchiveJob {
+
+	byte[] buffer = new byte[1024];
+
+	@Override
+	public void archive() {
+		Log.info("Starting archiving...");
+		String fileName = FileHandlerUtil.generateAuditFileName(FileHandlerUtil.substractDate(new Date(),
+				archiveDateDiff));
+		String archiveFileName = FileHandlerUtil.generateAuditArchiveFileName(FileHandlerUtil.substractDate(new Date(),
+				archiveDateDiff));
+
+		String filePath = FileHandlerUtil.generateOutputFilePath(path, fileName);
+		String archivePath = FileHandlerUtil.generateOutputFilePath(path, archiveFileName);
+
+		if (FileHandlerUtil.isFileAlreadyExists(filePath)) {
+
+			FileOutputStream fos = null;
+			try {
+				fos = new FileOutputStream(archivePath);
+			} catch (FileNotFoundException e) {
+				// TODO Auto-generated catch block
+				e.printStackTrace();
+			}
+			ZipOutputStream zos = new ZipOutputStream(fos);
+			ZipEntry ze = new ZipEntry(fileName);
+			try {
+				zos.putNextEntry(ze);
+
+				FileInputStream in = new FileInputStream(filePath);
+
+				int len;
+				while ((len = in.read(buffer)) > 0) {
+					zos.write(buffer, 0, len);
+				}
+
+				in.close();
+				zos.closeEntry();
+				zos.close();
+				System.out.println("Done");
+			} catch (IOException e) {
+				// TODO Auto-generated catch block
+				e.printStackTrace();
+			}
+		}
+	}
+
+}

src/main/java/org/audit4j/core/handler/file/FileAuditHandler.java

@@ -14,14 +14,26 @@
  */
 public class FileAuditHandler extends Handler {
 
-	/**
-	 * asdas
-	 */
+	/** asdas. */
 	private static final long serialVersionUID = 1L;
 
 	/** The writer. */
 	ZeroCopyFileWriter writer;
 
+	private String archive;
+	
+	/** The date pattern. */
+	private String datePattern;
+
+	/** The path. */
+	private String path;
+
+	/** The cron pattern. */
+	private String cronPattern;
+
+	/** The job. */
+	private ArchiveJob job;
+
 	/*
 	 * (non-Javadoc)
 	 * 
@@ -30,10 +42,19 @@ public class FileAuditHandler extends Handler {
 	@Override
 	public void init() throws InitializationException {
 		writer = new ZeroCopyFileWriter(getProperty("log.file.location"));
-		if (!hasDiskAccess(getProperty("log.file.location"))) {
-			throw new InitializationException(
-					"Can not write a file. Disk is not accessible. Please set read,write permission for "
-							+ getProperty("log.file.location"));
+		// if (!hasDiskAccess(getProperty("log.file.location"))) {
+		// throw new InitializationException(
+		// "Can not write a file. Disk is not accessible. Please set read,write permission for "
+		// + getProperty("log.file.location"));
+		// }
+		
+		if (null == archive || "true".equals(archive)) {
+			ArchiveManager manager = new ArchiveManager();
+			manager.setArchiveDate(datePattern);
+			manager.setCronPattern(cronPattern);
+			manager.setPath(getProperty("log.file.location"));
+			manager.setJob(job);
+			manager.init();
 		}
 	}
 
@@ -44,6 +65,8 @@ public void init() throws InitializationException {
 	 */
 	@Override
 	public void handle() {
+		System.out.println("inside handler");
+		System.out.println("writer" + writer);
 		writer.write(getQuery());
 	}
 
@@ -62,5 +85,37 @@ static boolean hasDiskAccess(final String path) {
 			return false;
 		}
 	}
+	
+	
+
+	public void setArchive(String archive) {
+		this.archive = archive;
+	}
 
+	/**
+	 * Sets the date pattern.
+	 *
+	 * @param datePattern the new date pattern
+	 */
+	public void setDatePattern(String datePattern) {
+		this.datePattern = datePattern;
+	}
+
+	/**
+	 * Sets the path.
+	 *
+	 * @param path the new path
+	 */
+	public void setPath(String path) {
+		this.path = path;
+	}
+
+	/**
+	 * Sets the cron pattern.
+	 *
+	 * @param cronPattern the new cron pattern
+	 */
+	public void setCronPattern(String cronPattern) {
+		this.cronPattern = cronPattern;
+	}
 }

src/test/java/org/audit4j/core/handler/file/ArchiveManagerTest.java

@@ -0,0 +1,31 @@
+package org.audit4j.core.handler.file;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+public class ArchiveManagerTest {
+
+	@Test
+	public void testExtractArchiveDateCount_d(){
+		ArchiveManager manager  = new ArchiveManager();
+		int numOfdates = manager.extractArchiveDateCount("1d");
+		Assert.assertEquals(1, numOfdates);
+		
+	}
+	
+	@Test
+	public void testExtractArchiveDateCount_d_M(){
+		ArchiveManager manager  = new ArchiveManager();
+		int numOfdates = manager.extractArchiveDateCount("1d1M");
+		Assert.assertEquals(31, numOfdates);
+		
+	}
+	
+	@Test
+	public void testExtractArchiveDateCount_d_M_y(){
+		ArchiveManager manager  = new ArchiveManager();
+		int numOfdates = manager.extractArchiveDateCount("1d1M1y");
+		Assert.assertEquals(396, numOfdates);
+		
+	}
+}