source: branches/3.3-HiveMigration/sources/HeuristicLab.Hive/HeuristicLab.Hive.Server.Core/3.3/DefaultScheduler.cs @ 4264

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using HeuristicLab.Hive.Server.Core.InternalInterfaces;
using HeuristicLab.Hive.Contracts.BusinessObjects;
using HeuristicLab.Hive.Server.DataAccess;
using System.Threading;
using HeuristicLab.DataAccess.Interfaces;
using System.Transactions;
using HeuristicLab.Hive.Contracts;

namespace HeuristicLab.Hive.Server.Core {
  internal class DefaultScheduler : IScheduler {

    private static object jobLock = new object();

    #region IScheduler Members

    public DefaultScheduler() { }

    public bool ExistsJobForSlave(HeartBeatData hbData) {
      List<JobDto> allOfflineJobsForSlave = new List<JobDto>(DaoLocator.JobDao.FindFittingJobsForSlave(JobState.Offline, hbData.FreeCores, hbData.FreeMemory, hbData.SlaveId));
      return (allOfflineJobsForSlave != null && allOfflineJobsForSlave.Count > 0);
    }

    public JobDto GetNextJobForSlave(Guid slaveId) {
      lock (jobLock) {
        /// Critical section ///
        JobDto jobToCalculate = null;
        using (TransactionScope scope = new TransactionScope(TransactionScopeOption.Required, new TransactionOptions { IsolationLevel = ApplicationConstants.ISOLATION_LEVEL_SCOPE })) {
          ClientDto slave = DaoLocator.ClientDao.FindById(slaveId);
          LinkedList<JobDto> allOfflineJobsForSlave = new LinkedList<JobDto>(DaoLocator.JobDao.FindFittingJobsForSlave(JobState.Offline, slave.NrOfFreeCores, slave.FreeMemory, slave.Id));
          if (allOfflineJobsForSlave != null && allOfflineJobsForSlave.Count > 0) {
            jobToCalculate = allOfflineJobsForSlave.First.Value;
            jobToCalculate.State = JobState.Calculating;
            jobToCalculate.Client = slave;
            jobToCalculate.Client.State = SlaveState.Calculating;
            jobToCalculate.DateCalculated = DateTime.Now;
            DaoLocator.JobDao.AssignSlaveToJob(slave.Id, jobToCalculate.Id);
            DaoLocator.JobDao.Update(jobToCalculate);
            DaoLocator.ClientDao.Update(jobToCalculate.Client);
          }
          scope.Complete();
        }
        /// End Critical section ///

        return jobToCalculate;
      }
    }

    #endregion
  }
}