using MES.Service.DB; using MES.Service.Dto.webApi; using MES.Service.Modes; using SqlSugar; namespace MES.Service.service.BasicData; public class MesStaffManager : Repository { // Save 方法用于保存单个员工记录,根据类型执行不同的操作 public bool Save(ErpStaff unit) { var entity = GetMesStaff(unit); // 将 ErpStaff 转换为 MesStaff return UseTransaction(db => { switch (unit.Type) { case "0": if (UpdateStaffStatus(db, entity.Id, "A")) // 启用员工 return 1; break; case "1": if (UpdateStaffStatus(db, entity.Id, "B")) // 禁用员工 return 1; break; case "2": case "4": if (InsertOrUpdateStaff(db, entity)) // 插入或更新员工 return 1; break; case "3": if (UpdateStaffStatus(db, entity.Id, "B")) // 禁用员工 return 1; break; default: throw new ArgumentNullException($"type没有{unit.Type}这个类型的参数"); } throw new NotImplementedException("操作失败"); }) > 0; } // 更新员工状态的方法 private bool UpdateStaffStatus(SqlSugarScope db, decimal staffId, string status) { var result = db.Updateable().SetColumns(s => s.FforbidStatus == status).Where(s => s.Id == staffId).ExecuteCommand(); return true; } // 插入或更新员工的方法 private bool InsertOrUpdateStaff(SqlSugarScope db, MesStaff 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; } } } // 将 ErpStaff 对象转换为 MesStaff 对象的方法 private MesStaff GetMesStaff(ErpStaff staff) { var entity = new MesStaff { Id = Convert.ToDecimal(staff.Id), StaffNo = staff.FStaffNumber, StaffName = staff.FName, DepartmentNo = staff.FPostDept, PositionCode = staff.FPost, PhoneNumber = staff.FMobile, CreateOrg = Convert.ToDecimal(staff.FCreateOrgId), UseOrg = Convert.ToDecimal(staff.FUseOrgId), CreateDate = DateTime.Now, Remark = staff.FDescription, FforbidStatus = staff.FForbidStatus }; if (staff.FStaffStartDate != null) entity.StartDate = DateTime.ParseExact(staff.FStaffStartDate, "yyyy-MM-dd HH:mm:ss", null); return entity; } // SaveList 方法用于保存多个员工记录,根据类型批量执行不同的操作 public bool SaveList(List staffs) { var list = staffs.Select(GetMesStaff).ToList(); var groupBy = list.GroupBy(s => s.Type).ToDictionary(g => g.Key, g => g.ToList()); return UseTransaction(db => { foreach (var staffGroup in groupBy) switch (staffGroup.Key) { case "0": if (!UpdateStaffStatusBatch(db, staffGroup.Value, "A")) // 批量启用员工 throw new NotImplementedException("启用失败"); break; case "1": if (!UpdateStaffStatusBatch(db, staffGroup.Value, "B")) // 批量禁用员工 throw new NotImplementedException("禁用失败"); break; case "2": if (!InsertStaffBatch(db, staffGroup.Value)) // 批量插入员工 throw new NotImplementedException("插入失败"); break; case "3": if (!UpdateStaffStatusBatch(db, staffGroup.Value, "B")) // 批量禁用员工 throw new NotImplementedException("删除失败"); break; case "4": if (!InsertOrUpdateBatch(db, staffGroup.Value)) // 批量插入或更新员工 throw new NotImplementedException("同步失败"); break; default: throw new ArgumentNullException( $"type没有{staffGroup.Key}这个类型的参数"); } return 1; }) > 0; } // 批量更新员工状态的方法 private bool UpdateStaffStatusBatch(SqlSugarScope db, List staffList, string status) { var ids = staffList.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 InsertStaffBatch(SqlSugarScope db, List staffList) { foreach (var entity in staffList) { if (!InsertOrUpdateStaff(db, entity)) { return false; } } return true; } // 批量插入或更新员工的方法 private bool InsertOrUpdateBatch(SqlSugarScope db, List staffList) { foreach (var entity in staffList) { if (!InsertOrUpdateStaff(db, entity)) { return false; } } return true; } }