Revert DataSize change for public-facing API (prestodb#24682)

* Revert API change for QueryStats

* Revert related tests

* Revert UI change

* Revert presto-spark-base change

Author: shangm2

presto-main/src/main/java/com/facebook/presto/event/QueryMonitor.java

@@ -436,21 +436,21 @@ private QueryStatistics createQueryStatistics(QueryInfo queryInfo)
                 Optional.of(ofMillis(queryStats.getAnalysisTime().toMillis())),
                 ofMillis(queryStats.getExecutionTime().toMillis()),
                 queryStats.getPeakRunningTasks(),
-                queryStats.getPeakUserMemoryReservationInBytes(),
-                queryStats.getPeakTotalMemoryReservationInBytes(),
-                queryStats.getPeakTaskUserMemoryInBytes(),
-                queryStats.getPeakTaskTotalMemoryInBytes(),
-                queryStats.getPeakNodeTotalMemoryInBytes(),
-                queryStats.getShuffledDataSizeInBytes(),
+                queryStats.getPeakUserMemoryReservation().toBytes(),
+                queryStats.getPeakTotalMemoryReservation().toBytes(),
+                queryStats.getPeakTaskUserMemory().toBytes(),
+                queryStats.getPeakTaskTotalMemory().toBytes(),
+                queryStats.getPeakNodeTotalMemory().toBytes(),
+                queryStats.getShuffledDataSize().toBytes(),
                 queryStats.getShuffledPositions(),
-                queryStats.getRawInputDataSizeInBytes(),
+                queryStats.getRawInputDataSize().toBytes(),
                 queryStats.getRawInputPositions(),
-                queryStats.getOutputDataSizeInBytes(),
+                queryStats.getOutputDataSize().toBytes(),
                 queryStats.getOutputPositions(),
-                queryStats.getWrittenOutputLogicalDataSizeInBytes(),
+                queryStats.getWrittenOutputLogicalDataSize().toBytes(),
                 queryStats.getWrittenOutputPositions(),
-                queryStats.getWrittenIntermediatePhysicalDataSizeInBytes(),
-                queryStats.getSpilledDataSizeInBytes(),
+                queryStats.getWrittenIntermediatePhysicalDataSize().toBytes(),
+                queryStats.getSpilledDataSize().toBytes(),
                 queryStats.getCumulativeUserMemory(),
                 queryStats.getCumulativeTotalMemory(),
                 queryStats.getCompletedDrivers(),

presto-main/src/main/java/com/facebook/presto/execution/QueryStateMachine.java

@@ -1304,33 +1304,33 @@ private static QueryStats pruneQueryStats(QueryStats queryStats)
                 queryStats.getCompletedDrivers(),
                 queryStats.getCumulativeUserMemory(),
                 queryStats.getCumulativeTotalMemory(),
-                queryStats.getUserMemoryReservationInBytes(),
-                queryStats.getTotalMemoryReservationInBytes(),
-                queryStats.getPeakUserMemoryReservationInBytes(),
-                queryStats.getPeakTotalMemoryReservationInBytes(),
-                queryStats.getPeakTaskUserMemoryInBytes(),
-                queryStats.getPeakTaskTotalMemoryInBytes(),
-                queryStats.getPeakNodeTotalMemoryInBytes(),
+                queryStats.getUserMemoryReservation(),
+                queryStats.getTotalMemoryReservation(),
+                queryStats.getPeakUserMemoryReservation(),
+                queryStats.getPeakTotalMemoryReservation(),
+                queryStats.getPeakTaskUserMemory(),
+                queryStats.getPeakTaskTotalMemory(),
+                queryStats.getPeakNodeTotalMemory(),
                 queryStats.isScheduled(),
                 queryStats.getTotalScheduledTime(),
                 queryStats.getTotalCpuTime(),
                 queryStats.getRetriedCpuTime(),
                 queryStats.getTotalBlockedTime(),
                 queryStats.isFullyBlocked(),
                 queryStats.getBlockedReasons(),
-                queryStats.getTotalAllocationInBytes(),
-                queryStats.getRawInputDataSizeInBytes(),
+                queryStats.getTotalAllocation(),
+                queryStats.getRawInputDataSize(),
                 queryStats.getRawInputPositions(),
-                queryStats.getProcessedInputDataSizeInBytes(),
+                queryStats.getProcessedInputDataSize(),
                 queryStats.getProcessedInputPositions(),
-                queryStats.getShuffledDataSizeInBytes(),
+                queryStats.getShuffledDataSize(),
                 queryStats.getShuffledPositions(),
-                queryStats.getOutputDataSizeInBytes(),
+                queryStats.getOutputDataSize(),
                 queryStats.getOutputPositions(),
                 queryStats.getWrittenOutputPositions(),
-                queryStats.getWrittenOutputLogicalDataSizeInBytes(),
-                queryStats.getWrittenOutputPhysicalDataSizeInBytes(),
-                queryStats.getWrittenIntermediatePhysicalDataSizeInBytes(),
+                queryStats.getWrittenOutputLogicalDataSize(),
+                queryStats.getWrittenOutputPhysicalDataSize(),
+                queryStats.getWrittenIntermediatePhysicalDataSize(),
                 queryStats.getStageGcStatistics(),
                 ImmutableList.of(), // Remove the operator summaries as OperatorInfo (especially ExchangeClientStatus) can hold onto a large amount of memory
                 queryStats.getRuntimeStats());