source: branches/HeuristicLab.Hive-3.3/sources/HeuristicLab.Hive/HeuristicLab.Hive.Server.LINQDataAccess/3.3/SlaveDao.cs @ 5511

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2010 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 HeuristicLab.Hive.Contracts.BusinessObjects;
using HeuristicLab.Hive.Server.DataAccess;

namespace HeuristicLab.Hive.Server.LINQDataAccess {
  public class SlaveDao: BaseDao<SlaveDto, Slave>, ISlaveDao {

    public SlaveDao() {      
    }

    public SlaveDto FindById(Guid id) {
      return (from slave in Context.Resources.OfType<Slave>()                         
              where slave.ResourceId.Equals(id)
              select EntityToDto(slave, null)
            ).SingleOrDefault();      
    }

    public IEnumerable<SlaveDto> FindAll() {
      return (from slave in Context.Resources.OfType<Slave>()
              select
                EntityToDto(slave, null)
             ).ToList();
    }

    public IEnumerable<SlaveDto> FindAllSlavesWithoutGroup() {
      return (from slave in Context.Resources.OfType<Slave>()
              where slave.SlaveGroup_Resources_Parents.Count == 0
              select EntityToDto(slave, null)).ToList();
    }

    public SlaveDto GetSlaveForJob(Guid jobId) {
      return (from job in Context.Jobs
              where job.JobId.Equals(jobId)
              select EntityToDto(job.Slave, null)).SingleOrDefault();
    }

    public void SetServerSideCalendar(SlaveDto slave, Guid slaveGroupId) {
      Slave dbslave = Context.Resources.OfType<Slave>().SingleOrDefault(c => c.ResourceId.Equals(slave.Id));
      dbslave.UseCalendarFromResourceId = slaveGroupId;
      dbslave.CalendarSyncStatus = Enum.GetName(typeof(CalendarState), CalendarState.Fetch);
      CommitChanges();
    }

    public SlaveDto Insert(SlaveDto info) {
      Slave c = DtoToEntity(info, null);      
      Context.Resources.InsertOnSubmit(c);
      CommitChanges();
      info.Id = c.ResourceId;
      return info;
    }

    //Cascading delete takes care of the rest
    public void Delete(Guid id) {
      Resource res = Context.Resources.SingleOrDefault(c => c.ResourceId.Equals(id));            
      Context.Resources.DeleteOnSubmit(res);
      CommitChanges();
    }

    public void Update(SlaveDto info) {
      Slave slave = Context.Resources.OfType<Slave>().SingleOrDefault(c => c.ResourceId.Equals(info.Id));
      DtoToEntity(info, slave);
      CommitChanges();      
    }

    public override Slave DtoToEntity(SlaveDto source, Slave target) {
      if (source == null)
        return null;
      if (target == null) 
        target = new Slave();
      
      target.CPUSpeed = source.CpuSpeedPerCore;
      
      target.FreeMemory = source.FreeMemory;
      target.Name = source.Name;
      target.ResourceId = source.Id;
      target.CalendarSyncStatus = Enum.GetName(typeof(CalendarState), source.CalendarSyncStatus);
      target.Login = source.Login;
      target.Memory = source.Memory;
      target.NumberOfCores = source.NrOfCores;
      target.NumberOfFreeCores = source.NrOfFreeCores;
      target.Status = source.State.ToString();
      target.LastHeartbeat = source.LastHeartbeat;
      return target;
    }

    public override SlaveDto EntityToDto(Slave source, SlaveDto target) {
      if (source == null)
        return null;
      if(target == null) 
        target = new SlaveDto();
      target.CpuSpeedPerCore = source.CPUSpeed;
      target.FreeMemory = source.FreeMemory ?? 0;
      target.Id = source.ResourceId;
      target.CalendarSyncStatus = (CalendarState) Enum.Parse(typeof (CalendarState), source.CalendarSyncStatus);
      target.Login = source.Login;
      target.Memory = source.Memory ?? 0;
      target.Name = source.Name;
      target.NrOfCores = source.NumberOfCores ?? 0;
      target.NrOfFreeCores = source.NumberOfFreeCores ?? 0;
      target.State = (SlaveState)Enum.Parse(typeof(SlaveState), source.Status, true);
      target.IsAllowedToCalculate = source.IsAllowedToCalculate;
      target.LastHeartbeat = source.LastHeartbeat;
      return target;
    } 
  }
}