[jboss-svn-commits] JBL Code SVN: r32810 - labs/jbossrules/trunk/drools-core/src/main/java/org/drools/audit.

[jboss-svn-commits at lists.jboss.org]

Author: tsurdilovic
Date: 2010-05-10 22:27:31 -0400 (Mon, 10 May 2010)
New Revision: 32810

Modified:
   labs/jbossrules/trunk/drools-core/src/main/java/org/drools/audit/WorkingMemoryFileLogger.java
Log:
JBRULES-2507

Modified: labs/jbossrules/trunk/drools-core/src/main/java/org/drools/audit/WorkingMemoryFileLogger.java
===================================================================
--- labs/jbossrules/trunk/drools-core/src/main/java/org/drools/audit/WorkingMemoryFileLogger.java	2010-05-11 01:36:18 UTC (rev 32809)
+++ labs/jbossrules/trunk/drools-core/src/main/java/org/drools/audit/WorkingMemoryFileLogger.java	2010-05-11 02:27:31 UTC (rev 32810)
@@ -54,6 +54,7 @@
     private int        nbOfFile          = 0;
     private boolean    split             = true;
     private boolean    initialized       = false;
+    private boolean    finalized         = false;
 
     public WorkingMemoryFileLogger() {
     }
@@ -133,8 +134,25 @@
         } finally {
             if( fileWriter != null ) { try { fileWriter.close(); } catch(Exception e) {} }
         }
+        
+        if (!finalized) {
+        	finalizeLog();
+        }
     }
 
+    private void finalizeLog() {
+    	try {
+            FileWriter writer = new FileWriter(this.fileName + (this.nbOfFile == 0 ? ".log" : this.nbOfFile + ".log"), true);
+            writer.append("</object-stream>\n");
+            writer.close();
+            finalized = true;
+        } catch ( final FileNotFoundException exc ) {
+            throw new RuntimeException( "Could not create the log file.  Please make sure that directory that the log file should be placed in does exist." );
+        } catch ( final Throwable t ) {
+            t.printStackTrace( System.err );
+        }
+    }
+    
     private void initializeLog() {
         try {
             FileWriter writer = new FileWriter(this.fileName + (this.nbOfFile == 0 ? ".log" : this.nbOfFile + ".log"), false);