hikalkan/MyDbContextTypeMatcher.cs

## MyDbContextTypeMatcher.cs
using System;
using System.Collections.Generic;
using System.Linq;
using Abp.Collections.Extensions;
using Abp.Dependency;
using Abp.Domain.Uow;
using Abp.EntityFramework.Repositories;
using Abp.MultiTenancy;

namespace Abp.EntityFramework
{
    public class MyDbContextTypeMatcher : IDbContextTypeMatcher
    {
        private readonly ICurrentUnitOfWorkProvider _currentUnitOfWorkProvider;
        private readonly Dictionary<Type, List<Type>> _dbContextTypes;

        public MyDbContextTypeMatcher(ICurrentUnitOfWorkProvider currentUnitOfWorkProvider)
        {
            _currentUnitOfWorkProvider = currentUnitOfWorkProvider;
            _dbContextTypes = new Dictionary<Type, List<Type>>();
        }

        public virtual void Add(Type sourceDbContextType, Type targetDbContextType)
        {
            if (!_dbContextTypes.ContainsKey(sourceDbContextType))
            {
                _dbContextTypes[sourceDbContextType] = new List<Type>();
            }

            _dbContextTypes[sourceDbContextType].Add(targetDbContextType);
        }

        public virtual Type GetConcreteType(Type sourceDbContextType)
        {
            //TODO: This can also get MultiTenancySide to filter dbcontexes

            //TODO: Can be optimized by extracting/caching MultiTenancySideAttribute attributes for DbContexes.

            //Get possible concrete types for given DbContext type
            var allTargetTypes = _dbContextTypes.GetOrDefault(sourceDbContextType);

            if (allTargetTypes.IsNullOrEmpty())
            {
                //Not found any target type, return the given type if it's not abstract
                if (sourceDbContextType.IsAbstract)
                {
                    throw new AbpException("Could not find a concrete implementation of given DbContext type: " + sourceDbContextType.AssemblyQualifiedName);
                }

                return sourceDbContextType;
            }

            if (allTargetTypes.Count == 1)
            {
                //Only one type does exists, return it
                return allTargetTypes[0];
            }

            //Will decide the target type with current UOW, so it should be in a UOW.
            if (_currentUnitOfWorkProvider.Current == null)
            {
                throw new AbpException("GetConcreteType method should be called in a UOW.");
            }

            var currentTenancySide = _currentUnitOfWorkProvider.Current.GetTenantId() == null
                ? MultiTenancySides.Host
                : MultiTenancySides.Tenant;

            var multiTenancySideContexes = allTargetTypes.Where(type =>
            {
                var attrs = type.GetCustomAttributes(typeof(MultiTenancySideAttribute), true);
                if (attrs.IsNullOrEmpty())
                {
                    return false;
                }

                return ((MultiTenancySideAttribute)attrs[0]).Side.HasFlag(currentTenancySide);
            }).ToList();

            //Try to get the DbContext which is for current multitenancy side.
            if (multiTenancySideContexes.Count == 1)
            {
                return multiTenancySideContexes[0];
            }

            //Try to get the DbContext which not defined AutoRepositoryTypesAttribute
            var defaultRepositoryContexes = allTargetTypes.Where(type => !type.IsDefined(typeof(AutoRepositoryTypesAttribute), true)).ToList();
            if (defaultRepositoryContexes.Count == 1)
            {
                return defaultRepositoryContexes[0];
            }

            //Generating exception

            if (multiTenancySideContexes.Count > 1)
            {
                throw new AbpException(string.Format(
                    "Found more than one concrete type for given DbContext Type ({0}) define MultiTenancySideAttribute with {1}. Found types: {2}.",
                    sourceDbContextType,
                    currentTenancySide,
                    multiTenancySideContexes.JoinAsString(", ")
                    ));
            }

            throw new AbpException(string.Format(
                "Found more than one concrete type for given DbContext Type ({0}) but none of them defines MultiTenancySideAttribute with {1}. Found types: {2}.",
                sourceDbContextType,
                currentTenancySide,
                multiTenancySideContexes.JoinAsString(", ")
                ));
        }
    }
}
	using System;
	using System.Collections.Generic;
	using System.Linq;
	using Abp.Collections.Extensions;
	using Abp.Dependency;
	using Abp.Domain.Uow;
	using Abp.EntityFramework.Repositories;
	using Abp.MultiTenancy;

	namespace Abp.EntityFramework
	{
	public class MyDbContextTypeMatcher : IDbContextTypeMatcher
	{
	private readonly ICurrentUnitOfWorkProvider _currentUnitOfWorkProvider;
	private readonly Dictionary<Type, List<Type>> _dbContextTypes;

	public MyDbContextTypeMatcher(ICurrentUnitOfWorkProvider currentUnitOfWorkProvider)
	{
	_currentUnitOfWorkProvider = currentUnitOfWorkProvider;
	_dbContextTypes = new Dictionary<Type, List<Type>>();
	}

	public virtual void Add(Type sourceDbContextType, Type targetDbContextType)
	{
	if (!_dbContextTypes.ContainsKey(sourceDbContextType))
	{
	_dbContextTypes[sourceDbContextType] = new List<Type>();
	}

	_dbContextTypes[sourceDbContextType].Add(targetDbContextType);
	}

	public virtual Type GetConcreteType(Type sourceDbContextType)
	{
	//TODO: This can also get MultiTenancySide to filter dbcontexes

	//TODO: Can be optimized by extracting/caching MultiTenancySideAttribute attributes for DbContexes.

	//Get possible concrete types for given DbContext type
	var allTargetTypes = _dbContextTypes.GetOrDefault(sourceDbContextType);

	if (allTargetTypes.IsNullOrEmpty())
	{
	//Not found any target type, return the given type if it's not abstract
	if (sourceDbContextType.IsAbstract)
	{
	throw new AbpException("Could not find a concrete implementation of given DbContext type: " + sourceDbContextType.AssemblyQualifiedName);
	}

	return sourceDbContextType;
	}

	if (allTargetTypes.Count == 1)
	{
	//Only one type does exists, return it
	return allTargetTypes[0];
	}

	//Will decide the target type with current UOW, so it should be in a UOW.
	if (_currentUnitOfWorkProvider.Current == null)
	{
	throw new AbpException("GetConcreteType method should be called in a UOW.");
	}

	var currentTenancySide = _currentUnitOfWorkProvider.Current.GetTenantId() == null
	? MultiTenancySides.Host
	: MultiTenancySides.Tenant;

	var multiTenancySideContexes = allTargetTypes.Where(type =>
	{
	var attrs = type.GetCustomAttributes(typeof(MultiTenancySideAttribute), true);
	if (attrs.IsNullOrEmpty())
	{
	return false;
	}

	return ((MultiTenancySideAttribute)attrs[0]).Side.HasFlag(currentTenancySide);
	}).ToList();

	//Try to get the DbContext which is for current multitenancy side.
	if (multiTenancySideContexes.Count == 1)
	{
	return multiTenancySideContexes[0];
	}

	//Try to get the DbContext which not defined AutoRepositoryTypesAttribute
	var defaultRepositoryContexes = allTargetTypes.Where(type => !type.IsDefined(typeof(AutoRepositoryTypesAttribute), true)).ToList();
	if (defaultRepositoryContexes.Count == 1)
	{
	return defaultRepositoryContexes[0];
	}

	//Generating exception

	if (multiTenancySideContexes.Count > 1)
	{
	throw new AbpException(string.Format(
	"Found more than one concrete type for given DbContext Type ({0}) define MultiTenancySideAttribute with {1}. Found types: {2}.",
	sourceDbContextType,
	currentTenancySide,
	multiTenancySideContexes.JoinAsString(", ")
	));
	}

	throw new AbpException(string.Format(
	"Found more than one concrete type for given DbContext Type ({0}) but none of them defines MultiTenancySideAttribute with {1}. Found types: {2}.",
	sourceDbContextType,
	currentTenancySide,
	multiTenancySideContexes.JoinAsString(", ")
	));
	}
	}
	}