using MES.Service.DB;
|
using MES.Service.Dto.webApi;
|
using MES.Service.Modes;
|
using SqlSugar;
|
|
namespace MES.Service.service.BasicData;
|
|
public class MesUnitManager : Repository<MesUnit>
|
{
|
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<ErpUnit> 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<MesUnit>()
|
.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<MesUnit>()
|
.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
|
{
|
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 existingUnit = Db.Queryable<MesUnit>()
|
.Where(s => s.Fnumber == entity.Fnumber)
|
.First();
|
|
if (existingUnit != null)
|
{
|
// 如果存在，使用现有的ID，后续将删除后重新插入
|
entity.Id = existingUnit.Id;
|
}
|
else
|
{
|
// 如果不存在，设为0，InsertOrUpdate方法将生成新ID
|
entity.Id = 0;
|
}
|
|
return entity;
|
}
|
|
private bool UpdateUnitStatusBatch(SqlSugarScope db, List<MesUnit> unitList,
|
string status)
|
{
|
var ids = unitList.Select(it => it.Id).ToArray();
|
var result = db.Updateable<MesUnit>()
|
.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<MesUnit> unitList)
|
{
|
var insertRange = db.Insertable(unitList).ExecuteCommand();
|
if (insertRange > 0)
|
return true;
|
|
throw new NotImplementedException("插入失败");
|
}
|
|
private bool DeleteUnitBatch(SqlSugarScope db, List<MesUnit> unitList)
|
{
|
var ids = unitList.Select(it => it.Id).ToArray();
|
var deleteByIds = db.Deleteable<MesUnit>()
|
.Where(s => ids.Contains(s.Id)).ExecuteCommand();
|
return deleteByIds > 0;
|
}
|
|
/// <summary>
|
/// 生成新的ID，确保不重复
|
/// </summary>
|
private decimal GenerateNewId()
|
{
|
// 处理空表的情况，从1开始
|
var maxId = Db.Queryable<MesUnit>().Max(x => (decimal?)x.Id) ?? 0;
|
var newId = maxId + 1;
|
|
// 双重检查，确保生成的ID不存在
|
while (Db.Queryable<MesUnit>().Where(x => x.Id == newId).Any())
|
{
|
newId++;
|
}
|
|
return newId;
|
}
|
|
private bool InsertOrUpdate(SqlSugarScope db, MesUnit entity)
|
{
|
if (entity.Id == 0)
|
{
|
// 新增情况：生成新ID并插入
|
var newId = GenerateNewId();
|
entity.Id = newId;
|
return db.Insertable(entity).IgnoreColumns(ignoreNullColumn:true).ExecuteCommand() > 0;
|
}
|
else
|
{
|
// 更新情况：删除后重新插入，保持原有ID
|
var originalId = entity.Id;
|
|
// 先删除原记录（如果存在）
|
db.Deleteable<MesUnit>().Where(s => s.Id == originalId).ExecuteCommand();
|
|
// 重新插入，保持原有ID
|
entity.Id = originalId;
|
return db.Insertable(entity).IgnoreColumns(ignoreNullColumn:true).ExecuteCommand() > 0;
|
}
|
}
|
|
private bool InsertOrUpdateBatch(SqlSugarScope db, List<MesUnit> unitList)
|
{
|
return unitList.All(entity => InsertOrUpdate(db, entity));
|
}
|
}
|
