using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; using Microsoft.Extensions.DependencyInjection; using Sln.Wcs.Common; using Sln.Wcs.Model.Domain; using Sln.Wcs.Serilog; using SqlSugar; namespace Sln.Wcs.Function.Functions { public class wcstaskSqlFunction : SqlFunctionFactory { private WcsTaskInstance? wcsTaskInstance; private ISqlSugarClient _sqlSugarClient; private readonly SerilogHelper _logger; public wcstaskSqlFunction(IServiceProvider serviceProvider) { _sqlSugarClient = serviceProvider.GetRequiredService(); _logger = serviceProvider.GetRequiredService(); } public override void deleteFunction(WcsTaskInstance _entity) { var wcsDb = _sqlSugarClient.AsTenant().GetConnection("wcs"); wcsTaskInstance = _entity; try { List baseConfigs = wcsDb.Queryable().Where(it => it.TaskCode == wcsTaskInstance.TaskCode || it.IsDeleted == true).ToList(); if (baseConfigs.Count == 0) { _logger.Info("不存在可删除任务"); return; } foreach (var item in baseConfigs) { wcsDb.Ado.BeginTran(); wcsDb.Deleteable(item).ExecuteCommand(); wcsDb.Ado.CommitTran(); } } catch (Exception ex) { wcsDb.AsTenant().RollbackTran(); _logger.Error($"删除任务异常{ex.Message}"); } } public override void insertFunction(WcsTaskInstance _entity) { var wcsDb = _sqlSugarClient.AsTenant().GetConnection("wcs"); wcsTaskInstance = _entity; try { bool hasTask = wcsDb.Queryable().Any(it => it.TaskCode == wcsTaskInstance.TaskCode); if (hasTask) { _logger.Info("已有该任务配置"); return; } wcsDb.Ado.BeginTran(); wcsDb.Insertable(wcsTaskInstance).ExecuteCommand(); wcsDb.Ado.CommitTran(); } catch (Exception ex) { wcsDb.AsTenant().RollbackTran(); _logger.Error($"插入任务设置异常{ex.Message}"); } } public override List searchFunction(WcsTaskInstance _entity) { var wcsDb = _sqlSugarClient.AsTenant().GetConnection("wcs"); wcsTaskInstance = _entity; try { // 根据 ConfigKey 查询配置信息 List result = wcsDb.Queryable() .Where(it => it.TaskCode == wcsTaskInstance.TaskCode) .ToList(); if (result.Count == 0) { _logger.Info($"未找到 ConfigKey 为 {wcsTaskInstance?.TaskCode} 的仓库配置"); // 可以根据需要返回 null 或者抛出异常 return null; } _logger.Info($"成功查询仓库配置: {wcsTaskInstance?.TaskCode}"); return result; } catch (Exception ex) { _logger.Error($"查询仓库设置异常: {ex.Message}"); return null; } } public override void updateFunction(WcsTaskInstance _entity) { var wcsDb = _sqlSugarClient.AsTenant().GetConnection("wcs"); wcsTaskInstance = _entity; try { // 检查是否存在该配置 bool exists = wcsDb.Queryable() .Any(it => it.TaskCode == wcsTaskInstance.TaskCode); if (!exists) { _logger.Info($"不存在 TaskCode 为 {wcsTaskInstance.TaskCode} 的任务,无法更新"); return; } // 开始事务 wcsDb.Ado.BeginTran(); var updateResult = wcsDb.Updateable(wcsTaskInstance).ExecuteCommand(); wcsDb.Ado.CommitTran(); if (updateResult > 0) { _logger.Info($"成功更新任务: {wcsTaskInstance.TaskCode}"); } else { _logger.Info($"更新任务行数为0: {wcsTaskInstance.TaskCode}"); } } catch (Exception ex) { wcsDb.AsTenant().RollbackTran(); _logger.Error($"更新任务异常: {ex.Message}"); throw; // 或者根据业务需求决定是否抛出 } } } }