org.javabits.yar.guice.BlockingSupplierRegistryAnnotatedBindingBuilderImpl Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2013 Romain Gilles
 *
 *    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.javabits.yar.guice;

import com.google.inject.Binder;
import com.google.inject.Inject;
import com.google.inject.Key;
import com.google.inject.TypeLiteral;
import org.javabits.yar.Id;

import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.concurrent.TimeUnit;

/**
 * TODO comment
 * Date: 3/4/13
 * Time: 10:11 PM
 *
 * @author Romain Gilles
 */
public class BlockingSupplierRegistryAnnotatedBindingBuilderImpl extends RegistryAnnotatedBindingBuilderImpl {

    public BlockingSupplierRegistryAnnotatedBindingBuilderImpl(Binder binder, Key key) {
        super(binder, key);
    }

    public BlockingSupplierRegistryAnnotatedBindingBuilderImpl(Binder binder, TypeLiteral typeLiteral) {
        super(binder, typeLiteral);
    }

    @Override
    RegistryProvider newRegistryProvider() {
        return new BlockingSupplierRegistryProvider<>(key());
    }

    @Override
    RegistryProvider newRegistryProvider(long timeout, TimeUnit unit) {
        return newRegistryProvider();
    }

    static class BlockingSupplierRegistryProvider implements RegistryProvider {

        private final Key key;
        private org.javabits.yar.BlockingSupplierRegistry registry;

        private BlockingSupplierRegistryProvider(Key key) {
            this.key = key;
        }

        @Override
        @SuppressWarnings("unchecked")
        public T get() {
            return (T) registry.get(getSupplierTypeParameter());
        }

        @SuppressWarnings("unchecked")
        private Id getSupplierTypeParameter() {
            Type type = key.getTypeLiteral().getType();
            checkParameterizedType(type);
            ParameterizedType parameterizedType = (ParameterizedType) type;
            Type[] actualTypeArguments = parameterizedType.getActualTypeArguments();
            if (actualTypeArguments.length != 1) {
                throw new IllegalArgumentException("Supplier type must be mono parameterized: " + type);
            }

            Type actualTypeArgument = actualTypeArguments[0];
            return GuiceId.of(actualTypeArgument, key);
        }

        @Inject
        public void setRegistry(org.javabits.yar.BlockingSupplierRegistry registry) {
            this.registry = registry;
        }

        @Override
        public void noWait() {
            //has no sens here in this context => nothing to do
        }
    }

    private static void checkParameterizedType(Type type) {
        if (!isParameterizedType(type)) {
            throw new IllegalArgumentException("Supplier type must be parameterized: " + type);
        }
    }

    private static boolean isParameterizedType(Type type) {
        return type instanceof ParameterizedType;
    }

}