public TableDefinition getTableDefinition(TableReference tableRef) { return tableDefinitions.get(tableRef.getTablename()); }
public IndexDefinition getIndex(TableReference indexRef) { return indexDefinitions.get(indexRef.getTablename()); }
public static boolean isHidden(TableReference tableReference) { return CASSANDRA_TABLES.contains(tableReference) || tableReference.getTablename().startsWith(AtlasDbConstants.LOCK_TABLE_PREFIX); } }
public Map<String, String> getTableNameTagFor(@Nullable TableReference tableRef) { String tableName = tableRef == null ? "unknown" : tableRef.getTablename(); if (!isSafeToLog.test(tableRef)) { tableName = "unsafeTable"; } return ImmutableMap.of("tableName", tableName); }
public static final byte[] getBytesForTableRef(TableReference tableRef) { byte[] nameSpace = EncodingUtils.encodeVarString(tableRef.getNamespace().getName()); byte[] table = PtBytes.toBytes(tableRef.getTablename()); return Bytes.concat(nameSpace, table); }
@Test // If you are intentionally making Table API changes, please manually regenerate the ApiTestSchema // and copy the new files to the ${EXPECTED_FILES_FOLDER_PATH} folder. public void checkAgainstAccidentalTableAPIChanges() throws IOException { // TODO (amarzoca): Add tests for schemas that use more of the rendering features (Triggers, StreamStores, etc) Schema schema = ApiTestSchema.getSchema(); schema.renderTables(testFolder.getRoot()); List<String> generatedTestTables = ApiTestSchema.getSchema().getAllTables().stream() .map(entry -> entry.getTablename() + "Table") .collect(Collectors.toList()); checkIfFilesAreTheSame(generatedTestTables); }
private TableReference truncateNamespace(TableReference tableRef) { if (tableRef.getNamespace().isEmptyNamespace()) { return tableRef; } String namespace = tableRef.getNamespace().getName(); namespace = truncate(namespace, MAX_NAMESPACE_LENGTH); return TableReference.create(Namespace.create(namespace), tableRef.getTablename()); }
@Test public void createLowerCasedWithoutNamespace() { String upperBar = "BAR"; TableReference upper = TableReference.createWithEmptyNamespace(upperBar); TableReference lower = TableReference.createLowerCased(upper); assertEquals(lower.getNamespace(), Namespace.EMPTY_NAMESPACE); assertEquals(lower.getTablename(), upperBar.toLowerCase()); }
@Test public void canSpecifyTableNameAsLoggable() { TableDefinition definition = new TableDefinition() {{ tableNameLogSafety(LogSafety.SAFE); javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING); noColumns(); }}; assertThat(definition.toTableMetadata().getNameLogSafety()).isEqualTo(LogSafety.SAFE); }
@Test public void createLowerCasedWithNamespace() { String upperFoo = "FOO"; String upperBar = "BAR"; TableReference upper = TableReference.create(Namespace.create(upperFoo), upperBar); TableReference lower = TableReference.createLowerCased(upper); assertEquals(lower.getNamespace().getName(), upperFoo.toLowerCase()); assertEquals(lower.getTablename(), upperBar.toLowerCase()); }
private TableDefinition getSimpleTableDefinition(TableReference tableRef) { return new TableDefinition() {{ javaTableName(tableRef.getTablename()); rowName(); rowComponent("rowName", ValueType.STRING); columns(); column("col1", "1", ValueType.VAR_LONG); }}; } }
@SuppressWarnings({"checkstyle:Indentation", "checkstyle:RightCurly"}) private TableDefinition getSimpleTableDefinition(TableReference tableRef) { return new TableDefinition() {{ javaTableName(tableRef.getTablename()); rowName(); rowComponent("rowName", ValueType.STRING); columns(); column("col1", "1", ValueType.VAR_LONG); }}; }
@SuppressWarnings({"checkstyle:Indentation", "checkstyle:RightCurly"}) private Schema getSchema() { Schema schema = new Schema(checkpointTable.getNamespace()); schema.addTableDefinition(checkpointTable.getTablename(), new TableDefinition() {{ rowName(); rowComponent("table_name", ValueType.VAR_STRING); rowComponent("range_id", ValueType.VAR_LONG); columns(); column("start", SHORT_COLUMN_NAME, ValueType.BLOB); rangeScanAllowed(); conflictHandler(ConflictHandler.IGNORE_ALL); }}); return schema; } }
@Test public void canSpecifyEverythingAsSafe() { TableDefinition definition = new TableDefinition() {{ allSafeForLoggingByDefault(); javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING, LogSafety.UNSAFE); columns(); column(COLUMN_NAME, COLUMN_SHORTNAME, ValueType.VAR_LONG); }}; assertThat(definition.toTableMetadata().getNameLogSafety()).isEqualTo(LogSafety.SAFE); assertRowComponentSafety(definition, LogSafety.UNSAFE); assertNamedColumnSafety(definition, LogSafety.SAFE); }
@Test public void canDeclareEverythingAsSafeByDefaultAndSpecificComponentsAsUnsafe() { TableDefinition definition = new TableDefinition() {{ allSafeForLoggingByDefault(); javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING, LogSafety.UNSAFE); columns(); column(COLUMN_NAME, COLUMN_SHORTNAME, ValueType.VAR_LONG); }}; assertThat(definition.toTableMetadata().getNameLogSafety()).isEqualTo(LogSafety.SAFE); assertRowComponentSafety(definition, LogSafety.UNSAFE); assertNamedColumnSafety(definition, LogSafety.SAFE); }
@Test public void canSpecifyTableNameAsUnsafeWithAllComponentsAsSafe() { TableDefinition definition = new TableDefinition() {{ tableNameLogSafety(LogSafety.UNSAFE); namedComponentsSafeByDefault(); javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING); columns(); column(COLUMN_NAME, COLUMN_SHORTNAME, ValueType.VAR_LONG); }}; assertThat(definition.toTableMetadata().getNameLogSafety()).isEqualTo(LogSafety.UNSAFE); assertRowComponentSafety(definition, LogSafety.SAFE); assertNamedColumnSafety(definition, LogSafety.SAFE); }
private TableDefinition getSimpleTableDefinition(TableReference tableRef) { return new TableDefinition() {{ javaTableName(tableRef.getTablename()); rowName(); rowComponent("rowName", ValueType.STRING); columns(); column("col1", "1", ValueType.VAR_LONG); column("col2", "2", ValueType.VAR_LONG); conflictHandler(ConflictHandler.IGNORE_ALL); }}; }
@Test public void canDeclareSpecificComponentsAsSafe() { TableDefinition definition = new TableDefinition() {{ javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING, LogSafety.SAFE); columns(); column(COLUMN_NAME, COLUMN_SHORTNAME, ValueType.VAR_LONG); }}; assertRowComponentSafety(definition, LogSafety.SAFE); assertNamedColumnSafety(definition, LogSafety.UNSAFE); }
@Test public void canSpecifyComponentsAsSafeByDefault() { TableDefinition definition = new TableDefinition() {{ namedComponentsSafeByDefault(); javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING); columns(); column(COLUMN_NAME, COLUMN_SHORTNAME, ValueType.VAR_LONG); }}; assertRowComponentSafety(definition, LogSafety.SAFE); assertNamedColumnSafety(definition, LogSafety.SAFE); }
@Test public void canDeclareComponentsAsSafeByDefaultAndSpecificComponentsAsUnsafe() { TableDefinition definition = new TableDefinition() {{ namedComponentsSafeByDefault(); javaTableName(TABLE_REF.getTablename()); rowName(); rowComponent(ROW_NAME, ValueType.STRING, LogSafety.UNSAFE); columns(); column(COLUMN_NAME, COLUMN_SHORTNAME, ValueType.VAR_LONG); }}; assertRowComponentSafety(definition, LogSafety.UNSAFE); assertNamedColumnSafety(definition, LogSafety.SAFE); }