using MES.Service.DB; using MES.Service.Dto.webApi; using MES.Service.Modes; using SqlSugar; namespace MES.Service.service.BasicData; public class MesDepotsManager : Repository { // Save 方法用于保存单个仓库记录,根据类型执行不同的操作 public bool Save(ErpDepots depots) { var entity = GetMesDepots(depots); // 将 ErpDepots 转换为 MesDepots return UseTransaction(db => { switch (depots.Type) { case "0": if (UpdateDepotStatus(db, entity.DepotId, "A")) return 1; break; case "1": if (UpdateDepotStatus(db, entity.DepotId, "B")) return 1; break; case "3": if (DeleteDepot(db, entity.DepotId)) return 1; break; case "2": case "4": if (InsertOrUpdate(db, entity)) return 1; break; default: throw new ArgumentNullException( $"type没有{depots.Type}这个类型的参数"); } throw new NotImplementedException("操作失败"); }) > 0; } // 更新仓库状态的方法 private bool UpdateDepotStatus(SqlSugarScope db, decimal depotId, string status) { var result = db.Updateable() .SetColumns(s => s.IsNg == status) .Where(s => s.DepotId == depotId).ExecuteCommand(); if (result > 0) return true; throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } // 插入新仓库的方法 private bool InsertDepot(SqlSugarScope db, MesDepots entity) { var insert = db.Insertable(entity).ExecuteCommand(); if (insert > 0) return true; throw new NotImplementedException("审核失败"); } // 删除仓库的方法 private bool DeleteDepot(SqlSugarScope db, decimal depotId) { var deleteById = db.Deleteable() .Where(s => s.DepotId == depotId).ExecuteCommand(); if (deleteById > 0) return true; throw new NotImplementedException("反审核失败"); } /// /// 生成新的DepotId,确保不重复 /// private decimal GenerateNewId() { // 处理空表的情况,从1开始 var maxId = Db.Queryable().Max(x => (decimal?)x.DepotId) ?? 0; var newId = maxId + 1; // 双重检查,确保生成的DepotId不存在 while (Db.Queryable().Where(x => x.DepotId == newId).Any()) { newId++; } return newId; } // 插入或更新仓库的方法 private bool InsertOrUpdate(SqlSugarScope db, MesDepots entity) { if (entity.DepotId == 0) { // 新增情况:生成新DepotId并插入 var newId = GenerateNewId(); entity.DepotId = newId; return db.Insertable(entity).IgnoreColumns(ignoreNullColumn:true).ExecuteCommand() > 0; } else { // 更新情况:删除后重新插入,保持原有DepotId var originalId = entity.DepotId; // 先删除原记录(如果存在) db.Deleteable().Where(s => s.DepotId == originalId).ExecuteCommand(); // 重新插入,保持原有DepotId entity.DepotId = originalId; return db.Insertable(entity).IgnoreColumns(ignoreNullColumn:true).ExecuteCommand() > 0; } } // 将 ErpDepots 对象转换为 MesDepots 对象的方法 private MesDepots GetMesDepots(ErpDepots depots) { var entity = new MesDepots { DepotCode = depots.FNumber, DepotName = depots.FName, IsFkc = depots.FAllowMinusQty, CreateBy = depots.FPrincipal, Depottype = depots.FStockProperty, IsNg = depots.FForbidStatus, Zuid = depots.FGroup, DocumentStatus = depots.FDocumentStatus, FSubsidiary = string.IsNullOrEmpty(depots.FUseOrgId) ? "1" : depots.FUseOrgId, Fumbrella = string.IsNullOrEmpty(depots.FCreateOrgId) ? "1" : depots.FCreateOrgId, CreateDate = DateTime.Now, LastupdateDate = DateTime.Now, Company = "1000", Factory = "1000" }; // 查找是否已存在相同仓库编码的记录 var existingDepot = Db.Queryable() .Where(s => s.DepotCode == entity.DepotCode) .First(); if (existingDepot != null) { // 如果存在,使用现有的DepotId,后续将删除后重新插入 entity.DepotId = existingDepot.DepotId; } else { // 如果不存在,设为0,InsertOrUpdate方法将生成新DepotId entity.DepotId = 0; } return entity; } // SaveList 方法用于保存多个仓库记录,根据类型批量执行不同的操作 public bool SaveList(List erpDepots) { if (erpDepots == null || !erpDepots.Any()) { Console.WriteLine("警告: 传入的列表为空"); return false; } // 逐条处理,全部成功才返回true(事务内批量处理更优,此处保持原有逻辑) var result = erpDepots.Select(Save).ToList(); return result.All(b => b); } // 批量更新仓库状态的方法 private bool UpdateDepotStatusBatch(SqlSugarScope db, List depotList, string status) { var ids = depotList.Select(it => it.DepotId).ToArray(); var result = db.Updateable() .SetColumns(s => s.IsNg == status) .Where(s => ids.Contains(s.DepotId)).ExecuteCommand(); if (result > 0) return true; throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } // 批量插入仓库的方法 private bool InsertDepotBatch(SqlSugarScope db, List depotList) { var insertRange = db.Insertable(depotList).ExecuteCommand(); if (insertRange > 0) return true; throw new NotImplementedException("审核失败"); } // 批量删除仓库的方法 private bool DeleteDepotBatch(SqlSugarScope db, List depotList) { var ids = depotList.Select(it => it.DepotId).ToArray(); var deleteByIds = db.Deleteable() .Where(s => ids.Contains(s.DepotId)).ExecuteCommand(); if (deleteByIds > 0) return true; throw new NotImplementedException("反审核失败"); } // 批量插入或更新仓库的方法 private bool InsertOrUpdateBatch(SqlSugarScope db, List depotList) { return depotList.All(entity => InsertOrUpdate(db, entity)); } }