Changeset 5327

Timestamp:

01/18/11 16:52:20 (14 years ago)

Author:

ascheibe

Message:

#1233 add auto cleanup to PluginCache

File:

• branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave/3.4/PluginCache.cs (modified) (5 diffs)

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave/3.4/PluginCache.cs

@@ -26 +26 @@
 using System.Reflection;
 using System.Runtime.CompilerServices;
+using System.Runtime.Serialization;
+using System.Runtime.Serialization.Formatters.Binary;
 using System.Threading;
 using HeuristicLab.Services.Hive.Common.DataTransfer;
@@ -33 +35 @@
     private static object locker = new object();
     public const string ConfigFileName = "Sandbox.config";
+    private const string lastUsedFileName = "lastUsed.dat";
+
+    //maximum number of days after which a plugin gets deleted if not used
+    private const int maxAge = 5;
 
     private static PluginCache instance = null;
@@ -95 +101 @@
           string curPath = Path.Combine(PluginCacheDir, requestedPlugin.Id.ToString());
           if (Directory.Exists(curPath)) {
+            writeDateLastUsed(curPath);
+
             foreach (string file in Directory.GetFiles(curPath)) {
               string fn = getFilenameFromPath(file);
-              File.Copy(file, Path.Combine(targetDir, fn));
+              if (fn != lastUsedFileName)
+                File.Copy(file, Path.Combine(targetDir, fn));
             }
           }
@@ -183 +192 @@
       }
     }
+
+    /// <summary>
+    /// creates a file in path with the current date;
+    /// this can later be used to find plugins which are outdated
+    /// </summary>    
+    private void writeDateLastUsed(string path) {
+      FileStream fs = new FileStream(Path.Combine(path, lastUsedFileName), FileMode.Create);
+      BinaryFormatter formatter = new BinaryFormatter();
+
+      try {
+        formatter.Serialize(fs, DateTime.Now);
+      }
+      catch (SerializationException) {
+        //rethrow...
+        throw;
+      }
+      finally {
+        fs.Close();
+      }
+    }
+
+    /// <summary>
+    /// checks the pluginCacheDirectory and deletes plugin folders which are not used anymore
+    /// </summary>
+    /// <param name="path"></param>
+    private void cleanPluginCache() {
+      FileStream fs = null;
+      DateTime luDate;
+      bool changed = false;
+
+      if (Directory.Exists(PluginCacheDir)) {
+        foreach (string dir in Directory.EnumerateDirectories(PluginCacheDir)) {
+          try {
+            fs = new FileStream(Path.Combine(dir, lastUsedFileName), FileMode.Open);
+            BinaryFormatter formatter = new BinaryFormatter();
+            luDate = (DateTime)formatter.Deserialize(fs);
+            fs.Close();
+
+            if (luDate.AddDays(maxAge) < DateTime.Now) {
+              Directory.Delete(dir, true);
+              changed = true;
+            }
+          }
+          catch (SerializationException) {
+            fs.Close();
+            throw;
+          }
+          catch (System.IO.FileNotFoundException) {
+            //nerver used
+            Directory.Delete(dir, true);
+            changed = true;
+          }
+          catch (Exception) {
+            throw;
+          }
+        }
+      }
+      if (changed)
+        DoUpdateRun();
+    }
+
 
     internal void DeletePluginsForJob(Guid id) {
@@ -203 +273 @@
         SlaveClientCom.Instance.ClientCom.LogMessage("failed while unloading " + id + " with exception " + ex);
       }
+      cleanPluginCache();
     }
   }