using MES.Service.DB; using MES.Service.Dto.webApi; using MES.Service.Modes; using SqlSugar; namespace MES.Service.service.BasicData; public class SysDepartmentManager : Repository { // 当前类已经继承了 Repository 增、删、查、改的方法 // Save 方法用于保存单个部门记录,根据类型执行不同的操作 public bool Save(ErpDepartment department) { var entity = GetSysDepartment(department); // 将 ErpDepartment 转换为 SysDepartment return UseTransaction(db => { switch (department.Type) { case "0": if (UpdateDepartmentStatus(db, entity.Id, "A")) // 启用部门 return 1; break; case "1": if (UpdateDepartmentStatus(db, entity.Id, "B")) // 禁用部门 return 1; break; case "3": if (DeleteDepartment(db, entity.Id)) // 删除部门 return 1; break; case "2": case "4": if (InsertOrUpdateDepartment(db, entity)) // 插入或更新部门 return 1; break; default: throw new ArgumentNullException( $"type没有{department.Type}这个类型的参数"); } throw new NotImplementedException("操作失败"); }) > 0; } // 更新部门状态的方法 private bool UpdateDepartmentStatus(SqlSugarScope db, decimal departmentId, string status) { var result = db.Updateable() .SetColumns(s => s.Depextr4 == status) .Where(s => s.Id == departmentId).ExecuteCommand(); if (result > 0) return true; throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } // 插入新部门的方法 private bool InsertDepartment(SqlSugarScope db, SysDepartment entity) { var insert = db.Insertable(entity).ExecuteCommand(); if (insert > 0) return true; throw new NotImplementedException("插入失败"); } // 删除部门的方法 private bool DeleteDepartment(SqlSugarScope db, decimal departmentId) { var deleteById = db.Deleteable() .Where(s => s.Id == departmentId).ExecuteCommand(); if (deleteById > 0) return true; throw new NotImplementedException("删除失败"); } /// /// 生成新的ID,确保不重复 /// private decimal GenerateNewId() { // 处理空表的情况,从1开始 var maxId = Db.Queryable().Max(x => (decimal?)x.Id) ?? 0; var newId = maxId + 1; // 双重检查,确保生成的ID不存在 while (Db.Queryable().Where(x => x.Id == newId).Any()) { newId++; } return newId; } // 插入或更新部门的方法 private bool InsertOrUpdateDepartment(SqlSugarScope db, SysDepartment entity) { if (entity.Id == 0) { // 新增情况:生成新ID并插入 var newId = GenerateNewId(); entity.Id = newId; return db.Insertable(entity).IgnoreColumns(ignoreNullColumn:true).ExecuteCommand() > 0; } else { // 更新情况:删除后重新插入,保持原有ID var originalId = entity.Id; // 先删除原记录(如果存在) db.Deleteable().Where(s => s.Id == originalId).ExecuteCommand(); // 重新插入,保持原有ID entity.Id = originalId; return db.Insertable(entity).IgnoreColumns(ignoreNullColumn:true).ExecuteCommand() > 0; } } // 将 ErpDepartment 对象转换为 SysDepartment 对象的方法 private SysDepartment GetSysDepartment(ErpDepartment department) { var fForbidStatus = department.FForbidStatus; if (department.FForbidStatus == "0") { fForbidStatus = "A"; } else if (department.FForbidStatus == "1") { fForbidStatus = "B"; } var entity = new SysDepartment { Departmentcode = department.FNumber, Departmentname = department.FName, Depextr1 = department.FDeptProperty, Depextr2 = department.FGroup, Depextr3 = department.FWIPStockID, Depextr4 = fForbidStatus, FParentID = department.FParentID, FSubsidiary = string.IsNullOrEmpty(department.FUseOrgId) ? "1" : department.FUseOrgId, Fumbrella = string.IsNullOrEmpty(department.FCreateOrgId) ? "1" : department.FCreateOrgId, CreateDate = DateTime.Now, LastupdateDate = DateTime.Now, Company = "1000", Factory = "1000" }; // 查找是否已存在相同部门编码的记录 var existingDepartment = Db.Queryable() .Where(s => s.Departmentcode == entity.Departmentcode) .First(); if (existingDepartment != null) { // 如果存在,使用现有的ID,后续将删除后重新插入 entity.Id = existingDepartment.Id; } else { // 如果不存在,设为0,InsertOrUpdate方法将生成新ID entity.Id = 0; } return entity; } // SaveList 方法用于保存多个部门记录,根据类型批量执行不同的操作 public bool SaveList(List departments) { if (departments == null || !departments.Any()) { Console.WriteLine("警告: 传入的物料列表为空"); return false; } // 逐条处理,全部成功才返回true(事务内批量处理更优,此处保持原有逻辑) var result = departments.Select(Save).ToList(); return result.All(b => b); } // 批量更新部门状态的方法 private bool UpdateDepartmentStatusBatch(SqlSugarScope db, List departmentList, string status) { var ids = departmentList.Select(it => it.Id).ToArray(); var result = db.Updateable() .SetColumns(s => s.Depextr4 == status) .Where(s => ids.Contains(s.Id)).ExecuteCommand(); if (result > 0) return true; throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } // 批量插入部门的方法 private bool InsertDepartmentBatch(SqlSugarScope db, List departmentList) { var insertRange = db.Insertable(departmentList).ExecuteCommand(); if (insertRange > 0) return true; throw new NotImplementedException("插入失败"); } // 批量删除部门的方法 private bool DeleteDepartmentBatch(SqlSugarScope db, List departmentList) { var ids = departmentList.Select(it => it.Id).ToArray(); var deleteByIds = db.Deleteable() .Where(s => ids.Contains(s.Id)).ExecuteCommand(); if (deleteByIds > 0) return true; throw new NotImplementedException("删除失败"); } // 批量插入或更新部门的方法 private bool InsertOrUpdateDepartmentBatch(SqlSugarScope db, List departmentList) { return departmentList.All(entity => InsertOrUpdateDepartment(db, entity)); } }