DataSimulation/DataSimulation.Repostory/TaskCache.cs

using System;
using System.Collections.Generic;
using System.Data.Entity;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using DataSimulation.Repostory.EFContext;
using DataSimulation.Repostory.Model;

namespace DataSimulation.Repostory
{
    public static class TaskCache
    {
        /// <summary>
        /// 获取数据库所有任务
        /// </summary>
        /// <returns></returns>
        public static async Task<List<TaskInfo>> GetAllAsync()
        {
            try
            {
                List<TaskInfo> list = new List<TaskInfo>();
                var dir = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "DbPart");
                if (!Directory.Exists(dir)) return list;
                var yearDirs = Directory.EnumerateDirectories(dir).OrderByDescending(p => Convert.ToInt32(new DirectoryInfo(p).Name));//年目录,倒叙排列
                foreach (var yearDir in yearDirs)
                {
                    //每一天的db文件,倒序排列
                    var dayFiles = Directory.EnumerateFiles(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, yearDir), "*.db").OrderByDescending(p => Convert.ToInt32(new DirectoryInfo(p).Name.Substring(0, 4)));
                    foreach (var dayFile in dayFiles)
                    {
                        using (SimulationPartContext db = SimulationPartContext.GetContext(dayFile))
                        {
                            list.AddRange(await db.TaskInfos.ToListAsync());
                        }
                    }
                }
                return list;
            }
            catch (Exception ex)
            {
                Serilog.Log.Error(ex, "加载任务信息异常");
                return null;
            }
        }

        /// <summary>
        /// 获取任务航迹
        /// </summary>
        /// <param name="taskId"></param>
        /// <returns></returns>
        public static async Task<List<TaskSimulation>> GetTaskSimulationAsync(long taskId)
        {
            try
            {
                List<TaskSimulation> list = new List<TaskSimulation>();
                var dir = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "DbPart");
                if (!Directory.Exists(dir)) return list;
                var yearDirs = Directory.EnumerateDirectories(dir).OrderByDescending(p => Convert.ToInt32(new DirectoryInfo(p).Name));//年目录,倒叙排列
                foreach (var yearDir in yearDirs)
                {
                    //每一天的db文件,倒序排列
                    var dayFiles = Directory.EnumerateFiles(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, yearDir), "*.db").OrderByDescending(p => Convert.ToInt32(new DirectoryInfo(p).Name.Substring(0, 4)));
                    foreach (var dayFile in dayFiles)
                    {
                        using (SimulationPartContext db = SimulationPartContext.GetContext(dayFile))
                        {
                            list.AddRange(await db.TaskSimulations.Where(w=>w.TaskId==taskId).ToListAsync());
                        }
                    }
                }
                return list;
            }
            catch (Exception ex)
            {
                Serilog.Log.Error(ex, "加载任务航迹信息异常");
                return null;
            }
        }
    }
}