Question

我正在使用SpringBatch从Oracle读取并写入ElasticSearch。

我的代码适用于静态查询。示例： select emp_id, emp_name from employee_table 我有一个RowMapper类，它将resultSet中的值映射到Employee POJO。

我的要求是

查询将由用户输入。所以查询可能如下

select emp_id, emp_name from employee_table
select cust_id, cust_name, cust_age from customer_table
select door_no, street_name, loc_name, city from address_table
Similar queries

我的问题是

我是否有办法根据用户提供的查询动态创建POJO？
如果查询在我的情况下不断变化，RowMapper概念会起作用吗？
是否有类似通用行标记的东西？

感谢您的时间。示例代码将非常感激。

Answer 1

如果您有对象，则需要映射到...

使用实际扩展RowMapper

的BeanWrapperFieldSetMapper自定义实现，考虑对SQL进行别名以匹配您的对象字段名称

所以如果您的POJO看起来像这样：

public class Employee {

    private String employeeId;
    private String employeeName;

    ...
    // getters and setters

}

然后你的SQL看起来像这样：

SELECT emp_id employeeId, emp_name employeeName from employee_table

然后你的包装RowMapper看起来像这样：

import org.springframework.jdbc.core.RowMapper
import org.springframework.batch.item.file.mapping.BeanWrapperFieldSetMapper

public class BeanWrapperRowMapper<T> extends BeanWrapperFieldSetMapper<T> implements RowMapper<T> {

    @Override
    public T mapRow(final ResultSet rs, final int rowNum) throws SQLException {
        final FieldSet fs = getFieldSet(rs);
        try {
            return super.mapFieldSet(fs);
        } catch (final BindException e) {
            throw new IllegalArgumentException("Could not bind bean to FieldSet", e);
        }
    }

    private FieldSet getFieldSet(final ResultSet rs) throws SQLException {
        final ResultSetMetaData metaData = rs.getMetaData();
        final int columnCount = metaData.getColumnCount();

        final List<String> tokens = new ArrayList<>();
        final List<String> names = new ArrayList<>();

        for (int i = 1; i <= columnCount; i++) {
            tokens.add(rs.getString(i));
            names.add(metaData.getColumnName(i));
        }

        return new DefaultFieldSet(tokens.toArray(new String[0]), names.toArray(new String[0]));    
    }

}

<强>另外...

如果您没有要映射的POJO，请使用开箱即用的ColumnMapRowMapper来获取列名称的地图（Map<String,Object>）（让＆＃39; s将它们称为COL_A，COL_B，COL_C）。然后，如果您的作者类似于JdbcBatchItemWriter，您可以将命名参数设置为：

INSERT TO ${schema}.TARGET_TABLE (COL_1, COL_2, COL_3) values (:COL_A, :COL_B, :COL_C)

然后您的ItemSqlParameterSourceProvider实施可能如此：

public class MapItemSqlParameterSourceProvider implements
        ItemSqlParameterSourceProvider<Map<String, Object>> {

    public SqlParameterSource createSqlParameterSource(Map<String, Object> item) {
        return new MapSqlParameterSource(item);
    }

}

Answer 2

回答你的问题：

有没有办法根据用户的查询动态创建POJO-即使有，我也不确定会有多少帮助。对于您的使用案例，我建议您使用Map。
如果查询不断变化，RowMapper概念是否有效 - 如果使用Map，则可以使用列名作为键，列值作为值。您应该能够创建一个可以执行此操作的RowMapper实现。
是否有像通用RowMapper这样的东西 - 但它只适用于POJO，因此您需要为此创建自己的。

Answer 3

我使用Spring的 ColumnMapRowMapper 找到了解决问题的方法。请从xml配置文件中找到一个片段。我没有生成任何POJO课程。我使用地图管理并将其插入ES。地图的密钥名称应与索引中的字段名称匹配。

<step id="slave" xmlns="http://www.springframework.org/schema/batch">
    <tasklet>
        <chunk reader="pagingItemReader" writer="elasticSearcItemWriter"
            processor="itemProcessor" commit-interval="10" />
    </tasklet>
</step>



<bean id="pagingItemReader"
        class="org.springframework.batch.item.database.JdbcPagingItemReader"
        scope="step">
        <property name="dataSource" ref="dataSource" />
        <property name="queryProvider">
            <bean
                class="org.springframework.batch.item.database.support.SqlPagingQueryProviderFactoryBean">
                <property name="dataSource" ref="dataSource" />
                <property name="selectClause" value="*******" />
                <property name="fromClause" value="*******" />

                <property name="whereClause" value="*******" />
                <property name="sortKey" value="*******" />
            </bean>

        </property>
        <!-- Inject via the ExecutionContext in rangePartitioner -->
        <property name="parameterValues">
            <map>
                <entry key="fromId" value="#{stepExecutionContext[fromId]}" />
                <entry key="toId" value="#{stepExecutionContext[toId]}" />
            </map>
        </property>
        <property name="pageSize" value="10" />
        <property name="rowMapper">
            <bean class="org.springframework.jdbc.core.ColumnMapRowMapper" />
        </property>
    </bean>

在我的elasticSearcItemWriter类中......

public class ElasticSearchItemWriter<T> extends AbstractItemStreamItemWriter<T>
        implements ResourceAwareItemWriterItemStream<T>, InitializingBean {
....
....
....
@Override
    public void write(List<? extends T> items) throws Exception {


        client = jestClient.getJestClient();
        if (items.size() > 0) {
            for (Object item : items) {
                @SuppressWarnings("unchecked")
                Map<String, Object> map = (Map<String, Object>) item;
                // Asynch index
                Index index = new Index.Builder(map).index(Start.prop.getProperty(Constants.ES_INDEX_NAME))
                    .type(Start.prop.getProperty(Constants.ES_INDEX_TYPE)).build();


                client.executeAsync(index, new JestResultHandler<JestResult>() {
                    public void failed(Exception ex) {
                    }

                    public void completed(JestResult result) {
                    }
                });
            }
        }
    }
.....
....
}

Answer 4

你可以像下面那样简单地做，

SettingsDto settings = SettingsDao.getById(1, new BeanPropertyRowMapper<>(SettingsDto.class));

一般来说，您可以传递您的 DTO 类，但请注意您必须使用与 SQL 列相同的名称，或者必须根据 DTO 在 SQL 查询中使用 ALIAS。

@Data
public class SettingsDto {

    private int id;

    private int retryCount;

    private int batchSize;

    private int retryPeriod;

    private int statusInitialDelay;

}

我的dao方法如下

SettingsDto getById(int id, final RowMapper<OMoneySettingsDto> mapper);

其实现如下，

 @Override
    public SettingsDto getById(final int id, final RowMapper<OMoneySettingsDto> mapper) {
        return new JdbcTemplate(YourDataSource).queryForObject(QUERY_SETTINGS_BY_ID,new Object[]{id}, mapper);
}

SQL 在这里，如下所示，您必须在 DTO 中使用相同的名称

private static final String OMONEY_SETTINGS_BY_ID = "SELECT AS id,retry_count AS retryCount FROM setttings WHERE id = ?";

Spring：用于动态查询的通用RowMapper

4 个答案: