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 (UpdateUnitStatus(db, entity.Id, "B")) 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 (!UpdateUnitStatusBatch(db, unitGroup.Value, "B")) 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(); return true; } private bool InsertUnit(SqlSugarScope db, MesUnit entity) { var exists = db.Queryable().Any(e => e.Id == entity.Id); if (exists) { var result = db.Updateable(entity).ExecuteCommand(); return true; } else { var insert = db.Insertable(entity).ExecuteCommand(); if (insert > 0) { return true; } else { throw new NotImplementedException("插入失败"); } } } 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(); return true; } else { var insert = db.Insertable(entity).ExecuteCommand(); if (insert > 0) { return true; } else { return false; } } } 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, CreateDate = DateTime.Now, 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(); return true; } private bool InsertUnitBatch(SqlSugarScope db, List unitList) { foreach (var entity in unitList) { if (!InsertUnit(db, entity)) { return false; } } return true; } private bool InsertOrUpdateBatch(SqlSugarScope db, List unitList) { foreach (var entity in unitList) { if (!InsertOrUpdate(db, entity)) { return false; } } return true; } }