DFrameWorkloadCollection(Assembly[] searchAssemblies, bool includesDefault, IServiceProviderIsService isService)
{
dframeTypes = new Dictionary <string, DFrameWorkloadTypeInfo>(StringComparer.InvariantCultureIgnoreCase);
foreach (var asm in searchAssemblies)
{
if (asm.FullName !.StartsWith("System") || asm.FullName.StartsWith("Microsoft.Extensions"))
{
continue;
}
Type[] types;
try
{
types = asm.GetTypes();
}
catch (ReflectionTypeLoadException ex)
{
types = ex.Types.Where(x => x != null).ToArray() !;
}
foreach (var workload in types)
{
if (typeof(Workload).IsAssignableFrom(workload) && !workload.IsAbstract)
{
#if !UNITY_2020_1_OR_NEWER
var isDefault = workload.GetCustomAttribute <DefaultWorkloadAttribute>(true);
if (isDefault != null)
{
if (!includesDefault)
{
continue;
}
}
#endif
var attr = workload.GetCustomAttribute <WorkloadAttribute>(false);
var name = attr?.Name ?? workload.Name;
var t = new DFrameWorkloadTypeInfo(workload, name, isService);
if (!dframeTypes.TryAdd(name, t))
{
throw new InvalidOperationException($"Worker name is duplicate. name:{name}, type:{t.WorkloadType.FullName}");
}
}
}
}
}
internal static bool TryAdd(this Dictionary <string, DFrameWorkloadTypeInfo> dict, string key, DFrameWorkloadTypeInfo value)
{
if (!dict.ContainsKey(key))
{
dict.Add(key, value);
return(true);
}
return(false);
}
