ti-enxame.com

Permitir um atributo personalizado apenas em um tipo específico

Existe uma maneira de forçar o compilador a restringir o uso de um atributo personalizado a ser usado apenas em tipos específicos de propriedade , como int, short, string (all os tipos primitivos)?
semelhante à AttributeUsageAttribute 's ValidOn - AttributeTargets enumeração.

33

Não, você não pode, basicamente. Você pode limitar a struct vs class vs interface, é isso. Mais: você não pode adicionar atributos a tipos fora do seu código (exceto por TypeDescriptor, que não é o mesmo).

27
Marc Gravell

Você pode executar este teste de unidade para verificá-lo.

Primeiro, declare o atributo de validação PropertyType:

  [AttributeUsage(AttributeTargets.Class)]
    // [JetBrains.Annotations.BaseTypeRequired(typeof(Attribute))] uncomment if you use JetBrains.Annotations
    public class PropertyTypeAttribute : Attribute
    {
        public Type[] Types { get; private set; }

        public PropertyTypeAttribute(params Type[] types)
        {
            Types = types;
        }
    }

Crie teste de unidade:

 [TestClass]
    public class TestPropertyType 
    {
        public static Type GetNullableUnderlying(Type nullableType)
        {
            return Nullable.GetUnderlyingType(nullableType) ?? nullableType;
        }

        [TestMethod]
        public void Test_PropertyType()
        {
            var allTypes = AppDomain.CurrentDomain.GetAssemblies().SelectMany(a => a.GetTypes());
            var allPropertyInfos = allTypes.SelectMany(a => a.GetProperties()).ToArray();

            foreach (var propertyInfo in allPropertyInfos)
            {
                var propertyType = GetNullableUnderlying(propertyInfo.PropertyType);
                foreach (var attribute in propertyInfo.GetCustomAttributes(true))
                {
                    var attributes = attribute.GetType().GetCustomAttributes(true).OfType<PropertyTypeAttribute>();
                    foreach (var propertyTypeAttr in attributes)
                        if (!propertyTypeAttr.Types.Contains(propertyType))
                            throw new Exception(string.Format(
                                "Property '{0}.{1}' has invalid type: '{2}'. Allowed types for attribute '{3}': {4}",
                                propertyInfo.DeclaringType,
                                propertyInfo.Name,
                                propertyInfo.PropertyType,
                                attribute.GetType(),
                                string.Join(",", propertyTypeAttr.Types.Select(x => "'" + x.ToString() + "'"))));
                }
            }
        }
    }

Seu atributo, por exemplo, permite apenas tipos de propriedades decimais:

 [AttributeUsage(AttributeTargets.Property)]
    [PropertyType(typeof(decimal))]
    public class PriceAttribute : Attribute
    {

    }

Modelo de exemplo:

public class TestModel  
{
    [Price]
    public decimal Price1 { get; set; } // ok

    [Price]
    public double Price2 { get; set; } // error
}
13
Sel

Você mesmo pode escrever um código para impor o uso correto da sua classe de atributo, mas é o máximo que você pode fazer.

4
Anders Tornblad

O código abaixo retornará um erro se o atributo tiver sido colocado em uma propriedade/campo que não seja Lista de sequência.

A linha if (!(value is List<string> list)) pode ser um recurso C # 6 ou 7.

[AttributeUsage(AttributeTargets.Property |
                AttributeTargets.Field, AllowMultiple = false)]
public sealed class RequiredStringListAttribute : ValidationAttribute
{
    protected override ValidationResult IsValid(object value, ValidationContext context)
    {
        if (!(value is List<string> list))
            return new ValidationResult($"The required attrribute must be of type List<string>");

        bool valid = false;
        foreach (var item in list)
        {
            if (!string.IsNullOrWhiteSpace(item))
                valid = true;
        }

        return valid
            ? ValidationResult.Success
            : new ValidationResult($"This field is required"); ;
    }

}
4
Ian