zhuangpei-mesbackend/Infrastructure/WebExtensions/EntityExtension.cs


using Infrastructure.Extensions;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Http.HttpResults;
using System;
using System.Reflection;

namespace Infrastructure
{
    public static class EntityExtension
    {
        public static TSource ToCreate<TSource>(this TSource source, HttpContext? context = null)
        {
            var types = source?.GetType();
            if (types == null || context == null) return source;
            BindingFlags flag = BindingFlags.Public | BindingFlags.IgnoreCase | BindingFlags.Instance;

            types.GetProperty("CreateTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("CreatedTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("AddTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("CreateBy", flag)?.SetValue(source, context.GetName(), null);
            types.GetProperty("CreatedBy", flag)?.SetValue(source, context.GetName(), null);
            types.GetProperty("Create_by", flag)?.SetValue(source, context.GetName(), null);
            types.GetProperty("UserId", flag)?.SetValue(source, context.GetUId(), null);

            return source;
        }
        /// <summary>
        /// 用昵称创造
        /// </summary>
        /// <typeparam name="TSource"></typeparam>
        /// <param name="source"></param>
        /// <param name="context"></param>
        /// <returns></returns>
        public static TSource ToCreate_nickName<TSource>(this TSource source, HttpContext? context = null)
        {
            var types = source?.GetType();
            if (types == null || context == null) return source;
            BindingFlags flag = BindingFlags.Public | BindingFlags.IgnoreCase | BindingFlags.Instance;

            types.GetProperty("CreateTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("CreatedTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("AddTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("CreateBy", flag)?.SetValue(source, context.GetNickName(), null);
            types.GetProperty("CreatedBy", flag)?.SetValue(source, context.GetNickName(), null);
            types.GetProperty("Create_by", flag)?.SetValue(source, context.GetNickName(), null);
            types.GetProperty("UserId", flag)?.SetValue(source, context.GetNickName(), null);

            return source;
        }

        public static TSource ToUpdate<TSource>(this TSource source, HttpContext? context = null)
        {
            var types = source?.GetType();
            if (types == null || context == null) return source;
            BindingFlags flag = BindingFlags.Public | BindingFlags.IgnoreCase | BindingFlags.Instance;

            types.GetProperty("UpdateTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("UpdatedTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("Update_time", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("UpdateBy", flag)?.SetValue(source, context.GetName(), null); 
            types.GetProperty("UpdatedBy", flag)?.SetValue(source, context.GetName(), null); 
            types.GetProperty("Update_by", flag)?.SetValue(source, context.GetName(), null);

            return source;
        }
        /// <summary>
        /// 用昵称更新
        /// </summary>
        /// <typeparam name="TSource"></typeparam>
        /// <param name="source"></param>
        /// <param name="context"></param>
        /// <returns></returns>
        public static TSource ToUpdate_nickName<TSource>(this TSource source, HttpContext? context = null)
        {
            var types = source?.GetType();
            if (types == null || context == null) return source;
            BindingFlags flag = BindingFlags.Public | BindingFlags.IgnoreCase | BindingFlags.Instance;

            types.GetProperty("UpdateTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("UpdatedTime", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("Update_time", flag)?.SetValue(source, DateTime.Now, null);
            types.GetProperty("UpdateBy", flag)?.SetValue(source, context.GetNickName(), null);
            types.GetProperty("UpdatedBy", flag)?.SetValue(source, context.GetNickName(), null);
            types.GetProperty("Update_by", flag)?.SetValue(source, context.GetNickName(), null);

            return source;
        }
        
    }
}