io.netlibs.bgp.config.nodes.impl.CapabilitiesImpl.java Source code

Introduction

Here is the source code for io.netlibs.bgp.config.nodes.impl.CapabilitiesImpl.java

Source

/**
 *  Copyright 2012 Rainer Bieniek (Rainer.Bieniek@web.de)
 *
 *  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.
 *
 * File: org.bgp4.config.nodes.impl.CapabilitiesImpl.java
 */
package io.netlibs.bgp.config.nodes.impl;

import java.util.Collections;
import java.util.Set;
import java.util.TreeSet;

import org.apache.commons.lang3.builder.HashCodeBuilder;

import io.netlibs.bgp.config.nodes.Capabilities;
import io.netlibs.bgp.protocol.capabilities.Capability;

/**
 * @author Rainer Bieniek (rainer@bgp4j.org)
 *
 */
public class CapabilitiesImpl implements Capabilities {

    private final TreeSet<Capability> requiredCapabilities = new TreeSet<Capability>();
    private final TreeSet<Capability> optionalCapabilities = new TreeSet<Capability>();

    public CapabilitiesImpl() {
    }

    public CapabilitiesImpl(final Capability[] requiredCaps) {
        if (requiredCaps != null) {
            for (final Capability cap : requiredCaps) {
                this.requiredCapabilities.add(cap);
            }
        }
    }

    public CapabilitiesImpl(final Capability[] requiredCaps, final Capability[] optionalCaps) {
        this(requiredCaps);

        if (optionalCaps != null) {
            for (final Capability cap : optionalCaps) {
                this.optionalCapabilities.add(cap);
            }
        }
    }

    @Override
    public Set<Capability> getRequiredCapabilities() {
        return Collections.unmodifiableSet(this.requiredCapabilities);
    }

    void addRequiredCapability(final Capability cap) {
        this.requiredCapabilities.add(cap);
    }

    /*
     * (non-Javadoc)
     *
     * @see java.lang.Object#hashCode()
     */
    @Override
    public int hashCode() {
        final HashCodeBuilder hcb = new HashCodeBuilder();

        for (final Capability cap : this.requiredCapabilities) {
            hcb.append(cap).append(false);
        }

        for (final Capability cap : this.optionalCapabilities) {
            hcb.append(cap).append(true);
        }

        return hcb.toHashCode();
    }

    /*
     * (non-Javadoc)
     *
     * @see java.lang.Object#equals(java.lang.Object)
     */
    @Override
    public boolean equals(final Object obj) {
        if (!(obj instanceof Capabilities)) {
            return false;
        }

        Set<Capability> otherCaps = ((Capabilities) obj).getRequiredCapabilities();

        if (otherCaps.size() != this.requiredCapabilities.size()) {
            return false;
        }

        for (final Capability cap : this.requiredCapabilities) {
            if (!otherCaps.contains(cap)) {
                return false;
            }
        }

        otherCaps = ((Capabilities) obj).getOptionalCapabilities();

        if (otherCaps.size() != this.optionalCapabilities.size()) {
            return false;
        }

        for (final Capability cap : this.optionalCapabilities) {
            if (!otherCaps.contains(cap)) {
                return false;
            }
        }
        return true;
    }

    @Override
    public Set<Capability> getOptionalCapabilities() {
        return Collections.unmodifiableSet(this.optionalCapabilities);
    }

    void addOptionalCapability(final Capability cap) {
        this.optionalCapabilities.add(cap);
    }

}