org.drools.impact.analysis.graph.Node Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2020 Red Hat, Inc. and/or its affiliates.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.drools.impact.analysis.graph;

import java.util.HashSet;
import java.util.Optional;
import java.util.Set;

import org.drools.impact.analysis.model.Rule;

public class Node {

    public enum Status {
        NONE,
        CHANGED,
        IMPACTED,
        TARGET,
        IMPACTING
    }

    private String packageName;
    private String ruleName;
    private Optional rule;
    protected Status status = Status.NONE;

    private Set incomingLinks;
    private Set outgoingLinks;

    public Node(Rule rule) {
        this.packageName = rule.getPkg();
        this.ruleName = rule.getName();
        this.rule = Optional.of(rule);
        incomingLinks = new HashSet<>();
        outgoingLinks = new HashSet<>();
    }

    /**
     * This constructor is used for a case where Node to Rule is not 1-to-1 (e.g. collapseWithRuleNamePrefix)
     * @param packageName
     * @param ruleName
     */
    public Node(String packageName, String ruleName) {
        this.packageName = packageName;
        this.ruleName = ruleName;
        this.rule = Optional.empty();
        incomingLinks = new HashSet<>();
        outgoingLinks = new HashSet<>();
    }

    public Status getStatus() {
        return status;
    }

    public void setStatus(Status status) {
        this.status = status;
    }

    public String getId() {
        return getFqdn();
    }

    public String getFqdn() {
        return packageName + "." + ruleName;
    }

    public String getPackageName() {
        return packageName;
    }

    public String getRuleName() {
        return ruleName;
    }

    public Optional getRule() {
        return rule;
    }

    public Set getIncomingLinks() {
        return incomingLinks;
    }

    public Set getOutgoingLinks() {
        return outgoingLinks;
    }

    public static void linkNodes(Node source, Node target, ReactivityType type) {
        // TODO: We may omit a link to oneself (Or it may be decided when rendering)
        Link link = new Link(source, target, type);
        source.getOutgoingLinks().add(link);
        target.getIncomingLinks().add(link);
    }

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((packageName == null) ? 0 : packageName.hashCode());
        result = prime * result + ((ruleName == null) ? 0 : ruleName.hashCode());
        return result;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        Node other = (Node) obj;
        if (packageName == null) {
            if (other.packageName != null)
                return false;
        } else if (!packageName.equals(other.packageName))
            return false;
        if (ruleName == null) {
            if (other.ruleName != null)
                return false;
        } else if (!ruleName.equals(other.ruleName))
            return false;
        return true;
    }

    @Override
    public String toString() {
        return "Node [packageName=" + packageName + ", ruleName=" + ruleName + ", outgoingLinks=" + outgoingLinks + "]";
    }
}