using MES.Service.DB; using MES.Service.Dto.webApi; using MES.Service.Modes; using SqlSugar; namespace MES.Service.service.BasicData; public class MesUnitManager : Repository { public bool Save(ErpUnit unit) { var entity = GetMesUnit(unit); return UseTransaction(db => { switch (unit.Type) { case "0": if (UpdateUnitStatus(db, entity.Id, "A")) return 1; break; case "1": if (UpdateUnitStatus(db, entity.Id, "B")) return 1; break; case "3": if (DeleteUnit(db, entity.Id)) return 1; break; case "2": case "4": if (InsertOrUpdate(db, entity)) return 1; break; default: throw new ArgumentNullException( $"type没有{unit.Type}这个类型的参数"); } throw new NotImplementedException("操作失败"); }) > 0; } public bool SaveList(List units) { if (units == null || !units.Any()) { Console.WriteLine("警告: 传入的列表为空"); return false; } // 逐条处理,全部成功才返回true(事务内批量处理更优,此处保持原有逻辑) var result = units.Select(Save).ToList(); return result.All(b => b); } private bool UpdateUnitStatus(SqlSugarScope db, decimal unitId, string status) { var result = db.Updateable() .SetColumns(s => s.Fforbidstatus == status) .Where(s => s.Id == unitId).ExecuteCommand(); if (result > 0) return true; throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } private bool InsertUnit(SqlSugarScope db, MesUnit entity) { var insert = db.Insertable(entity).ExecuteCommand(); if (insert > 0) return true; throw new NotImplementedException("插入失败"); } private bool DeleteUnit(SqlSugarScope db, decimal unitId) { var deleteById = db.Deleteable() .Where(s => s.Id == unitId).ExecuteCommand(); if (deleteById > 0) return true; throw new NotImplementedException("删除失败"); } private MesUnit GetMesUnit(ErpUnit unit) { var fForbidStatus = unit.FForbidStatus; if (unit.FForbidStatus == "0") { fForbidStatus = "A"; } else if (unit.FForbidStatus == "1") { fForbidStatus = "B"; } var entity = new MesUnit { Id = string.IsNullOrEmpty(unit.Id) ? DateTimeOffset.UtcNow.ToUnixTimeSeconds() : Convert.ToDecimal(unit.Id), Fnumber = unit.FNumber, Fname = unit.FName, Funitgroupid = unit.FUnitGroupId, Fconvertdenominator = Convert.ToDecimal(unit.FConvertDenominator), Fcurrentunitid = unit.FCurrentUnitId, Fconvertnumerator = Convert.ToDecimal(unit.FConvertNumerator), Fbaseunit = unit.FBaseUnit, Froundtype = unit.FRoundType, Fprecision = unit.FPrecision, Fdescription = unit.FDescription, Fforbidstatus = fForbidStatus, CreateDate = DateTime.Now, LastupdateDate = DateTime.Now, Company = "1000", Factory = "1000", Type = unit.Type }; var mesUnit = Db.Queryable() .Where(s => s.Fnumber == entity.Fnumber) .First(); if (mesUnit != null) { entity.Id = mesUnit.Id; } return entity; } private bool UpdateUnitStatusBatch(SqlSugarScope db, List unitList, string status) { var ids = unitList.Select(it => it.Id).ToArray(); var result = db.Updateable() .SetColumns(s => s.Fforbidstatus == status) .Where(s => ids.Contains(s.Id)).ExecuteCommand(); if (result > 0) return true; throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } private bool InsertUnitBatch(SqlSugarScope db, List unitList) { var insertRange = db.Insertable(unitList).ExecuteCommand(); if (insertRange > 0) return true; throw new NotImplementedException("插入失败"); } private bool DeleteUnitBatch(SqlSugarScope db, List unitList) { var ids = unitList.Select(it => it.Id).ToArray(); var deleteByIds = db.Deleteable() .Where(s => ids.Contains(s.Id)).ExecuteCommand(); return deleteByIds > 0; } private bool InsertOrUpdate(SqlSugarScope db, MesUnit entity) { db.Deleteable() .Where(s => s.Id == entity.Id).ExecuteCommand(); var insert = db.Insertable(entity).ExecuteCommand(); return insert > 0; } private bool InsertOrUpdateBatch(SqlSugarScope db, List unitList) { return unitList.All(entity => InsertOrUpdate(db, entity)); } }