src/main/java/org/hnaves/dfs/repository/Container.java

   1 package org.hnaves.dfs.repository;
   2
   3 import java.io.File;
   4 import java.io.Serializable;
   5 import java.util.ArrayList;
   6 import java.util.HashMap;
   7 import java.util.logging.Logger;
   8
   9 import org.hnaves.dfs.Configurator;
  10 import org.hnaves.dfs.election.Node;
  11
  12 public class Container implements Serializable {
  13         private static final long serialVersionUID = 4190763929963328455L;
  14         private static final Logger LOG = Logger.getLogger(Container.class.getName());
  15
  16         private final Node location;
  17         private final ArrayList<ContentItem> content = new ArrayList<ContentItem>();
  18         private transient long lastUpdate;
  19
  20         public Container(Node location) {
  21                 if (location == null) throw new NullPointerException("Location is null");
  22                 this.location = location;
  23         }
  24
  25         public Node getLocation() {
  26                 return location;
  27         }
  28
  29         public ContentItem[] getContent() {
  30                 return content.toArray(new ContentItem[0]);
  31         }
  32
  33         public int getSize() {
  34                 return content.size();
  35         }
  36
  37         public void clearContent() {
  38                 content.clear();
  39         }
  40
  41         public void addItem(ContentItem item) {
  42                 if (item == null) throw new NullPointerException("Item is null");
  43                 content.add(item);
  44         }
  45
  46         public boolean containsItem(ContentItem item) {
  47                 return content.contains(item);
  48         }
  49
  50         public void addAllItems(Container c) {
  51                 if (c == null) throw new NullPointerException("Container is null");
  52                 content.addAll(c.content);
  53         }
  54
  55         public void update() {
  56                 this.lastUpdate = System.currentTimeMillis();
  57         }
  58
  59         public boolean hasExpired() {
  60                 return (System.currentTimeMillis() > lastUpdate + Configurator.getExpireTime());
  61         }
  62
  63         public void updateFiles(File directory) {
  64                 if (directory == null) throw new NullPointerException("Directory is null");
  65                 if (!directory.isDirectory()) throw new IllegalArgumentException("Invalid directory " + directory);
  66
  67                 HashMap<String, ContentItem> oldContent = new HashMap<String, ContentItem>();
  68                 for(ContentItem item : content) {
  69                         oldContent.put(item.getName(), item);
  70                 }
  71
  72                 content.clear();
  73                 for(File file : directory.listFiles()) {
  74                         if (!file.isFile()) continue;
  75                         if (file.getName().startsWith(".")) continue;
  76                         if (file.getName().startsWith("dfs.log")) continue;
  77                         if (file.getName().equals("index.dat") || file.getName().equals("config.ini")) continue;
  78
  79                         LOG.fine("Found file " + file.getName());
  80                         ContentItem oldItem = oldContent.get(file.getName());
  81                         if (oldItem != null && oldItem.getLastModified().getTime() == file.lastModified() &&
  82                                         (oldItem.getSize() == (int) file.length())) {
  83                                 LOG.fine("Using old hash " + oldItem.getMd5Hash());
  84                                 content.add(oldItem);
  85                         } else {
  86                                 LOG.fine("New item!");
  87                                 ContentItem item = new ContentItem(file);
  88                                 content.add(item);
  89                         }
  90                 }
  91         }
  92
  93         @Override
  94         public boolean equals(Object obj) {
  95                 if (obj == this) return true;
  96                 if (!(obj instanceof Container)) return false;
  97                 Container other = (Container) obj;
  98                 return location.equals(other.location);
  99         }
 100
 101         @Override
 102         public int hashCode() {
 103                 return location.hashCode();
 104         }
 105
 106         @Override
 107         public String toString() {
 108                 StringBuilder sb = new StringBuilder();
 109                 sb.append("Location: ").append(location).append("\n");
 110                 for(ContentItem item : content) {
 111                         sb.append(item).append('\n');
 112                 }
 113                 return sb.toString();
 114         }
 115 }