合并冲突
xiyunfei 编写于 2021-08-15 10:21:33
NewLife.Cube
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Mvc.ModelBinding;
using Microsoft.AspNetCore.Mvc.ModelBinding.Binders;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using NewLife.Collections;
using NewLife.Cube.Common;
using NewLife.Cube.Extensions;
using NewLife.Log;
using NewLife.Reflection;
using NewLife.Serialization;
using XCode;

namespace NewLife.Cube
{
    /// <summary>实体模型绑定器</summary>
    class EntityModelBinder : ComplexTypeModelBinder
    {
        /// <summary>实例化实体模型绑定器</summary>
        /// <param name="propertyBinders"></param>
        /// <param name="loggerFactory"></param>
        public EntityModelBinder(IDictionary<ModelMetadata, IModelBinder> propertyBinders, ILoggerFactory loggerFactory)
            : base(propertyBinders, loggerFactory) { }

        /// <summary>创建模型。对于有Key的请求,使用FindByKeyForEdit方法先查出来数据,而不是直接反射实例化实体对象</summary>
        /// <param name="bindingContext"></param>
        /// <returns></returns>
        protected override Object CreateModel(ModelBindingContext bindingContext)
        {
            var modelType = bindingContext.ModelType;
            if (!modelType.As<IEntity>()) return base.CreateModel(bindingContext);

            var fact = EntityFactory.CreateOperate(modelType);

            if (fact == null) return base.CreateModel(bindingContext);

            // 尝试从body读取json格式的参数
            var ctx = bindingContext.HttpContext;
            var request = ctx.Request;
            if (request.GetRequestBody<Object>() != null)
            {
                ctx.Items["EntityBody"] = ctx.Items["RequestBody"];
                var cubeBodyValueProvider = new CubeBodyValueProvider(bindingContext.ValueProvider,
                    ctx.Items["EntityBody"] as NullableDictionary<String, Object>);

                // 添加body提供者,从body中取值,只取第一层,
                // 下面的BindProperty方法,以前从body中并没有处理值的格式,
                // 强行绑定会出错记录在ModelState,在api中返回400错误,mvc不会
                bindingContext.ValueProvider = cubeBodyValueProvider;
            }

            var pks = fact.Table.PrimaryKeys;
            var uk = fact.Unique;

            IEntity entity = null;
            if (uk != null)
            {
                // 查询实体对象用于编辑
                var id = bindingContext.ValueProvider.GetValue(uk.Name);
                if (id != ValueProviderResult.None) entity = fact.FindByKeyForEdit(id.ToString());
            }
            else if (pks.Length > 0)
            {
                // 查询实体对象用于编辑
                var exp = new WhereExpression();
                foreach (var item in pks)
                {
                    var v = bindingContext.ValueProvider.GetValue(item.Name);
                    if(v == ValueProviderResult.None) continue;
                    exp &= item.Equal(v.ChangeType(item.Type));
                }

                entity = fact.Find(exp);
            }

            return entity ?? fact.Create(true);
        }

        protected override Boolean CanBindProperty(ModelBindingContext bindingContext, ModelMetadata propertyMetadata)
        {
            // 不要绑定复杂类型,那是扩展属性
            if (propertyMetadata.ModelType.GetTypeCode() == TypeCode.Object) return false;

            return base.CanBindProperty(bindingContext, propertyMetadata);
        }

        /// <summary>
        /// 绑定属性,在这里赋值
        /// </summary>
        /// <param name="bindingContext"></param>
        /// <returns></returns>
        protected override Task BindProperty(ModelBindingContext bindingContext)
        {
            var metadata = bindingContext.ModelMetadata;

            switch (metadata.ModelType.GetTypeCode())
            {
                case TypeCode.DateTime:
                    // 客户端可能提交空时间,不要绑定属性,以免出现空时间验证失败
                    //if (result.Model is not DateTime) return Task.CompletedTask;
                    var dt = bindingContext.ValueProvider.GetValue(metadata.Name).Values;
                    if (dt.Count == 0 || dt.ToString().IsNullOrEmpty()) return Task.CompletedTask;

                    break;
            }

            return base.BindProperty(bindingContext);
        }

        /// <summary>
        /// 设置属性,二次处理
        /// </summary>
        /// <param name="bindingContext"></param>
        /// <param name="modelName"></param>
        /// <param name="propertyMetadata"></param>
        /// <param name="result"></param>
        protected override void SetProperty(ModelBindingContext bindingContext, String modelName, ModelMetadata propertyMetadata, ModelBindingResult result)
        {
            switch (propertyMetadata.ModelType.GetTypeCode())
            {
                case TypeCode.String:
                    // 如果有多个值,则修改结果,避免 3,2,5 变成只有3
                    var vs = bindingContext.ValueProvider.GetValue(modelName).Values;
                    if (vs.Count > 1) result = ModelBindingResult.Success(vs.ToString());
                    break;
            }

            base.SetProperty(bindingContext, modelName, propertyMetadata, result);
        }
    }

    /// <summary>实体模型绑定器提供者,为所有XCode实体类提供实体模型绑定器</summary>
    public class EntityModelBinderProvider : IModelBinderProvider
    {
        /// <summary>获取绑定器</summary>
        /// <param name="context"></param>
        /// <returns></returns>
        public IModelBinder GetBinder(ModelBinderProviderContext context)
        {
            if (context == null) throw new ArgumentNullException(nameof(context));

            if (!context.Metadata.ModelType.As<IEntity>()) return null;

            var loggerFactory = context.Services.GetRequiredService<ILoggerFactory>();
            var propertyBinders = new Dictionary<ModelMetadata, IModelBinder>();
            foreach (var property in context.Metadata.Properties)
            {
                propertyBinders.Add(property, context.CreateBinder(property));
            }

            return new EntityModelBinder(propertyBinders, loggerFactory);
        }

        /// <summary>实例化</summary>
        public EntityModelBinderProvider() => XTrace.WriteLine("注册实体模型绑定器:{0}", typeof(EntityModelBinderProvider).FullName);
    }
}