From 5fc7f882d6f8ff963ef793a0c74b539f4f0e57c4 Mon Sep 17 00:00:00 2001
From: Kamil Endruszkiewicz <kamil.endruszkiewicz@starburstdata.com>
Date: Thu, 27 Oct 2022 08:46:56 +0200
Subject: [PATCH] Fix EventDrivenTaskSource for record split time

---
 .../io/trino/execution/scheduler/EventDrivenTaskSource.java   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/core/trino-main/src/main/java/io/trino/execution/scheduler/EventDrivenTaskSource.java b/core/trino-main/src/main/java/io/trino/execution/scheduler/EventDrivenTaskSource.java
index d6ffa62e943f..98c28d40b47b 100644
--- a/core/trino-main/src/main/java/io/trino/execution/scheduler/EventDrivenTaskSource.java
+++ b/core/trino-main/src/main/java/io/trino/execution/scheduler/EventDrivenTaskSource.java
@@ -464,7 +464,7 @@ private synchronized void processNext()
                 return;
             }
             verify(splitLoadingFuture == null || splitLoadingFuture.isDone(), "splitLoadingFuture is still running");
-            long start = System.currentTimeMillis();
+            long start = System.nanoTime();
             splitLoadingFuture = splitSource.getNextBatch(splitBatchSize);
             Futures.addCallback(splitLoadingFuture, new FutureCallback<>()
             {
@@ -472,7 +472,7 @@ private synchronized void processNext()
                 public void onSuccess(SplitBatch result)
                 {
                     try {
-                        getSplitTimeRecorder.accept(System.currentTimeMillis() - start);
+                        getSplitTimeRecorder.accept(start);
                         ListMultimap<Integer, Split> splits = result.getSplits().stream()
                                 .collect(toImmutableListMultimap(splitToPartition::applyAsInt, Function.identity()));
                         callback.update(splits, result.isLastBatch());