Пример #1
0
        public static string TryRemoveAssessmentTaskById(int id)
        {
            using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
            {
                // Try to get task
                var task = db.AssessmentTasks.Where(r => r.Id == id).FirstOrDefault();

                if (task == null)
                {
                    return(null);
                }

                // Remove task
                db.AssessmentTasks.Remove(task);

                AssessmentReport report = null;

                // Report id exists
                if (!string.IsNullOrWhiteSpace(task.ReportId))
                {
                    // Try to get report
                    report = db.AssessmentReports.Where(r => r.ReportId == task.ReportId).FirstOrDefault();

                    // Report exists
                    if (report != null)
                    {
                        db.AssessmentReports.Remove(report);
                    }
                }

                db.SaveChanges();

                return(report?.FileName);
            }
        }
Пример #2
0
        public static int CreateNewTask(
            string tenantId,
            string userObjectId,
            string subscriptionId,
            string subscriptionName,
            string targetRegion)
        {
            using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
            {
                var task = new AssessmentTask
                {
                    DeploymentId     = DeploymentId,
                    TenantId         = tenantId,
                    UserObjectId     = userObjectId,
                    SubscriptionId   = subscriptionId,
                    SubscriptionName = subscriptionName,
                    TargetRegion     = targetRegion,
                    Status           = AssessmentTask.TaskStatus.NotStarted,
                    CreatedTime      = DateTime.UtcNow
                };

                db.AssessmentTasks.Add(task);
                db.SaveChanges();

                return(task.Id);
            }
        }
Пример #3
0
        public static string SaveReport(
            string tenantId,
            string userObjectId,
            string localFilePath)
        {
            var reportId = Guid.NewGuid().ToString();
            var fileName = StorageAccess.UploadFile(localFilePath);

            var report = new AssessmentReport
            {
                DeploymentId = DeploymentId,
                TenantId     = tenantId,
                UserObjectId = userObjectId,
                ReportId     = reportId,
                FileName     = fileName
            };

            using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
            {
                db.AssessmentReports.Add(report);
                db.SaveChanges();
            }

            return(reportId);
        }
Пример #4
0
 public static IEnumerable <AssessmentTask> ListTasks(
     string tenantId,
     string userObjectId)
 {
     using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
     {
         return(db.AssessmentTasks.Where(
                    t => t.DeploymentId.Equals(DeploymentId) &&
                    t.TenantId.Equals(tenantId) &&
                    t.UserObjectId.Equals(userObjectId))
                .ToList());
     }
 }
Пример #5
0
 public static string GetReportFileName(
     string tenantId,
     string userObjectId,
     string reportId)
 {
     using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
     {
         return(db.AssessmentReports
                .Where(r => r.DeploymentId.Equals(DeploymentId) &&
                       r.TenantId.Equals(tenantId) &&
                       r.UserObjectId.Equals(userObjectId) &&
                       r.ReportId.Equals(reportId))
                .Select(r => r.FileName)
                .FirstOrDefault());
     }
 }
Пример #6
0
        public static void UpdateTaskStatusProcessing(int taskId)
        {
            using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
            {
                var task = db.AssessmentTasks
                           .Where(t => t.Id.Equals(taskId))
                           .FirstOrDefault();

                if (task != null)
                {
                    task.Status = AssessmentTask.TaskStatus.Processing;

                    db.Entry(task).State = EntityState.Modified;
                    db.SaveChanges();
                }
            }
        }
Пример #7
0
        public static IEnumerable <AssessmentTask> ListAllNotStartedTasks()
        {
            using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
            {
                var tasks = db.AssessmentTasks.Where(
                    t => t.DeploymentId.Equals(DeploymentId) &&
                    t.Status == AssessmentTask.TaskStatus.NotStarted)
                            .ToList();

                if (tasks.Count() < 2)
                {
                    return(tasks);
                }

                var result = new List <AssessmentTask>();

                var groups = tasks.GroupBy(
                    t => $"{t.TenantId}|{t.UserObjectId}|{t.SubscriptionId}|{t.TargetRegion}");

                foreach (var group in groups)
                {
                    if (group.Count() == 1)
                    {
                        result.Add(group.First());
                        continue;
                    }

                    var orderedTasks = group.OrderByDescending(t => t.CreatedTime);

                    result.Add(orderedTasks.First());

                    foreach (var task in orderedTasks.Skip(1))
                    {
                        task.Status          = AssessmentTask.TaskStatus.Failed;
                        task.CompletedTime   = DateTime.UtcNow;
                        task.FailedReason    = @"Abandoned.";
                        db.Entry(task).State = EntityState.Modified;
                    }
                }

                db.SaveChanges();

                return(result.OrderBy(t => t.CreatedTime));
            }
        }
Пример #8
0
        public static void UpdateTaskStatusCompleted(
            int taskId,
            string reportId)
        {
            using (var db = new AssessmentDbContext(ConfigHelper.DatabaseConnectionString))
            {
                var task = db.AssessmentTasks
                           .Where(t => t.Id.Equals(taskId))
                           .FirstOrDefault();

                if (task != null)
                {
                    task.Status        = AssessmentTask.TaskStatus.Completed;
                    task.CompletedTime = DateTime.UtcNow;
                    task.ReportId      = reportId;

                    db.Entry(task).State = EntityState.Modified;
                    db.SaveChanges();
                }
            }
        }