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 "2": if (InsertUnit(db, entity)) return 1; break; case "3": if (DeleteUnit(db, entity.Id)) return 1; break; 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) { var list = units.Select(GetMesUnit).ToList(); var groupBy = list.GroupBy(s => s.Type) .ToDictionary(g => g.Key, g => g.ToList()); return UseTransaction(db => { foreach (var unitGroup in groupBy) switch (unitGroup.Key) { case "0": if (!UpdateUnitStatusBatch(db, unitGroup.Value, "A")) throw new NotImplementedException("启用失败"); break; case "1": if (!UpdateUnitStatusBatch(db, unitGroup.Value, "B")) throw new NotImplementedException("禁用失败"); break; case "2": if (!InsertUnitBatch(db, unitGroup.Value)) throw new NotImplementedException("插入失败"); break; case "3": if (!DeleteUnitBatch(db, unitGroup.Value)) throw new NotImplementedException("删除失败"); break; case "4": if (!InsertOrUpdateBatch(db, unitGroup.Value)) throw new NotImplementedException("同步失败"); break; default: throw new ArgumentNullException( $"type没有{unitGroup.Key}这个类型的参数"); } return 1; }) > 0; } 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().In(unitId).ExecuteCommand(); if (deleteById > 0) return true; throw new NotImplementedException("删除失败"); } private MesUnit GetMesUnit(ErpUnit unit) { return new MesUnit { Id = Convert.ToDecimal(unit.Id), Fnumber = unit.FNumber, Fname = unit.FName, Funitgroupid = unit.FUnitGroupId, Fconvertdenominator = unit.FConvertDenominator, Fcurrentunitid = unit.FCurrentUnitId, Fconvertnumerator = unit.FConvertNumerator, Fbaseunit = unit.FBaseUnit, Froundtype = unit.FRoundType, Fprecision = unit.FPrecision, Fdescription = unit.FDescription, Fforbidstatus = unit.FForbidStatus, Company = "1000", Factory = "1000" }; } 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().In(ids).ExecuteCommand(); return deleteByIds > 0; } private bool InsertOrUpdate(SqlSugarScope db, MesUnit entity) { var exists = db.Queryable().Any(e => e.Id == entity.Id); if (exists) { var update = db.Updateable(entity).ExecuteCommand(); if (update > 0) return true; } else { var insert = db.Insertable(entity).ExecuteCommand(); if (insert > 0) return true; } return false; } private bool InsertOrUpdateBatch(SqlSugarScope db, List unitList) { foreach (var entity in unitList) if (!InsertOrUpdate(db, entity)) return false; return true; } }