private boolean shouldDenyPrivilege(String userName, String entityName, TestingPrivilegeType type) { TestingPrivilege testPrivilege = privilege(userName, entityName, type); for (TestingPrivilege denyPrivilege : denyPrivileges) { if (denyPrivilege.matches(testPrivilege)) { return true; } } return false; }
@Override protected List<Driver> createDrivers(TaskContext taskContext) { return localQueryRunner.createDrivers(session, query, new NullOutputFactory(), taskContext); } }
public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, TaskStateMachine taskStateMachine) { return builder(notificationExecutor, yieldExecutor, session) .setTaskStateMachine(taskStateMachine) .build(); }
public static MaterializedResult toMaterializedResult(ConnectorSession session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
public synchronized Builder rows(Object[][] rows) { for (Object[] row : rows) { row(row); } return this; }
@Override public void close() { runner.close(); } }
public static TaskContext createTaskContext(Executor notificationExecutor, ScheduledExecutorService yieldExecutor, Session session, DataSize maxMemory) { return builder(notificationExecutor, yieldExecutor, session) .setQueryMaxMemory(maxMemory) .build(); }
public synchronized Builder pages(Iterable<Page> pages) { for (Page page : pages) { this.page(page); } return this; }
@Override public MaterializedResultWithPlan executeWithPlan(Session session, String sql, WarningCollector warningCollector) { return inTransaction(session, transactionSession -> executeInternal(transactionSession, sql)); }
@Override public Session getSession() { return queryRunner.getDefaultSession(); }
@Override public OperatorFactory duplicate() { return new NullOutputOperatorFactory(operatorId, planNodeId); } }
@Override public WarningCollector create() { return new TestingWarningCollector(config, testConfig); } }
private static SqlTimestamp intToSqlTimestamp(Integer input) { if (input == null) { return null; } return sqlTimestampOf(input, SESSION); }
@Override public void close() { localQueryRunner.close(); } }
@Override public ConnectorHandleResolver getHandleResolver() { return new TestingHandleResolver(); }
@Override public ConnectorOutputTableHandle beginCreateTable(ConnectorSession session, ConnectorTableMetadata tableMetadata, Optional<ConnectorNewTableLayout> layout) { createTable(session, tableMetadata, false); return TestingHandle.INSTANCE; }
@Override public ConnectorPageSinkProvider getPageSinkProvider() { return new TestingPageSinkProvider(); } }
public static MaterializedResult toMaterializedResult(Session session, List<Type> types, List<Page> pages) { // materialize pages MaterializedResult.Builder resultBuilder = MaterializedResult.resultBuilder(session, types); for (Page outputPage : pages) { resultBuilder.page(outputPage); } return resultBuilder.build(); }
@Override public OperatorFactory createOutputOperator(int operatorId, PlanNodeId planNodeId, List<Type> types, Function<Page, Page> pagePreprocessor, PagesSerdeFactory serdeFactory) { return new NullOutputOperatorFactory(operatorId, planNodeId); } }