Convert DataSize for presto-spark

Author: shangm2

presto-spark-base/src/main/java/com/facebook/presto/spark/PrestoSparkQueryExecutionFactory.java

@@ -156,7 +156,6 @@
 import static com.google.common.base.Ticker.systemTicker;
 import static com.google.common.base.Verify.verify;
 import static com.google.common.collect.ImmutableList.toImmutableList;
-import static io.airlift.units.DataSize.succinctBytes;
 import static java.lang.Math.max;
 import static java.lang.String.format;
 import static java.nio.charset.StandardCharsets.UTF_8;
@@ -333,11 +332,11 @@ public static QueryInfo createQueryInfo(
                 rootStage,
                 allStages,
                 peakRunningTasks,
-                succinctBytes(peakUserMemoryReservationInBytes),
-                succinctBytes(peakTotalMemoryReservationInBytes),
-                succinctBytes(peakTaskUserMemoryInBytes),
-                succinctBytes(peakTaskTotalMemoryInBytes),
-                succinctBytes(peakNodeTotalMemoryInBytes),
+                peakUserMemoryReservationInBytes,
+                peakTotalMemoryReservationInBytes,
+                peakTaskUserMemoryInBytes,
+                peakTaskTotalMemoryInBytes,
+                peakNodeTotalMemoryInBytes,
                 session.getRuntimeStats());
 
         Optional<PrestoSparkExecutionContext> prestoSparkExecutionContext = Optional.empty();
@@ -427,8 +426,8 @@ public static StageInfo createStageInfo(QueryId queryId, SubPlan plan, ListMulti
                 DateTime.now(),
                 new Distribution().snapshot(),
                 new RuntimeStats(),
-                succinctBytes(peakUserMemoryReservationInBytes),
-                succinctBytes(peakNodeTotalMemoryReservationInBytes),
+                peakUserMemoryReservationInBytes,
+                peakNodeTotalMemoryReservationInBytes,
                 1,
                 1);
         return new StageInfo(

presto-spark-base/src/main/java/com/facebook/presto/spark/node/PrestoSparkTaskManager.java

@@ -29,7 +29,6 @@
 import com.facebook.presto.metadata.MetadataUpdates;
 import com.facebook.presto.sql.planner.PlanFragment;
 import com.google.common.util.concurrent.ListenableFuture;
-import io.airlift.units.DataSize;
 
 import java.util.List;
 import java.util.Optional;
@@ -98,7 +97,7 @@ public TaskInfo abortTask(TaskId taskId)
     }
 
     @Override
-    public ListenableFuture<BufferResult> getTaskResults(TaskId taskId, OutputBuffers.OutputBufferId bufferId, long startingSequenceId, DataSize maxSize)
+    public ListenableFuture<BufferResult> getTaskResults(TaskId taskId, OutputBuffers.OutputBufferId bufferId, long startingSequenceId, long maxSizeInBytes)
     {
         throw new UnsupportedOperationException();
     }