io.liftwizard.dropwizard.bundle.prioritized.PrioritizedBundle Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of liftwizard-bundle-prioritized-api Show documentation

There is a newer version: 2.1.23

Show newest version

/*
 * Copyright 2020 Craig Motlin
 *
 * 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.liftwizard.dropwizard.bundle.prioritized;

import javax.annotation.Nonnull;

import io.dropwizard.ConfiguredBundle;
import io.dropwizard.setup.Bootstrap;
import io.dropwizard.setup.Environment;
import org.slf4j.MDC;
import org.slf4j.MDC.MDCCloseable;

public interface PrioritizedBundle
        extends ConfiguredBundle
{
    String MDC_BUNDLE   = "liftwizard.bundle";
    String MDC_PRIORITY = "liftwizard.priority";

    default int getPriority()
    {
        return 0;
    }

    default  C safeCastConfiguration(Class aClass, Object configuration)
    {
        if (aClass.isInstance(configuration))
        {
            return aClass.cast(configuration);
        }

        String message = String.format(
                "Expected configuration to implement %s but found %s",
                aClass.getCanonicalName(),
                configuration.getClass().getCanonicalName());
        throw new IllegalStateException(message);
    }

    @Override
    default void initialize(@Nonnull Bootstrap bootstrap)
    {
        try (
                MDCCloseable mdc1 = MDC.putCloseable(MDC_BUNDLE, this.getClass().getSimpleName());
                MDCCloseable mdc2 = MDC.putCloseable(MDC_PRIORITY, String.valueOf(this.getPriority())))
        {
            this.initializeWithMdc(bootstrap);
        }
    }

    default void initializeWithMdc(@Nonnull Bootstrap bootstrap)
    {
    }

    @Override
    default void run(@Nonnull T configuration, @Nonnull Environment environment) throws Exception
    {
        try (
                MDCCloseable mdc1 = MDC.putCloseable(MDC_BUNDLE, this.getClass().getSimpleName());
                MDCCloseable mdc2 = MDC.putCloseable(MDC_PRIORITY, String.valueOf(this.getPriority())))
        {
            this.runWithMdc(configuration, environment);
        }
    }

    void runWithMdc(@Nonnull T configuration, @Nonnull Environment environment) throws Exception;
}