de.malkusch.whoisServerList.publicSuffixList.index.tree.Node Maven / Gradle / Ivy

Go to download
package de.malkusch.whoisServerList.publicSuffixList.index.tree;

import java.util.Arrays;
import java.util.Collection;
import java.util.Deque;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.Map;

import de.malkusch.whoisServerList.publicSuffixList.rule.Rule;
import de.malkusch.whoisServerList.publicSuffixList.util.DomainUtil;

/**
 * Tree Node.
 *
 * Operations on the node are case insensitive.
 *
 * @author [email protected]
 * @param  {@code Node} implementation
 * @see Donations
 */
abstract class Node> {

    /**
     * Domain label. This is the search index in its canonical form.
     *
     * @see TreeIndex#getCanonicalLabel(String)
     */
    private final String label;

    /**
     * The children mapped by their canonical labels.
     *
     * Accessing {@code Map#get(Object)} is thread-safe on this map.
     */
    private final Map children;

    /**
     * Sets the domain label.
     *
     * @param label  the domain label, may be null for the root
     */
    Node(final String label) {
        this(label, new HashMap());
    }

    /**
     * Sets the domain label and existing children.
     *
     * @param label     the domain label, may be null for the root
     * @param children  the children
     */
    Node(final String label, final Map children) {
        this.label = TreeIndex.getCanonicalLabel(label);
        this.children = children;
    }

    /**
     * Returns a child.
     *
     * @param childLabel  the case insensitive domain label, null returns null
     * @return the child, or null if the child doesn't exist
     */
    T getChild(final String childLabel) {
        return children.get(TreeIndex.getCanonicalLabel(childLabel));
    }

    /**
     * Returns the children.
     *
     * @return the children, not null
     */
    Collection getChildren() {
        return children.values();
    }

    /**
     * Returns the canonical label.
     *
     * @return the canonical label, or null for the root
     */
    String getLabel() {
        return label;
    }

    /**
     * Adds a child.
     *
     * This is done only during building the tree in
     * {@code TreeIndexFactory#build(java.util.List)}.
     *
     * @param node  the child, not null
     */
    void addChild(final T node) {
        addChild(node, children);
    }

    /**
     * Adds a child to a children map of a parent node.
     *
     * @param child     the new child, not null
     * @param children  the children map of the parent, not null
     * @param        the {@code Node} implementation
     */
    static > void addChild(
            final T child, final Map children) {

        children.put(child.getLabel(), child);
    }

    /**
     * Returns the rule.
     *
     * @return the rule, may be null
     */
    abstract Rule getRule();

    /**
     * Returns the wildcard child.
     *
     * @return the wildcard, may be null.
     * @see Rule#WILDCARD
     */
    T getWildcard() {
        return children.get(Rule.WILDCARD);
    }

    /**
     * Converts a domain name into its labels.
     *
     * @param domain  the domain name, not null
     * @return the domain labels
     */
    Deque convertDomain(final String domain) {
        String[] labels = DomainUtil.splitLabels(domain);
        return new LinkedList(Arrays.asList(labels));
    }

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

}