public Dataset execute()

in src/main/java/com/uber/uberscriptquery/execution/JdbcSqlInputStatementExecutor.java [34:45]

• 12 lines of code
• 2 McCabe index (conditional complexity)

    public Dataset<Row> execute(SparkSession sparkSession, StatementAssignment statementAssignment, CredentialProvider credentialManager) {
        logger.info("Running query by sql jdbc: " + statementAssignment);
        Map<String, String> queryConfig = statementAssignment.getQueryConfig();
        String connectionString = queryConfig.get(StatementAssignment.QUERY_CONFIG_CONNECTION_STRING);
        String passwordFile = queryConfig.get(StatementAssignment.QUERY_CONFIG_PASSWORD_FILE);
        String passwordEntry = queryConfig.get(StatementAssignment.QUERY_CONFIG_PASSWORD_ENTRY);
        String password = credentialManager.getPassword(passwordFile, passwordEntry);
        if (password != null) {
            connectionString = connectionString.replace("[password]", password);
        }
        return SparkUtils.readJdbc(connectionString, statementAssignment.getQueryStatement(), sparkSession);
    }