public String getPrefixedOverflowTableName(TableReference tableRef) { return overflowTablePrefix + DbKvs.internalTableName(tableRef); }
public String getPrefixedTableName(TableReference tableRef) { return tablePrefix + DbKvs.internalTableName(tableRef); }
private String getSimpleRowSelectOneQueryPostgres( TableReference tableRef, String extraWhere, String order, ConnectionSupplier conns) { return String.format( SIMPLE_ROW_SELECT_ONE_POSTGRES_TEMPLATE, DbKvs.internalTableName(tableRef), getPrefixedTableName(tableRef, conns), extraWhere, order); }
public String get(TableReference tableRef) { return config.tablePrefix() + DbKvs.internalTableName(tableRef); } }
private String prefixedTableName() { return config.tablePrefix() + DbKvs.internalTableName(tableName); }
public String getShortPrefixedTableName( ConnectionSupplier connectionSupplier, String tablePrefix, TableReference tableRef) { Preconditions.checkState(tablePrefix.length() <= AtlasDbConstants.MAX_TABLE_PREFIX_LENGTH, "The tablePrefix can be at most %s characters long", AtlasDbConstants.MAX_TABLE_PREFIX_LENGTH); TableReference shortenedNamespaceTableRef = truncateNamespace(tableRef); String prefixedTableName = tablePrefix + DbKvs.internalTableName(shortenedNamespaceTableRef); String truncatedTableName = truncate(prefixedTableName, NAMESPACED_TABLE_NAME_LENGTH); String fullTableName = tablePrefix + DbKvs.internalTableName(tableRef); return truncatedTableName + "_" + getTableNumber(connectionSupplier, fullTableName, truncatedTableName); }
private String getSimpleRowSelectOneQueryOracle( TableReference tableRef, String minMax, String extraWhere, ConnectionSupplier conns) { return String.format( SIMPLE_ROW_SELECT_ONE_ORACLE_TEMPLATE, DbKvs.internalTableName(tableRef), getPrimaryKeyConstraintName(tableRef, conns), minMax, getPrefixedTableName(tableRef, conns), extraWhere); }
@Override public Iterator<List<CellTsPairInfo>> createPageIterator(TableReference tableRef, CandidateCellForSweepingRequest request) { return new PageIterator( connectionPool, request, Math.max(1, request.batchSizeHint().orElse(DEFAULT_BATCH_SIZE)), DbKvs.internalTableName(tableRef), prefixedTableNames.get(tableRef), request.startRowInclusive()); }
@Override public DbReadTable createRead(TableReference tableRef, ConnectionSupplier conns) { return new DbReadTable( conns, new PostgresQueryFactory(DbKvs.internalTableName(tableRef), config)); }
@SuppressWarnings("checkstyle:NestedTryDepth") public String getShortTableNameFromMappingTable( ConnectionSupplier connectionSupplier, String tablePrefix, TableReference tableRef) throws TableMappingNotFoundException { String fullTableName = tablePrefix + DbKvs.internalTableName(tableRef); try { return unmappingCache.get(fullTableName, () -> { SqlConnection conn = connectionSupplier.get(); AgnosticResultSet results = conn.selectResultSetUnregisteredQuery( "SELECT short_table_name " + "FROM " + AtlasDbConstants.ORACLE_NAME_MAPPING_TABLE + " WHERE table_name = ?", fullTableName); if (results.size() == 0) { throw new TableMappingNotFoundException( "The table " + fullTableName + " does not have a mapping." + "This might be because the table does not exist."); } return Iterables.getOnlyElement(results.rows()).getString("short_table_name"); }); } catch (ExecutionException e) { throw new TableMappingNotFoundException(e.getCause()); } }
String query = String.format( SIMPLE_ROW_SELECT_TEMPLATE, DbKvs.internalTableName(tableRef), getPrimaryKeyConstraintName(tableRef, conns), getPrefixedTableName(tableRef, conns),
private void assertThatVacuumWasNotPerformed(SqlConnection sqlConnection) { assertFalse(postgresDdlTable.shouldRunCompaction()); postgresDdlTable.compactInternally(false); verify(sqlConnection, times(2)).selectResultSetUnregisteredQuery(startsWith("SELECT FLOOR"), any()); verify(sqlConnection, never()).executeUnregisteredQuery( eq("VACUUM ANALYZE " + DbKvs.internalTableName(TEST_TABLE))); } }
private void assertThatVacuumWasPerformed(SqlConnection sqlConnection, boolean assertThatTimestampsWereChecked) { assertTrue(postgresDdlTable.shouldRunCompaction()); postgresDdlTable.compactInternally(false); if (assertThatTimestampsWereChecked) { verify(sqlConnection, times(2)).selectResultSetUnregisteredQuery(startsWith("SELECT FLOOR"), any()); } verify(sqlConnection).executeUnregisteredQuery(eq("VACUUM ANALYZE " + DbKvs.internalTableName(TEST_TABLE))); }
@Override public Iterator<RowResult<Value>> getRange(TableReference tableRef, RangeRequest rangeRequest, long timestamp) { int maxRowsPerPage = RangeHelpers.getMaxRowsPerPage(rangeRequest); int cellsPerRowEstimate = getCellsPerRowEstimate(tableRef, rangeRequest); int maxCellsPerPage = Math.min( AtlasDbPerformanceConstants.MAX_BATCH_SIZE, maxRowsPerPage * cellsPerRowEstimate) + 1; String tableName = DbKvs.internalTableName(tableRef); Iterator<Iterator<RowResult<Value>>> pageIterator = new PageIterator( rangeRequest.getStartInclusive(), rangeRequest.getEndExclusive(), rangeRequest.getColumnNames(), rangeRequest.isReverse(), timestamp, maxRowsPerPage, maxCellsPerPage, tableName, prefixedTableNames.get(tableRef)); return Iterators.concat(pageIterator); }
private String getSimpleRowSelectOneQueryPostgres( TableReference tableRef, String extraWhere, String order, ConnectionSupplier conns) { return String.format( SIMPLE_ROW_SELECT_ONE_POSTGRES_TEMPLATE, DbKvs.internalTableName(tableRef), getPrefixedTableName(tableRef, conns), extraWhere, order); }
public String get(TableReference tableRef) { return config.tablePrefix() + DbKvs.internalTableName(tableRef); } }
public String getShortPrefixedTableName( ConnectionSupplier connectionSupplier, String tablePrefix, TableReference tableRef) { Preconditions.checkState(tablePrefix.length() <= AtlasDbConstants.MAX_TABLE_PREFIX_LENGTH, "The tablePrefix can be at most %s characters long", AtlasDbConstants.MAX_TABLE_PREFIX_LENGTH); TableReference shortenedNamespaceTableRef = truncateNamespace(tableRef); String prefixedTableName = tablePrefix + DbKvs.internalTableName(shortenedNamespaceTableRef); String truncatedTableName = truncate(prefixedTableName, NAMESPACED_TABLE_NAME_LENGTH); String fullTableName = tablePrefix + DbKvs.internalTableName(tableRef); return truncatedTableName + "_" + getTableNumber(connectionSupplier, fullTableName, truncatedTableName); }
private String getSimpleRowSelectOneQueryOracle( TableReference tableRef, String minMax, String extraWhere, ConnectionSupplier conns) { return String.format( SIMPLE_ROW_SELECT_ONE_ORACLE_TEMPLATE, DbKvs.internalTableName(tableRef), getPrimaryKeyConstraintName(tableRef, conns), minMax, getPrefixedTableName(tableRef, conns), extraWhere); }
@Override public Iterator<List<CellTsPairInfo>> createPageIterator(TableReference tableRef, CandidateCellForSweepingRequest request) { return new PageIterator( connectionPool, request, Math.max(1, request.batchSizeHint().orElse(DEFAULT_BATCH_SIZE)), DbKvs.internalTableName(tableRef), prefixedTableNames.get(tableRef), request.startRowInclusive()); }
@Override public DbReadTable createRead(TableReference tableRef, ConnectionSupplier conns) { return new DbReadTable( conns, new PostgresQueryFactory(DbKvs.internalTableName(tableRef), config)); }