TypeRegistrationBuilder.cs
74 lines
| 2.9 KiB
| text/x-csharp
|
CSharpLexer
cin
|
r274 | using System; | |
using System.Collections.Generic; | |||
using System.Linq; | |||
using Unity.Injection; | |||
using Unity.Registration; | |||
namespace Implab.ServiceHost.Unity { | |||
public class TypeRegistrationBuilder : RegistrationBuilder { | |||
readonly TypeResolver m_resolver; | |||
readonly List<InjectionMember> m_injections = new List<InjectionMember>(); | |||
cin
|
r277 | internal InjectionMember[] Injections { get { return m_injections.ToArray(); } } | |
cin
|
r274 | ||
public Type ImplementationType { | |||
get; | |||
private set; | |||
} | |||
internal TypeRegistrationBuilder(TypeResolver resolver, Type registrationType, Type implementationType) : base(registrationType) { | |||
ImplementationType = implementationType; | |||
// when registering a generic mapping, register all generic parameter names as local types | |||
if (ImplementationType.IsGenericTypeDefinition) { | |||
m_resolver = new TypeResolver(resolver); | |||
foreach (var p in ImplementationType.GetGenericArguments()) | |||
m_resolver.AddMapping(p.Name, p); | |||
} else { | |||
m_resolver = resolver; | |||
} | |||
} | |||
internal void Visit(ConstructorInjectionElement constructorInjection) { | |||
var parameters = constructorInjection.Parameters? | |||
.Select(x => { | |||
cin
|
r278 | var valueBuilder = new InjectionParameterBuilder(m_resolver, null); | |
cin
|
r274 | x.Visit(valueBuilder); | |
return valueBuilder.Injection; | |||
}) | |||
.ToArray(); | |||
var injection = parameters != null ? new InjectionConstructor(parameters) : new InjectionConstructor(); | |||
m_injections.Add(injection); | |||
} | |||
internal void Visit(MethodInjectionElement methodInjection) { | |||
var parameters = methodInjection.Parameters? | |||
.Select(x => { | |||
cin
|
r278 | var valueBuilder = new InjectionParameterBuilder(m_resolver, null); | |
cin
|
r274 | x.Visit(valueBuilder); | |
return valueBuilder.Injection; | |||
}) | |||
.ToArray(); | |||
var injection = parameters != null ? new InjectionMethod(methodInjection.Name, parameters) : new InjectionMethod(methodInjection.Name); | |||
m_injections.Add(injection); | |||
} | |||
internal void Visit(PropertyInjectionElement propertyInjection) { | |||
if (propertyInjection.Value == null) | |||
throw new Exception($"A value value must be specified for the property '{propertyInjection.Name}'"); | |||
cin
|
r278 | var propertyType = ImplementationType.GetProperty(propertyInjection.Name)?.PropertyType; | |
var valueContext = new InjectionParameterBuilder(m_resolver, propertyType); | |||
cin
|
r274 | ||
propertyInjection.Value.Visit(valueContext); | |||
var injection = new InjectionProperty(propertyInjection.Name, valueContext.Injection); | |||
m_injections.Add(injection); | |||
} | |||
} | |||
} |