source: branches/2994-AutoDiffForIntervals/HeuristicLab.Clients.Hive/3.3/JobResultPoller.cs @ 17209

Last change on this file since 17209 was 17209, checked in by gkronber, 4 months ago

#2994: merged r17132:17198 from trunk to branch

File size: 4.3 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) Heuristic and Evolutionary Algorithms Laboratory (HEAL)
4 *
5 * This file is part of HeuristicLab.
6 *
7 * HeuristicLab is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * HeuristicLab is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
19 */
20#endregion
21
22using System;
23using System.Collections.Generic;
24using System.Threading;
25using HeuristicLab.Common;
26
27namespace HeuristicLab.Clients.Hive {
28  public class JobResultPoller {
29    private bool stopRequested;
30    private AutoResetEvent waitHandle;
31    private Thread thread;
32
33    private Guid jobId;
34    public Guid JobId {
35      get { return jobId; }
36      set { jobId = value; }
37    }
38
39    private TimeSpan interval;
40    public TimeSpan Interval {
41      get { return interval; }
42      set { interval = value; }
43    }
44
45    private bool isPolling;
46    public bool IsPolling {
47      get { return isPolling; }
48      set {
49        if (isPolling != value) {
50          isPolling = value;
51          OnIsPollingChanged();
52        }
53      }
54    }
55
56    private bool autoResumeOnException = false;
57    public bool AutoResumeOnException {
58      get { return autoResumeOnException; }
59      set { autoResumeOnException = value; }
60    }
61
62
63    public JobResultPoller(Guid jobId, TimeSpan interval) {
64      this.isPolling = false;
65      this.jobId = jobId;
66      this.interval = interval;
67    }
68
69    public void Start() {
70      IsPolling = true;
71      stopRequested = false;
72      thread = new Thread(RunPolling);
73      thread.Start();
74    }
75
76    public void Stop() {
77      stopRequested = true;
78      waitHandle.Set();
79      IsPolling = false;
80      thread = null;
81    }
82
83    private void RunPolling() {
84      IsPolling = true;
85      while (true) {
86        try {
87          waitHandle = new AutoResetEvent(false);
88          while (!stopRequested) {
89            OnPollingStarted();
90            FetchJobResults();
91            OnPollingFinished();
92            waitHandle.WaitOne(Interval);
93          }
94
95          waitHandle.Close();
96          IsPolling = false;
97          return;
98        }
99        catch (Exception e) {
100          OnExceptionOccured(e);
101          if (!autoResumeOnException) {
102            waitHandle.Close();
103            IsPolling = false;
104            return;
105          }
106        }
107      }
108    }
109
110    public IEnumerable<LightweightTask> FetchJobResults() {
111      return HiveServiceLocator.Instance.CallHiveService(service => {
112        var responses = new List<LightweightTask>();
113        responses.AddRange(service.GetLightweightJobTasks(jobId));
114        OnJobResultsReceived(responses);
115        return responses;
116      });
117    }
118
119    public event EventHandler<EventArgs<IEnumerable<LightweightTask>>> JobResultsReceived;
120    private void OnJobResultsReceived(IEnumerable<LightweightTask> lightweightJobs) {
121      var handler = JobResultsReceived;
122      if (handler != null) handler(this, new EventArgs<IEnumerable<LightweightTask>>(lightweightJobs));
123    }
124
125    public event EventHandler<EventArgs<Exception>> ExceptionOccured;
126    private void OnExceptionOccured(Exception e) {
127      var handler = ExceptionOccured;
128      if (handler != null) handler(this, new EventArgs<Exception>(e));
129    }
130
131    public event EventHandler IsPollingChanged;
132    private void OnIsPollingChanged() {
133      var handler = IsPollingChanged;
134      if (handler != null) handler(this, EventArgs.Empty);
135    }
136    public event EventHandler PollingStarted;
137    private void OnPollingStarted() {
138      var handler = PollingStarted;
139      if (handler != null) handler(this, EventArgs.Empty);
140    }
141    public event EventHandler PollingFinished;
142    private void OnPollingFinished() {
143      var handler = PollingFinished;
144      if (handler != null) handler(this, EventArgs.Empty);
145    }
146  }
147}
Note: See TracBrowser for help on using the repository browser.