XdCxRhDW/XdCxRhDw.Dto/Attribute/RangeAttribute.cs

using System;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace XdCxRhDw.Dto
{
    /// <summary>
    /// int或枚举类型范围限制,默认不包含最小值,包含最大值
    /// </summary>
    public class RangeIntAttribute : ValidationAttribute
    {
        /// <summary>
        /// 
        /// </summary>
        public RangeIntAttribute()
        {
        }
        /// <summary>
        /// 
        /// </summary>
        /// <param name="min"></param>
        public RangeIntAttribute(int min)
        {
            this.Min = min;
        }
        /// <summary>
        /// 
        /// </summary>
        /// <param name="min"></param>
        /// <param name="max"></param>
        public RangeIntAttribute(int min, int max)
        {
            this.Min = min;
            this.Max = max;
        }
        /// <summary>
        /// 最小值(为Null不会验证最小值)
        /// </summary>
        public int? Min { get; set; }

        /// <summary>
        /// 最大值(为Null不会验证最大值)
        /// </summary>
        public int? Max { get; set; }

        /// <summary>
        /// 是否包含最小值(默认不包含最小值)
        /// </summary>
        public bool IncludeMin { get; set; } = false;

        /// <summary>
        /// 是否包含最大值(默认包含最大值)
        /// </summary>
        public bool IncludeMax { get; set; } = true;

        /// <summary>
        /// Range验证
        /// </summary>
        /// <param name="value"></param>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public override bool IsValid(object value)
        {
            if (value == null) return true;
            if (value.GetType() != typeof(int) && !value.GetType().IsEnum)
            {
                throw new Exception($"{nameof(RangeIntAttribute)}只能用于int或枚举类型!");
            }
            var val = (int)value;
            if (Min == null && Max == null) return true;
            else if (Min != null && Max != null)
            {

                if (IncludeMin && IncludeMax)
                {
                    ErrorMessage = "字段 {0} 范围" + $"[{Min},{Max}]";
                    return val >= Min && val <= Max;
                }
                else if (IncludeMin && !IncludeMax)
                {
                    ErrorMessage = "字段 {0} 范围" + $"[{Min},{Max})";
                    return val >= Min && val < Max;
                }
                else if (!IncludeMin && IncludeMax)
                {
                    ErrorMessage = "字段 {0} 范围" + $"({Min},{Max}]";
                    return val > Min && val <= Max;
                }
                else
                {
                    ErrorMessage = "字段 {0} 范围" + $"({Min},{Max})";
                    return val > Min && val < Max;
                }

            }
            else if (Min != null && Max == null)
            {
                if (IncludeMin)
                {
                    ErrorMessage = "字段 {0} 必须大于等于" + $"{Min}";
                    return val >= Min;
                }
                else
                {
                    ErrorMessage = "字段 {0} 必须大于" + $"{Min}";
                    return val > Min;
                }
            }
            else//Min==null&&Max!=null
            {
                if (IncludeMax)
                {
                    ErrorMessage = "字段 {0} 必须小于等于" + $"{Max}";
                    return val <= Max;
                }
                else
                {
                    ErrorMessage = "字段 {0} 必须小于" + $"{Max}";
                    return val < Max;
                }
            }
        }
    }

    /// <summary>
    /// double类型范围限制,默认不包含最小值,包含最大值
    /// </summary>
    public class RangeDoubleAttribute : ValidationAttribute
    {
        /// <summary>
        /// 
        /// </summary>
        public RangeDoubleAttribute()
        {
        }
        /// <summary>
        /// 
        /// </summary>
        /// <param name="min"></param>
        public RangeDoubleAttribute(double min)
        {
            this.Min = min;
        }
        /// <summary>
        /// 
        /// </summary>
        /// <param name="min"></param>
        /// <param name="max"></param>
        public RangeDoubleAttribute(double min, double max)
        {
            this.Min = min;
            this.Max = max;
        }
        /// <summary>
        /// 最小值(为Null不会验证最小值)
        /// </summary>
        public double? Min { get; set; }

        /// <summary>
        /// 最大值(为Null不会验证最大值)
        /// </summary>
        public double? Max { get; set; }

        /// <summary>
        /// 是否包含最小值(默认不包含最小值)
        /// </summary>
        public bool IncludeMin { get; set; } = false;

        /// <summary>
        /// 是否包含最大值(默认包含最大值)
        /// </summary>
        public bool IncludeMax { get; set; } = true;

        /// <summary>
        /// Range验证
        /// </summary>
        /// <param name="value"></param>
        /// <returns></returns>
        /// <exception cref="Exception"></exception>
        public override bool IsValid(object value)
        {
            if (value == null) return true;
            if (value.GetType() != typeof(double))
            {
                throw new Exception($"{nameof(RangeDoubleAttribute)}只能用于double类型!");
            }
            var val = (double)value;
            if (Min == null && Max == null) return true;
            else if (Min != null && Max != null)
            {

                if (IncludeMin && IncludeMax)
                {
                    ErrorMessage = "字段 {0} 范围" + $"[{Min},{Max}]";
                    return val >= Min && val <= Max;
                }
                else if (IncludeMin && !IncludeMax)
                {
                    ErrorMessage = "字段 {0} 范围" + $"[{Min},{Max})";
                    return val >= Min && val < Max;
                }
                else if (!IncludeMin && IncludeMax)
                {
                    ErrorMessage = "字段 {0} 范围" + $"({Min},{Max}]";
                    return val > Min && val <= Max;
                }
                else
                {
                    ErrorMessage = "字段 {0} 范围" + $"({Min},{Max})";
                    return val > Min && val < Max;
                }

            }
            else if (Min != null && Max == null)
            {
                if (IncludeMin)
                {
                    ErrorMessage = "字段 {0} 必须大于等于" + $"{Min}";
                    return val >= Min;
                }
                else
                {
                    ErrorMessage = "字段 {0} 必须大于" + $"{Min}";
                    return val > Min;
                }
            }
            else//Min==null&&Max!=null
            {
                if (IncludeMax)
                {
                    ErrorMessage = "字段 {0} 必须小于等于" + $"{Max}";
                    return val <= Max;
                }
                else
                {
                    ErrorMessage = "字段 {0} 必须小于" + $"{Max}";
                    return val < Max;
                }
            }
        }
    }
}