Statistics
| Branch: | Tag: | Revision:

root / scripts / gnt-cluster @ cea881e5

History | View | Annotate | Download (25.1 kB)

1
#!/usr/bin/python
2
#
3

    
4
# Copyright (C) 2006, 2007 Google Inc.
5
#
6
# This program is free software; you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation; either version 2 of the License, or
9
# (at your option) any later version.
10
#
11
# This program is distributed in the hope that it will be useful, but
12
# WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14
# General Public License for more details.
15
#
16
# You should have received a copy of the GNU General Public License
17
# along with this program; if not, write to the Free Software
18
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19
# 02110-1301, USA.
20

    
21
"""Cluster related commands"""
22

    
23
# pylint: disable-msg=W0401,W0613,W0614,C0103
24
# W0401: Wildcard import ganeti.cli
25
# W0613: Unused argument, since all functions follow the same API
26
# W0614: Unused import %s from wildcard import (since we need cli)
27
# C0103: Invalid name gnt-cluster
28

    
29
import sys
30
import os.path
31
import time
32
import OpenSSL
33

    
34
from ganeti.cli import *
35
from ganeti import opcodes
36
from ganeti import constants
37
from ganeti import errors
38
from ganeti import utils
39
from ganeti import bootstrap
40
from ganeti import ssh
41
from ganeti import objects
42
from ganeti import uidpool
43
from ganeti import compat
44

    
45

    
46
@UsesRPC
47
def InitCluster(opts, args):
48
  """Initialize the cluster.
49

    
50
  @param opts: the command line options selected by the user
51
  @type args: list
52
  @param args: should contain only one element, the desired
53
      cluster name
54
  @rtype: int
55
  @return: the desired exit code
56

    
57
  """
58
  if not opts.lvm_storage and opts.vg_name:
59
    ToStderr("Options --no-lvm-storage and --vg-name conflict.")
60
    return 1
61

    
62
  vg_name = opts.vg_name
63
  if opts.lvm_storage and not opts.vg_name:
64
    vg_name = constants.DEFAULT_VG
65

    
66
  hvlist = opts.enabled_hypervisors
67
  if hvlist is None:
68
    hvlist = constants.DEFAULT_ENABLED_HYPERVISOR
69
  hvlist = hvlist.split(",")
70

    
71
  hvparams = dict(opts.hvparams)
72
  beparams = opts.beparams
73
  nicparams = opts.nicparams
74

    
75
  # prepare beparams dict
76
  beparams = objects.FillDict(constants.BEC_DEFAULTS, beparams)
77
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
78

    
79
  # prepare nicparams dict
80
  nicparams = objects.FillDict(constants.NICC_DEFAULTS, nicparams)
81
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)
82

    
83
  # prepare hvparams dict
84
  for hv in constants.HYPER_TYPES:
85
    if hv not in hvparams:
86
      hvparams[hv] = {}
87
    hvparams[hv] = objects.FillDict(constants.HVC_DEFAULTS[hv], hvparams[hv])
88
    utils.ForceDictType(hvparams[hv], constants.HVS_PARAMETER_TYPES)
89

    
90
  if opts.candidate_pool_size is None:
91
    opts.candidate_pool_size = constants.MASTER_POOL_SIZE_DEFAULT
92

    
93
  if opts.mac_prefix is None:
94
    opts.mac_prefix = constants.DEFAULT_MAC_PREFIX
95

    
96
  uid_pool = opts.uid_pool
97
  if uid_pool is not None:
98
    uid_pool = uidpool.ParseUidPool(uid_pool)
99

    
100
  bootstrap.InitCluster(cluster_name=args[0],
101
                        secondary_ip=opts.secondary_ip,
102
                        vg_name=vg_name,
103
                        mac_prefix=opts.mac_prefix,
104
                        master_netdev=opts.master_netdev,
105
                        file_storage_dir=opts.file_storage_dir,
106
                        enabled_hypervisors=hvlist,
107
                        hvparams=hvparams,
108
                        beparams=beparams,
109
                        nicparams=nicparams,
110
                        candidate_pool_size=opts.candidate_pool_size,
111
                        modify_etc_hosts=opts.modify_etc_hosts,
112
                        modify_ssh_setup=opts.modify_ssh_setup,
113
                        maintain_node_health=opts.maintain_node_health,
114
                        uid_pool=uid_pool,
115
                        )
116
  op = opcodes.OpPostInitCluster()
117
  SubmitOpCode(op, opts=opts)
118
  return 0
119

    
120

    
121
@UsesRPC
122
def DestroyCluster(opts, args):
123
  """Destroy the cluster.
124

    
125
  @param opts: the command line options selected by the user
126
  @type args: list
127
  @param args: should be an empty list
128
  @rtype: int
129
  @return: the desired exit code
130

    
131
  """
132
  if not opts.yes_do_it:
133
    ToStderr("Destroying a cluster is irreversible. If you really want"
134
             " destroy this cluster, supply the --yes-do-it option.")
135
    return 1
136

    
137
  op = opcodes.OpDestroyCluster()
138
  master = SubmitOpCode(op, opts=opts)
139
  # if we reached this, the opcode didn't fail; we can proceed to
140
  # shutdown all the daemons
141
  bootstrap.FinalizeClusterDestroy(master)
142
  return 0
143

    
144

    
145
def RenameCluster(opts, args):
146
  """Rename the cluster.
147

    
148
  @param opts: the command line options selected by the user
149
  @type args: list
150
  @param args: should contain only one element, the new cluster name
151
  @rtype: int
152
  @return: the desired exit code
153

    
154
  """
155
  name = args[0]
156
  if not opts.force:
157
    usertext = ("This will rename the cluster to '%s'. If you are connected"
158
                " over the network to the cluster name, the operation is very"
159
                " dangerous as the IP address will be removed from the node"
160
                " and the change may not go through. Continue?") % name
161
    if not AskUser(usertext):
162
      return 1
163

    
164
  op = opcodes.OpRenameCluster(name=name)
165
  SubmitOpCode(op, opts=opts)
166
  return 0
167

    
168

    
169
def RedistributeConfig(opts, args):
170
  """Forces push of the cluster configuration.
171

    
172
  @param opts: the command line options selected by the user
173
  @type args: list
174
  @param args: empty list
175
  @rtype: int
176
  @return: the desired exit code
177

    
178
  """
179
  op = opcodes.OpRedistributeConfig()
180
  SubmitOrSend(op, opts)
181
  return 0
182

    
183

    
184
def ShowClusterVersion(opts, args):
185
  """Write version of ganeti software to the standard output.
186

    
187
  @param opts: the command line options selected by the user
188
  @type args: list
189
  @param args: should be an empty list
190
  @rtype: int
191
  @return: the desired exit code
192

    
193
  """
194
  cl = GetClient()
195
  result = cl.QueryClusterInfo()
196
  ToStdout("Software version: %s", result["software_version"])
197
  ToStdout("Internode protocol: %s", result["protocol_version"])
198
  ToStdout("Configuration format: %s", result["config_version"])
199
  ToStdout("OS api version: %s", result["os_api_version"])
200
  ToStdout("Export interface: %s", result["export_version"])
201
  return 0
202

    
203

    
204
def ShowClusterMaster(opts, args):
205
  """Write name of master node to the standard output.
206

    
207
  @param opts: the command line options selected by the user
208
  @type args: list
209
  @param args: should be an empty list
210
  @rtype: int
211
  @return: the desired exit code
212

    
213
  """
214
  master = bootstrap.GetMaster()
215
  ToStdout(master)
216
  return 0
217

    
218

    
219
def _PrintGroupedParams(paramsdict, level=1):
220
  """Print Grouped parameters (be, nic, disk) by group.
221

    
222
  @type paramsdict: dict of dicts
223
  @param paramsdict: {group: {param: value, ...}, ...}
224
  @type level: int
225
  @param level: Level of indention
226

    
227
  """
228
  indent = "  " * level
229
  for item, val in sorted(paramsdict.items()):
230
    if isinstance(val, dict):
231
      ToStdout("%s- %s:", indent, item)
232
      _PrintGroupedParams(val, level=level + 1)
233
    else:
234
      ToStdout("%s  %s: %s", indent, item, val)
235

    
236

    
237
def ShowClusterConfig(opts, args):
238
  """Shows cluster information.
239

    
240
  @param opts: the command line options selected by the user
241
  @type args: list
242
  @param args: should be an empty list
243
  @rtype: int
244
  @return: the desired exit code
245

    
246
  """
247
  cl = GetClient()
248
  result = cl.QueryClusterInfo()
249

    
250
  ToStdout("Cluster name: %s", result["name"])
251
  ToStdout("Cluster UUID: %s", result["uuid"])
252

    
253
  ToStdout("Creation time: %s", utils.FormatTime(result["ctime"]))
254
  ToStdout("Modification time: %s", utils.FormatTime(result["mtime"]))
255

    
256
  ToStdout("Master node: %s", result["master"])
257

    
258
  ToStdout("Architecture (this node): %s (%s)",
259
           result["architecture"][0], result["architecture"][1])
260

    
261
  if result["tags"]:
262
    tags = utils.CommaJoin(utils.NiceSort(result["tags"]))
263
  else:
264
    tags = "(none)"
265

    
266
  ToStdout("Tags: %s", tags)
267

    
268
  ToStdout("Default hypervisor: %s", result["default_hypervisor"])
269
  ToStdout("Enabled hypervisors: %s",
270
           utils.CommaJoin(result["enabled_hypervisors"]))
271

    
272
  ToStdout("Hypervisor parameters:")
273
  _PrintGroupedParams(result["hvparams"])
274

    
275
  ToStdout("OS specific hypervisor parameters:")
276
  _PrintGroupedParams(result["os_hvp"])
277

    
278
  ToStdout("Cluster parameters:")
279
  ToStdout("  - candidate pool size: %s", result["candidate_pool_size"])
280
  ToStdout("  - master netdev: %s", result["master_netdev"])
281
  ToStdout("  - lvm volume group: %s", result["volume_group_name"])
282
  ToStdout("  - file storage path: %s", result["file_storage_dir"])
283
  ToStdout("  - maintenance of node health: %s",
284
           result["maintain_node_health"])
285
  ToStdout("  - uid pool: %s", uidpool.FormatUidPool(result["uid_pool"]))
286

    
287
  ToStdout("Default instance parameters:")
288
  _PrintGroupedParams(result["beparams"])
289

    
290
  ToStdout("Default nic parameters:")
291
  _PrintGroupedParams(result["nicparams"])
292

    
293
  return 0
294

    
295

    
296
def ClusterCopyFile(opts, args):
297
  """Copy a file from master to some nodes.
298

    
299
  @param opts: the command line options selected by the user
300
  @type args: list
301
  @param args: should contain only one element, the path of
302
      the file to be copied
303
  @rtype: int
304
  @return: the desired exit code
305

    
306
  """
307
  filename = args[0]
308
  if not os.path.exists(filename):
309
    raise errors.OpPrereqError("No such filename '%s'" % filename,
310
                               errors.ECODE_INVAL)
311

    
312
  cl = GetClient()
313

    
314
  cluster_name = cl.QueryConfigValues(["cluster_name"])[0]
315

    
316
  results = GetOnlineNodes(nodes=opts.nodes, cl=cl, filter_master=True,
317
                           secondary_ips=opts.use_replication_network)
318

    
319
  srun = ssh.SshRunner(cluster_name=cluster_name)
320
  for node in results:
321
    if not srun.CopyFileToNode(node, filename):
322
      ToStderr("Copy of file %s to node %s failed", filename, node)
323

    
324
  return 0
325

    
326

    
327
def RunClusterCommand(opts, args):
328
  """Run a command on some nodes.
329

    
330
  @param opts: the command line options selected by the user
331
  @type args: list
332
  @param args: should contain the command to be run and its arguments
333
  @rtype: int
334
  @return: the desired exit code
335

    
336
  """
337
  cl = GetClient()
338

    
339
  command = " ".join(args)
340

    
341
  nodes = GetOnlineNodes(nodes=opts.nodes, cl=cl)
342

    
343
  cluster_name, master_node = cl.QueryConfigValues(["cluster_name",
344
                                                    "master_node"])
345

    
346
  srun = ssh.SshRunner(cluster_name=cluster_name)
347

    
348
  # Make sure master node is at list end
349
  if master_node in nodes:
350
    nodes.remove(master_node)
351
    nodes.append(master_node)
352

    
353
  for name in nodes:
354
    result = srun.Run(name, "root", command)
355
    ToStdout("------------------------------------------------")
356
    ToStdout("node: %s", name)
357
    ToStdout("%s", result.output)
358
    ToStdout("return code = %s", result.exit_code)
359

    
360
  return 0
361

    
362

    
363
def VerifyCluster(opts, args):
364
  """Verify integrity of cluster, performing various test on nodes.
365

    
366
  @param opts: the command line options selected by the user
367
  @type args: list
368
  @param args: should be an empty list
369
  @rtype: int
370
  @return: the desired exit code
371

    
372
  """
373
  skip_checks = []
374
  if opts.skip_nplusone_mem:
375
    skip_checks.append(constants.VERIFY_NPLUSONE_MEM)
376
  op = opcodes.OpVerifyCluster(skip_checks=skip_checks,
377
                               verbose=opts.verbose,
378
                               error_codes=opts.error_codes,
379
                               debug_simulate_errors=opts.simulate_errors)
380
  if SubmitOpCode(op, opts=opts):
381
    return 0
382
  else:
383
    return 1
384

    
385

    
386
def VerifyDisks(opts, args):
387
  """Verify integrity of cluster disks.
388

    
389
  @param opts: the command line options selected by the user
390
  @type args: list
391
  @param args: should be an empty list
392
  @rtype: int
393
  @return: the desired exit code
394

    
395
  """
396
  op = opcodes.OpVerifyDisks()
397
  result = SubmitOpCode(op, opts=opts)
398
  if not isinstance(result, (list, tuple)) or len(result) != 3:
399
    raise errors.ProgrammerError("Unknown result type for OpVerifyDisks")
400

    
401
  bad_nodes, instances, missing = result
402

    
403
  retcode = constants.EXIT_SUCCESS
404

    
405
  if bad_nodes:
406
    for node, text in bad_nodes.items():
407
      ToStdout("Error gathering data on node %s: %s",
408
               node, utils.SafeEncode(text[-400:]))
409
      retcode |= 1
410
      ToStdout("You need to fix these nodes first before fixing instances")
411

    
412
  if instances:
413
    for iname in instances:
414
      if iname in missing:
415
        continue
416
      op = opcodes.OpActivateInstanceDisks(instance_name=iname)
417
      try:
418
        ToStdout("Activating disks for instance '%s'", iname)
419
        SubmitOpCode(op, opts=opts)
420
      except errors.GenericError, err:
421
        nret, msg = FormatError(err)
422
        retcode |= nret
423
        ToStderr("Error activating disks for instance %s: %s", iname, msg)
424

    
425
  if missing:
426
    for iname, ival in missing.iteritems():
427
      all_missing = compat.all(ival, lambda x: x[0] in bad_nodes)
428
      if all_missing:
429
        ToStdout("Instance %s cannot be verified as it lives on"
430
                 " broken nodes", iname)
431
      else:
432
        ToStdout("Instance %s has missing logical volumes:", iname)
433
        ival.sort()
434
        for node, vol in ival:
435
          if node in bad_nodes:
436
            ToStdout("\tbroken node %s /dev/xenvg/%s", node, vol)
437
          else:
438
            ToStdout("\t%s /dev/xenvg/%s", node, vol)
439
    ToStdout("You need to run replace_disks for all the above"
440
           " instances, if this message persist after fixing nodes.")
441
    retcode |= 1
442

    
443
  return retcode
444

    
445

    
446
def RepairDiskSizes(opts, args):
447
  """Verify sizes of cluster disks.
448

    
449
  @param opts: the command line options selected by the user
450
  @type args: list
451
  @param args: optional list of instances to restrict check to
452
  @rtype: int
453
  @return: the desired exit code
454

    
455
  """
456
  op = opcodes.OpRepairDiskSizes(instances=args)
457
  SubmitOpCode(op, opts=opts)
458

    
459

    
460
@UsesRPC
461
def MasterFailover(opts, args):
462
  """Failover the master node.
463

    
464
  This command, when run on a non-master node, will cause the current
465
  master to cease being master, and the non-master to become new
466
  master.
467

    
468
  @param opts: the command line options selected by the user
469
  @type args: list
470
  @param args: should be an empty list
471
  @rtype: int
472
  @return: the desired exit code
473

    
474
  """
475
  if opts.no_voting:
476
    usertext = ("This will perform the failover even if most other nodes"
477
                " are down, or if this node is outdated. This is dangerous"
478
                " as it can lead to a non-consistent cluster. Check the"
479
                " gnt-cluster(8) man page before proceeding. Continue?")
480
    if not AskUser(usertext):
481
      return 1
482

    
483
  return bootstrap.MasterFailover(no_voting=opts.no_voting)
484

    
485

    
486
def SearchTags(opts, args):
487
  """Searches the tags on all the cluster.
488

    
489
  @param opts: the command line options selected by the user
490
  @type args: list
491
  @param args: should contain only one element, the tag pattern
492
  @rtype: int
493
  @return: the desired exit code
494

    
495
  """
496
  op = opcodes.OpSearchTags(pattern=args[0])
497
  result = SubmitOpCode(op, opts=opts)
498
  if not result:
499
    return 1
500
  result = list(result)
501
  result.sort()
502
  for path, tag in result:
503
    ToStdout("%s %s", path, tag)
504

    
505

    
506
def _RenewCrypto(new_cluster_cert, new_rapi_cert, rapi_cert_filename,
507
                 new_confd_hmac_key, force):
508
  """Renews cluster certificates, keys and secrets.
509

    
510
  @type new_cluster_cert: bool
511
  @param new_cluster_cert: Whether to generate a new cluster certificate
512
  @type new_rapi_cert: bool
513
  @param new_rapi_cert: Whether to generate a new RAPI certificate
514
  @type rapi_cert_filename: string
515
  @param rapi_cert_filename: Path to file containing new RAPI certificate
516
  @type new_confd_hmac_key: bool
517
  @param new_confd_hmac_key: Whether to generate a new HMAC key
518
  @type force: bool
519
  @param force: Whether to ask user for confirmation
520

    
521
  """
522
  if new_rapi_cert and rapi_cert_filename:
523
    ToStderr("Only one of the --new-rapi-certficate and --rapi-certificate"
524
             " options can be specified at the same time.")
525
    return 1
526

    
527
  if rapi_cert_filename:
528
    # Read and verify new certificate
529
    try:
530
      rapi_cert_pem = utils.ReadFile(rapi_cert_filename)
531

    
532
      OpenSSL.crypto.load_certificate(OpenSSL.crypto.FILETYPE_PEM,
533
                                      rapi_cert_pem)
534
    except Exception, err: # pylint: disable-msg=W0703
535
      ToStderr("Can't load new RAPI certificate from %s: %s" %
536
               (rapi_cert_filename, str(err)))
537
      return 1
538

    
539
    try:
540
      OpenSSL.crypto.load_privatekey(OpenSSL.crypto.FILETYPE_PEM, rapi_cert_pem)
541
    except Exception, err: # pylint: disable-msg=W0703
542
      ToStderr("Can't load new RAPI private key from %s: %s" %
543
               (rapi_cert_filename, str(err)))
544
      return 1
545

    
546
  else:
547
    rapi_cert_pem = None
548

    
549
  if not force:
550
    usertext = ("This requires all daemons on all nodes to be restarted and"
551
                " may take some time. Continue?")
552
    if not AskUser(usertext):
553
      return 1
554

    
555
  def _RenewCryptoInner(ctx):
556
    ctx.feedback_fn("Updating certificates and keys")
557
    bootstrap.GenerateClusterCrypto(new_cluster_cert, new_rapi_cert,
558
                                    new_confd_hmac_key,
559
                                    rapi_cert_pem=rapi_cert_pem)
560

    
561
    files_to_copy = []
562

    
563
    if new_cluster_cert:
564
      files_to_copy.append(constants.NODED_CERT_FILE)
565

    
566
    if new_rapi_cert or rapi_cert_pem:
567
      files_to_copy.append(constants.RAPI_CERT_FILE)
568

    
569
    if new_confd_hmac_key:
570
      files_to_copy.append(constants.CONFD_HMAC_KEY)
571

    
572
    if files_to_copy:
573
      for node_name in ctx.nonmaster_nodes:
574
        ctx.feedback_fn("Copying %s to %s" %
575
                        (", ".join(files_to_copy), node_name))
576
        for file_name in files_to_copy:
577
          ctx.ssh.CopyFileToNode(node_name, file_name)
578

    
579
  RunWhileClusterStopped(ToStdout, _RenewCryptoInner)
580

    
581
  ToStdout("All requested certificates and keys have been replaced."
582
           " Running \"gnt-cluster verify\" now is recommended.")
583

    
584
  return 0
585

    
586

    
587
def RenewCrypto(opts, args):
588
  """Renews cluster certificates, keys and secrets.
589

    
590
  """
591
  return _RenewCrypto(opts.new_cluster_cert,
592
                      opts.new_rapi_cert,
593
                      opts.rapi_cert,
594
                      opts.new_confd_hmac_key,
595
                      opts.force)
596

    
597

    
598
def SetClusterParams(opts, args):
599
  """Modify the cluster.
600

    
601
  @param opts: the command line options selected by the user
602
  @type args: list
603
  @param args: should be an empty list
604
  @rtype: int
605
  @return: the desired exit code
606

    
607
  """
608
  if not (not opts.lvm_storage or opts.vg_name or
609
          opts.enabled_hypervisors or opts.hvparams or
610
          opts.beparams or opts.nicparams or
611
          opts.candidate_pool_size is not None or
612
          opts.uid_pool is not None or
613
          opts.maintain_node_health is not None or
614
          opts.add_uids is not None or
615
          opts.remove_uids is not None):
616
    ToStderr("Please give at least one of the parameters.")
617
    return 1
618

    
619
  vg_name = opts.vg_name
620
  if not opts.lvm_storage and opts.vg_name:
621
    ToStderr("Options --no-lvm-storage and --vg-name conflict.")
622
    return 1
623

    
624
  if not opts.lvm_storage:
625
    vg_name = ""
626

    
627
  hvlist = opts.enabled_hypervisors
628
  if hvlist is not None:
629
    hvlist = hvlist.split(",")
630

    
631
  # a list of (name, dict) we can pass directly to dict() (or [])
632
  hvparams = dict(opts.hvparams)
633
  for hv_params in hvparams.values():
634
    utils.ForceDictType(hv_params, constants.HVS_PARAMETER_TYPES)
635

    
636
  beparams = opts.beparams
637
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
638

    
639
  nicparams = opts.nicparams
640
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)
641

    
642

    
643
  mnh = opts.maintain_node_health
644

    
645
  uid_pool = opts.uid_pool
646
  if uid_pool is not None:
647
    uid_pool = uidpool.ParseUidPool(uid_pool)
648

    
649
  add_uids = opts.add_uids
650
  if add_uids is not None:
651
    add_uids = uidpool.ParseUidPool(add_uids)
652

    
653
  remove_uids = opts.remove_uids
654
  if remove_uids is not None:
655
    remove_uids = uidpool.ParseUidPool(remove_uids)
656

    
657
  op = opcodes.OpSetClusterParams(vg_name=vg_name,
658
                                  enabled_hypervisors=hvlist,
659
                                  hvparams=hvparams,
660
                                  os_hvp=None,
661
                                  beparams=beparams,
662
                                  nicparams=nicparams,
663
                                  candidate_pool_size=opts.candidate_pool_size,
664
                                  maintain_node_health=mnh,
665
                                  uid_pool=uid_pool,
666
                                  add_uids=add_uids,
667
                                  remove_uids=remove_uids)
668
  SubmitOpCode(op, opts=opts)
669
  return 0
670

    
671

    
672
def QueueOps(opts, args):
673
  """Queue operations.
674

    
675
  @param opts: the command line options selected by the user
676
  @type args: list
677
  @param args: should contain only one element, the subcommand
678
  @rtype: int
679
  @return: the desired exit code
680

    
681
  """
682
  command = args[0]
683
  client = GetClient()
684
  if command in ("drain", "undrain"):
685
    drain_flag = command == "drain"
686
    client.SetQueueDrainFlag(drain_flag)
687
  elif command == "info":
688
    result = client.QueryConfigValues(["drain_flag"])
689
    if result[0]:
690
      val = "set"
691
    else:
692
      val = "unset"
693
    ToStdout("The drain flag is %s" % val)
694
  else:
695
    raise errors.OpPrereqError("Command '%s' is not valid." % command,
696
                               errors.ECODE_INVAL)
697

    
698
  return 0
699

    
700

    
701
def _ShowWatcherPause(until):
702
  if until is None or until < time.time():
703
    ToStdout("The watcher is not paused.")
704
  else:
705
    ToStdout("The watcher is paused until %s.", time.ctime(until))
706

    
707

    
708
def WatcherOps(opts, args):
709
  """Watcher operations.
710

    
711
  @param opts: the command line options selected by the user
712
  @type args: list
713
  @param args: should contain only one element, the subcommand
714
  @rtype: int
715
  @return: the desired exit code
716

    
717
  """
718
  command = args[0]
719
  client = GetClient()
720

    
721
  if command == "continue":
722
    client.SetWatcherPause(None)
723
    ToStdout("The watcher is no longer paused.")
724

    
725
  elif command == "pause":
726
    if len(args) < 2:
727
      raise errors.OpPrereqError("Missing pause duration", errors.ECODE_INVAL)
728

    
729
    result = client.SetWatcherPause(time.time() + ParseTimespec(args[1]))
730
    _ShowWatcherPause(result)
731

    
732
  elif command == "info":
733
    result = client.QueryConfigValues(["watcher_pause"])
734
    _ShowWatcherPause(result[0])
735

    
736
  else:
737
    raise errors.OpPrereqError("Command '%s' is not valid." % command,
738
                               errors.ECODE_INVAL)
739

    
740
  return 0
741

    
742

    
743
commands = {
744
  'init': (
745
    InitCluster, [ArgHost(min=1, max=1)],
746
    [BACKEND_OPT, CP_SIZE_OPT, ENABLED_HV_OPT, GLOBAL_FILEDIR_OPT,
747
     HVLIST_OPT, MAC_PREFIX_OPT, MASTER_NETDEV_OPT, NIC_PARAMS_OPT,
748
     NOLVM_STORAGE_OPT, NOMODIFY_ETCHOSTS_OPT, NOMODIFY_SSH_SETUP_OPT,
749
     SECONDARY_IP_OPT, VG_NAME_OPT, MAINTAIN_NODE_HEALTH_OPT,
750
     UIDPOOL_OPT],
751
    "[opts...] <cluster_name>", "Initialises a new cluster configuration"),
752
  'destroy': (
753
    DestroyCluster, ARGS_NONE, [YES_DOIT_OPT],
754
    "", "Destroy cluster"),
755
  'rename': (
756
    RenameCluster, [ArgHost(min=1, max=1)],
757
    [FORCE_OPT],
758
    "<new_name>",
759
    "Renames the cluster"),
760
  'redist-conf': (
761
    RedistributeConfig, ARGS_NONE, [SUBMIT_OPT],
762
    "", "Forces a push of the configuration file and ssconf files"
763
    " to the nodes in the cluster"),
764
  'verify': (
765
    VerifyCluster, ARGS_NONE,
766
    [VERBOSE_OPT, DEBUG_SIMERR_OPT, ERROR_CODES_OPT, NONPLUS1_OPT],
767
    "", "Does a check on the cluster configuration"),
768
  'verify-disks': (
769
    VerifyDisks, ARGS_NONE, [],
770
    "", "Does a check on the cluster disk status"),
771
  'repair-disk-sizes': (
772
    RepairDiskSizes, ARGS_MANY_INSTANCES, [],
773
    "", "Updates mismatches in recorded disk sizes"),
774
  'masterfailover': (
775
    MasterFailover, ARGS_NONE, [NOVOTING_OPT],
776
    "", "Makes the current node the master"),
777
  'version': (
778
    ShowClusterVersion, ARGS_NONE, [],
779
    "", "Shows the cluster version"),
780
  'getmaster': (
781
    ShowClusterMaster, ARGS_NONE, [],
782
    "", "Shows the cluster master"),
783
  'copyfile': (
784
    ClusterCopyFile, [ArgFile(min=1, max=1)],
785
    [NODE_LIST_OPT, USE_REPL_NET_OPT],
786
    "[-n node...] <filename>", "Copies a file to all (or only some) nodes"),
787
  'command': (
788
    RunClusterCommand, [ArgCommand(min=1)],
789
    [NODE_LIST_OPT],
790
    "[-n node...] <command>", "Runs a command on all (or only some) nodes"),
791
  'info': (
792
    ShowClusterConfig, ARGS_NONE, [],
793
    "", "Show cluster configuration"),
794
  'list-tags': (
795
    ListTags, ARGS_NONE, [], "", "List the tags of the cluster"),
796
  'add-tags': (
797
    AddTags, [ArgUnknown()], [TAG_SRC_OPT],
798
    "tag...", "Add tags to the cluster"),
799
  'remove-tags': (
800
    RemoveTags, [ArgUnknown()], [TAG_SRC_OPT],
801
    "tag...", "Remove tags from the cluster"),
802
  'search-tags': (
803
    SearchTags, [ArgUnknown(min=1, max=1)],
804
    [], "", "Searches the tags on all objects on"
805
    " the cluster for a given pattern (regex)"),
806
  'queue': (
807
    QueueOps,
808
    [ArgChoice(min=1, max=1, choices=["drain", "undrain", "info"])],
809
    [], "drain|undrain|info", "Change queue properties"),
810
  'watcher': (
811
    WatcherOps,
812
    [ArgChoice(min=1, max=1, choices=["pause", "continue", "info"]),
813
     ArgSuggest(min=0, max=1, choices=["30m", "1h", "4h"])],
814
    [],
815
    "{pause <timespec>|continue|info}", "Change watcher properties"),
816
  'modify': (
817
    SetClusterParams, ARGS_NONE,
818
    [BACKEND_OPT, CP_SIZE_OPT, ENABLED_HV_OPT, HVLIST_OPT,
819
     NIC_PARAMS_OPT, NOLVM_STORAGE_OPT, VG_NAME_OPT, MAINTAIN_NODE_HEALTH_OPT,
820
     UIDPOOL_OPT, ADD_UIDS_OPT, REMOVE_UIDS_OPT],
821
    "[opts...]",
822
    "Alters the parameters of the cluster"),
823
  "renew-crypto": (
824
    RenewCrypto, ARGS_NONE,
825
    [NEW_CLUSTER_CERT_OPT, NEW_RAPI_CERT_OPT, RAPI_CERT_OPT,
826
     NEW_CONFD_HMAC_KEY_OPT, FORCE_OPT],
827
    "[opts...]",
828
    "Renews cluster certificates, keys and secrets"),
829
  }
830

    
831

    
832
if __name__ == '__main__':
833
  sys.exit(GenericMain(commands, override={"tag_type": constants.TAG_CLUSTER}))