Examples with QueryMemoryInfo com.facebook.presto.memory.LowMemoryKiller.QueryMemoryInfo used on opensource projects

Example 1 with QueryMemoryInfo

use of com.facebook.presto.memory.LowMemoryKiller.QueryMemoryInfo in project presto by prestodb.

the class ClusterMemoryManager method callOomKiller.

private synchronized void callOomKiller(Iterable<QueryExecution> runningQueries) {
    List<QueryMemoryInfo> queryMemoryInfoList = Streams.stream(runningQueries).map(this::createQueryMemoryInfo).collect(toImmutableList());
    List<MemoryInfo> nodeMemoryInfos = nodes.values().stream().map(RemoteNodeMemory::getInfo).filter(Optional::isPresent).map(Optional::get).collect(toImmutableList());
    Optional<QueryId> chosenQueryId = lowMemoryKiller.chooseQueryToKill(queryMemoryInfoList, nodeMemoryInfos);
    if (chosenQueryId.isPresent()) {
        log.debug("Low memory killer chose %s", chosenQueryId.get());
        Optional<QueryExecution> chosenQuery = Streams.stream(runningQueries).filter(query -> chosenQueryId.get().equals(query.getQueryId())).collect(toOptional());
        if (chosenQuery.isPresent()) {
            // See comments in  isLastKilledQueryGone for why chosenQuery might be absent.
            chosenQuery.get().fail(new PrestoException(CLUSTER_OUT_OF_MEMORY, "Query killed because the cluster is out of memory. Please try again in a few minutes."));
            queriesKilledDueToOutOfMemory.incrementAndGet();
            lastKilledQuery = chosenQueryId.get();
            logQueryKill(chosenQueryId.get(), nodeMemoryInfos);
        }
    }
}