com.avanza.astrix.beans.factory.AstrixFactoryBeanRegistry Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2014 Avanza Bank AB
 *
 * 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.avanza.astrix.beans.factory;

import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import com.avanza.astrix.beans.core.AstrixBeanKey;

/**
 * 
 * @author Elias Lindholm (elilin)
 *
 */
final class AstrixFactoryBeanRegistry {
	
	private final ConcurrentMap, StandardFactoryBean> factoryByBeanKey = new ConcurrentHashMap<>();
	private final ConcurrentMap, DynamicFactoryBean> dynamicFactoryByBeanType = new ConcurrentHashMap<>();

	
	/* (non-Javadoc)
	 * @see com.avanza.astrix.beans.factory.AstrixFactoryBeanRegistry#getFactoryBean(com.avanza.astrix.beans.factory.AstrixBeanKey)
	 */
	@SuppressWarnings("unchecked")
	public  StandardFactoryBean getFactoryBean(AstrixBeanKey  beanKey) {
		StandardFactoryBean factoryBean = (StandardFactoryBean) this.factoryByBeanKey.get(beanKey);
		if (factoryBean != null) {
			return factoryBean;
		}
		DynamicFactoryBean dynamicFactoryBean = (DynamicFactoryBean) this.dynamicFactoryByBeanType.get(beanKey.getBeanType());
		if (dynamicFactoryBean != null) {
			return new SynthesizedFactoryBean(dynamicFactoryBean, beanKey);
		}
		throw new MissingBeanProviderException(beanKey);
	}
	
	@SuppressWarnings("unchecked")
	public  Set> getBeansOfType(Class type) {
		Set> result = new HashSet<>();
		for (AstrixBeanKey key : this.factoryByBeanKey.keySet()) {
			if (key.getBeanType().equals(type)) {
				result.add((AstrixBeanKey) key);
			}
		}
		return result;
	}
	
	public  void registerFactory(FactoryBean factory) {
		if (factory instanceof DynamicFactoryBean) {
			registerFactory((DynamicFactoryBean) factory);
			return;
		}
		if (factory instanceof StandardFactoryBean) {
			registerFactory((StandardFactoryBean) factory);
			return;
		}
		throw new IllegalArgumentException("Unknown FactoryBean type: " + factory);
	}
	
	
	public  void registerFactory(StandardFactoryBean factory) {
		StandardFactoryBean duplicateFactory = factoryByBeanKey.putIfAbsent(factory.getBeanKey(), factory);
		if (duplicateFactory != null) {
			throw new MultipleBeanFactoryException(factory.getBeanKey());
		}
	}
	
	public  void registerFactory(DynamicFactoryBean dynamicFactoryBean) {
		DynamicFactoryBean duplicateFactory = dynamicFactoryByBeanType.putIfAbsent(dynamicFactoryBean.getType(), dynamicFactoryBean);
		if (duplicateFactory != null) {
			throw new MultipleBeanFactoryException(dynamicFactoryBean.getType());
		}
	}

	public  AstrixBeanKey resolveBean(AstrixBeanKey beanKey) {
		return beanKey;
	}
	
	private static class SynthesizedFactoryBean implements StandardFactoryBean {
		private final DynamicFactoryBean factory;
		private final AstrixBeanKey key;
		
		public SynthesizedFactoryBean(DynamicFactoryBean factory,
				AstrixBeanKey key) {
			this.factory = factory;
			this.key = key;
		}
		@Override
		public T create(AstrixBeans beans) {
			return factory.create(key);
		}
		@Override
		public AstrixBeanKey getBeanKey() {
			return key;
		}
	}

	
	
}