Show count of Active domains on grafana.

Author: Sina Kashipazha

engine/schema/src/main/java/com/cloud/user/dao/AccountDao.java

@@ -78,4 +78,5 @@ public interface AccountDao extends GenericDao<AccountVO, Long> {
      */
     long getDomainIdForGivenAccountId(long id);
 
+    int getActiveDomains();
 }

engine/schema/src/main/java/com/cloud/user/dao/AccountDaoImpl.java

@@ -28,6 +28,7 @@
 import com.cloud.utils.db.GenericSearchBuilder;
 import com.cloud.utils.db.SearchBuilder;
 import com.cloud.utils.db.SearchCriteria;
+import com.cloud.utils.db.SearchCriteria.Func;
 import com.cloud.utils.db.SearchCriteria.Op;
 import org.apache.commons.lang3.StringUtils;
 import com.cloud.utils.db.TransactionLegacy;
@@ -54,6 +55,7 @@ public class AccountDaoImpl extends GenericDaoBase<AccountVO, Long> implements A
     protected final SearchBuilder<AccountVO> NonProjectAccountSearch;
     protected final SearchBuilder<AccountVO> AccountByRoleSearch;
     protected final GenericSearchBuilder<AccountVO, Long> AccountIdsSearch;
+    protected final GenericSearchBuilder<AccountVO, Long> ActiveDomainCount;
 
     public AccountDaoImpl() {
         AllFieldsSearch = createSearchBuilder();
@@ -101,6 +103,13 @@ public AccountDaoImpl() {
         AccountByRoleSearch = createSearchBuilder();
         AccountByRoleSearch.and("roleId", AccountByRoleSearch.entity().getRoleId(), SearchCriteria.Op.EQ);
         AccountByRoleSearch.done();
+
+        ActiveDomainCount = createSearchBuilder(Long.class);
+        ActiveDomainCount.select(null, Func.COUNT, null);
+        ActiveDomainCount.and("domain", ActiveDomainCount.entity().getDomainId(), SearchCriteria.Op.EQ);
+        ActiveDomainCount.and("state", ActiveDomainCount.entity().getState(), SearchCriteria.Op.EQ);
+        ActiveDomainCount.groupBy(ActiveDomainCount.entity().getDomainId());
+        ActiveDomainCount.done();
     }
 
     @Override
@@ -318,5 +327,10 @@ public long getDomainIdForGivenAccountId(long id) {
         }
     }
 
-
+    @Override
+    public int getActiveDomains() {
+        SearchCriteria<Long> sc = ActiveDomainCount.create();
+        sc.setParameters("state", "enabled");
+        return customSearch(sc, null).size();
+    }
 }

plugins/integrations/prometheus/src/main/java/org/apache/cloudstack/metrics/PrometheusExporterImpl.java

@@ -54,7 +54,6 @@
 import com.cloud.host.HostVO;
 import com.cloud.host.Status;
 import com.cloud.host.dao.HostDao;
-import com.cloud.host.dao.HostTagsDao;
 import com.cloud.network.dao.IPAddressDao;
 import com.cloud.storage.ImageStore;
 import com.cloud.storage.StorageStats;
@@ -66,6 +65,7 @@
 import com.cloud.utils.component.Manager;
 import com.cloud.utils.component.ManagerBase;
 import com.cloud.vm.VirtualMachine.State;
+import com.cloud.vm.dao.UserVmDao;
 import com.cloud.vm.dao.VMInstanceDao;
 
 public class PrometheusExporterImpl extends ManagerBase implements PrometheusExporter, Manager {
@@ -89,6 +89,8 @@ public class PrometheusExporterImpl extends ManagerBase implements PrometheusExp
     @Inject
     private VMInstanceDao vmDao;
     @Inject
+    private UserVmDao uservmDao;
+    @Inject
     private VolumeDao volumeDao;
     @Inject
     private IPAddressDao publicIpAddressDao;
@@ -395,6 +397,10 @@ private void addDomainResourceCount(final List<Item> metricsList) {
         }
     }
 
+    private void addDomainMetrics(final List<Item> metricsList, final String zoneName, final String zoneUuid) {
+        metricsList.add(new ItemActiveDomains(zoneName, zoneUuid, _accountDao.getActiveDomains()));
+    }
+
     @Override
     public void updateMetrics() {
         final List<Item> latestMetricsItems = new ArrayList<Item>();
@@ -409,6 +415,7 @@ public void updateMetrics() {
                 addStorageMetrics(latestMetricsItems, dc.getId(), zoneName, zoneUuid);
                 addIpAddressMetrics(latestMetricsItems, dc.getId(), zoneName, zoneUuid);
                 addVlanMetrics(latestMetricsItems, dc.getId(), zoneName, zoneUuid);
+                addDomainMetrics(latestMetricsItems, zoneName, zoneUuid);
             }
             addDomainLimits(latestMetricsItems);
             addDomainResourceCount(latestMetricsItems);
@@ -834,6 +841,24 @@ public String toMetricsString() {
 
     }
 
+    class ItemActiveDomains extends Item {
+        String zoneName;
+        String zoneUuid;
+        int total;
+
+        public ItemActiveDomains(final String zn, final String zu, final int cnt) {
+            super("cloudstack_active_domains_total");
+            zoneName = zn;
+            zoneUuid = zu;
+            total = cnt;
+        }
+
+        @Override
+        public String toMetricsString() {
+            return String.format("%s{zone=\"%s\"} %d", name, zoneName, total);
+        }
+    }
+
     class ItemHostDedicatedToAccount extends Item {
         String zoneName;
         String hostName;