/*
 * Copyright 2014-2015 Open Networking Laboratory
 *
 * 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.cordvtn;

import com.google.common.base.MoreObjects;

import java.util.Objects;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * Representation of service identifier.
 */
public final class CordServiceId {

    private final String id;

    /**
     * Default constructor.
     *
     * @param id service identifier
     */
    private CordServiceId(String id) {
        this.id = id;
    }

    /**
     * Returns the CordServiceId with value.
     *
     * @param id service id
     * @return CordServiceId
     */
    public static CordServiceId of(String id) {
        checkNotNull(id);
        return new CordServiceId(id);
    }

    /**
     * Returns service identifier.
     *
     * @return service id
     */
    public String id() {
        return id;
    }

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

    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (!(obj instanceof CordServiceId)) {
            return false;
        }
        final CordServiceId other = (CordServiceId) obj;
        return Objects.equals(this.id, other.id);
    }

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