From 0b49fee10340ea40a6762da657239675d3db22f6 Mon Sep 17 00:00:00 2001
From: sunpengfei <i@angelzzz.com>
Date: 星期四, 07 八月 2025 09:41:31 +0800
Subject: [PATCH] feat:权限和请求日志

---
 FlexJobApi.Core/Utils/DbUtils/DbUtils.cs |  227 ++++++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 187 insertions(+), 40 deletions(-)

diff --git a/FlexJobApi.Core/Utils/DbUtils/DbUtils.cs b/FlexJobApi.Core/Utils/DbUtils/DbUtils.cs
index e6a1389..03c260d 100644
--- a/FlexJobApi.Core/Utils/DbUtils/DbUtils.cs
+++ b/FlexJobApi.Core/Utils/DbUtils/DbUtils.cs
@@ -1,4 +1,5 @@
-锘縰sing Furion;
+锘縰sing Consul.Filtering;
+using Furion;
 using Furion.DatabaseAccessor;
 using Furion.DistributedIDGenerator;
 using Furion.FriendlyException;
@@ -7,6 +8,7 @@
 using Microsoft.EntityFrameworkCore.ChangeTracking;
 using Microsoft.EntityFrameworkCore.Diagnostics;
 using Microsoft.EntityFrameworkCore.Metadata.Builders;
+using Microsoft.EntityFrameworkCore.Migrations;
 using Microsoft.Extensions.Configuration;
 using Newtonsoft.Json;
 using System;
@@ -16,6 +18,7 @@
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using System.Timers;
 
 namespace FlexJobApi.Core
 {
@@ -24,6 +27,105 @@
     /// </summary>
     public static class DbUtils
     {
+        /// <summary>
+        /// 鑾峰彇鏍戝舰鏁版嵁璺緞
+        /// </summary>
+        /// <typeparam name="TEntity"></typeparam>
+        /// <param name="parentId"></param>
+        /// <param name="cancellationToken"></param>
+        /// <returns></returns>
+        public static async Task<string> GetTreeDataPath<TEntity>(
+             Guid? parentId,
+            CancellationToken cancellationToken = default)
+            where TEntity : CommonEntity, ITreeData<TEntity>, new()
+        {
+            var rep = Db.GetRepository<TEntity>();
+            if (parentId.HasValue)
+            {
+                var parent = await rep.AsQueryable().AsNoTracking()
+                    .Where(it => it.Id == parentId)
+                    .Select(it => new
+                    {
+                        it.Path,
+                        it.Code
+                    })
+                    .FirstOrDefaultAsync(cancellationToken);
+                var summary = typeof(TEntity).GetSummary();
+                if (parent == null) throw Oops.Oh(EnumErrorCodeType.s404, $"涓婄骇{summary}");
+                return $"{parent.Path}{parent.Code}/";
+            }
+            else
+            {
+                return "/";
+            }
+        }
+
+        /// <summary>
+        /// 鏇存柊鏍戝舰鏁版嵁涓嬬骇璺緞
+        /// </summary>
+        /// <param name="oldPath"></param>
+        /// <param name="newPath"></param>
+        /// <param name="cancellationToken"></param>
+        /// <returns></returns>
+        public static async Task UpdateTreeDataChildrenPath<TEntity>(
+            string oldPath,
+            string newPath,
+            CancellationToken cancellationToken)
+            where TEntity : CommonEntity, ITreeData<TEntity>, new()
+        {
+            var rep = Db.GetRepository<TEntity>();
+            var models = await rep.AsQueryable()
+                .Where(it => it.Path.StartsWith(oldPath))
+                .ToListAsync(cancellationToken);
+            if (models.IsNotNull())
+            {
+                foreach (var model in models)
+                {
+                    model.Path = model.Path.Replace(oldPath, newPath);
+                }
+                await rep.UpdateAsync(models);
+            }
+        }
+
+        /// <summary>
+        /// 鏌ヨ閫夋嫨鍣ㄦ暟鎹�
+        /// </summary>
+        /// <typeparam name="TEntity"></typeparam>
+        /// <typeparam name="TValue"></typeparam>
+        /// <typeparam name="TData"></typeparam>
+        /// <param name="request"></param>
+        /// <param name="getValue"></param>
+        /// <param name="getLabel"></param>
+        /// <param name="query"></param>
+        /// <param name="cancellationToken"></param>
+        /// <returns></returns>
+        public static async Task<List<SelectOption<TValue, TData>>> GetSelect<TEntity, TValue, TData>(
+            this SelectQuery<TValue, TData> request,
+            Func<TData, TValue> getValue,
+            Func<TData, string> getLabel,
+            Func<IQueryable<TEntity>, IQueryable<TEntity>> query = null,
+            CancellationToken cancellationToken = default)
+            where TEntity : CommonEntity, new()
+        {
+            var rep = Db.GetRepository<TEntity>();
+            var q = rep.AsQueryable().AsNoTracking();
+            if (query != null) q = query(q);
+            else q = q.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime);
+            var models = await q
+                .ProjectToType<TData>()
+                .ToListAsync(cancellationToken);
+            var options = new List<SelectOption<TValue, TData>>();
+            foreach (var model in models)
+            {
+                var option = new SelectOption<TValue, TData>();
+                option.Data = model;
+                option.Value = getValue(model);
+                option.Label = getLabel(model);
+                options.Add(option);
+            }
+            return options;
+        }
+
         /// <summary>
         /// 鏌ヨ鍒嗛〉鍒楄〃鏁版嵁
         /// </summary>
@@ -34,8 +136,8 @@
         /// <param name="selector"></param>
         /// <param name="cancellationToken"></param>
         /// <returns></returns>
-        public static async Task<PagedListQueryResult<TItem>> ToPagedListAsync<TEntity, TItem>(
-            this PagedListQueryPageModel page, 
+        public static async Task<PagedListQueryResult<TItem>> GetPagedListAsync<TEntity, TItem>(
+            this PagedListQueryPageModel page,
             Func<IQueryable<TEntity>, IQueryable<TEntity>> query = null,
             Expression<Func<TEntity, TItem>> selector = null,
             CancellationToken cancellationToken = default)
@@ -45,11 +147,36 @@
             var rep = Db.GetRepository<TEntity>();
             var q = rep.AsQueryable().AsNoTracking();
             if (query != null) q = query(q);
+            else q = q.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime);
             q = q.OrderBy(page.OrderInput);
             var s = selector == null
                 ? q.ProjectToType<TItem>()
                 : q.Select(selector);
             var pagedList = await s.ToPagedListAsync(page.Page, page.Rows, cancellationToken);
+            var result = new PagedListQueryResult<TItem>();
+            result.PageModel = page.Adapt<PagedListQueryResultPageModel>();
+            result.PageModel.TotalCount = pagedList.TotalCount;
+            result.PageModel.TotalPage = pagedList.TotalPages;
+            result.Data = pagedList.Items.ToList();
+            return result;
+        }
+
+        /// <summary>
+        /// 鏌ヨ鍒嗛〉鍒楄〃鏁版嵁
+        /// </summary>
+        /// <typeparam name="TItem"></typeparam>
+        /// <param name="page"></param>
+        /// <param name="q"></param>
+        /// <param name="cancellationToken"></param>
+        /// <returns></returns>
+        public static async Task<PagedListQueryResult<TItem>> GetPagedListAsync<TItem>(
+            this PagedListQueryPageModel page,
+            IQueryable<TItem> q,
+            CancellationToken cancellationToken = default)
+            where TItem : class, new()
+        {
+            q = q.OrderBy(page.OrderInput);
+            var pagedList = await q.ToPagedListAsync(page.Page, page.Rows, cancellationToken);
             var result = new PagedListQueryResult<TItem>();
             result.PageModel = page.Adapt<PagedListQueryResultPageModel>();
             result.PageModel.TotalCount = pagedList.TotalCount;
@@ -65,7 +192,9 @@
         /// <param name="q"></param>
         /// <param name="orders"></param>
         /// <returns></returns>
-        public static IQueryable<T> OrderBy<T>(this IQueryable<T> q, List<PagedListQueryPageModelOrderInput> orders)
+        public static IQueryable<T> OrderBy<T>(
+            this IQueryable<T> q,
+            List<PagedListQueryPageModelOrderInput> orders)
         {
             if (orders.IsNull()) return q;
 
@@ -126,7 +255,10 @@
         /// <param name="query"></param>
         /// <param name="cancellationToken"></param>
         /// <returns></returns>
-        public static async Task<int> SetIsDisable<TEntity>(this SetIsDisabledCommand request, Func<IQueryable<TEntity>, IQueryable<TEntity>> query = null, CancellationToken cancellationToken = default)
+        public static async Task<int> SetIsDisabled<TEntity>(
+            this SetIsDisabledCommand request,
+            Func<IQueryable<TEntity>, IQueryable<TEntity>> query = null,
+            CancellationToken cancellationToken = default)
             where TEntity : CommonEntity, IIsDisabled, new()
         {
             var rep = Db.GetRepository<TEntity>();
@@ -150,7 +282,10 @@
         /// <param name="query"></param>
         /// <param name="cancellationToken"></param>
         /// <returns></returns>
-        public static async Task<int> DeleteData<TEntity>(this DeleteDataCommand request, Func<IQueryable<TEntity>, IQueryable<TEntity>> query = null, CancellationToken cancellationToken = default)
+        public static async Task<int> DeleteData<TEntity>(
+            this DeleteDataCommand request, Func<IQueryable<TEntity>,
+            IQueryable<TEntity>> query = null,
+            CancellationToken cancellationToken = default)
             where TEntity : CommonEntity, new()
         {
             var rep = Db.GetRepository<TEntity>();
@@ -171,29 +306,36 @@
         /// <typeparam name="TRequest"></typeparam>
         /// <param name="request"></param>
         /// <param name="checkExist"></param>
+        /// <param name="update"></param>
         /// <param name="cancellationToken"></param>
         /// <returns></returns>
-        public static async Task<Guid> SaveData<TEntity, TRequest>(this TRequest request, Func<IQueryable<TEntity>, TEntity, TRequest, bool> checkExist = null, CancellationToken cancellationToken = default)
+        public static async Task<Guid> SaveData<TEntity, TRequest>(
+            this TRequest request,
+            Expression<Func<TEntity, bool>> checkExist = null,
+            Action<TEntity> update = null,
+            CancellationToken cancellationToken = default)
             where TEntity : CommonEntity, new()
             where TRequest : SaveDataCommand, new()
         {
             var xmlDoc = await XmlDocUtils.GetXmlDocAsync();
             var summary = await typeof(TEntity).GetSummary(xmlDoc);
             var rep = Db.GetRepository<TEntity>();
+            if (checkExist != null && await rep.AsQueryable().AsNoTracking().AnyAsync(checkExist))
+                throw Oops.Oh(EnumErrorCodeType.s405, $"璇summary ?? "淇℃伅"}");
             if (request.Id.HasValue)
             {
                 var entity = await rep.AsQueryable().FirstOrDefaultAsync(it => it.Id == request.Id, cancellationToken);
                 if (entity == null) throw Oops.Oh(EnumErrorCodeType.s404, $"璇summary ?? "淇℃伅"}");
-                if (checkExist != null && checkExist(rep.AsQueryable().AsNoTracking(), entity, request)) throw Oops.Oh(EnumErrorCodeType.s405, $"璇summary ?? "淇℃伅"}");
-                request.Adapt(entity);
+                if (update != null) update(entity);
+                else request.Adapt(entity);
                 await rep.UpdateAsync(entity);
                 return entity.Id;
             }
             else
             {
                 var entity = new TEntity();
-                if (checkExist != null && checkExist(rep.AsQueryable().AsNoTracking(), entity, request)) throw Oops.Oh(EnumErrorCodeType.s405, $"璇summary ?? "淇℃伅"}");
-                request.Adapt(entity);
+                if (update != null) update(entity);
+                else request.Adapt(entity);
                 await rep.InsertAsync(entity);
                 return entity.Id;
             }
@@ -207,35 +349,41 @@
         /// <returns></returns>
         public static async Task BuildEntity(ModelBuilder modelBuilder, Type dbContextLocator = null)
         {
-            var xmlDoc = await XmlDocUtils.GetXmlDocAsync();
-            foreach (var entityType in modelBuilder.Model.GetEntityTypes())
+            if (App.GetService<IMigrator>() != null)
             {
-                Console.WriteLine($"姝e湪鐢熸垚琛細{entityType.Name}");
-                // 鑾峰彇瀹炰綋绫荤殑XML娉ㄩ噴锛屽苟璁剧疆涓鸿〃娉ㄩ噴
-                var entityBuilder = modelBuilder.Entity(entityType.ClrType);
-                string typeComment = (await entityType.ClrType.GetXmlDocMemberAsync(xmlDoc))?.Summary;
-                if (!string.IsNullOrEmpty(typeComment))
+                var xmlDoc = await XmlDocUtils.GetXmlDocAsync();
+                foreach (var entityType in modelBuilder.Model.GetEntityTypes())
                 {
-                    Console.WriteLine($"姝e湪鐢熸垚琛ㄦ敞閲婏細{entityType.Name}-{typeComment}");
-                    entityBuilder.ToTable(it => it.HasComment(typeComment));
-                }
-                // 鑾峰彇瀹炰綋灞炴�х殑XML娉ㄩ噴锛屽苟璁剧疆涓哄垪娉ㄩ噴
-                var properties = entityType.ClrType.GetProperties()
-                    .Where(p =>
-                        p.CanRead
-                        && p.CanWrite
-                        && !typeof(System.Collections.ICollection).IsAssignableFrom(p.PropertyType)
-                        && (p.PropertyType.IsClass ? p.PropertyType.FullName.Contains("System.") : true));
-                foreach (var property in properties)
-                {
-                    string propComment = (await property.GetXmlDocMemberAsync(xmlDoc))?.Summary;
-                    if (!string.IsNullOrEmpty(propComment))
+                    Console.WriteLine($"姝e湪鐢熸垚琛細{entityType.Name}");
+                    // 鑾峰彇瀹炰綋绫荤殑XML娉ㄩ噴锛屽苟璁剧疆涓鸿〃娉ㄩ噴
+                    var entityBuilder = modelBuilder.Entity(entityType.ClrType);
+                    string typeComment = (await entityType.ClrType.GetXmlDocMemberAsync(xmlDoc))?.Summary;
+                    if (!string.IsNullOrEmpty(typeComment))
                     {
-                        Console.WriteLine($"姝e湪鐢熸垚灞炴�ф敞閲婏細{property.Name}-{propComment}");
-                        entityBuilder.Property(property.Name).HasComment(propComment);
+                        Console.WriteLine($"姝e湪鐢熸垚琛ㄦ敞閲婏細{entityType.Name}-{typeComment}");
+                        entityBuilder.ToTable(it => it.HasComment(typeComment));
+                    }
+                    // 鑾峰彇瀹炰綋灞炴�х殑XML娉ㄩ噴锛屽苟璁剧疆涓哄垪娉ㄩ噴
+                    var properties = entityType.ClrType.GetProperties()
+                        .Where(p =>
+                            p.CanRead
+                            && p.CanWrite
+                            && !typeof(System.Collections.ICollection).IsAssignableFrom(p.PropertyType)
+                            && (p.PropertyType.IsClass ? p.PropertyType.FullName.Contains("System.") : true));
+                    foreach (var property in properties)
+                    {
+                        string propComment = (await property.GetXmlDocMemberAsync(xmlDoc))?.Summary;
+                        if (!string.IsNullOrEmpty(propComment))
+                        {
+                            Console.WriteLine($"姝e湪鐢熸垚灞炴�ф敞閲婏細{property.Name}-{propComment}");
+                            entityBuilder.Property(property.Name).HasComment(propComment);
+                        }
                     }
                 }
+            }
 
+            foreach (var entityType in modelBuilder.Model.GetEntityTypes())
+            {
                 if (typeof(CommonEntity).IsAssignableFrom(entityType.ClrType))
                 {
                     // 鏋勫缓绛涢�夋潯浠讹細IsDeleted == false
@@ -249,8 +397,6 @@
                     modelBuilder.Entity(entityType.ClrType).HasQueryFilter(lambda);
                 }
             }
-
-            Console.WriteLine("鏁版嵁搴撻摼鎺ュ湴鍧�锛�" + App.Configuration.GetConnectionString("FlexJobApi"));
         }
 
         /// <summary>
@@ -279,6 +425,7 @@
 
             // 閫氳繃璇锋眰涓幏鍙栧綋鍓嶆搷浣滀汉
             var logier = JwtUtils.GetCurrentLogier();
+            var traceId = App.GetTraceId();
 
             // 鑾峰彇鎵�鏈夊凡鏇存敼鐨勫疄浣�
             foreach (var entity in entities)
@@ -312,9 +459,9 @@
 
                     // 璧嬪�艰窡韪狪d
                     prop = entity.Property(nameof(CommonEntity.TraceId));
-                    if (prop != null)
+                    if (prop != null && prop.CurrentValue == null && traceId.IsNotNull())
                     {
-                        prop.CurrentValue = App.GetTraceId();
+                        prop.CurrentValue = traceId;
                     }
                 }
                 else
@@ -335,9 +482,9 @@
 
                     // 璧嬪�艰窡韪狪d
                     prop = entity.Property(nameof(CommonEntity.TraceId));
-                    if (prop != null)
+                    if (prop != null && traceId.IsNotNull())
                     {
-                        prop.CurrentValue = App.GetTraceId();
+                        prop.CurrentValue = traceId;
                     }
 
                     // 杞垹闄�

--
Gitblit v1.9.1