fix:bug

sunpengfei

2025-08-08 c69ab569de2794952a864ba2a855b4bc395d4972

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
using FlexJobApi.Core;
using Furion.DatabaseAccessor;
using Furion.DistributedIDGenerator;
using Furion.FriendlyException;
using Mapster;
using MediatR;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace FlexJobApi.FlexJobServer.Application
{
    /// <summary>
    /// 任务命令处理器
    /// </summary>
    public class TaskInfoCommandHandler(
            IRepository<TaskInfo> rep
        ) :
        IRequestHandler<SaveTaskInfoCommand, Guid>,
        IRequestHandler<SetTaskInfoReleaseStatusCommand, int>,
        IRequestHandler<SetTaskInfoRecommendStatusCommand, int>
    {
        private readonly IRepository<TaskInfo> rep = rep;
 
        /// <summary>
        /// 保存任务
        /// </summary>
        /// <param name="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        /// <exception cref="NotImplementedException"></exception>
        [UnitOfWork]
        public async Task<Guid> Handle(SaveTaskInfoCommand request, CancellationToken cancellationToken)
        {
            var logier = JwtUtils.GetCurrentLogier();
            if (logier.EnterpriseId == null) throw Oops.Oh(EnumErrorCodeType.s400, "当前登录用户必须为企业用户才可以创建任务");
            return await request.SaveData<TaskInfo, SaveTaskInfoCommand>(
                 q =>
                 {
                     q = TaskInfoRepository.GetQueryable(rep, false)
                          .Include(it => it.Benefits)
                          .Include(it => it.CredentialLimits);
                     return q;
                 },
                 null,
                 (entity) =>
                 {
                     if (request.Id == null)
                     {
                         entity.EnterpriseId = logier.EnterpriseId.Value;
                         entity.Status = EnumTaskStatus.Wait;
                         entity.ReleaseStatus = EnumTaskReleaseStatus.InProcess;
                         entity.RecommendStatus = EnumTaskRecommendStatus.No;
                         BuildCode(entity).Wait();
                     }
                     request.Adapt(entity);
                     entity.Benefits = request.Benefits
                         .Select(it => new TaskInfoBenefit
                         {
                             BenefitCode = it
                         })
                         .ToList();
                     entity.CredentialLimits = request.CredentialLimits
                         .Select(it => new TaskInfoCredentialLimit
                         {
                             TypeCode = it
                         })
                         .ToList();
                 },
                 cancellationToken);
        }
 
        /// <summary>
        /// 设置任务发布状态
        /// </summary>
        /// <param name="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public async Task<int> Handle(SetTaskInfoReleaseStatusCommand request, CancellationToken cancellationToken)
        {
            var entities = await TaskInfoRepository.GetQueryable(rep, false)
                .Where(it => request.Ids.Contains(it.Id))
                .ToListAsync();
            foreach (var entity in entities)
            {
                entity.ReleaseStatus = request.ReleaseStatus;
            }
            await rep.UpdateAsync(entities);
            return entities.Count;
        }
 
        /// <summary>
        /// 设置任务是否推荐
        /// </summary>
        /// <param name="request"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public async Task<int> Handle(SetTaskInfoRecommendStatusCommand request, CancellationToken cancellationToken)
        {
            var entities = await TaskInfoRepository.GetQueryable(rep, false)
                .Where(it => request.Ids.Contains(it.Id))
                .ToListAsync();
            foreach (var entity in entities)
            {
                entity.RecommendStatus = request.RecommendStatus;
            }
            await rep.UpdateAsync(entities);
            return entities.Count;
        }
 
        private async Task BuildCode(TaskInfo entity)
        {
            entity.Code = $"{DateTime.Now:yyyyMMddHHmm}{new Random(IDGen.NextID().GetHashCode()).Next(1000, 9999)}";
            var exist = await rep.AsQueryable().AsNoTracking()
               .AnyAsync(it => it.EnterpriseId == entity.EnterpriseId && it.Code == entity.Code);
            if (exist) await BuildCode(entity);
        }
    }
}