jprdonnelly/airbyte
1
0
Fork 0
mirror of synced 2025-12-25 02:09:19 -05:00
Code Issues Packages Projects Releases Wiki Activity

Close all unsafe queries (#12495)

Browse Source 
* Add helper methods to return lists of json nodes

* Close all unsafe queries

* Add one more helper method

* Simplify helper names

* Format code
This commit is contained in:
LiRen Tu
2022-05-03 13:45:02 -07:00
committed by GitHub
parent 265eddcc08
commit e15ae56389
21 changed files with 235 additions and 245 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
airbyte-integrations/connectors/source-db2/src/main/java/io.airbyte.integrations.source.db2/Db2Source.java
View File

@@ -26,6 +26,7 @@ import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import org.apache.commons.lang3.RandomStringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -89,10 +90,9 @@ public class Db2Source extends AbstractJdbcSource<JDBCType> implements Source {
@Override
public Set<JdbcPrivilegeDto> getPrivilegesTableForCurrentUser(final JdbcDatabase database, final String schema) throws SQLException {
return database
.unsafeQuery(getPrivileges(), sourceOperations::rowToJson)
.map(this::getPrivilegeDto)
.collect(Collectors.toSet());
try (final Stream<JsonNode> stream = database.unsafeQuery(getPrivileges(), sourceOperations::rowToJson)) {
return stream.map(this::getPrivilegeDto).collect(Collectors.toSet());
}
}
@Override