From b66bf0730a328620ccb08d5a94612abcf50b5c65 Mon Sep 17 00:00:00 2001
From: Nikolay Shirokovskiy <nshirokovskiy@virtuozzo.com>
Date: Mon, 12 Dec 2016 10:56:38 +0300
Subject: [PATCH] vz: add state group to all domain stats

---
 src/vz/vz_driver.c | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/vz/vz_driver.c b/src/vz/vz_driver.c
index b518c827d2..59bfb7a53d 100644
--- a/src/vz/vz_driver.c
+++ b/src/vz/vz_driver.c
@@ -3739,6 +3739,28 @@ vzDomainGetBalloonStats(virDomainObjPtr dom,
     return 0;
 }
 
+static int
+vzDomainGetStateStats(virDomainObjPtr dom,
+                      virDomainStatsRecordPtr record,
+                      int *maxparams)
+{
+    if (virTypedParamsAddInt(&record->params,
+                             &record->nparams,
+                             maxparams,
+                             "state.state",
+                             dom->state.state) < 0)
+        return -1;
+
+    if (virTypedParamsAddInt(&record->params,
+                             &record->nparams,
+                             maxparams,
+                             "state.reason",
+                             dom->state.reason) < 0)
+        return -1;
+
+    return 0;
+}
+
 static virDomainStatsRecordPtr
 vzDomainGetAllStats(virConnectPtr conn,
                     virDomainObjPtr dom)
@@ -3749,6 +3771,9 @@ vzDomainGetAllStats(virConnectPtr conn,
     if (VIR_ALLOC(stat) < 0)
         return NULL;
 
+    if (vzDomainGetStateStats(dom, stat, &maxparams) < 0)
+        goto error;
+
     if (vzDomainGetBlockStats(dom, stat, &maxparams) < 0)
         goto error;