mirror of
				https://github.com/nsnail/FreeSql.git
				synced 2025-11-04 17:20:49 +08:00 
			
		
		
		
	v0.8.5
This commit is contained in:
		
							
								
								
									
										164
									
								
								Extensions/FreeSql.Extensions.BaseEntity/BaseEntityTree.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										164
									
								
								Extensions/FreeSql.Extensions.BaseEntity/BaseEntityTree.cs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,164 @@
 | 
			
		||||
using FreeSql;
 | 
			
		||||
using FreeSql.DataAnnotations;
 | 
			
		||||
using System;
 | 
			
		||||
using System.Collections.Generic;
 | 
			
		||||
using System.Linq;
 | 
			
		||||
using System.Text;
 | 
			
		||||
using System.Threading.Tasks;
 | 
			
		||||
 | 
			
		||||
namespace FreeSql
 | 
			
		||||
{
 | 
			
		||||
    /// <summary>
 | 
			
		||||
    /// 树状基类
 | 
			
		||||
    /// </summary>
 | 
			
		||||
    /// <typeparam name="TEntity"></typeparam>
 | 
			
		||||
    /// <typeparam name="TKey"></typeparam>
 | 
			
		||||
    [Table(DisableSyncStructure = true)]
 | 
			
		||||
    public abstract class BaseEntityTree<TEntity, TKey> : BaseEntity<TEntity, TKey> where TEntity : class
 | 
			
		||||
    {
 | 
			
		||||
        /// <summary>
 | 
			
		||||
        /// 父级id
 | 
			
		||||
        /// </summary>
 | 
			
		||||
        public TKey ParentId
 | 
			
		||||
        {
 | 
			
		||||
            get => _ParentId;
 | 
			
		||||
            set
 | 
			
		||||
            {
 | 
			
		||||
                if (Equals(value, default(TKey)) == false && Equals(value, Id))
 | 
			
		||||
                    throw new ArgumentException("ParentId 值不能与 Id 相同");
 | 
			
		||||
                _ParentId = value;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        public TEntity Parent { get; set; }
 | 
			
		||||
        private TKey _ParentId;
 | 
			
		||||
 | 
			
		||||
        /// <summary>
 | 
			
		||||
        /// 下级列表
 | 
			
		||||
        /// </summary>
 | 
			
		||||
        [Navigate("ParentId")]
 | 
			
		||||
        public List<TEntity> Childs { get; set; }
 | 
			
		||||
 | 
			
		||||
        /// <summary>
 | 
			
		||||
        /// 名称
 | 
			
		||||
        /// </summary>
 | 
			
		||||
        public string Name { get; set; }
 | 
			
		||||
        /// <summary>
 | 
			
		||||
        /// 名称:技术部-前端
 | 
			
		||||
        /// </summary>
 | 
			
		||||
        public string FullName { get; set; }
 | 
			
		||||
 | 
			
		||||
        public List<TEntity> GetAllChilds() => Select.WhereDynamic(this)
 | 
			
		||||
            .IncludeMany(a => (a as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t1 => t1.IncludeMany(a1 => (a1 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t2 => t2.IncludeMany(a2 => (a2 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t3 => t3.IncludeMany(a3 => (a3 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t4 => t4.IncludeMany(a4 => (a4 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t5 => t5.IncludeMany(a5 => (a5 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t6 => t6.IncludeMany(a6 => (a6 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t7 => t7.IncludeMany(a7 => (a7 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t8 => t8.IncludeMany(a8 => (a8 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t9 => t9.IncludeMany(a9 => (a9 as BaseEntityTree<TEntity, TKey>).Childs,
 | 
			
		||||
                t10 => t10.IncludeMany(a10 => (a10 as BaseEntityTree<TEntity, TKey>).Childs))))))))))).ToList()
 | 
			
		||||
                .SelectMany(a => (a as BaseEntityTree<TEntity, TKey>).Childs
 | 
			
		||||
                .SelectMany(a1 => (a1 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a2 => (a2 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a3 => (a3 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a4 => (a4 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a5 => (a5 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a6 => (a6 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a7 => (a7 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a8 => (a8 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a9 => (a9 as BaseEntityTree<TEntity, TKey>)?.Childs
 | 
			
		||||
                .SelectMany(a10 => (a10 as BaseEntityTree<TEntity, TKey>)?.Childs))))))))))).Where(a => a != null).ToList();
 | 
			
		||||
 | 
			
		||||
        protected void RefershFullName()
 | 
			
		||||
        {
 | 
			
		||||
            var buf = new List<TEntity>();
 | 
			
		||||
            buf.Add(this as TEntity);
 | 
			
		||||
            buf.AddRange(this.GetAllChilds());
 | 
			
		||||
            var repo = Orm.GetRepository<TEntity>();
 | 
			
		||||
            repo.UnitOfWork = UnitOfWork.Current.Value;
 | 
			
		||||
            buf = repo.Select.WhereDynamic(buf)
 | 
			
		||||
                .Include(a => ((((((((((a as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent
 | 
			
		||||
                    as BaseEntityTree<TEntity, TKey>).Parent).ToList(true);
 | 
			
		||||
            foreach (var item in buf)
 | 
			
		||||
            {
 | 
			
		||||
                var up = item as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
                up.Name = up.Name;
 | 
			
		||||
                var cur = up.Parent as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
                while (cur != null)
 | 
			
		||||
                {
 | 
			
		||||
                    up.Name = $"{cur.Name}-{up.Name}";
 | 
			
		||||
                    cur = cur.Parent as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            repo.Update(buf);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        T UpdateIsDelete<T>(bool value, Func<BaseRepository<TEntity>, List<TEntity>, T> func)
 | 
			
		||||
        {
 | 
			
		||||
            var childs = GetAllChilds();
 | 
			
		||||
            childs.Add(this as TEntity);
 | 
			
		||||
            var repo = Orm.GetRepository<TEntity>();
 | 
			
		||||
            repo.UnitOfWork = UnitOfWork.Current.Value;
 | 
			
		||||
            repo.Attach(childs);
 | 
			
		||||
            foreach (var item in childs)
 | 
			
		||||
                (item as BaseEntity).IsDeleted = false;
 | 
			
		||||
            return func(repo, childs);
 | 
			
		||||
        }
 | 
			
		||||
        public override bool Delete() => UpdateIsDelete(true, (repo, chis) => repo.Update(chis)) > 0;
 | 
			
		||||
        async public override Task<bool> DeleteAsync() => await UpdateIsDelete(true, (repo, chis) => repo.UpdateAsync(chis)) > 0;
 | 
			
		||||
 | 
			
		||||
        public override bool Restore() => UpdateIsDelete(false, (repo, chis) => repo.Update(chis)) > 0;
 | 
			
		||||
        async public override Task<bool> RestoreAsync() => await UpdateIsDelete(false, (repo, chis) => repo.UpdateAsync(chis)) > 0;
 | 
			
		||||
 | 
			
		||||
        public override TEntity Insert()
 | 
			
		||||
        {
 | 
			
		||||
            var ret = base.Insert();
 | 
			
		||||
            RefershFullName();
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
        async public override Task<TEntity> InsertAsync()
 | 
			
		||||
        {
 | 
			
		||||
            var ret = await base.InsertAsync();
 | 
			
		||||
            RefershFullName();
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
        public override bool Update()
 | 
			
		||||
        {
 | 
			
		||||
            var old = Find(this.Id) as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
            var ret = base.Update();
 | 
			
		||||
            if (old.Name != this.Name || Equals(old.ParentId, old.ParentId) == false) RefershFullName();
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
        async public override Task<bool> UpdateAsync()
 | 
			
		||||
        {
 | 
			
		||||
            var old = Find(this.Id) as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
            var ret = await base.UpdateAsync();
 | 
			
		||||
            if (old.Name != this.Name || Equals(old.ParentId, old.ParentId) == false) RefershFullName();
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
        public override TEntity Save()
 | 
			
		||||
        {
 | 
			
		||||
            var old = Find(this.Id) as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
            var ret = base.Save();
 | 
			
		||||
            if (old.Name != this.Name || Equals(old.ParentId, old.ParentId) == false) RefershFullName();
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
        async public override Task<TEntity> SaveAsync()
 | 
			
		||||
        {
 | 
			
		||||
            var old = Find(this.Id) as BaseEntityTree<TEntity, TKey>;
 | 
			
		||||
            var ret = await base.SaveAsync();
 | 
			
		||||
            if (old.Name != this.Name || Equals(old.ParentId, old.ParentId) == false) RefershFullName();
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user