Skip to content

Dev #58

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
matteobruni merged 4 commits into from
Apr 28, 2016
Merged

Dev #58

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
5ce413c
fix gettypes exception
matteobruni Apr 28, 2016
9a4d0af
fix
matteobruni Apr 28, 2016
47e0629
fix
matteobruni Apr 28, 2016
8ea68c6
Merge branch 'master' into dev
matteobruni Apr 28, 2016
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
231 changes: 126 additions & 105 deletions UnitOfWork.NET/Classes/UnitOfWork.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,110 +8,131 @@

namespace UnitOfWork.NET.Classes
{
public class UnitOfWork : IUnitOfWork
{
private readonly ObservableCollection<Assembly> _assemblies;
private readonly IContainer _container;

public UnitOfWork()
{
_assemblies = new ObservableCollection<Assembly>();

var cb = new ContainerBuilder();

cb.Register(t => this).AsImplementedInterfaces().AsSelf().As<UnitOfWork>();
cb.RegisterType<Repository>().AsSelf().As<IRepository>().PreserveExistingDefaults();
cb.RegisterGeneric(typeof(Repository<>)).AsSelf().As(typeof(IRepository<>));
cb.RegisterGeneric(typeof(Repository<,>)).AsSelf().As(typeof(IRepository<,>));
cb.RegisterGeneric(typeof(Repository<,,>)).AsSelf().As(typeof(IListRepository<,,>));

_container = cb.Build();

cb = new ContainerBuilder();

var fields = GetType().GetFields().ToArray();
var properties = GetType().GetProperties().ToArray();

foreach (var type in fields.Select(t => t.FieldType).Union(properties.Select(t => t.PropertyType)).Where(IsRepository)) cb.RegisterType(type).AsSelf().AsImplementedInterfaces();

cb.Update(_container);

foreach (var field in fields.Where(t => t.FieldType.IsAssignableTo<IRepository>())) field.SetValue(this, _container.ResolveOptional(field.FieldType));
foreach (var property in properties.Where(t => t.PropertyType.IsAssignableTo<IRepository>())) property.SetValue(this, _container.ResolveOptional(property.PropertyType));

_assemblies.CollectionChanged += (sender, args) => RegisterAssembly(args.NewItems.Cast<Assembly>().ToArray());

foreach (var assembly in AppDomain.CurrentDomain.GetAssemblies()) _assemblies.Add(assembly);
}

private bool IsRepository(Type t)
{
try
{
return t.IsAssignableTo<IRepository>() && !t.IsInterface && !t.IsAbstract && t != typeof(Repository) && ((t.IsGenericType && t.GetGenericTypeDefinition() != typeof(Repository<>) && t.GetGenericTypeDefinition() != typeof(Repository<,>) && t.GetGenericTypeDefinition() != typeof(Repository<,,>)) || !t.IsGenericType) && !_container.IsRegistered(t);
}
catch
{
return false;
}
}

private void RegisterAssembly(Assembly[] assemblyArr)
{
var cb = new ContainerBuilder();

cb.RegisterAssemblyTypes(assemblyArr.Where(t => t.GetTypes().Any(IsRepository)).ToArray()).Where(IsRepository).AsSelf().AsImplementedInterfaces();
cb.Update(_container);

foreach (var assembly in assemblyArr.SelectMany(t => t.GetReferencedAssemblies()).Select(Assembly.Load).Where(t => !_assemblies.Contains(t) && t.GetTypes().Any(IsRepository)))
_assemblies.Add(assembly);
}

public void RegisterRepository<TRepository>() where TRepository : IRepository => RegisterRepository(typeof(TRepository));

public void RegisterRepositories(Type[] repositoryTypes)
{
var cb = new ContainerBuilder();

foreach (var repositoryType in repositoryTypes)
{
if (repositoryType.IsInterface || repositoryType.IsAbstract || _container.IsRegistered(repositoryType)) continue;

if (repositoryType.IsGenericTypeDefinition)
cb.RegisterGeneric(repositoryType).AsSelf().AsImplementedInterfaces();
else
cb.RegisterType(repositoryType).AsSelf().AsImplementedInterfaces();
}

cb.Update(_container);
}

public void RegisterRepository(Type repositoryType)
{
RegisterRepositories(new[] { repositoryType });
}

private TRepository GetRepository<TRepository>() where TRepository : IRepository
{
RegisterRepository<TRepository>();

return _container.Resolve<TRepository>();
}

public TRepository CustomRepository<TRepository>() where TRepository : IRepository => GetRepository<TRepository>();

public IRepository<T> Repository<T>() where T : class => GetRepository<IRepository<T>>();

public IRepository<TSource, TDestination> Repository<TSource, TDestination>() where TSource : class where TDestination : class => GetRepository<IRepository<TSource, TDestination>>();

public IListRepository<TSource, TDestination, TListDestination> Repository<TSource, TDestination, TListDestination>() where TSource : class where TDestination : class where TListDestination : class => GetRepository<IListRepository<TSource, TDestination, TListDestination>>();

public virtual IEnumerable<T> Data<T>() where T : class => Enumerable.Empty<T>();

public virtual void Dispose()
{
_container.Dispose();
}
}
public class UnitOfWork : IUnitOfWork
{
private readonly ObservableCollection<Assembly> _assemblies;
private readonly IContainer _container;

public UnitOfWork()
{
_assemblies = new ObservableCollection<Assembly>();

var cb = new ContainerBuilder();

cb.Register(t => this).AsImplementedInterfaces().AsSelf().As<UnitOfWork>();
cb.RegisterType<Repository>().AsSelf().As<IRepository>().PreserveExistingDefaults();
cb.RegisterGeneric(typeof(Repository<>)).AsSelf().As(typeof(IRepository<>));
cb.RegisterGeneric(typeof(Repository<,>)).AsSelf().As(typeof(IRepository<,>));
cb.RegisterGeneric(typeof(Repository<,,>)).AsSelf().As(typeof(IListRepository<,,>));

_container = cb.Build();

cb = new ContainerBuilder();

var fields = GetType().GetFields().ToArray();
var properties = GetType().GetProperties().ToArray();

foreach (var type in fields.Select(t => t.FieldType).Union(properties.Select(t => t.PropertyType)).Where(IsRepository)) cb.RegisterType(type).AsSelf().AsImplementedInterfaces();

cb.Update(_container);

foreach (var field in fields.Where(t => t.FieldType.IsAssignableTo<IRepository>())) field.SetValue(this, _container.ResolveOptional(field.FieldType));
foreach (var property in properties.Where(t => t.PropertyType.IsAssignableTo<IRepository>())) property.SetValue(this, _container.ResolveOptional(property.PropertyType));

_assemblies.CollectionChanged += (sender, args) => RegisterAssembly(args.NewItems.Cast<Assembly>().ToArray());

foreach (var assembly in AppDomain.CurrentDomain.GetAssemblies()) _assemblies.Add(assembly);
}

private bool IsRepository(Type t)
{
try
{
return t.IsAssignableTo<IRepository>() && !t.IsInterface && !t.IsAbstract && t != typeof(Repository) && ((t.IsGenericType && t.GetGenericTypeDefinition() != typeof(Repository<>) && t.GetGenericTypeDefinition() != typeof(Repository<,>) && t.GetGenericTypeDefinition() != typeof(Repository<,,>)) || !t.IsGenericType) && !_container.IsRegistered(t);
}
catch
{
return false;
}
}

private void RegisterAssembly(Assembly[] assemblyArr)
{
var cb = new ContainerBuilder();

cb.RegisterAssemblyTypes(assemblyArr.Where(t =>
{
try
{
return t.GetTypes().Any(IsRepository);
}
catch
{
return false;
}
}).ToArray()).Where(IsRepository).AsSelf().AsImplementedInterfaces();
cb.Update(_container);

foreach (var assembly in assemblyArr.SelectMany(t => t.GetReferencedAssemblies()).Select(Assembly.Load).Where(
t =>
{
try
{
return !_assemblies.Contains(t) && t.GetTypes().Any(IsRepository);
}
catch
{
return false;
}
}))
_assemblies.Add(assembly);
}

public void RegisterRepository<TRepository>() where TRepository : IRepository => RegisterRepository(typeof(TRepository));

public void RegisterRepositories(Type[] repositoryTypes)
{
var cb = new ContainerBuilder();

foreach (var repositoryType in repositoryTypes)
{
if (repositoryType.IsInterface || repositoryType.IsAbstract || _container.IsRegistered(repositoryType)) continue;

if (repositoryType.IsGenericTypeDefinition)
cb.RegisterGeneric(repositoryType).AsSelf().AsImplementedInterfaces();
else
cb.RegisterType(repositoryType).AsSelf().AsImplementedInterfaces();
}

cb.Update(_container);
}

public void RegisterRepository(Type repositoryType)
{
RegisterRepositories(new[] { repositoryType });
}

private TRepository GetRepository<TRepository>() where TRepository : IRepository
{
RegisterRepository<TRepository>();

return _container.Resolve<TRepository>();
}

public TRepository CustomRepository<TRepository>() where TRepository : IRepository => GetRepository<TRepository>();

public IRepository<T> Repository<T>() where T : class => GetRepository<IRepository<T>>();

public IRepository<TSource, TDestination> Repository<TSource, TDestination>() where TSource : class where TDestination : class => GetRepository<IRepository<TSource, TDestination>>();

public IListRepository<TSource, TDestination, TListDestination> Repository<TSource, TDestination, TListDestination>() where TSource : class where TDestination : class where TListDestination : class => GetRepository<IListRepository<TSource, TDestination, TListDestination>>();

public virtual IEnumerable<T> Data<T>() where T : class => Enumerable.Empty<T>();

public virtual void Dispose()
{
_container.Dispose();
}
}
}