gw.util.cache.FqnCacheNode Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gosu-core-api Show documentation
There is a newer version: 1.18.2
/*
 * Copyright 2014 Guidewire Software, Inc.
 */

package gw.util.cache;

import gw.util.Predicate;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

public class FqnCacheNode {
  private final String name;
  private final FqnCacheNode parent;
  private K userData;
  private Map> children;

  public FqnCacheNode( String text, FqnCacheNode parent ) {
    this.name = text;
    this.parent = parent;
  }

  public final FqnCacheNode getChild( String segment ) {
    if (children != null) {
      return children.get(segment);
    } else {
      return null;
    }
  }

  public final void clear() {
    children = null;
  }

  public FqnCacheNode getOrCreateChild( String segment ) {
    if (children == null) {
      children = new ConcurrentHashMap>(2);
    }
    FqnCacheNode node = children.get(segment);
    if (node == null) {
      node = new FqnCacheNode(segment, this);
      children.put(segment, node);
    }
    return node;
  }

  public final void delete() {
    parent.deleteChild(this);
  }

  private void deleteChild(FqnCacheNode child) {
    if (children != null) {
      children.remove(child.name);
      if (children.isEmpty()) {
        children = null;
      }
    }
  }

  public final K getUserData() {
    return userData;
  }

  public final void setUserData( K userData ) {
    this.userData = userData;
  }

  public final boolean isLeaf() {
    return children == null || children.isEmpty();
  }

  public void collectNames( Set names, String s ) {
    if (children != null) {
      for (FqnCacheNode child : children.values()) {
        String path = s.length() == 0
                      ? child.name
                      : s + child.separator() + child.name;
        if (child.isLeaf()) {
          names.add(path);
        } else {
          child.collectNames(names, path);
        }
      }
    }
  }

  public final Collection> getChildren() {
    if (children != null) {
      return children.values();
    } else {
      return Collections.emptySet();
    }
  }

  public final boolean visitDepthFirst( Predicate visitor ) {
    if (children != null) {
      List> copy = new ArrayList>( children.values() );
      for( FqnCacheNode child: copy ) {
        if( !child.visitDepthFirst( visitor ) ) {
          return false;
        }
      }
    }
    return visitor.evaluate( getUserData() );
  }

  public final boolean visitNodeDepthFirst( Predicate visitor ) {
    if (children != null) {
      List> copy = new ArrayList>( children.values() );
      for( FqnCacheNode child: copy ) {
        if( !child.visitNodeDepthFirst( visitor ) ) {
          return false;
        }
      }
    }
    return visitor.evaluate( this );
  }

  public final boolean visitBreadthFirst( Predicate visitor ) {
    if( !visitor.evaluate( getUserData() ) ) {
      return false;
    }
    if (children != null) {
      List> copy = new ArrayList>( children.values() );
      for( FqnCacheNode child: copy ) {
        child.visitBreadthFirst( visitor );
      }
    }
    return true;
  }

  public final boolean visitNodeBreadthFirst( Predicate visitor ) {
    if( !visitor.evaluate( this ) ) {
      return false;
    }
    if( children != null ) {
      List> copy = new ArrayList>( children.values() );
      for( FqnCacheNode child: copy ) {
        child.visitNodeBreadthFirst( visitor );
      }
    }
    return true;
  }

  public final String getName() {
    return name;
  }

  public final String getFqn() {
    StringBuilder sb = new StringBuilder();
    FqnCacheNode node = this;
    while( node.parent != null ) {
        String str = node.name + (sb.length() == 0 ? "" : separator());
        sb.insert( 0, str);
        node = node.parent;
    }
    return sb.toString();
  }

  private String separator() {
      char c = name.charAt(0);
      return c == '[' || c == '<'? "" : ".";
  }

  @Override
  public String toString() {
    return name;
  }
}