wanshenmean
5 小时以前 96adc295cb04fd135d63d3a907f2732274f90965
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
#region << 版 本 注 释 >>
/*----------------------------------------------------------------
 * 命名空间:WIDESEAWCS_QuartzJob
 * 创建者:胡童庆
 * 创建时间:2024/8/2 16:13:36
 * 版本:V1.0.0
 * 描述:Job工厂类
 *
 * ----------------------------------------------------------------
 * 修改人:
 * 修改时间:
 * 版本:V1.0.1
 * 修改说明:
 * 
 *----------------------------------------------------------------*/
#endregion << 版 本 注 释 >>
 
using Microsoft.Extensions.DependencyInjection;
using Quartz.Spi;
using Quartz;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using WIDESEAWCS_Core;
 
namespace WIDESEAWCS_QuartzJob
{
    /// <summary>
    /// Job注入工厂,管理 Job 实例的 DI 作用域生命周期
    /// </summary>
    public class JobFactory : IJobFactory
    {
        /// <summary>
        /// 注入反射获取依赖对象
        /// </summary>
        private readonly IServiceProvider _serviceProvider;
 
        /// <summary>
        /// Job 实例与对应的 DI 作用域映射,用于在 ReturnJob 时正确释放资源
        /// </summary>
        private readonly ConcurrentDictionary<IJob, IServiceScope> _scopes = new();
 
        /// <summary>
        /// Job注入
        /// </summary>
        /// <param name="serviceProvider">服务提供者</param>
        public JobFactory(IServiceProvider serviceProvider)
        {
            _serviceProvider = serviceProvider;
        }
 
        /// <summary>
        /// 创建 Job 实例,并为每个 Job 创建独立的 DI 作用域
        /// </summary>
        /// <param name="bundle">触发器触发上下文</param>
        /// <param name="scheduler">调度器实例</param>
        /// <returns>Job 实例</returns>
        public IJob NewJob(TriggerFiredBundle bundle, IScheduler scheduler)
        {
            try
            {
                // 验证有效期,使用 TotalSeconds 避免 .Seconds 取模导致间歇性判断错误
                if (App.ExpDateTime != null && (DateTime.Now - App.ExpDateTime.GetValueOrDefault()).TotalSeconds > 0)
                {
                    throw new InvalidOperationException($"验证错误");
                }
 
                // 为每个 Job 创建独立的 DI 作用域,确保 Scoped 服务正确释放
                IServiceScope serviceScope = _serviceProvider.CreateScope();
                IJob? job = serviceScope.ServiceProvider.GetService(bundle.JobDetail.JobType) as IJob;
 
                if (job == null)
                {
                    // Job 解析失败时立即释放作用域,避免泄漏
                    serviceScope.Dispose();
                    throw new InvalidOperationException($"无法解析 Job 类型: {bundle.JobDetail.JobType.Name}");
                }
 
                // 保存 scope 引用,以便 ReturnJob 时释放
                _scopes[job] = serviceScope;
                return job;
            }
            catch (Exception ex)
            {
                Console.Out.WriteLine(ex.ToString());
                throw;
            }
        }
 
        /// <summary>
        /// 释放 Job 实例及其关联的 DI 作用域
        /// </summary>
        /// <param name="job">待释放的 Job 实例</param>
        public void ReturnJob(IJob job)
        {
            // 先释放关联的 DI 作用域(包括其中所有 Scoped 服务)
            if (_scopes.TryRemove(job, out IServiceScope? scope))
            {
                scope.Dispose();
            }
 
            // 再释放 Job 本身
            (job as IDisposable)?.Dispose();
        }
    }
}