Synnefo » snf-ganeti

#!/usr/bin/python

#

# Copyright (C) 2006, 2007 Google Inc.

#

# This program 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 2 of the License, or

# (at your option) any later version.

#

# This program 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 this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

import sys

import os

import itertools

import simplejson

import time

from optparse import make_option

from cStringIO import StringIO

from ganeti.cli import *

from ganeti import opcodes

from ganeti import logger

from ganeti import constants

from ganeti import utils

from ganeti import errors

def Delay(opts, args):

  """Sleeps for a while

  """

  delay = float(args[0])

  op = opcodes.OpTestDelay(duration=delay,

                           on_master=opts.on_master,

                           on_nodes=opts.on_nodes)

  job = opcodes.Job(op_list=[op])

  jid = SubmitJob(job)

  print "Job id", jid

  return 0

def GenericOpCodes(opts, args):

  """Send any opcode to the master

  """

  fname = args[0]

  op_data = simplejson.loads(open(fname).read())

  op_list = [opcodes.OpCode.LoadOpCode(val) for val in op_data]

  job = opcodes.Job(op_list=op_list)

  jid = SubmitJob(job)

  print "Job id:", jid

  query = {

    "object": "jobs",

    "fields": ["status"],

    "names": [jid],

    }

  # wait for job to complete (either by success or failure)

  while True:

    jdata = SubmitQuery(query)

    if not jdata:

      # job not found, gone away!

      print "Job lost!"

      return 1

    status = jdata[0][0]

    print status

    if status in (opcodes.Job.STATUS_SUCCESS, opcodes.Job.STATUS_FAIL):

      break

    # sleep between checks

    time.sleep(0.5)

  # job has finished, get and process its results

  query["fields"].extend(["op_list", "op_status", "op_result"])

  jdata = SubmitQuery(query)

  if not jdata:

    # job not found, gone away!

    print "Job lost!"

    return 1

  print jdata[0]

  status, op_list, op_status, op_result = jdata[0]

  for idx, op in enumerate(op_list):

    print idx, op.OP_ID, op_status[idx], op_result[idx]

  return 0

commands = {

  'delay': (Delay, ARGS_ONE,

            [DEBUG_OPT,

             make_option("--no-master", dest="on_master", default=True,

                         action="store_false",

                         help="Do not sleep in the master code"),

             make_option("-n", dest="on_nodes", default=[],

                         action="append",

                         help="Select nodes to sleep on"),

             ],

            "[opts...] <duration>", "Executes a TestDelay OpCode"),

  'submit-job': (GenericOpCodes, ARGS_ONE,

                 [DEBUG_OPT,

                  ],

                 "<op_list_file>", "Submits a job built from a json-file"

                 " with a list of serialized opcodes"),

  }

if __name__ == '__main__':

  sys.exit(GenericMain(commands))