@Override public DataSize getTotalMemoryReservation() { // acquire reference to scheduler before checking finalQueryInfo, because // state change listener sets finalQueryInfo and then clears scheduler when // the query finishes. SqlQueryScheduler scheduler = queryScheduler.get(); Optional<QueryInfo> finalQueryInfo = stateMachine.getFinalQueryInfo(); if (finalQueryInfo.isPresent()) { return finalQueryInfo.get().getQueryStats().getTotalMemoryReservation(); } if (scheduler == null) { return new DataSize(0, BYTE); } return succinctBytes(scheduler.getTotalMemoryReservation()); }
assertEquals(actual.getTotalMemoryReservation(), new DataSize(19, BYTE)); assertEquals(actual.getPeakUserMemoryReservation(), new DataSize(20, BYTE)); assertEquals(actual.getPeakTotalMemoryReservation(), new DataSize(21, BYTE));
public BasicQueryStats(QueryStats queryStats) { this(queryStats.getCreateTime(), queryStats.getEndTime(), queryStats.getQueuedTime(), queryStats.getElapsedTime(), queryStats.getExecutionTime(), queryStats.getTotalDrivers(), queryStats.getQueuedDrivers(), queryStats.getRunningDrivers(), queryStats.getCompletedDrivers(), queryStats.getRawInputDataSize(), queryStats.getRawInputPositions(), queryStats.getCumulativeUserMemory(), queryStats.getUserMemoryReservation(), queryStats.getTotalMemoryReservation(), queryStats.getPeakUserMemoryReservation(), queryStats.getTotalCpuTime(), queryStats.getTotalScheduledTime(), queryStats.isFullyBlocked(), queryStats.getBlockedReasons(), queryStats.getProgressPercentage()); }
queryStats.getCumulativeUserMemory(), queryStats.getUserMemoryReservation(), queryStats.getTotalMemoryReservation(), queryStats.getPeakUserMemoryReservation(), queryStats.getPeakTotalMemoryReservation(),
@Override public long getTotalMemoryReservation() { // acquire reference to outputStage before checking finalQueryInfo, because // state change listener sets finalQueryInfo and then clears outputStage when // the query finishes. SqlQueryScheduler scheduler = queryScheduler.get(); QueryInfo queryInfo = finalQueryInfo.get(); if (queryInfo != null) { return queryInfo.getQueryStats().getTotalMemoryReservation().toBytes(); } return scheduler.getTotalMemoryReservation(); }
assertEquals(actual.getTotalMemoryReservation(), new DataSize(18, BYTE)); assertEquals(actual.getPeakMemoryReservation(), new DataSize(19, BYTE));