Adding blkiotune command to virsh tool
Signed-off-by: Gui Jianfeng <guijianfeng(a)cn.fujitsu.com>
---
tools/virsh.c | 127 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 127 insertions(+), 0 deletions(-)
diff --git a/tools/virsh.c b/tools/virsh.c
index ce275f1..5d6e06e 100644
--- a/tools/virsh.c
+++ b/tools/virsh.c
@@ -3010,6 +3010,132 @@ cmdSetmaxmem(vshControl *ctl, const vshCmd *cmd)
}
/*
+ * "blkiotune" command
+ */
+static const vshCmdInfo info_blkiotune[] = {
+ {"help", N_("Get or set blkio parameters")},
+ {"desc", N_("Get or set the current blkio parameters for a guest"
\
+ " domain.\n" \
+ " To get the blkio parameters use following command: \n\n"
\
+ " virsh # blkiotune <domain>")},
+ {NULL, NULL}
+};
+
+static const vshCmdOptDef opts_blkiotune[] = {
+ {"domain", VSH_OT_DATA, VSH_OFLAG_REQ, N_("domain name, id or
uuid")},
+ {"weight", VSH_OT_INT, VSH_OFLAG_NONE,
+ N_("IO Weight in range [100, 1000]")},
+ {NULL, 0, 0, NULL}
+};
+
+static int
+cmdBlkiotune(vshControl * ctl, const vshCmd * cmd)
+{
+ virDomainPtr dom;
+ int weight;
+ int nparams = 0;
+ unsigned int i = 0;
+ virBlkioParameterPtr params = NULL, temp = NULL;
+ int ret = FALSE;
+ int found;
+
+ if (!vshConnectionUsability(ctl, ctl->conn))
+ return FALSE;
+
+ if (!(dom = vshCommandOptDomain(ctl, cmd, NULL)))
+ return FALSE;
+
+ weight = vshCommandOptInt(cmd, "weight", &found);
+ if (found) {
+ nparams++;
+ if (weight < 0) {
+ virDomainFree(dom);
+ vshError(ctl, _("Invalid value of %d for I/O weight"), weight);
+ return FALSE;
+ }
+ }
+
+ if (nparams == 0) {
+ /* get the number of blkio parameters */
+ if (virDomainGetBlkioParameters(dom, NULL, &nparams, 0) != 0) {
+ vshError(ctl, "%s",
+ _("Unable to get number of blkio parameters"));
+ goto cleanup;
+ }
+
+ if (nparams == 0) {
+ /* nothing to output */
+ ret = TRUE;
+ goto cleanup;
+ }
+
+ /* now go get all the blkio parameters */
+ params = vshCalloc(ctl, nparams, sizeof(*params));
+ if (virDomainGetBlkioParameters(dom, params, &nparams, 0) != 0) {
+ vshError(ctl, "%s", _("Unable to get blkio
parameters"));
+ goto cleanup;
+ }
+
+ for (i = 0; i < nparams; i++) {
+ switch (params[i].type) {
+ case VIR_DOMAIN_BLKIO_PARAM_INT:
+ vshPrint(ctl, "%-15s: %d\n", params[i].field,
+ params[i].value.i);
+ break;
+ case VIR_DOMAIN_BLKIO_PARAM_UINT:
+ vshPrint(ctl, "%-15s: %u\n", params[i].field,
+ params[i].value.ui);
+ break;
+ case VIR_DOMAIN_BLKIO_PARAM_LLONG:
+ vshPrint(ctl, "%-15s: %lld\n", params[i].field,
+ params[i].value.l);
+ break;
+ case VIR_DOMAIN_BLKIO_PARAM_ULLONG:
+ vshPrint(ctl, "%-15s: %llu\n", params[i].field,
+ params[i].value.ul);
+ break;
+ case VIR_DOMAIN_BLKIO_PARAM_DOUBLE:
+ vshPrint(ctl, "%-15s: %f\n", params[i].field,
+ params[i].value.d);
+ break;
+ case VIR_DOMAIN_BLKIO_PARAM_BOOLEAN:
+ vshPrint(ctl, "%-15s: %d\n", params[i].field,
+ params[i].value.b);
+ break;
+ default:
+ vshPrint(ctl, "unimplemented blkio parameter type\n");
+ }
+ }
+
+ ret = TRUE;
+ } else {
+ /* set the blkio parameters */
+ params = vshCalloc(ctl, nparams, sizeof(*params));
+
+ for (i = 0; i < nparams; i++) {
+ temp = ¶ms[i];
+ temp->type = VIR_DOMAIN_BLKIO_PARAM_UINT;
+
+ if (weight) {
+ temp->value.ui = (unsigned int)weight;
+ strncpy(temp->field, VIR_DOMAIN_BLKIO_WEIGHT,
+ sizeof(temp->field));
+ weight = 0;
+ }
+ }
+ if (virDomainSetBlkioParameters(dom, params, nparams, 0) != 0)
+ vshError(ctl, "%s", _("Unable to change blkio
parameters"));
+ else
+ ret = TRUE;
+ }
+
+ cleanup:
+ VIR_FREE(params);
+ virDomainFree(dom);
+ return ret;
+}
+
+/*
* "memtune" command
*/
static const vshCmdInfo info_memtune[] = {
@@ -10324,6 +10450,7 @@ static const vshCmdDef domManagementCmds[] = {
{"managedsave", cmdManagedSave, opts_managedsave, info_managedsave},
{"managedsave-remove", cmdManagedSaveRemove, opts_managedsaveremove,
info_managedsaveremove},
{"maxvcpus", cmdMaxvcpus, opts_maxvcpus, info_maxvcpus},
+ {"blkiotune", cmdBlkiotune, opts_blkiotune, info_blkiotune},
{"memtune", cmdMemtune, opts_memtune, info_memtune},
{"migrate", cmdMigrate, opts_migrate, info_migrate},
{"migrate-setmaxdowntime", cmdMigrateSetMaxDowntime,
opts_migrate_setmaxdowntime, info_migrate_setmaxdowntime},
--
1.7.1