encompass-cs/encompass-cs/Attributes/WritesImmediate.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Encompass.Exceptions;

namespace Encompass
{
    [AttributeUsage(AttributeTargets.Class)]
    public class WritesImmediate : Attribute
    {
        public readonly HashSet<Type> WriteImmediateTypes = new HashSet<Type>();

        public WritesImmediate(params Type[] writeImmediateTypes)
        {
            foreach (var writeImmediateType in writeImmediateTypes)
            {
                var isComponent = writeImmediateType.GetInterfaces().Contains(typeof(IComponent));
                if (!isComponent)
                {
                    throw new IllegalWriteImmediateTypeException("{0} must be a Component", writeImmediateType.Name);
                }

                this.WriteImmediateTypes.Add(writeImmediateType);
            }
        }
    }
}