ZrAdminNetCore/ZR.Admin.WebApi/Extensions/EntityExtension.cs

using Microsoft.AspNetCore.Http;
using Snowflake.Core;
using System;

namespace ZR.Admin.WebApi.Extensions
{
    public static class EntityExtension
    {
        public static TSource ToCreate<TSource>(this TSource source, HttpContext context = null)
        {
            var types = source.GetType();

            //var worker = new IdWorker(1, 1);
            //if (types.GetProperty("ID") != null)
            //{
            //    long id = worker.NextId();

            //    types.GetProperty("ID").SetValue(source, id.ToString(), null);
            //}

            if (types.GetProperty("CreateTime") != null)
            {
                types.GetProperty("CreateTime").SetValue(source, DateTime.Now, null);
            }

            if (types.GetProperty("UpdateTime") != null)
            {
                types.GetProperty("UpdateTime").SetValue(source, DateTime.Now, null);
            }

            if (types.GetProperty("Create_by") != null && context != null)
            {
                types.GetProperty("Create_by").SetValue(source, context.GetName(), null);

                // types.GetProperty("CreateName").SetValue(source, userSession.UserName, null);
            }
            if (types.GetProperty("UserId") != null && context != null)
            {
                types.GetProperty("UserId").SetValue(source, context.GetUId(), null);
            }
            //if (types.GetProperty("UpdateID") != null)
            //{
            //    types.GetProperty("UpdateID").SetValue(source, userSession.UserID, null);

            //    types.GetProperty("UpdateName").SetValue(source, userSession.UserName, null);
            //}


            return source;
        }

        public static TSource ToUpdate<TSource>(this TSource source, HttpContext context = null)
        {
            var types = source.GetType();

            if (types.GetProperty("UpdateTime") != null)
            {
                types.GetProperty("UpdateTime").SetValue(source, DateTime.Now, null);
            }
            if (types.GetProperty("Update_Time") != null)
            {
                types.GetProperty("Update_Time").SetValue(source, DateTime.Now, null);
            }
            //if (types.GetProperty("UpdateID") != null)
            //{
            //    types.GetProperty("UpdateID").SetValue(source, userSession.UserID, null);
            //}

            if (types.GetProperty("UpdateBy") != null)
            {
                types.GetProperty("UpdateBy").SetValue(source,context.GetName(), null);
            }
            if (types.GetProperty("Update_by") != null)
            {
                types.GetProperty("Update_by").SetValue(source, context.GetName(), null);
            }

            return source;
        }

    }
}