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 (DeleteStaff(db, entity.Id)) // 删除员工 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(); if (result > 0) { return true; } throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } // 插入或更新员工的方法 private bool InsertOrUpdateStaff(SqlSugarScope db,MesStaff entity) { var exists = db.Queryable().Any(u => u.StaffNo == entity.StaffNo); if (exists) { var updateStaff = db.Updateable(entity).ExecuteCommand(); if (updateStaff > 0) { return true; } } else { var insertStaff = db.Insertable(entity).ExecuteCommand(); if (insertStaff > 0) { return true; } } return false; } // 删除员工的方法 private bool DeleteStaff(SqlSugarScope db,decimal staffId) { var deleteStaff =db.Deleteable().In(staffId).ExecuteCommand(); if (deleteStaff > 0) { return true; } throw new NotImplementedException("反审核失败"); } // 将 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 departments) { var list = new List(); departments.ForEach(s => { var entity = GetMesStaff(s); entity.Type = s.Type; list.Add(entity); }); 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 (!DeleteStaffBatch(db,staffGroup.Value)) // 批量删除员工 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(); if (result > 0) { return true; } throw new NotImplementedException(status == "A" ? "启用失败" : "禁用失败"); } // 批量插入员工的方法 private bool InsertStaffBatch(SqlSugarScope db,List staffList) { var userInsert = staffList.FindAll(s => s.Type == "INSERT"); var executeCommand = db.Insertable(userInsert).ExecuteCommand(); if (executeCommand > 0) if (db.Insertable(staffList).ExecuteCommand() > 0) return true; throw new ArgumentNullException("审核失败"); } // 批量删除员工的方法 private bool DeleteStaffBatch(SqlSugarScope db, List staffList) { var ids = staffList.Select(it => it.Id).ToArray(); if (db.Deleteable().In(ids).ExecuteCommand() > 0) { return true; } throw new ArgumentNullException("反审核失败"); } // 批量插入或更新员工的方法 private bool InsertOrUpdateBatch(SqlSugarScope db, List staffList) { foreach (var staff in staffList) { var entity = staffList.First(s => s.StaffNo == staff.StaffNo); if (!InsertOrUpdateStaff(db, entity)) { return false; } } return true; } }