io.prestosql.server.BasicQueryInfo.getQueryStats java code examples

private static boolean isLeaked(Map<QueryId, BasicQueryInfo> queryIdToInfo, QueryId queryId)
{
  BasicQueryInfo queryInfo = queryIdToInfo.get(queryId);
  if (queryInfo == null) {
    return true;
  }
  DateTime queryEndTime = queryInfo.getQueryStats().getEndTime();
  if (queryInfo.getState() == RUNNING || queryEndTime == null) {
    return false;
  }
  return secondsBetween(queryEndTime, now()).getSeconds() >= DEFAULT_LEAK_CLAIM_DELTA_SEC;
}

private static void logQueryTimeline(BasicQueryInfo queryInfo)
{
  DateTime queryStartTime = queryInfo.getQueryStats().getCreateTime();
  DateTime queryEndTime = queryInfo.getQueryStats().getEndTime();
  // query didn't finish cleanly
  if (queryStartTime == null || queryEndTime == null) {
    return;
  }
  long elapsed = max(queryEndTime.getMillis() - queryStartTime.getMillis(), 0);
  logQueryTimeline(
      queryInfo.getQueryId(),
      queryInfo.getSession().getTransactionId().map(TransactionId::toString).orElse(""),
      elapsed,
      elapsed,
      0,
      0,
      0,
      queryStartTime,
      queryEndTime);
}

private static boolean isLeaked(Map<QueryId, BasicQueryInfo> queryIdToInfo, QueryId queryId)
{
  BasicQueryInfo queryInfo = queryIdToInfo.get(queryId);
  if (queryInfo == null) {
    return true;
  }
  DateTime queryEndTime = queryInfo.getQueryStats().getEndTime();
  if (queryInfo.getState() == RUNNING || queryEndTime == null) {
    return false;
  }
  return secondsBetween(queryEndTime, now()).getSeconds() >= DEFAULT_LEAK_CLAIM_DELTA_SEC;
}

public static QueryStateInfo createQueryStateInfo(BasicQueryInfo queryInfo, Optional<ResourceGroupId> group)
{
  Optional<QueryProgressStats> progress = Optional.empty();
  if (!queryInfo.getState().isDone() && queryInfo.getState() != QUEUED) {
    progress = Optional.of(createQueryProgressStats(queryInfo.getQueryStats()));
  }
  return createQueryStateInfo(queryInfo, group, Optional.empty(), progress);
}

public static QueryStateInfo createQueryStateInfo(BasicQueryInfo queryInfo, Optional<ResourceGroupId> group)
{
  Optional<QueryProgressStats> progress = Optional.empty();
  if (!queryInfo.getState().isDone() && queryInfo.getState() != QUEUED) {
    progress = Optional.of(createQueryProgressStats(queryInfo.getQueryStats()));
  }
  return createQueryStateInfo(queryInfo, group, Optional.empty(), progress);
}

private static void logQueryTimeline(BasicQueryInfo queryInfo)
{
  DateTime queryStartTime = queryInfo.getQueryStats().getCreateTime();
  DateTime queryEndTime = queryInfo.getQueryStats().getEndTime();
  // query didn't finish cleanly
  if (queryStartTime == null || queryEndTime == null) {
    return;
  }
  long elapsed = max(queryEndTime.getMillis() - queryStartTime.getMillis(), 0);
  logQueryTimeline(
      queryInfo.getQueryId(),
      queryInfo.getSession().getTransactionId().map(TransactionId::toString).orElse(""),
      elapsed,
      elapsed,
      0,
      0,
      0,
      queryStartTime,
      queryEndTime);
}

createQueryFailureInfo(failure, Optional.empty()),
ImmutableList.of(),
ofEpochMilli(queryInfo.getQueryStats().getCreateTime().getMillis()),
ofEpochMilli(queryInfo.getQueryStats().getCreateTime().getMillis()),
ofEpochMilli(queryInfo.getQueryStats().getEndTime().getMillis())));

createQueryFailureInfo(failure, Optional.empty()),
ImmutableList.of(),
ofEpochMilli(queryInfo.getQueryStats().getCreateTime().getMillis()),
ofEpochMilli(queryInfo.getQueryStats().getCreateTime().getMillis()),
ofEpochMilli(queryInfo.getQueryStats().getEndTime().getMillis())));

public void queryCreatedEvent(BasicQueryInfo queryInfo)
{
  eventListenerManager.queryCreated(
      new QueryCreatedEvent(
          queryInfo.getQueryStats().getCreateTime().toDate().toInstant(),
          createQueryContext(queryInfo.getSession(), queryInfo.getResourceGroupId()),
          new QueryMetadata(
              queryInfo.getQueryId().toString(),
              queryInfo.getSession().getTransactionId().map(TransactionId::toString),
              queryInfo.getQuery(),
              QUEUED.toString(),
              queryInfo.getSelf(),
              Optional.empty(),
              Optional.empty())));
}

public void queryCreatedEvent(BasicQueryInfo queryInfo)
{
  eventListenerManager.queryCreated(
      new QueryCreatedEvent(
          queryInfo.getQueryStats().getCreateTime().toDate().toInstant(),
          createQueryContext(queryInfo.getSession(), queryInfo.getResourceGroupId()),
          new QueryMetadata(
              queryInfo.getQueryId().toString(),
              queryInfo.getSession().getTransactionId().map(TransactionId::toString),
              queryInfo.getQuery(),
              QUEUED.toString(),
              queryInfo.getSelf(),
              Optional.empty(),
              Optional.empty())));
}

private static boolean isBlockedWaitingForMemory(BasicQueryInfo info)
{
  BasicQueryStats stats = info.getQueryStats();
  boolean isWaitingForMemory = stats.getBlockedReasons().contains(WAITING_FOR_MEMORY);
  if (!isWaitingForMemory) {
    return false;
  }
  // queries are not marked as fully blocked if there are no running drivers
  return stats.isFullyBlocked() || stats.getRunningDrivers() == 0;
}

if (query.getQueryStats().isFullyBlocked()) {
  blockedQueries++;
totalInputBytes += query.getQueryStats().getRawInputDataSize().toBytes();
totalInputRows += query.getQueryStats().getRawInputPositions();
totalCpuTimeSecs += query.getQueryStats().getTotalCpuTime().getValue(SECONDS);
memoryReservation += query.getQueryStats().getUserMemoryReservation().toBytes();
runningDrivers += query.getQueryStats().getRunningDrivers();

if (query.getQueryStats().isFullyBlocked()) {
  blockedQueries++;
totalInputBytes += query.getQueryStats().getRawInputDataSize().toBytes();
totalInputRows += query.getQueryStats().getRawInputPositions();
totalCpuTimeSecs += query.getQueryStats().getTotalCpuTime().getValue(SECONDS);
memoryReservation += query.getQueryStats().getUserMemoryReservation().toBytes();
runningDrivers += query.getQueryStats().getRunningDrivers();

assertEquals(basicInfo.getQuery(), "SELECT 4");
assertEquals(basicInfo.getQueryStats().getCreateTime(), DateTime.parse("1991-09-06T05:00-05:30"));
assertEquals(basicInfo.getQueryStats().getEndTime(), DateTime.parse("1991-09-06T06:00-05:30"));
assertEquals(basicInfo.getQueryStats().getElapsedTime(), Duration.valueOf("8m"));
assertEquals(basicInfo.getQueryStats().getExecutionTime(), Duration.valueOf("44m"));
assertEquals(basicInfo.getQueryStats().getTotalDrivers(), 16);
assertEquals(basicInfo.getQueryStats().getQueuedDrivers(), 17);
assertEquals(basicInfo.getQueryStats().getRunningDrivers(), 18);
assertEquals(basicInfo.getQueryStats().getCompletedDrivers(), 19);
assertEquals(basicInfo.getQueryStats().getCumulativeUserMemory(), 20.0);
assertEquals(basicInfo.getQueryStats().getUserMemoryReservation(), DataSize.valueOf("21GB"));
assertEquals(basicInfo.getQueryStats().getPeakUserMemoryReservation(), DataSize.valueOf("23GB"));
assertEquals(basicInfo.getQueryStats().getTotalCpuTime(), Duration.valueOf("24m"));
assertEquals(basicInfo.getQueryStats().isFullyBlocked(), true);
assertEquals(basicInfo.getQueryStats().getBlockedReasons(), ImmutableSet.of(BlockedReason.WAITING_FOR_MEMORY));
assertEquals(basicInfo.getQueryStats().getProgressPercentage(), OptionalDouble.of(100));

assertEquals(basicInfo.getQuery(), "SELECT 4");
assertEquals(basicInfo.getQueryStats().getCreateTime(), DateTime.parse("1991-09-06T05:00-05:30"));
assertEquals(basicInfo.getQueryStats().getEndTime(), DateTime.parse("1991-09-06T06:00-05:30"));
assertEquals(basicInfo.getQueryStats().getElapsedTime(), Duration.valueOf("8m"));
assertEquals(basicInfo.getQueryStats().getExecutionTime(), Duration.valueOf("44m"));
assertEquals(basicInfo.getQueryStats().getTotalDrivers(), 16);
assertEquals(basicInfo.getQueryStats().getQueuedDrivers(), 17);
assertEquals(basicInfo.getQueryStats().getRunningDrivers(), 18);
assertEquals(basicInfo.getQueryStats().getCompletedDrivers(), 19);
assertEquals(basicInfo.getQueryStats().getCumulativeUserMemory(), 20.0);
assertEquals(basicInfo.getQueryStats().getUserMemoryReservation(), DataSize.valueOf("21GB"));
assertEquals(basicInfo.getQueryStats().getPeakUserMemoryReservation(), DataSize.valueOf("23GB"));
assertEquals(basicInfo.getQueryStats().getTotalCpuTime(), Duration.valueOf("24m"));
assertEquals(basicInfo.getQueryStats().isFullyBlocked(), true);
assertEquals(basicInfo.getQueryStats().getBlockedReasons(), ImmutableSet.of(BlockedReason.WAITING_FOR_MEMORY));
assertEquals(basicInfo.getQueryStats().getProgressPercentage(), OptionalDouble.of(100));

private static QueryStateInfo createQueryStateInfo(
    BasicQueryInfo queryInfo,
    Optional<ResourceGroupId> groupId,
    Optional<List<ResourceGroupInfo>> pathToRoot,
    Optional<QueryProgressStats> progress)
{
  return new QueryStateInfo(
      queryInfo.getQueryId(),
      queryInfo.getState(),
      groupId,
      queryInfo.getQuery(),
      queryInfo.getQueryStats().getCreateTime(),
      queryInfo.getSession().getUser(),
      queryInfo.getSession().getSource(),
      queryInfo.getSession().getClientInfo(),
      queryInfo.getSession().getCatalog(),
      queryInfo.getSession().getSchema(),
      pathToRoot,
      progress);
}

private static QueryStateInfo createQueryStateInfo(
    BasicQueryInfo queryInfo,
    Optional<ResourceGroupId> groupId,
    Optional<List<ResourceGroupInfo>> pathToRoot,
    Optional<QueryProgressStats> progress)
{
  return new QueryStateInfo(
      queryInfo.getQueryId(),
      queryInfo.getState(),
      groupId,
      queryInfo.getQuery(),
      queryInfo.getQueryStats().getCreateTime(),
      queryInfo.getSession().getUser(),
      queryInfo.getSession().getSource(),
      queryInfo.getSession().getClientInfo(),
      queryInfo.getSession().getCatalog(),
      queryInfo.getSession().getSchema(),
      pathToRoot,
      progress);
}

How to use getQueryStatsmethodin io.prestosql.server.BasicQueryInfo

Best Java code snippets using io.prestosql.server.BasicQueryInfo.getQueryStats (Showing top 17 results out of 315)

How to use
getQueryStats
method
in
io.prestosql.server.BasicQueryInfo