io.prestosql.memory.QueryContext.getMemoryPool()方法的使用及代码示例

x33g5p2x  于2022-01-29 转载在 其他  
字(3.6k)|赞(0)|评价(0)|浏览(76)

本文整理了Java中io.prestosql.memory.QueryContext.getMemoryPool方法的一些代码示例,展示了QueryContext.getMemoryPool的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。QueryContext.getMemoryPool方法的具体详情如下:
包路径:io.prestosql.memory.QueryContext
类名称:QueryContext
方法名:getMemoryPool

QueryContext.getMemoryPool介绍

暂无

代码示例

代码示例来源:origin: prestosql/presto

private long getMemoryAlreadyBeingRevoked(Collection<SqlTask> sqlTasks, MemoryPool memoryPool)
{
  return sqlTasks.stream()
      .filter(task -> task.getTaskStatus().getState() == TaskState.RUNNING)
      .filter(task -> task.getQueryContext().getMemoryPool() == memoryPool)
      .mapToLong(task -> task.getQueryContext().accept(new TraversingQueryContextVisitor<Void, Long>()
      {
        @Override
        public Long visitOperatorContext(OperatorContext operatorContext, Void context)
        {
          if (operatorContext.isMemoryRevokingRequested()) {
            return operatorContext.getReservedRevocableBytes();
          }
          return 0L;
        }
        @Override
        public Long mergeResults(List<Long> childrenResults)
        {
          return childrenResults.stream()
              .mapToLong(i -> i).sum();
        }
      }, null))
      .sum();
}

代码示例来源:origin: io.prestosql/presto-main

private long getMemoryAlreadyBeingRevoked(Collection<SqlTask> sqlTasks, MemoryPool memoryPool)
{
  return sqlTasks.stream()
      .filter(task -> task.getTaskStatus().getState() == TaskState.RUNNING)
      .filter(task -> task.getQueryContext().getMemoryPool() == memoryPool)
      .mapToLong(task -> task.getQueryContext().accept(new TraversingQueryContextVisitor<Void, Long>()
      {
        @Override
        public Long visitOperatorContext(OperatorContext operatorContext, Void context)
        {
          if (operatorContext.isMemoryRevokingRequested()) {
            return operatorContext.getReservedRevocableBytes();
          }
          return 0L;
        }
        @Override
        public Long mergeResults(List<Long> childrenResults)
        {
          return childrenResults.stream()
              .mapToLong(i -> i).sum();
        }
      }, null))
      .sum();
}

代码示例来源:origin: io.prestosql/presto-main

sqlTasks.stream()
    .filter(task -> task.getTaskStatus().getState() == TaskState.RUNNING)
    .filter(task -> task.getQueryContext().getMemoryPool() == memoryPool)
    .sorted(ORDER_BY_CREATE_TIME)
    .forEach(task -> task.getQueryContext().accept(new VoidTraversingQueryContextVisitor<AtomicLong>()

代码示例来源:origin: prestosql/presto

sqlTasks.stream()
    .filter(task -> task.getTaskStatus().getState() == TaskState.RUNNING)
    .filter(task -> task.getQueryContext().getMemoryPool() == memoryPool)
    .sorted(ORDER_BY_CREATE_TIME)
    .forEach(task -> task.getQueryContext().accept(new VoidTraversingQueryContextVisitor<AtomicLong>()

代码示例来源:origin: prestosql/presto

@BeforeMethod
public void setUp()
{
  executor = newCachedThreadPool(daemonThreadsNamed("test-executor-%s"));
  scheduledExecutor = newScheduledThreadPool(2, daemonThreadsNamed("test-scheduledExecutor-%s"));
  TaskContext taskContext = TestingTaskContext.builder(executor, scheduledExecutor, TEST_SESSION)
      .setQueryMaxMemory(DataSize.valueOf("100MB"))
      .setMemoryPoolSize(DataSize.valueOf("10B"))
      .setQueryId(QUERY_ID)
      .build();
  memoryPool = taskContext.getQueryContext().getMemoryPool();
  driverContext = taskContext
      .addPipelineContext(0, true, true, false)
      .addDriverContext();
}

代码示例来源:origin: io.prestosql/presto-main

@BeforeMethod
public void setUp()
{
  executor = newCachedThreadPool(daemonThreadsNamed("test-executor-%s"));
  scheduledExecutor = newScheduledThreadPool(2, daemonThreadsNamed("test-scheduledExecutor-%s"));
  TaskContext taskContext = TestingTaskContext.builder(executor, scheduledExecutor, TEST_SESSION)
      .setQueryMaxMemory(DataSize.valueOf("100MB"))
      .setMemoryPoolSize(DataSize.valueOf("10B"))
      .setQueryId(QUERY_ID)
      .build();
  memoryPool = taskContext.getQueryContext().getMemoryPool();
  driverContext = taskContext
      .addPipelineContext(0, true, true, false)
      .addDriverContext();
}

相关文章

微信公众号

最新文章

更多