TEAMModelOS/TEAMModelOS.Extension/HTEX.Complex/Services/TaskService.cs

using Azure.Storage.Sas;
using StackExchange.Redis;
using System.Web;
using TEAMModelOS.SDK.DI;
using TEAMModelOS.SDK;
using TEAMModelOS.SDK.Extension;

namespace HTEX.Complex.Services
{
    public class TaskService
    {
        public static async Task<(PDFGenRedis genRedis, PDFGenQueue genQueue, string msg)> SentTask(AzureRedisFactory _azureRedis, AzureStorageFactory _azureStorage)
        {
            string msg = string.Empty;
            //从尾部弹出元素，队列先进先出
            long now = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
            var queueValue = await _azureRedis.GetRedisClient(8).ListRightPopAsync("PDFGen:Queue");
            PDFGenRedis genRedis = null;
            PDFGenQueue genQueue = null;
            if (queueValue!=default && queueValue.HasValue)
            {
                genQueue = queueValue.ToString().ToObject<PDFGenQueue>();
                RedisValue redisValue = await _azureRedis.GetRedisClient(8).HashGetAsync($"PDFGen:{genQueue.sessionId}", genQueue.id);
                if (redisValue!=default)
                {
                    genRedis = redisValue.ToString().ToObject<PDFGenRedis>();
                    //计算等待了多久的时间才开始生成。
                    var wait = now- genRedis.join;
                    genRedis.wait  = wait;
                    genRedis.status = 1;
                    try
                    {
                        Uri uri = new Uri(genQueue.pageUrl);
                        var query = HttpUtility.ParseQueryString(uri.Query);
                        string? url = query["url"];
                        if (!string.IsNullOrWhiteSpace(url))
                        {
                            url=  HttpUtility.UrlDecode(url);
                            uri = new Uri(url);
                            string host = uri.Host;
                            string azure_region = genQueue.env.Equals(ScreenConstant.env_develop) ? "Test" : "Default";
                            var blobServiceClient = _azureStorage.GetBlobServiceClient(azure_region);
                            if (host.Equals(blobServiceClient.Uri.Host))
                            {
                                // 获取容器名，它是路径的第一个部分
                                string containerName = uri.Segments[1].TrimEnd('/');
                                // 获取文件的完整同级目录，这是文件路径中除了文件名和扩展名之外的部分
                                // 由于文件名是路径的最后一个部分，我们可以通过连接除了最后一个部分之外的所有部分来获取目录路径
                                string directoryPath = string.Join("", uri.Segments, 2, uri.Segments.Length - 3);
                                string? fileName = Path.GetFileNameWithoutExtension(uri.AbsolutePath);
                                string blobPath = $"{directoryPath}{fileName}.pdf";
                                var urlSas = _azureStorage.GetBlobSAS(containerName, blobPath, BlobSasPermissions.Write|BlobSasPermissions.Read, hour: 1, name: azure_region);
                                genQueue.blobSas =  urlSas.sas;
                                genQueue.blobName=blobPath;
                                genQueue.cntName=containerName;
                                genQueue.blobFullUrl=urlSas.fullUri;
                            }
                            else
                            {
                                msg=$"数据地址与服务提供的站点不一致！{host},{blobServiceClient.Uri.Host}";
                                genRedis.status=3;
                                genRedis.msg = msg;

                            }
                        }
                        else
                        {
                            msg="数据地址解析异常！";
                            genRedis.status=3;
                            genRedis.msg = msg;

                        }
                    }
                    catch (Exception ex)
                    {
                        msg=$"数据地址处理异常，异常信息：{ex.Message}";
                        genRedis.status=3;
                        genRedis.msg = ex.Message;
                    }
                    await _azureRedis.GetRedisClient(8).HashSetAsync($"PDFGen:{genQueue.sessionId}", genQueue.id, genRedis.ToJsonString());
                }
                else
                {
                    msg="队列任务关联数据为空！";
                }
            }
            else
            {
                msg="队列暂无任务！";
            }
            return (genRedis, genQueue, msg);
        }
    }
}