/*
 * Copyright (c) 2021 PANTHEON.tech, s.r.o. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.controller.eos.akka;

import static java.util.Objects.requireNonNull;

import com.google.common.base.MoreObjects;
import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipListener;
import org.opendaylight.yangtools.concepts.AbstractObjectRegistration;

final class ListenerRegistration extends AbstractObjectRegistration<DOMEntityOwnershipListener> {
    private final AkkaEntityOwnershipService service;
    private final @NonNull String entityType;

    ListenerRegistration(final DOMEntityOwnershipListener listener, final String entityType,
            final AkkaEntityOwnershipService service) {
        super(listener);
        this.entityType = requireNonNull(entityType);
        this.service = requireNonNull(service);
    }

    public String entityType() {
        return entityType;
    }

    @Override
    protected void removeRegistration() {
        service.unregisterListener(entityType, getInstance());
    }

    @Override
    protected MoreObjects.ToStringHelper addToStringAttributes(final MoreObjects.ToStringHelper toStringHelper) {
        return toStringHelper.add("entityType", entityType);
    }
}