@Override public int hashCode() { int result = (isCreateTables() ? 1 : 0); result = 31 * result + (getHost() != null ? getHost().hashCode() : 0); result = 31 * result + getPort(); result = 31 * result + (getConnectURI() != null ? getConnectURI().hashCode() : 0); result = 31 * result + (getUser() != null ? getUser().hashCode() : 0); result = 31 * result + (passwordProvider != null ? passwordProvider.hashCode() : 0); result = 31 * result + (dbcpProperties != null ? dbcpProperties.hashCode() : 0); return result; } }
private DBI ensureDBI(CacheScheduler.EntryImpl<JdbcExtractionNamespace> id, JdbcExtractionNamespace namespace) { final CacheScheduler.EntryImpl<JdbcExtractionNamespace> key = id; DBI dbi = null; if (dbiCache.containsKey(key)) { dbi = dbiCache.get(key); } if (dbi == null) { final DBI newDbi = new DBI( namespace.getConnectorConfig().getConnectURI(), namespace.getConnectorConfig().getUser(), namespace.getConnectorConfig().getPassword() ); dbiCache.putIfAbsent(key, newDbi); dbi = dbiCache.get(key); } return dbi; }
protected BasicDataSource getDatasource() { MetadataStorageConnectorConfig connectorConfig = getConfig(); BasicDataSource dataSource; try { Properties dbcpProperties = connectorConfig.getDbcpProperties(); if (dbcpProperties != null) { dataSource = BasicDataSourceFactory.createDataSource(dbcpProperties); } else { dataSource = new BasicDataSource(); } } catch (Exception e) { throw new RuntimeException(e); } dataSource.setUsername(connectorConfig.getUser()); dataSource.setPassword(connectorConfig.getPassword()); String uri = connectorConfig.getConnectURI(); dataSource.setUrl(uri); dataSource.setValidationQuery(getValidationQuery()); dataSource.setTestOnBorrow(true); return dataSource; }
private void testMetadataStorageConnectionConfig( boolean createTables, String host, int port, String connectURI, String user, String pwdString, String pwd ) throws Exception { MetadataStorageConnectorConfig config = jsonMapper.readValue( "{" + "\"createTables\": \"" + createTables + "\"," + "\"host\": \"" + host + "\"," + "\"port\": \"" + port + "\"," + "\"connectURI\": \"" + connectURI + "\"," + "\"user\": \"" + user + "\"," + "\"password\": " + pwdString + "}", MetadataStorageConnectorConfig.class ); Assert.assertEquals(host, config.getHost()); Assert.assertEquals(port, config.getPort()); Assert.assertEquals(connectURI, config.getConnectURI()); Assert.assertEquals(user, config.getUser()); Assert.assertEquals(pwd, config.getPassword()); Assert.assertNull(config.getDbcpProperties()); }
@Override public void createDataSourceTable() { if (config.get().isCreateTables()) { createDataSourceTable(tablesConfigSupplier.get().getDataSourceTable()); } }
@Override public String toString() { return "DbConnectorConfig{" + "createTables=" + createTables + ", connectURI='" + getConnectURI() + '\'' + ", user='" + user + '\'' + ", passwordProvider=" + passwordProvider + ", dbcpProperties=" + dbcpProperties + '}'; }
@Test public void testEquals() throws IOException { MetadataStorageConnectorConfig metadataStorageConnectorConfig = createMetadataStorageConfig( true, "testHost", 4000, "url", "user", "\"nothing\"" ); MetadataStorageConnectorConfig metadataStorageConnectorConfig2 = createMetadataStorageConfig( true, "testHost", 4000, "url", "user", "\"nothing\"" ); Assert.assertTrue(metadataStorageConnectorConfig.equals(metadataStorageConnectorConfig2)); Assert.assertTrue(metadataStorageConnectorConfig.hashCode() == metadataStorageConnectorConfig2.hashCode()); }
taskLockbox = new TaskLockbox(taskStorage); testDerbyConnector = new TestDerbyConnector( Suppliers.ofInstance(new MetadataStorageConnectorConfig()), Suppliers.ofInstance(metadataStorageTablesConfig) );
@Override public boolean equals(Object o) { if (this == o) { return true; } if (!(o instanceof JdbcDataFetcher)) { return false; } JdbcDataFetcher that = (JdbcDataFetcher) o; if (!connectorConfig.equals(that.connectorConfig)) { return false; } if (!table.equals(that.table)) { return false; } if (!keyColumn.equals(that.keyColumn)) { return false; } return valueColumn.equals(that.valueColumn); }
); Assert.assertEquals(host, config.getHost()); Assert.assertEquals(port, config.getPort()); Assert.assertEquals(connectURI, config.getConnectURI()); Assert.assertEquals(user, config.getUser()); Assert.assertEquals(pwd, config.getPassword()); Properties dbcpProperties = config.getDbcpProperties(); Assert.assertEquals(dbcpProperties.getProperty("maxConnLifetimeMillis"), "1200000"); Assert.assertEquals(dbcpProperties.getProperty("defaultQueryTimeout"), "30000");
@Override public void createSegmentTable() { if (config.get().isCreateTables()) { createSegmentTable(tablesConfigSupplier.get().getSegmentsTable()); } }
@JsonCreator public JdbcExtractionNamespace( @NotNull @JsonProperty(value = "connectorConfig", required = true) final MetadataStorageConnectorConfig connectorConfig, @NotNull @JsonProperty(value = "table", required = true) final String table, @NotNull @JsonProperty(value = "keyColumn", required = true) final String keyColumn, @NotNull @JsonProperty(value = "valueColumn", required = true) final String valueColumn, @Nullable @JsonProperty(value = "tsColumn", required = false) final String tsColumn, @Nullable @JsonProperty(value = "filter", required = false) final String filter, @Min(0) @Nullable @JsonProperty(value = "pollPeriod", required = false) final Period pollPeriod ) { this.connectorConfig = Preconditions.checkNotNull(connectorConfig, "connectorConfig"); Preconditions.checkNotNull(connectorConfig.getConnectURI(), "connectorConfig.connectURI"); this.table = Preconditions.checkNotNull(table, "table"); this.keyColumn = Preconditions.checkNotNull(keyColumn, "keyColumn"); this.valueColumn = Preconditions.checkNotNull(valueColumn, "valueColumn"); this.tsColumn = tsColumn; this.filter = filter; this.pollPeriod = pollPeriod == null ? new Period(0L) : pollPeriod; }
private void createAndUpdateTable(final String tableName) derbyFirehoseConnector = new TestDerbyFirehoseConnector(new MetadataStorageConnectorConfig(), derbyConnector.getDBI()); derbyConnector.createTable(
if (isCreateTables() != that.isCreateTables()) { return false; if (getPort() != that.getPort()) { return false; if (getHost() != null ? !getHost().equals(that.getHost()) : that.getHost() != null) { return false; if (getConnectURI() != null ? !getConnectURI().equals(that.getConnectURI()) : that.getConnectURI() != null) { return false; if (getUser() != null ? !getUser().equals(that.getUser()) : that.getUser() != null) { return false; if (getDbcpProperties() == null ? that.getDbcpProperties() != null : !getDbcpProperties().equals(that.getDbcpProperties())) { return false;
protected BasicDataSource getDatasource(MetadataStorageConnectorConfig connectorConfig) { BasicDataSource dataSource = new BasicDataSource(); dataSource.setUsername(connectorConfig.getUser()); dataSource.setPassword(connectorConfig.getPassword()); String uri = connectorConfig.getConnectURI(); dataSource.setUrl(uri); dataSource.setTestOnBorrow(true); dataSource.setValidationQuery(getValidationQuery()); return dataSource; }
@Override public void createConfigTable() { if (config.get().isCreateTables()) { createConfigTable(tablesConfigSupplier.get().getConfigTable()); } }
@Override public String toString() { return "DbConnectorConfig{" + "createTables=" + createTables + ", connectURI='" + getConnectURI() + '\'' + ", user='" + user + '\'' + ", passwordProvider=" + passwordProvider + '}'; }
@Override public int hashCode() { int result = (isCreateTables() ? 1 : 0); result = 31 * result + (getHost() != null ? getHost().hashCode() : 0); result = 31 * result + getPort(); result = 31 * result + (getConnectURI() != null ? getConnectURI().hashCode() : 0); result = 31 * result + (getUser() != null ? getUser().hashCode() : 0); result = 31 * result + (passwordProvider != null ? passwordProvider.hashCode() : 0); return result; } }