com.facebook.presto.verifier.resolver.FailureResolverModule Maven / Gradle / Ivy

Go to download
/*
 * 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 com.facebook.presto.verifier.resolver;

import com.facebook.airlift.configuration.AbstractConfigurationAwareModule;
import com.google.common.collect.ImmutableList;
import com.google.inject.Binder;
import com.google.inject.Module;

import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

import static com.facebook.airlift.configuration.ConfigBinder.configBinder;
import static com.google.inject.Scopes.SINGLETON;
import static com.google.inject.multibindings.Multibinder.newSetBinder;
import static com.google.inject.name.Names.named;
import static java.util.Objects.requireNonNull;

public class FailureResolverModule
        extends AbstractConfigurationAwareModule
{
    public static final FailureResolverModule BUILT_IN = FailureResolverModule.builder()
            // Query Failure Resolvers
            .bind(ExceededGlobalMemoryLimitFailureResolver.NAME, ExceededGlobalMemoryLimitFailureResolver.class, Optional.empty())
            .bind(ExceededTimeLimitFailureResolver.NAME, ExceededTimeLimitFailureResolver.class, Optional.empty())
            .bind(ChecksumExceededTimeLimitFailureResolver.NAME, ChecksumExceededTimeLimitFailureResolver.class, Optional.empty())
            .bind(VerifierLimitationFailureResolver.NAME, VerifierLimitationFailureResolver.class, Optional.empty())
            .bindFactory(
                    TooManyOpenPartitionsFailureResolver.NAME,
                    TooManyOpenPartitionsFailureResolver.Factory.class,
                    Optional.of(TooManyOpenPartitionsFailureResolverConfig.class),
                    ImmutableList.of(new ClusterSizeFetcherModule()))
            // Result Mismatch Resolvers
            .bind(StructuredColumnMismatchResolver.NAME, StructuredColumnMismatchResolver.class, Optional.empty())
            .bind(IgnoredFunctionsMismatchResolver.NAME, IgnoredFunctionsMismatchResolver.class, Optional.of(IgnoredFunctionsMismatchResolverConfig.class))
            .build();

    private final List resolvers;
    private final List factories;

    private FailureResolverModule(
            List resolvers,
            List factories)
    {
        this.resolvers = ImmutableList.copyOf(resolvers);
        this.factories = ImmutableList.copyOf(factories);
    }

    @Override
    protected void setup(Binder binder)
    {
        configBinder(binder).bindConfig(FailureResolverConfig.class);
        binder.bind(FailureResolverManagerFactory.class).in(SINGLETON);
        newSetBinder(binder, FailureResolver.class);
        newSetBinder(binder, FailureResolverFactory.class);
        if (!buildConfigObject(FailureResolverConfig.class).isEnabled()) {
            return;
        }

        for (FailureResolverBinding binding : resolvers) {
            bind(binder, binding.getName(), Optional.of(binding.getResolverClass()), Optional.empty(), binding.getConfigClass(), binding.getAdditionalModules());
        }
        for (FailureResolverFactoryBinding binding : factories) {
            bind(binder, binding.getName(), Optional.empty(), Optional.of(binding.getFactoryClass()), binding.getConfigClass(), binding.getAdditionalModules());
        }
    }

    private void bind(
            Binder binder,
            String name,
            Optional> failureResolverClass,
            Optional> failureResolverFactoryClass,
            Optional> failureResolverConfigClass,
            List additionalModules)
    {
        configBinder(binder).bindConfig(FailureResolverConfig.class, named(name), name);
        if (buildConfigObject(FailureResolverConfig.class, name).isEnabled()) {
            failureResolverClass.ifPresent(clazz -> newSetBinder(binder, FailureResolver.class).addBinding().to(clazz).in(SINGLETON));
            failureResolverFactoryClass.ifPresent(clazz -> newSetBinder(binder, FailureResolverFactory.class).addBinding().to(clazz).in(SINGLETON));
            failureResolverConfigClass.ifPresent(clazz -> configBinder(binder).bindConfig(clazz, name));
            additionalModules.forEach(this::install);
        }
    }

    public static Builder builder()
    {
        return new Builder();
    }

    public static class Builder
    {
        private final List resolvers = new ArrayList<>();
        private final List factories = new ArrayList<>();

        public Builder bind(String name, Class failureResolverClass, Optional> configClass)
        {
            resolvers.add(new FailureResolverBinding(name, failureResolverClass, configClass));
            return this;
        }

        public Builder bindFactory(
                String name,
                Class failureResolverFactoryClass,
                Optional> configClass,
                List additionalModules)
        {
            factories.add(new FailureResolverFactoryBinding(name, failureResolverFactoryClass, configClass, additionalModules));
            return this;
        }

        public FailureResolverModule build()
        {
            return new FailureResolverModule(resolvers, factories);
        }
    }

    private static class FailureResolverBinding
            extends AbstractFailureResolverBinding
    {
        private final Class resolverClass;

        public FailureResolverBinding(String name, Class resolverClass, Optional> configClass)
        {
            super(name, configClass, ImmutableList.of());
            this.resolverClass = resolverClass;
        }

        public Class getResolverClass()
        {
            return resolverClass;
        }
    }

    private static class FailureResolverFactoryBinding
            extends AbstractFailureResolverBinding
    {
        private final Class factoryClass;

        public FailureResolverFactoryBinding(String name, Class factoryClass, Optional> configClass, List additionalModules)
        {
            super(name, configClass, additionalModules);
            this.factoryClass = factoryClass;
        }

        public Class getFactoryClass()
        {
            return factoryClass;
        }
    }

    private abstract static class AbstractFailureResolverBinding
    {
        private final String name;
        private final Optional> configClass;
        private final List additionalModules;

        public AbstractFailureResolverBinding(String name, Optional> configClass, List additionalModules)
        {
            this.name = requireNonNull(name, "name is null");
            this.configClass = requireNonNull(configClass, "configClass is null");
            this.additionalModules = ImmutableList.copyOf(additionalModules);
        }

        public String getName()
        {
            return name;
        }

        public Optional> getConfigClass()
        {
            return configClass;
        }

        public List getAdditionalModules()
        {
            return additionalModules;
        }
    }
}