presto-main/src/main/java/io/prestosql/execution/SqlQueryManager.java/SqlQueryManager/cancelStage
class SqlQueryManager:    @Override
    public void cancelStage(StageId stageId)
    {
        requireNonNull(stageId, "stageId is null");

        log.debug("Cancel stage %s", stageId);

        queryTracker.tryGetQuery(stageId.getQueryId())
                .ifPresent(query -> query.cancelStage(stageId));
    }