jinko.yikatong/Src/Dlhis.ScheduledTask/TaskAssignment/Services/Implements/DeviceTaskAssignmentService.cs

using System.Linq.Expressions;
using Dlhis.Entity.Basic.Entity;
using Dlhis.Entity.Device.Entities;
using Dlhis.Entity.Device.Enums;
using Dlhis.ScheduledTask.TaskAssignment.Request;
using Dlhis.ScheduledTask.TaskAssignment.Response;
using Kurisu.AspNetCore.Scope;
using Kurisu.AspNetCore.UnifyResultAndValidation;
using Kurisu.AspNetCore.Utils.Extensions;
using Kurisu.Core.DataAccess.Dto;
using Kurisu.SqlSugar.Extensions;
using Kurisu.SqlSugar.Services;
using Microsoft.AspNetCore.DataProtection;
using Microsoft.Extensions.Options;
using SqlSugar;

namespace Dlhis.ScheduledTask.TaskAssignment.Services.Implements;

public class DeviceTaskAssignmentService : IDeviceTaskAssignmentService
{
    private readonly IDbContext _dbContext;
    private readonly ILogger<DeviceTaskAssignmentService> _logger;

    public DeviceTaskAssignmentService(IDbContext dbContext, ILogger<DeviceTaskAssignmentService> logger)
    {
        _dbContext = dbContext;
        _logger = logger;
    }

    public async Task<Pagination<DeviceTaskAssignmentResponse>> GetPageAsync(DeviceTaskAssignmentRequest request)
    {
        var page = await _dbContext.Queryable<DeviceTaskEntity>()
            .LeftJoin<GainfoEntity>((deviceTask, ga) => deviceTask.UserNo == ga.No)
              .Select((deviceTask, ga) => new DeviceTaskAssignmentResponse
              {
                  DeviceUniqueNo = deviceTask.DeviceUniqueNo,
                  ResultMessage = deviceTask.ResultMessage,
                  RetryCount = deviceTask.RetryCount,
                  Status = deviceTask.Status,
                  UserNo = deviceTask.UserNo,
                  UserName = ga.Name,
                  UserCode = deviceTask.UserCode,
                  Type = deviceTask.Type,
                  LastUpdateTime = deviceTask.ModifiedTime
              })
              .ToPageListAsync(request);

        return page;
    }

    public async Task<List<DeviceTaskToDeviceByDeviceResponse>> DeviceGetTasksAsync(DeviceTaskGetRequest request)
    {
        var data = await ContinueUpdateAsync(async () =>
          {
              var page = await _dbContext.Queryable<DeviceTaskEntity>()
               .LeftJoin<DeviceEntity>((deviceTask, deivce) => deviceTask.DeviceUniqueNo == deivce.UniqueNo)
               .LeftJoin<DeviceTemplateEntity>((deviceTask, deivce, deivceTamplate) => deivceTamplate.Code == deivce.TemplateCode)
               .LeftJoin<GainfoEntity>((deviceTask, deivce, deivceTamplate, ga) => deviceTask.UserNo == ga.No)
               .LeftJoin<DepartmentEntity>((deviceTask, deivce, deivceTamplate, ga, department) => department.No == ga.DepartmentNo)
               .WhereIF(request.Brand.IsPresent(), (deviceTask, deivce, deivceTamplate) => deivceTamplate.Brand == request.Brand)
               .WhereIF(request.DeviceUniqueNo.IsPresent(), (deviceTask, deivce) => deviceTask.DeviceUniqueNo == request.DeviceUniqueNo)
               .Where(deviceTask => deviceTask.Status != DeviceTaskStatusEnumType.Success)
               .OrderBy(deviceTask => deviceTask.CreateTime)
               .Select((deviceTask, deivce, deivceTamplate, ga, department) => new
               {
                   deviceTask.Id,
                   deviceTask.Type,
                   deviceTask.DeviceUniqueNo,
                   DeviceConnectionInfo = deivce.ConnectionInfo,
                   deivce.Brand,
                   deviceTask.UserNo,
                   deviceTask.UserCode,
                   UserName = ga.Name,
                   ga.DepartmentNo,
                   DepartmentName = department.Name,
                   ga.CardNo,
                   deviceTask.Version
               })
               .ToPageListAsync(request);

              var taskIdVs = page.Data.Select(x => new { x.Id, x.Version }).ToList();
              Expression<Func<DeviceTaskEntity, bool>> where = LinqExpressionExtension.False<DeviceTaskEntity>();
              taskIdVs.ForEach(p =>
              {
                  where = where.Or(x => x.Id == p.Id && x.Version == p.Version && x.Status != DeviceTaskStatusEnumType.Success);
              });

              await _dbContext.Client.Ado.BeginTranAsync();
              try
              {
                  var effecet = await _dbContext.Updateable<DeviceTaskEntity>()
                   .SetColumns(x => x.Version == x.Version + 1)
                   .SetColumns(x => x.Status == DeviceTaskStatusEnumType.Process)
                   .SetColumns(x => x.ModifiedTime == DateTime.Now)
                   .Where(where)
                   .ExecuteCommandAsync();

                  if (taskIdVs.Count != effecet)
                  {
                      throw new VersionExceptions("更新错误");
                  }

                  await _dbContext.Client.Ado.CommitTranAsync();
              }
              catch
              {
                  await _dbContext.Client.Ado.RollbackTranAsync();
                  throw;
              }

              return page.Data;
          }, 5);

        return Scoped.Request.Value.Create((sp) =>
          {
              var faceUrlOptions = sp.GetService<IOptions<FaceUrlOptions>>().Value;
              var dataProtectionProvider = sp.GetService<IDataProtectionProvider>();

              var timeLimitedDataProtector = dataProtectionProvider
                 .CreateProtector("down:face")
                 .ToTimeLimitedDataProtector();

              return data.GroupBy(x => new { x.Brand, x.DeviceUniqueNo, x.DeviceConnectionInfo })
                .Select(x =>
                {
                    var b = x.Key.Brand;
                    var d = x.Key.DeviceUniqueNo;
                    var dc = x.Key.DeviceConnectionInfo;

                    return new DeviceTaskToDeviceByDeviceResponse
                    {
                        Brand = b,
                        DeviceUniqueNo = d,
                        DeviceConnectionInfo = dc,
                        Tasks = x.AsEnumerable().Select(item =>
                        {
                            var key = $"face:{item.UserCode}";
                            var token = timeLimitedDataProtector.Protect(key, TimeSpan.FromSeconds(faceUrlOptions.Limit));

                            return new TaskCombineUserDimensionDto
                            {
                                TaskId = item.Id,
                                Type = item.Type,
                                UserInfo = new UserToDeviceResponse
                                {
                                    UserNo = item.UserNo,
                                    UserCode = item.UserCode,
                                    UserName = item.UserName,
                                    DepartmentNo = item.DepartmentNo,
                                    DepartmentName = item.DepartmentName,
                                    CardNo = item.CardNo,
                                    Face = faceUrlOptions.Url + token
                                }
                            };
                        }).ToList()
                    };

                }).ToList();
          });
    }


    private async Task<T> ContinueUpdateAsync<T>(Func<Task<T>> func, int retry)
    {
        for (int i = 0; i < retry; i++)
        {
            try
            {
                return await func();
            }
            catch (Exception ex) when (ex is VersionExceptions)
            {
                _logger.LogWarning(ex, ex.Message);
            }
            catch (Exception)
            {
                throw;
            }
        }

        throw new UserFriendlyException("操作失败,请重试");
    }
}