util/FileStock.java

   1
   2 package org.de.metux.util;
   3
   4 import java.util.Hashtable;
   5 import java.util.Enumeration;
   6 import java.io.IOException;
   7 import java.io.File;
   8 import java.io.FileNotFoundException;
   9 import java.io.BufferedReader;
  10 import java.io.FileReader;
  11
  12 public class FileStock
  13 {
  14     class record
  15     {
  16         String filename;
  17         String md5;
  18     }
  19
  20     Hashtable ht;
  21
  22     public FileStock()
  23     {
  24         ht = new Hashtable();
  25     }
  26
  27     public String getmd5(String fn)
  28     {
  29         return getmd5(new File(fn));
  30     }
  31
  32     public String getmd5(File f)
  33     {
  34         return FileOps.md5(f);
  35     }
  36
  37     private void add(record r_new)
  38     {
  39         record r_old;
  40
  41         if ((r_old = (record)ht.get(r_new.md5))==null)
  42         {
  43             ht.put(r_new.md5,r_new);
  44             return;
  45         }
  46
  47         if (r_new.md5.equals(r_old.md5))
  48             return;
  49
  50         throw new RuntimeException("COLLISSION "+r_old.filename+" vs. "+r_new.filename);
  51     }
  52
  53     public void add(String [] fns)
  54         throws FileNotFoundException, IOException
  55     {
  56         if (fns==null)
  57             return;
  58
  59         for (int x=0; x<fns.length; x++)
  60             add(fns[x]);
  61     }
  62
  63     public void add(String fn)
  64         throws FileNotFoundException, IOException
  65     {
  66         record r = new record();
  67         r.filename = fn;
  68         r.md5      = getmd5(fn);
  69         add(r);
  70     }
  71
  72     public String[] getFilenames()
  73     {
  74         int count = ht.size();
  75         String fns[] = new String[count];
  76
  77         int x = 0;
  78         for (Enumeration e=ht.keys(); e.hasMoreElements(); )
  79         {
  80             record r = (record)ht.get((String)e.nextElement());
  81             fns[x++] = r.filename;
  82         }
  83         System.runFinalization();
  84         return fns;
  85     }
  86 }