ds8-solution-pro/ds-wms-service/DS.WMS.Core/Sys/Method/SysOrgService.cs

using DS.Module.Core;
using DS.Module.Core.Extensions;
using DS.Module.UserModule;
using DS.WMS.Core.Sys.Dtos;
using DS.WMS.Core.Sys.Entity;
using DS.WMS.Core.Sys.Interface;
using Mapster;
using Microsoft.Extensions.DependencyInjection;
using SqlSugar;

namespace DS.WMS.Core.Sys.Method;

/// <summary>
/// 
/// </summary>
public class SysOrgService:ISysOrgService
{
    
    private readonly IServiceProvider _serviceProvider;
    private readonly ISqlSugarClient db;
    private readonly IUser user;
    private readonly ICommonService _commonService;
    /// <summary>
    ///
    /// </summary>
    /// <param name="serviceProvider"></param>
    public SysOrgService(IServiceProvider serviceProvider)
    {
        _serviceProvider = serviceProvider;
        db = _serviceProvider.GetRequiredService<ISqlSugarClient>();
        user = _serviceProvider.GetRequiredService<IUser>();
        _commonService = _serviceProvider.GetRequiredService<ICommonService>();
    }
    public DataResult<List<SysOrgRes>> GetListByPage(PageRequest request)
    {
        //序列化查询条件
        var whereList = db.ConfigQuery.Context.Utilities.JsonToConditionalModels(request.QueryCondition);
        var data = db.Queryable<SysOrg>()
            .Select<SysOrgRes>()
            .Where(whereList).ToQueryPage(request.PageCondition);
        return data;
    }
    /// <summary>
    /// 
    /// </summary>
    /// <param name="req"></param>
    /// <returns></returns>
    public DataResult EditOrg(SysOrgReq req)
    {
        if (req.Id == 0)
        {
            var isExist = db.Queryable<SysOrg>().Where(x => x.OrgName == req.OrgName).First();
            if (isExist != null)
            {
                return DataResult.Failed("机构名称已存在!",MultiLanguageConst.OrgNameExist);
            }

            var data = req.Adapt<SysOrg>();

            var entity = db.Insertable(data).ExecuteReturnEntity();

            return DataResult.Successed("添加成功!", entity.Id,MultiLanguageConst.DataCreateSuccess);
        }
        else
        {
            var info = db.Queryable<SysOrg>().Where(x => x.Id == req.Id).First();

            info = req.Adapt(info);

            db.Updateable(info).IgnoreColumns(ignoreAllNullColumns: true).ExecuteCommand();
            return DataResult.Successed("更新成功!",MultiLanguageConst.DataUpdateSuccess);
        }
    }
    /// <summary>
    /// 获取机构信息
    /// </summary>
    /// <param name="id"></param>
    /// <returns></returns>
    public DataResult<SysOrgRes> GetOrgInfo(string id)
    {
        var data = db.Queryable<SysOrg>()
            .Where(a => a.Id == long.Parse(id))
            .Select<SysOrgRes>()
            .First();
        return DataResult<SysOrgRes>.Success(data,MultiLanguageConst.DataQuerySuccess);
    }
    
    /// <summary>
    /// 获取树型机构
    /// </summary>
    /// <returns></returns>
    public DataResult<List<OrgTree>> GetOrgTree()
    { 
        var list = db.Queryable<SysOrg>().Select(a => new OrgTree
        {
            Title = a.OrgName,
            Value = a.Id,
            ParentId = a.ParentId
        }).ToList();

        var orgList = BuildTree(list);

        return DataResult<List<OrgTree>>.Success("获取数据成功!", orgList);
    }
    
    /// <summary>
    /// 双层循环
    /// </summary>
    /// <param name="treeNodes"></param>
    /// <returns></returns>
    public static List<OrgTree> BuildTree(List<OrgTree> treeNodes)
    {
        try
        {
            List<OrgTree> trees = new List<OrgTree>();

            foreach (var treeNode in treeNodes)
            {
                if (0 == (treeNode.ParentId))
                {
                    trees.Add(treeNode);
                }

                foreach (var it in treeNodes)
                {
                    if (it.ParentId == treeNode.Value)
                    {
                        treeNode.Children.Add(it);
                    }
                }
            }
            return trees;
        }
        catch (Exception ex)
        {
            throw ex;
        }
    }
}