@Override protected void updateEntityForRemove(Quota entity, Guid id) { // since we're mocking remove using update, we'll throw 404 if the object isn't found boolean throw404 = false; // global cluster (has same id as quota) if (entity.getId().equals(id)) { if (entity.getGlobalQuotaCluster() == null) { throw404 = true; } else { entity.setGlobalQuotaCluster(null); } // specific cluster (has same id as cluster) } else { if (entity.getQuotaClusters() != null) { Iterator<QuotaCluster> iterator = entity.getQuotaClusters().iterator(); while (iterator.hasNext()) { if (iterator.next().getClusterId().equals(id)) { iterator.remove(); return; } } throw404 = true; } } if (throw404) { throw new WebApplicationException(Response.Status.NOT_FOUND); } } }
/** * Returns initialized entity with quota result set. */ private RowMapper<Quota> getQuotaFromResultSet() { return (rs, rowNum) -> { Quota entity = getQuotaMetaDataFromResultSet(rs); // Check if memory size is not null, this is an indication if global limitation for vds group exists or // not, since global limitation must be for all the quota vds group parameters. if (rs.getObject("mem_size_mb") != null) { // Set global vds group quota. QuotaCluster clusterEntity = new QuotaCluster(); clusterEntity.setMemSizeMB((Long) rs.getObject("mem_size_mb")); clusterEntity.setMemSizeMBUsage((Long) rs.getObject("mem_size_mb_usage")); clusterEntity.setVirtualCpu((Integer) rs.getObject("virtual_cpu")); clusterEntity.setVirtualCpuUsage((Integer) rs.getObject("virtual_cpu_usage")); entity.setGlobalQuotaCluster(clusterEntity); } // Check if storage limit size is not null, this is an indication if global limitation for storage // exists or // not. if (rs.getObject("storage_size_gb") != null) { // Set global storage quota. QuotaStorage storageEntity = new QuotaStorage(); storageEntity.setStorageSizeGB((Long) rs.getObject("storage_size_gb")); storageEntity.setStorageSizeGBUsage((Double) rs.getObject("storage_size_gb_usage")); entity.setGlobalQuotaStorage(storageEntity); } return entity; }; }
if (quota != null) { if (quotaCluster.getClusterId() == null || quotaCluster.getClusterId().equals(Guid.Empty)) { quota.setGlobalQuotaCluster(quotaCluster); } else { if (quota.getQuotaClusters() == null) {
@Test public void testListGlobalLimit() { Quota quota = getQuota(); quota.setGlobalQuotaCluster(getClusterGlobalCpuLimit()); setUpGetEntityExpectations(quota); QuotaClusterLimits clusterLimits = collection.list(); assertClusterLimitsFound(clusterLimits, 1); QuotaClusterLimit clusterLimit = clusterLimits.getQuotaClusterLimits().get(0); assertEquals(20, clusterLimit.getVcpuLimit().intValue()); }
@Mapping(from = QuotaClusterLimit.class, to = org.ovirt.engine.core.common.businessentities.Quota.class) public static org.ovirt.engine.core.common.businessentities.Quota map(QuotaClusterLimit model, org.ovirt.engine.core.common.businessentities.Quota template) { org.ovirt.engine.core.common.businessentities.Quota entity = template != null ? template : new org.ovirt.engine.core.common.businessentities.Quota(); QuotaCluster quotaCluster = new QuotaCluster(); if (model.isSetVcpuLimit()) { quotaCluster.setVirtualCpu(model.getVcpuLimit()); } if (model.isSetMemoryLimit()) { double limit = model.getMemoryLimit(); quotaCluster.setMemSizeMB( (limit < 0) ? -1 : (long) (limit * 1024) ); } // specific cluster if (model.isSetCluster() && model.getCluster().isSetId()) { quotaCluster.setClusterId(GuidUtils.asGuid(model.getCluster().getId())); entity.getQuotaClusters().add(quotaCluster); } else { // global entity.setGlobalQuotaCluster(quotaCluster); } return entity; } }
private static void setQuotaGlobalLimitations(Quota quota) { QuotaStorage quotaStorage = new QuotaStorage(); QuotaCluster quotaCluster = new QuotaCluster(); // Set Quota storage capacity definition. quotaStorage.setStorageSizeGB(10000L); quotaStorage.setStorageSizeGBUsage(0d); // Set Quota cluster virtual memory definition. quotaCluster.setMemSizeMB(16000000L); quotaCluster.setMemSizeMBUsage(0L); // Set Quota cluster virtual CPU definition. quotaCluster.setVirtualCpu(2880); quotaCluster.setVirtualCpuUsage(0); quota.setGlobalQuotaStorage(quotaStorage); quota.setGlobalQuotaCluster(quotaCluster); }
QuotaCluster quotaCluster; for (QuotaCluster iter : model.getQuotaClusters().getItems()) { quota.setGlobalQuotaCluster(new QuotaCluster()); quota.getGlobalQuotaCluster().setMemSizeMB(iter.getMemSizeMB()); quota.getGlobalQuotaCluster().setVirtualCpu(iter.getVirtualCpu()); quota.setGlobalQuotaCluster(null); ArrayList<QuotaCluster> quotaClusterList = new ArrayList<>(); QuotaCluster quotaCluster;