RegionSizeStore getRegionSizeStore() { return rs.getRegionServerSpaceQuotaManager().getRegionSizeStore(); }
/** * Creates an HRegionServer using the given Configuration. */ private HRegionServer mockRegionServer(Configuration conf) { final HRegionServer rs = mock(HRegionServer.class); final RegionServerSpaceQuotaManager quotaManager = mock(RegionServerSpaceQuotaManager.class); when(rs.getConfiguration()).thenReturn(conf); when(rs.getRegionServerSpaceQuotaManager()).thenReturn(quotaManager); when(quotaManager.getRegionSizeStore()).thenReturn(new RegionSizeStoreImpl()); return rs; }
@VisibleForTesting void replaceStoreFiles(Collection<HStoreFile> compactedFiles, Collection<HStoreFile> result) throws IOException { this.lock.writeLock().lock(); try { this.storeEngine.getStoreFileManager().addCompactionResults(compactedFiles, result); synchronized (filesCompacting) { filesCompacting.removeAll(compactedFiles); } // These may be null when the RS is shutting down. The space quota Chores will fix the Region // sizes later so it's not super-critical if we miss these. RegionServerServices rsServices = region.getRegionServerServices(); if (rsServices != null && rsServices.getRegionServerSpaceQuotaManager() != null) { updateSpaceQuotaAfterFileReplacement( rsServices.getRegionServerSpaceQuotaManager().getRegionSizeStore(), getRegionInfo(), compactedFiles, result); } } finally { this.lock.writeLock().unlock(); } }
void _chore() { final RegionServerSpaceQuotaManager quotaManager = rsServices.getRegionServerSpaceQuotaManager(); // Get the HRegionInfo for each online region HashSet<RegionInfo> onlineRegionInfos = getOnlineRegionInfos(rsServices.getRegions()); RegionSizeStore store = quotaManager.getRegionSizeStore(); // Remove all sizes for non-online regions removeNonOnlineRegions(store, onlineRegionInfos); rsServices.reportRegionSizesForQuotas(store); }
RegionServerSpaceQuotaManager quotaManager = rsServices.getRegionServerSpaceQuotaManager(); if (quotaManager != null) { quotaManager.getRegionSizeStore().incrementRegionSize( this.getRegionInfo(), flushedOutputFileSize);
getSpaceQuotaManager().getRegionSizeStore().incrementRegionSize( region.getRegionInfo(), sizeToBeLoaded);