如何从Spring Batch中的ItemReader动态设置MultiResourceItemWriter的资源

时间:2013-12-03 14:15:59

标签: java spring-batch

批处理作业是:

  • 从csv文件中读取

  • 为csv中的每个记录(行)创建一个xml文件,名称为Patent.ID.xml(其中ID是csv中的字段,Patent是我的模型类),例如:1.xml,2 .XML

问题是我找不到从csv文件动态设置文件名到每个ID的方法

这是我的配置:

<beans xmlns="http://www.springframework.org/schema/beans"
    xmlns:batch="http://www.springframework.org/schema/batch"xmlns:task="http://www.springframework.org/schema/task"
    xmlns:util="http://www.springframework.org/schema/util"xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://www.springframework.org/schema/batch
        http://www.springframework.org/schema/batch/spring-batch-2.2.xsd
        http://www.springframework.org/schema/beans 
        http://www.springframework.org/schema/beans/spring-beans-3.2.xsd
        http://www.springframework.org/schema/util 
        http://www.springframework.org/schema/util/spring-util-3.2.xsd">


    <!-- My beans -->   
    <bean id="patent" class="com.example.model.Patent" scope="prototype" />
    <bean id="xmlsuffix" class="com.example.filename.PatentFileSuffixCreator"/>
    <bean id="noInputException"     class="com.example.listener.NoWorkFoundStepExecutionListener"/>

    <batch:job id="csvtoxml">
        <batch:step id="step1">
            <batch:tasklet>             
                <batch:chunk reader="fileReader"     writer="multiResourceItemWriter" commit-interval="1">
                </batch:chunk>
                <batch:listeners>
                    <batch:listener ref="noInputException"/>
                </batch:listeners>  
            </batch:tasklet>
        </batch:step>
    </batch:job>

    <bean id="multiResourceItemWriter"
    class="org.springframework.batch.item.file.MultiResourceItemWriter">
        <property name="resource" value="file:xml/#{patent.ID}" />
        <property name="delegate" ref="XMLwriter"/>
        <property name="itemCountLimitPerResource" value="1"/>
        <property name="resourceSuffixCreator" ref="xmlsuffix"/>
    </bean>

    <bean id="XMLwriter" 
                class="org.springframework.batch.item.xml.StaxEventItemWriter">
        <property name="marshaller" ref="patentUnmarshaller" />
        <property name="rootTagName" value="Patents" />
    </bean>

    <bean id="patentUnmarshaller" 
     class="org.springframework.oxm.xstream.XStreamMarshaller">
        <property name="aliases">
            <map>
                <entry key="Patent"
                value="com.example.model.Patent" />
            </map>
        </property>
    </bean>

  <bean id="fileReader"
        class="org.springframework.batch.item.file.FlatFileItemReader" scope="step">
        <property name="lineMapper" ref="lineMapper"/>
        <property name="resource" value="file:dbbrev_sample_m.csv"/>
        <property name="linesToSkip" value="1"/>
    </bean>

    <bean id="lineMapper"
        class="org.springframework.batch.item.file.mapping.DefaultLineMapper">
        <property name="fieldSetMapper" ref="fieldSetMapper"/>
        <property name="lineTokenizer" ref="lineTokenizer"/>
    </bean>

    <bean id="lineTokenizer"
        class="org.springframework.batch.item.file.transform.DelimitedLineTokenizer">
        <property name="delimiter" value=","/>
        <property name="names" value="ID,TYPE,NOPUBLICATION" />
    </bean>

    <bean id="fieldSetMapper"
        class="org.springframework.batch.item.file.mapping.BeanWrapperFieldSetMapper">
        <property name="targetType" value="com.example.model.Patent"/>
    </bean>
</beans>

2 个答案:

答案 0 :(得分:2)

创建一个ItemWriteListener,注入multiResourceItemWriter bean并将监听器绑定到您的步骤 在ItemWriteListener.beforeWrite()中使用您要写入的项目创建一个新的ResourceSuffixCreator对象,以创建资源扩展名(后缀)。
MultiResourceItemWriter.resource可能需要更改为file:xml/,因为1.xml,2.xml等将使用为您正在编写的每个项目创建的自定义ResourceSuffixCreator附加。

此解决方案很脏,并且(可能)因commit-interval=1而起作用;如果你改变我的答案(可能)会失败。

我希望我很清楚,英语不是我的母语。

答案 1 :(得分:0)

我找不到让Luca的解决方案工作的方法,所以我用这种方式扩展了MultiResourceItemWriter:

public class MyMultipleItemWriter extends MultiResourceItemWriter<MyBeanXml>
{
    @Override
    public void write(List<? extends MyBeanXml> items) throws Exception
    {
        this.setResourceSuffixCreator(new MySuffixCreator(items.get(0).getField()));
        super.write(items);
    }
}