Ejemplo n.º 1
0
 /**
  * update the quota for the given path
  *
  * @param path the path to be used
  */
 private void updateQuotaForPath(String path) {
   Counts c = new Counts();
   getCounts(path, c);
   StatsTrack strack = new StatsTrack();
   strack.setBytes(c.bytes);
   strack.setCount(c.count);
   String statPath = Quotas.quotaZookeeper + path + "/" + Quotas.statNode;
   DataNode node = getNode(statPath);
   // it should exist
   if (node == null) {
     LOG.warn("Missing quota stat node " + statPath);
     return;
   }
   synchronized (node) {
     node.data = strack.toString().getBytes();
   }
 }
Ejemplo n.º 2
0
 /**
  * this method gets the count of nodes and the bytes under a subtree
  *
  * @param path the path to be used
  * @param bytes the long bytes
  * @param count the int count
  */
 private void getCounts(String path, Counts counts) {
   DataNode node = getNode(path);
   if (node == null) {
     return;
   }
   String[] children = null;
   synchronized (node) {
     children = node.children.toArray(new String[node.children.size()]);
   }
   // add itself
   counts.count += 1;
   counts.bytes += (long) node.data.length;
   if (children.length == 0) {
     return;
   }
   for (String child : children) {
     getCounts(path + "/" + child, counts);
   }
 }