/*
 * Copyright 2015-present Open Networking Foundation
 *
 * 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.onosproject.incubator.net.virtual;

import org.onosproject.net.TenantId;

import java.util.Objects;

import static com.google.common.base.MoreObjects.toStringHelper;

/**
 * Default implementation of the virtual network descriptor.
 */
public final class DefaultVirtualNetwork implements VirtualNetwork {

    private final NetworkId id;
    private final TenantId tenantId;

    /**
     * Creates a new virtual network descriptor.
     *
     * @param id       network identifier
     * @param tenantId tenant identifier
     */
    public DefaultVirtualNetwork(NetworkId id, TenantId tenantId) {
        this.id = id;
        this.tenantId = tenantId;
    }

    @Override
    public NetworkId id() {
        return id;
    }

    @Override
    public TenantId tenantId() {
        return tenantId;
    }

    @Override
    public int hashCode() {
        return Objects.hash(id, tenantId);
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj instanceof DefaultVirtualNetwork) {
            DefaultVirtualNetwork that = (DefaultVirtualNetwork) obj;
            return Objects.equals(this.id, that.id)
                    && Objects.equals(this.tenantId, that.tenantId);
        }
        return false;
    }

    @Override
    public String toString() {
        return toStringHelper(this)
                .add("id", id)
                .add("tenantId", tenantId)
                .toString();
    }
}
