root / src / Ganeti / Jobs.hs @ ace37e24
History | View | Annotate | Download (2.1 kB)
1 |
{-| Generic code to work with jobs, e.g. submit jobs and check their status. |
---|---|
2 |
|
3 |
-} |
4 |
|
5 |
{- |
6 |
|
7 |
Copyright (C) 2009, 2010, 2011, 2012 Google Inc. |
8 |
|
9 |
This program is free software; you can redistribute it and/or modify |
10 |
it under the terms of the GNU General Public License as published by |
11 |
the Free Software Foundation; either version 2 of the License, or |
12 |
(at your option) any later version. |
13 |
|
14 |
This program is distributed in the hope that it will be useful, but |
15 |
WITHOUT ANY WARRANTY; without even the implied warranty of |
16 |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
17 |
General Public License for more details. |
18 |
|
19 |
You should have received a copy of the GNU General Public License |
20 |
along with this program; if not, write to the Free Software |
21 |
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
22 |
02110-1301, USA. |
23 |
|
24 |
-} |
25 |
|
26 |
module Ganeti.Jobs |
27 |
( execJobsWait |
28 |
, waitForJobs |
29 |
) where |
30 |
|
31 |
import Control.Concurrent (threadDelay) |
32 |
|
33 |
import Ganeti.BasicTypes |
34 |
import Ganeti.Errors |
35 |
import qualified Ganeti.Luxi as L |
36 |
import Ganeti.OpCodes |
37 |
import Ganeti.Types |
38 |
|
39 |
-- | Executes a set of jobs and waits for their completion, returning their |
40 |
-- status. |
41 |
execJobsWait :: [[MetaOpCode]] -- ^ The list of jobs |
42 |
-> ([L.JobId] -> IO ()) -- ^ Post-submission callback |
43 |
-> L.Client -- ^ The Luxi client |
44 |
-> IO (Result [(L.JobId, JobStatus)]) |
45 |
execJobsWait opcodes callback client = do |
46 |
jids <- L.submitManyJobs client opcodes |
47 |
case jids of |
48 |
Bad e -> return . Bad $ "Job submission error: " ++ formatError e |
49 |
Ok jids' -> do |
50 |
callback jids' |
51 |
waitForJobs jids' client |
52 |
|
53 |
-- | Polls a set of jobs at a fixed interval until all are finished |
54 |
-- one way or another. |
55 |
waitForJobs :: [L.JobId] -> L.Client -> IO (Result [(L.JobId, JobStatus)]) |
56 |
waitForJobs jids client = do |
57 |
sts <- L.queryJobsStatus client jids |
58 |
case sts of |
59 |
Bad e -> return . Bad $ "Checking job status: " ++ formatError e |
60 |
Ok sts' -> if any (<= JOB_STATUS_RUNNING) sts' |
61 |
then do |
62 |
-- TODO: replace hardcoded value with a better thing |
63 |
threadDelay (1000000 * 15) |
64 |
waitForJobs jids client |
65 |
else return . Ok $ zip jids sts' |