gnt-cluster 24.2 KB
Newer Older
Iustin Pop's avatar
Iustin Pop committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/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.

Iustin Pop's avatar
Iustin Pop committed
21
"""Cluster related commands"""
Iustin Pop's avatar
Iustin Pop committed
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)
Iustin Pop's avatar
Iustin Pop committed
27
# C0103: Invalid name gnt-cluster
28

Iustin Pop's avatar
Iustin Pop committed
29
import sys
30
import os.path
31
import time
32
import OpenSSL
Iustin Pop's avatar
Iustin Pop committed
33
34
35

from ganeti.cli import *
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
Iustin Pop's avatar
Iustin Pop committed
42
43


44
@UsesRPC
Iustin Pop's avatar
Iustin Pop committed
45
46
47
def InitCluster(opts, args):
  """Initialize the cluster.

48
49
50
51
52
53
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the desired
      cluster name
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
54
55

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

  vg_name = opts.vg_name
  if opts.lvm_storage and not opts.vg_name:
    vg_name = constants.DEFAULT_VG

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

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

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

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

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

88
89
90
  if opts.candidate_pool_size is None:
    opts.candidate_pool_size = constants.MASTER_POOL_SIZE_DEFAULT

91
92
93
  if opts.mac_prefix is None:
    opts.mac_prefix = constants.DEFAULT_MAC_PREFIX

94
95
96
97
98
  bootstrap.InitCluster(cluster_name=args[0],
                        secondary_ip=opts.secondary_ip,
                        vg_name=vg_name,
                        mac_prefix=opts.mac_prefix,
                        master_netdev=opts.master_netdev,
99
100
101
                        file_storage_dir=opts.file_storage_dir,
                        enabled_hypervisors=hvlist,
                        hvparams=hvparams,
102
                        beparams=beparams,
103
                        nicparams=nicparams,
104
                        candidate_pool_size=opts.candidate_pool_size,
105
                        modify_etc_hosts=opts.modify_etc_hosts,
106
                        modify_ssh_setup=opts.modify_ssh_setup,
107
                        maintain_node_health=opts.maintain_node_health,
108
                        )
109
  op = opcodes.OpPostInitCluster()
110
  SubmitOpCode(op, opts=opts)
Iustin Pop's avatar
Iustin Pop committed
111
112
113
  return 0


114
@UsesRPC
Iustin Pop's avatar
Iustin Pop committed
115
116
117
def DestroyCluster(opts, args):
  """Destroy the cluster.

118
119
120
121
122
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code
123

Iustin Pop's avatar
Iustin Pop committed
124
125
  """
  if not opts.yes_do_it:
126
127
    ToStderr("Destroying a cluster is irreversible. If you really want"
             " destroy this cluster, supply the --yes-do-it option.")
Iustin Pop's avatar
Iustin Pop committed
128
129
130
    return 1

  op = opcodes.OpDestroyCluster()
131
  master = SubmitOpCode(op, opts=opts)
Iustin Pop's avatar
Iustin Pop committed
132
133
134
  # if we reached this, the opcode didn't fail; we can proceed to
  # shutdown all the daemons
  bootstrap.FinalizeClusterDestroy(master)
Iustin Pop's avatar
Iustin Pop committed
135
136
137
  return 0


138
139
140
def RenameCluster(opts, args):
  """Rename the cluster.

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

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

  op = opcodes.OpRenameCluster(name=name)
158
  SubmitOpCode(op, opts=opts)
159
160
161
  return 0


162
163
164
165
166
167
168
169
170
171
def RedistributeConfig(opts, args):
  """Forces push of the cluster configuration.

  @param opts: the command line options selected by the user
  @type args: list
  @param args: empty list
  @rtype: int
  @return: the desired exit code

  """
172
  op = opcodes.OpRedistributeConfig()
173
174
175
176
  SubmitOrSend(op, opts)
  return 0


Iustin Pop's avatar
Iustin Pop committed
177
178
179
def ShowClusterVersion(opts, args):
  """Write version of ganeti software to the standard output.

180
181
182
183
184
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
185
186

  """
187
188
  cl = GetClient()
  result = cl.QueryClusterInfo()
189
190
191
192
193
  ToStdout("Software version: %s", result["software_version"])
  ToStdout("Internode protocol: %s", result["protocol_version"])
  ToStdout("Configuration format: %s", result["config_version"])
  ToStdout("OS api version: %s", result["os_api_version"])
  ToStdout("Export interface: %s", result["export_version"])
Iustin Pop's avatar
Iustin Pop committed
194
195
196
197
198
199
  return 0


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

200
201
202
203
204
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
205
206

  """
207
208
  master = bootstrap.GetMaster()
  ToStdout(master)
Iustin Pop's avatar
Iustin Pop committed
209
210
  return 0

211

212
def _PrintGroupedParams(paramsdict, level=1):
213
214
215
216
  """Print Grouped parameters (be, nic, disk) by group.

  @type paramsdict: dict of dicts
  @param paramsdict: {group: {param: value, ...}, ...}
217
218
  @type level: int
  @param level: Level of indention
219
220

  """
221
222
223
224
225
226
227
  indent = "  " * level
  for item, val in paramsdict.items():
    if isinstance(val, dict):
      ToStdout("%s- %s:", indent, item)
      _PrintGroupedParams(val, level=level + 1)
    else:
      ToStdout("%s  %s: %s", indent, item, val)
Iustin Pop's avatar
Iustin Pop committed
228

229

Iustin Pop's avatar
Iustin Pop committed
230
231
232
def ShowClusterConfig(opts, args):
  """Shows cluster information.

233
234
235
236
237
238
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code

Iustin Pop's avatar
Iustin Pop committed
239
  """
240
241
  cl = GetClient()
  result = cl.QueryClusterInfo()
Iustin Pop's avatar
Iustin Pop committed
242

243
  ToStdout("Cluster name: %s", result["name"])
244
  ToStdout("Cluster UUID: %s", result["uuid"])
Iustin Pop's avatar
Iustin Pop committed
245

246
247
248
  ToStdout("Creation time: %s", utils.FormatTime(result["ctime"]))
  ToStdout("Modification time: %s", utils.FormatTime(result["mtime"]))

249
  ToStdout("Master node: %s", result["master"])
Iustin Pop's avatar
Iustin Pop committed
250

251
252
  ToStdout("Architecture (this node): %s (%s)",
           result["architecture"][0], result["architecture"][1])
Iustin Pop's avatar
Iustin Pop committed
253

254
  if result["tags"]:
255
    tags = utils.CommaJoin(utils.NiceSort(result["tags"]))
256
257
258
259
260
  else:
    tags = "(none)"

  ToStdout("Tags: %s", tags)

261
  ToStdout("Default hypervisor: %s", result["default_hypervisor"])
262
263
  ToStdout("Enabled hypervisors: %s",
           utils.CommaJoin(result["enabled_hypervisors"]))
264

265
  ToStdout("Hypervisor parameters:")
266
  _PrintGroupedParams(result["hvparams"])
267

268
269
270
  ToStdout("OS specific hypervisor parameters:")
  _PrintGroupedParams(result["os_hvp"])

271
  ToStdout("Cluster parameters:")
272
  ToStdout("  - candidate pool size: %s", result["candidate_pool_size"])
273
274
275
  ToStdout("  - master netdev: %s", result["master_netdev"])
  ToStdout("  - lvm volume group: %s", result["volume_group_name"])
  ToStdout("  - file storage path: %s", result["file_storage_dir"])
276
277
  ToStdout("  - maintenance of node health: %s",
           result["maintain_node_health"])
278
279

  ToStdout("Default instance parameters:")
280
281
282
283
  _PrintGroupedParams(result["beparams"])

  ToStdout("Default nic parameters:")
  _PrintGroupedParams(result["nicparams"])
284

Iustin Pop's avatar
Iustin Pop committed
285
286
287
288
289
290
  return 0


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

291
292
293
294
295
296
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the path of
      the file to be copied
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
297
298

  """
299
300
  filename = args[0]
  if not os.path.exists(filename):
301
302
    raise errors.OpPrereqError("No such filename '%s'" % filename,
                               errors.ECODE_INVAL)
303

Iustin Pop's avatar
Iustin Pop committed
304
305
306
307
  cl = GetClient()

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

308
309
  results = GetOnlineNodes(nodes=opts.nodes, cl=cl, filter_master=True,
                           secondary_ips=opts.use_replication_network)
Michael Hanselmann's avatar
Michael Hanselmann committed
310

Iustin Pop's avatar
Iustin Pop committed
311
  srun = ssh.SshRunner(cluster_name=cluster_name)
312
313
  for node in results:
    if not srun.CopyFileToNode(node, filename):
314
      ToStderr("Copy of file %s to node %s failed", filename, node)
315

Iustin Pop's avatar
Iustin Pop committed
316
317
318
319
320
321
  return 0


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

322
323
324
325
326
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain the command to be run and its arguments
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
327
328

  """
Iustin Pop's avatar
Iustin Pop committed
329
  cl = GetClient()
Michael Hanselmann's avatar
Michael Hanselmann committed
330

Iustin Pop's avatar
Iustin Pop committed
331
  command = " ".join(args)
332
333

  nodes = GetOnlineNodes(nodes=opts.nodes, cl=cl)
Iustin Pop's avatar
Iustin Pop committed
334
335
336

  cluster_name, master_node = cl.QueryConfigValues(["cluster_name",
                                                    "master_node"])
337

Iustin Pop's avatar
Iustin Pop committed
338
  srun = ssh.SshRunner(cluster_name=cluster_name)
339

Michael Hanselmann's avatar
Michael Hanselmann committed
340
  # Make sure master node is at list end
341
342
343
344
345
346
  if master_node in nodes:
    nodes.remove(master_node)
    nodes.append(master_node)

  for name in nodes:
    result = srun.Run(name, "root", command)
347
348
349
350
    ToStdout("------------------------------------------------")
    ToStdout("node: %s", name)
    ToStdout("%s", result.output)
    ToStdout("return code = %s", result.exit_code)
351
352

  return 0
Iustin Pop's avatar
Iustin Pop committed
353
354
355
356
357


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

358
359
360
361
362
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
363
364

  """
Iustin Pop's avatar
Iustin Pop committed
365
  skip_checks = []
366
367
  if opts.skip_nplusone_mem:
    skip_checks.append(constants.VERIFY_NPLUSONE_MEM)
368
369
  op = opcodes.OpVerifyCluster(skip_checks=skip_checks,
                               verbose=opts.verbose,
370
371
                               error_codes=opts.error_codes,
                               debug_simulate_errors=opts.simulate_errors)
372
  if SubmitOpCode(op, opts=opts):
373
374
375
    return 0
  else:
    return 1
Iustin Pop's avatar
Iustin Pop committed
376
377


378
379
380
def VerifyDisks(opts, args):
  """Verify integrity of cluster disks.

381
382
383
384
385
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code
386
387
388

  """
  op = opcodes.OpVerifyDisks()
389
  result = SubmitOpCode(op, opts=opts)
390
  if not isinstance(result, (list, tuple)) or len(result) != 3:
391
392
    raise errors.ProgrammerError("Unknown result type for OpVerifyDisks")

393
  bad_nodes, instances, missing = result
394

395
  retcode = constants.EXIT_SUCCESS
396

397
398
399
  if bad_nodes:
    for node, text in bad_nodes.items():
      ToStdout("Error gathering data on node %s: %s",
400
               node, utils.SafeEncode(text[-400:]))
401
      retcode |= 1
402
      ToStdout("You need to fix these nodes first before fixing instances")
403

404
405
  if instances:
    for iname in instances:
406
407
      if iname in missing:
        continue
408
409
      op = opcodes.OpActivateInstanceDisks(instance_name=iname)
      try:
410
        ToStdout("Activating disks for instance '%s'", iname)
411
        SubmitOpCode(op, opts=opts)
412
413
414
      except errors.GenericError, err:
        nret, msg = FormatError(err)
        retcode |= nret
415
        ToStderr("Error activating disks for instance %s: %s", iname, msg)
416
417
418

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

  return retcode


438
439
440
441
442
443
444
445
446
447
448
def RepairDiskSizes(opts, args):
  """Verify sizes of cluster disks.

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

  """
  op = opcodes.OpRepairDiskSizes(instances=args)
449
  SubmitOpCode(op, opts=opts)
450
451


452
@UsesRPC
Iustin Pop's avatar
Iustin Pop committed
453
454
455
456
457
458
459
def MasterFailover(opts, args):
  """Failover the master node.

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

460
461
462
463
464
465
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code

Iustin Pop's avatar
Iustin Pop committed
466
  """
467
468
469
470
471
472
473
474
475
  if opts.no_voting:
    usertext = ("This will perform the failover even if most other nodes"
                " are down, or if this node is outdated. This is dangerous"
                " as it can lead to a non-consistent cluster. Check the"
                " gnt-cluster(8) man page before proceeding. Continue?")
    if not AskUser(usertext):
      return 1

  return bootstrap.MasterFailover(no_voting=opts.no_voting)
Iustin Pop's avatar
Iustin Pop committed
476
477


Iustin Pop's avatar
Iustin Pop committed
478
479
480
def SearchTags(opts, args):
  """Searches the tags on all the cluster.

481
482
483
484
485
486
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the tag pattern
  @rtype: int
  @return: the desired exit code

Iustin Pop's avatar
Iustin Pop committed
487
488
  """
  op = opcodes.OpSearchTags(pattern=args[0])
489
  result = SubmitOpCode(op, opts=opts)
Iustin Pop's avatar
Iustin Pop committed
490
491
492
493
494
  if not result:
    return 1
  result = list(result)
  result.sort()
  for path, tag in result:
495
    ToStdout("%s %s", path, tag)
Iustin Pop's avatar
Iustin Pop committed
496
497


498
def _RenewCrypto(new_cluster_cert, new_rapi_cert, rapi_cert_filename,
499
                 new_confd_hmac_key, force):
500
501
502
503
504
505
506
507
  """Renews cluster certificates, keys and secrets.

  @type new_cluster_cert: bool
  @param new_cluster_cert: Whether to generate a new cluster certificate
  @type new_rapi_cert: bool
  @param new_rapi_cert: Whether to generate a new RAPI certificate
  @type rapi_cert_filename: string
  @param rapi_cert_filename: Path to file containing new RAPI certificate
508
509
  @type new_confd_hmac_key: bool
  @param new_confd_hmac_key: Whether to generate a new HMAC key
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
  @type force: bool
  @param force: Whether to ask user for confirmation

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

  if rapi_cert_filename:
    # Read and verify new certificate
    try:
      rapi_cert_pem = utils.ReadFile(rapi_cert_filename)

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

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

  else:
    rapi_cert_pem = None

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

  def _RenewCryptoInner(ctx):
    ctx.feedback_fn("Updating certificates and keys")
    bootstrap.GenerateClusterCrypto(new_cluster_cert, new_rapi_cert,
550
                                    new_confd_hmac_key,
551
552
553
554
555
                                    rapi_cert_pem=rapi_cert_pem)

    files_to_copy = []

    if new_cluster_cert:
556
      files_to_copy.append(constants.NODED_CERT_FILE)
557
558
559
560

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

561
562
    if new_confd_hmac_key:
      files_to_copy.append(constants.CONFD_HMAC_KEY)
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585

    if files_to_copy:
      for node_name in ctx.nonmaster_nodes:
        ctx.feedback_fn("Copying %s to %s" %
                        (", ".join(files_to_copy), node_name))
        for file_name in files_to_copy:
          ctx.ssh.CopyFileToNode(node_name, file_name)

  RunWhileClusterStopped(ToStdout, _RenewCryptoInner)

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

  return 0


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

  """
  return _RenewCrypto(opts.new_cluster_cert,
                      opts.new_rapi_cert,
                      opts.rapi_cert,
586
                      opts.new_confd_hmac_key,
587
588
589
                      opts.force)


590
591
592
def SetClusterParams(opts, args):
  """Modify the cluster.

593
594
595
596
597
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should be an empty list
  @rtype: int
  @return: the desired exit code
598
599

  """
600
601
  if not (not opts.lvm_storage or opts.vg_name or
          opts.enabled_hypervisors or opts.hvparams or
602
          opts.beparams or opts.nicparams or
603
604
          opts.candidate_pool_size is not None or
          opts.maintain_node_health is not None):
605
    ToStderr("Please give at least one of the parameters.")
606
607
608
609
    return 1

  vg_name = opts.vg_name
  if not opts.lvm_storage and opts.vg_name:
610
    ToStderr("Options --no-lvm-storage and --vg-name conflict.")
611
    return 1
612
613
614

  if not opts.lvm_storage:
    vg_name = ""
615

616
617
618
619
  hvlist = opts.enabled_hypervisors
  if hvlist is not None:
    hvlist = hvlist.split(",")

620
621
  # a list of (name, dict) we can pass directly to dict() (or [])
  hvparams = dict(opts.hvparams)
Iustin Pop's avatar
Iustin Pop committed
622
  for hv_params in hvparams.values():
623
    utils.ForceDictType(hv_params, constants.HVS_PARAMETER_TYPES)
624
625

  beparams = opts.beparams
626
  utils.ForceDictType(beparams, constants.BES_PARAMETER_TYPES)
627

628
629
630
  nicparams = opts.nicparams
  utils.ForceDictType(nicparams, constants.NICS_PARAMETER_TYPES)

631
632
  mnh = opts.maintain_node_health

633
  op = opcodes.OpSetClusterParams(vg_name=vg_name,
634
635
                                  enabled_hypervisors=hvlist,
                                  hvparams=hvparams,
636
                                  os_hvp=None,
637
                                  beparams=beparams,
638
                                  nicparams=nicparams,
639
640
                                  candidate_pool_size=opts.candidate_pool_size,
                                  maintain_node_health=mnh)
641
  SubmitOpCode(op, opts=opts)
642
643
644
  return 0


645
646
647
def QueueOps(opts, args):
  """Queue operations.

648
649
650
651
652
653
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the subcommand
  @rtype: int
  @return: the desired exit code

654
655
656
657
658
659
660
661
662
  """
  command = args[0]
  client = GetClient()
  if command in ("drain", "undrain"):
    drain_flag = command == "drain"
    client.SetQueueDrainFlag(drain_flag)
  elif command == "info":
    result = client.QueryConfigValues(["drain_flag"])
    if result[0]:
663
      val = "set"
664
    else:
665
666
      val = "unset"
    ToStdout("The drain flag is %s" % val)
667
  else:
668
669
    raise errors.OpPrereqError("Command '%s' is not valid." % command,
                               errors.ECODE_INVAL)
670

671
672
  return 0

673

674
675
676
677
678
679
680
def _ShowWatcherPause(until):
  if until is None or until < time.time():
    ToStdout("The watcher is not paused.")
  else:
    ToStdout("The watcher is paused until %s.", time.ctime(until))


681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
def WatcherOps(opts, args):
  """Watcher operations.

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

  """
  command = args[0]
  client = GetClient()

  if command == "continue":
    client.SetWatcherPause(None)
696
    ToStdout("The watcher is no longer paused.")
697
698
699

  elif command == "pause":
    if len(args) < 2:
700
      raise errors.OpPrereqError("Missing pause duration", errors.ECODE_INVAL)
701

702
703
    result = client.SetWatcherPause(time.time() + ParseTimespec(args[1]))
    _ShowWatcherPause(result)
704
705
706

  elif command == "info":
    result = client.QueryConfigValues(["watcher_pause"])
707
    _ShowWatcherPause(result[0])
708
709

  else:
710
711
    raise errors.OpPrereqError("Command '%s' is not valid." % command,
                               errors.ECODE_INVAL)
712
713
714
715

  return 0


Iustin Pop's avatar
Iustin Pop committed
716
commands = {
717
718
  'init': (
    InitCluster, [ArgHost(min=1, max=1)],
719
    [BACKEND_OPT, CP_SIZE_OPT, ENABLED_HV_OPT, GLOBAL_FILEDIR_OPT,
720
     HVLIST_OPT, MAC_PREFIX_OPT, MASTER_NETDEV_OPT, NIC_PARAMS_OPT,
721
     NOLVM_STORAGE_OPT, NOMODIFY_ETCHOSTS_OPT, NOMODIFY_SSH_SETUP_OPT,
722
     SECONDARY_IP_OPT, VG_NAME_OPT, MAINTAIN_NODE_HEALTH_OPT],
723
724
    "[opts...] <cluster_name>", "Initialises a new cluster configuration"),
  'destroy': (
725
    DestroyCluster, ARGS_NONE, [YES_DOIT_OPT],
726
727
728
    "", "Destroy cluster"),
  'rename': (
    RenameCluster, [ArgHost(min=1, max=1)],
729
    [FORCE_OPT],
730
731
732
    "<new_name>",
    "Renames the cluster"),
  'redist-conf': (
733
    RedistributeConfig, ARGS_NONE, [SUBMIT_OPT],
734
735
736
737
    "", "Forces a push of the configuration file and ssconf files"
    " to the nodes in the cluster"),
  'verify': (
    VerifyCluster, ARGS_NONE,
738
    [VERBOSE_OPT, DEBUG_SIMERR_OPT, ERROR_CODES_OPT, NONPLUS1_OPT],
739
740
    "", "Does a check on the cluster configuration"),
  'verify-disks': (
741
    VerifyDisks, ARGS_NONE, [],
742
743
    "", "Does a check on the cluster disk status"),
  'repair-disk-sizes': (
744
    RepairDiskSizes, ARGS_MANY_INSTANCES, [],
745
746
    "", "Updates mismatches in recorded disk sizes"),
  'masterfailover': (
747
    MasterFailover, ARGS_NONE, [NOVOTING_OPT],
748
749
    "", "Makes the current node the master"),
  'version': (
750
    ShowClusterVersion, ARGS_NONE, [],
751
752
    "", "Shows the cluster version"),
  'getmaster': (
753
    ShowClusterMaster, ARGS_NONE, [],
754
755
756
    "", "Shows the cluster master"),
  'copyfile': (
    ClusterCopyFile, [ArgFile(min=1, max=1)],
757
    [NODE_LIST_OPT, USE_REPL_NET_OPT],
758
759
760
    "[-n node...] <filename>", "Copies a file to all (or only some) nodes"),
  'command': (
    RunClusterCommand, [ArgCommand(min=1)],
761
    [NODE_LIST_OPT],
762
763
    "[-n node...] <command>", "Runs a command on all (or only some) nodes"),
  'info': (
764
    ShowClusterConfig, ARGS_NONE, [],
765
766
    "", "Show cluster configuration"),
  'list-tags': (
767
    ListTags, ARGS_NONE, [], "", "List the tags of the cluster"),
768
  'add-tags': (
769
    AddTags, [ArgUnknown()], [TAG_SRC_OPT],
770
771
    "tag...", "Add tags to the cluster"),
  'remove-tags': (
772
    RemoveTags, [ArgUnknown()], [TAG_SRC_OPT],
773
774
775
    "tag...", "Remove tags from the cluster"),
  'search-tags': (
    SearchTags, [ArgUnknown(min=1, max=1)],
776
    [], "", "Searches the tags on all objects on"
777
778
779
780
    " the cluster for a given pattern (regex)"),
  'queue': (
    QueueOps,
    [ArgChoice(min=1, max=1, choices=["drain", "undrain", "info"])],
781
    [], "drain|undrain|info", "Change queue properties"),
782
783
784
785
  'watcher': (
    WatcherOps,
    [ArgChoice(min=1, max=1, choices=["pause", "continue", "info"]),
     ArgSuggest(min=0, max=1, choices=["30m", "1h", "4h"])],
786
    [],
787
788
789
    "{pause <timespec>|continue|info}", "Change watcher properties"),
  'modify': (
    SetClusterParams, ARGS_NONE,
790
    [BACKEND_OPT, CP_SIZE_OPT, ENABLED_HV_OPT, HVLIST_OPT,
791
     NIC_PARAMS_OPT, NOLVM_STORAGE_OPT, VG_NAME_OPT, MAINTAIN_NODE_HEALTH_OPT],
792
793
    "[opts...]",
    "Alters the parameters of the cluster"),
794
795
  "renew-crypto": (
    RenewCrypto, ARGS_NONE,
796
797
    [NEW_CLUSTER_CERT_OPT, NEW_RAPI_CERT_OPT, RAPI_CERT_OPT,
     NEW_CONFD_HMAC_KEY_OPT, FORCE_OPT],
798
799
    "[opts...]",
    "Renews cluster certificates, keys and secrets"),
Iustin Pop's avatar
Iustin Pop committed
800
801
  }

802

Iustin Pop's avatar
Iustin Pop committed
803
if __name__ == '__main__':
804
  sys.exit(GenericMain(commands, override={"tag_type": constants.TAG_CLUSTER}))