コード例 #1
0
ファイル: ModuleWeaver.cs プロジェクト: mdabbagh88/YALF
        private IEnumerable<MethodDefinition> SelectMethods(ModuleDefinition moduleDefinition, string excludeAttributeName)
        {
            var noLogAssembly = moduleDefinition.ContainsAttribute(excludeAttributeName);

            if (noLogAssembly)
            {
                LogInfo(string.Format("Skipping '{0}' assembly due to {1} attribute", moduleDefinition.Name, excludeAttributeName));
                return Enumerable.Empty<MethodDefinition>();
            }

            LogInfo(string.Format("Searching for Methods in assembly ({0}).", moduleDefinition.Name));

            var definitions = new HashSet<MethodDefinition>(
                moduleDefinition
                .Types
                .Where(x => x.IsClass && !x.ContainsAttribute(excludeAttributeName))
                .SelectMany(x => x.Methods)
                .Where(x => ValidMethod(moduleDefinition, excludeAttributeName, x))
                );

            // Remove NoLogAttribute
            moduleDefinition.Types.SelectMany(x => x.Methods).ToList().ForEach(x => x.RemoveAttribute(excludeAttributeName));
            moduleDefinition.Types.ToList().ForEach(x => x.RemoveAttribute(excludeAttributeName));
            moduleDefinition.Assembly.RemoveAttribute(excludeAttributeName);

            return definitions;
        }