source: branches/HiveStatistics/sources/HeuristicLab.Services.WebApp.Statistics/3.3/WebApi/ClientController.cs @ 12516

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Linq;
using System.Web.Http;
using HeuristicLab.Services.Hive;
using HeuristicLab.Services.Hive.DataAccess;
using HeuristicLab.Services.Hive.DataAccess.Interfaces;
using HeuristicLab.Services.WebApp.Status.WebApi;
using DT = HeuristicLab.Services.WebApp.Statistics.WebApi.DataTransfer;

namespace HeuristicLab.Services.WebApp.Statistics.WebApi {

  [Authorize]
  public class ClientController : ApiController {
    private IPersistenceManager PersistenceManager {
      get { return ServiceLocator.Instance.PersistenceManager; }
    }

    public DT.ClientDetails GetClientDetails(Guid id) {
      using (var pm = PersistenceManager) {
        var dimClientDao = pm.DimClientDao;
        var factClientInfoDao = pm.FactClientInfoDao;
        var factTaskDao = pm.FactTaskDao;
        return pm.UseTransaction(() => {
          var timeData = factClientInfoDao.GetByClientId(id)
            .GroupBy(x => x.ClientId)
            .Select(x => new {
              TotalIdleTime = (long)x.Sum(y => y.TotalTimeIdle),
              TotalOfflineTime = (long)x.Sum(y => y.TotalTimeOffline),
              TotalUnavailableTime = (long)x.Sum(y => y.TotalTimeUnavailable)
            }).FirstOrDefault();
          var taskTimeData = factTaskDao.GetByClientId(id)
            .GroupBy(x => x.LastClientId)
            .Select(x => new {
              TotalCalculatingTime = (long)x.Sum(y => y.CalculatingTime),
              TotalTransferringTime = (long)x.Sum(y => y.TransferTime)
            }).FirstOrDefault();
          var startDate = factClientInfoDao.GetByClientId(id)
            .Where(x => x.SlaveState == SlaveState.Offline)
            .OrderByDescending(x => x.Time)
            .Select(x => x.Time).FirstOrDefault();
          long upTime = 0;
          if (startDate == default(DateTime)) {
            startDate = factClientInfoDao.GetByClientId(id)
              .OrderByDescending(x => x.Time)
              .Select(x => x.Time)
              .FirstOrDefault();
          }
          if (startDate != default(DateTime)) {
            upTime = (long)(DateTime.Now - startDate).TotalSeconds;
          }
          return (from client in dimClientDao.GetAll().Where(x => x.Id == id)
                  join info in factClientInfoDao.GetAll()
                    on client.Id equals info.ClientId into clientInfoJoin
                  from clientInfo in clientInfoJoin.OrderByDescending(x => x.Time).Take(1)
                  let offline = (client.ExpirationTime != null || clientInfo.SlaveState == SlaveState.Offline)
                  select new DT.ClientDetails {
                    Id = client.Id,
                    Name = client.Name,
                    TotalCores = clientInfo.NumTotalCores,
                    UsedCores = offline ? 0 : clientInfo.NumUsedCores,
                    TotalMemory = clientInfo.TotalMemory,
                    UsedMemory = offline ? 0 : clientInfo.UsedMemory,
                    CpuUtilization = offline ? 0 : clientInfo.CpuUtilization,
                    State = clientInfo.SlaveState.ToString(),
                    LastUpdate = clientInfo.Time,
                    UpTime = offline ? 0 : upTime,
                    TotalUnavailableTime = timeData != null ? timeData.TotalUnavailableTime : 0,
                    TotalCalculatingTime = taskTimeData != null ? taskTimeData.TotalCalculatingTime : 0,
                    TotalIdleTime = timeData != null ? timeData.TotalIdleTime : 0,
                    TotalOfflineTime = timeData != null ? timeData.TotalOfflineTime : 0,
                    TotalTransferringTime = taskTimeData != null ? taskTimeData.TotalTransferringTime : 0,
                    TasksStates = factTaskDao.GetByClientId(id)
                                    .GroupBy(x => x.TaskState)
                                    .Select(x => new DT.TaskStateCount {
                                      State = x.Key.ToString(),
                                      Count = x.Count()
                                    }).ToList()
                  })
                  .FirstOrDefault();
        });
      }
    }

    public DT.ClientPage GetClients(int page, int size, bool expired = false) {
      using (var pm = PersistenceManager) {
        var dimClientDao = pm.DimClientDao;
        var factClientInfoDao = pm.FactClientInfoDao;
        return pm.UseTransaction(() => {
          var clients = expired ? dimClientDao.GetExpiredClients() : dimClientDao.GetActiveClients();
          var query = (from client in clients
                       join info in factClientInfoDao.GetAll()
                         on client.Id equals info.ClientId into clientInfoJoin
                       from clientInfo in clientInfoJoin.OrderByDescending(x => x.Time).Take(1)
                       let offline = (expired || clientInfo.SlaveState == SlaveState.Offline)
                       select new DT.Client {
                         Id = client.Id,
                         Name = client.Name,
                         TotalCores = clientInfo.NumTotalCores,
                         UsedCores = offline ? 0 : clientInfo.NumUsedCores,
                         TotalMemory = clientInfo.TotalMemory,
                         UsedMemory = offline ? 0 : clientInfo.UsedMemory,
                         CpuUtilization = offline ? 0 : clientInfo.CpuUtilization,
                         State = clientInfo.SlaveState.ToString(),
                       });
          return new DT.ClientPage {
            TotalClients = query.Count(),
            Clients = query.Skip((page - 1) * size)
              .Take(size)
              .ToList()
          };
        });
      }
    }

    public IEnumerable<DT.ClientStatus> GetClientHistory(Guid id, DateTime start, DateTime end) {
      TimeSpan ts = end - start;
      int increment = 1;
      double totalMinutes = ts.TotalMinutes;
      while (totalMinutes > 5761) {
        totalMinutes -= 5761;
        increment += 5;
      }
      using (var pm = PersistenceManager) {
        var factClientInfo = pm.FactClientInfoDao;
        var clientInfos = factClientInfo.GetByClientId(id)
          .Where(x => x.Time >= start && x.Time <= end)
          .OrderBy(x => x.Time)
          .ToList();
        var clientStatus = new DT.ClientStatus {
          CpuUtilization = 0,
          TotalCores = 0,
          UsedCores = 0,
          TotalMemory = 0,
          UsedMemory = 0
        };
        int i = 1;
        foreach (var clientInfo in clientInfos) {
          clientStatus.CpuUtilization += clientInfo.CpuUtilization;
          clientStatus.TotalCores += clientInfo.NumTotalCores;
          clientStatus.UsedCores += clientInfo.NumUsedCores;
          clientStatus.TotalMemory += clientInfo.TotalMemory;
          clientStatus.UsedMemory += clientInfo.UsedMemory;
          if (i >= increment) {
            clientStatus.Timestamp = JavascriptUtils.ToTimestamp(clientInfo.Time);
            clientStatus.CpuUtilization /= i;
            clientStatus.TotalCores /= i;
            clientStatus.UsedCores /= i;
            clientStatus.TotalMemory /= i;
            clientStatus.UsedMemory /= i;
            yield return clientStatus;
            clientStatus = new DT.ClientStatus {
              CpuUtilization = 0,
              TotalCores = 0,
              UsedCores = 0,
              TotalMemory = 0,
              UsedMemory = 0
            };
            i = 1;
          } else {
            ++i;
          }
        }
      }
    }

  }
}