/*
 * Copyright 2014-2017 Real Logic Ltd.
 *
 * 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 io.aeron.agent;

import io.aeron.driver.SubscriptionLink;
import io.aeron.driver.NetworkPublication;
import io.aeron.driver.PublicationImage;
import io.aeron.driver.media.ReceiveChannelEndpoint;
import net.bytebuddy.asm.Advice;

import static io.aeron.agent.EventLogger.LOGGER;

public class CleanupInterceptor
{
    public static class DriverConductorInterceptor
    {
        public static class CleanupImage
        {
            @Advice.OnMethodEnter
            public static void cleanupImageInterceptor(final PublicationImage image)
            {
                LOGGER.logImageRemoval(
                    image.channelUriString(), image.sessionId(), image.streamId(), image.correlationId());
            }
        }

        public static class CleanupPublication
        {
            @Advice.OnMethodEnter
            public static void cleanupPublication(final NetworkPublication publication)
            {
                LOGGER.logPublicationRemoval(
                    publication.channelEndpoint().originalUriString(),
                    publication.sessionId(),
                    publication.streamId());
            }
        }

        public static class CleanupSubscriptionLink
        {
            @Advice.OnMethodEnter
            public static void cleanupSubscriptionLink(final SubscriptionLink subscriptionLink)
            {
                final ReceiveChannelEndpoint channelEndpoint = subscriptionLink.channelEndpoint();

                if (null != channelEndpoint)
                {
                    LOGGER.logSubscriptionRemoval(
                        channelEndpoint.originalUriString(),
                        subscriptionLink.streamId(),
                        subscriptionLink.registrationId());
                }
            }
        }
    }
}