/*
 * 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;

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

    private final long serviceId;

    /**
     * Default constructor.
     *
     * @param serviceId service identifier
     */
    private ServiceId(long serviceId) {
        this.serviceId = serviceId;
    }

    /**
     * Returns the ServiceId with value.
     *
     * @param serviceId service id
     * @return ServiceId
     */
    public static ServiceId of(long serviceId) {
        return new ServiceId(serviceId);
    }

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

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

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

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