Synnefo » snf-ganeti

import Text.JSON (encode, showJSON, JSValue(..))

import Ganeti.Utils (lockFile, exitIfBad, watchFile)

handleCall _ _ cfg (WaitForJobChange jid fields prev_job prev_log tmout) = do

  let compute_fn = computeJobUpdate cfg jid fields prev_log 

  qDir <- queueDir

  -- verify if the job is finalized, and return immediately in this case

  jobresult <- loadJobFromDisk qDir False jid

  case jobresult of

    Ok (job, _) | not (jobFinalized job) -> do

      let jobfile = liveJobFile qDir jid

      answer <- watchFile jobfile (min tmout C.luxiWfjcTimeout)

                  (prev_job, JSArray []) compute_fn

      return . Ok $ showJSON answer

    _ -> liftM (Ok . showJSON) compute_fn

-- | Query the status of a job and return the requested fields

-- and the logs newer than the given log number.

computeJobUpdate :: ConfigData -> JobId -> [String] -> JSValue 

                    -> IO (JSValue, JSValue)

computeJobUpdate cfg jid fields prev_log = do

  let sjid = show $ fromJobId jid

  logDebug $ "Inspecting fields " ++ show fields ++ " of job " ++ sjid

  let fromJSArray (JSArray xs) = xs

      fromJSArray _ = []

  let logFilter JSNull (JSArray _) = True

      logFilter (JSRational _ n) (JSArray (JSRational _ m:_)) = n < m

      logFilter _ _ = False

  let filterLogs n logs = JSArray (filter (logFilter n) (logs >>= fromJSArray))

  jobQuery <- handleClassicQuery cfg (Qlang.ItemTypeLuxi Qlang.QRJob)

                [Right . fromIntegral $ fromJobId jid] ("oplog" : fields) False

  let (rfields, rlogs) = case jobQuery of

        Ok (JSArray [JSArray (JSArray logs : answer)]) ->

          (answer, filterLogs prev_log logs)

        _ -> (map (const JSNull) fields, JSArray [])

  logDebug $ "Updates for job " ++ sjid ++ " are " ++ encode (rfields, rlogs)

  return (JSArray rfields, rlogs)