|
16 | 16 |
|
17 | 17 | package org.springframework.integration.sftp.inbound;
|
18 | 18 |
|
| 19 | +import java.io.File; |
| 20 | +import java.io.IOException; |
19 | 21 | import java.io.InputStream;
|
| 22 | +import java.nio.charset.StandardCharsets; |
20 | 23 | import java.util.Arrays;
|
21 | 24 | import java.util.Comparator;
|
22 | 25 | import java.util.concurrent.ConcurrentHashMap;
|
23 | 26 | import java.util.concurrent.ConcurrentMap;
|
24 | 27 |
|
25 | 28 | import com.jcraft.jsch.ChannelSftp.LsEntry;
|
| 29 | +import org.apache.commons.io.FileUtils; |
26 | 30 | import org.junit.jupiter.api.Test;
|
27 | 31 |
|
28 | 32 | import org.springframework.beans.factory.annotation.Autowired;
|
|
38 | 42 | import org.springframework.integration.endpoint.SourcePollingChannelAdapter;
|
39 | 43 | import org.springframework.integration.file.FileHeaders;
|
40 | 44 | import org.springframework.integration.file.filters.AcceptAllFileListFilter;
|
| 45 | +import org.springframework.integration.file.filters.ChainFileListFilter; |
41 | 46 | import org.springframework.integration.file.remote.session.SessionFactory;
|
42 | 47 | import org.springframework.integration.metadata.SimpleMetadataStore;
|
43 | 48 | import org.springframework.integration.scheduling.PollerMetadata;
|
44 | 49 | import org.springframework.integration.sftp.SftpTestSupport;
|
45 | 50 | import org.springframework.integration.sftp.filters.SftpPersistentAcceptOnceFileListFilter;
|
| 51 | +import org.springframework.integration.sftp.filters.SftpSimplePatternFileListFilter; |
| 52 | +import org.springframework.integration.sftp.filters.SftpSystemMarkerFilePresentFileListFilter; |
46 | 53 | import org.springframework.integration.sftp.session.SftpFileInfo;
|
47 | 54 | import org.springframework.integration.sftp.session.SftpRemoteFileTemplate;
|
48 | 55 | import org.springframework.integration.transformer.StreamTransformer;
|
@@ -167,6 +174,69 @@ public void testMaxFetchLambdaFilter() throws Exception {
|
167 | 174 | StaticMessageHeaderAccessor.getCloseableResource(received).close();
|
168 | 175 | }
|
169 | 176 |
|
| 177 | + |
| 178 | + @Test |
| 179 | + public void maxFetchIsAdjustedWhenNoSupportsSingleFileFiltering() throws Exception { |
| 180 | + SftpStreamingMessageSource messageSource = buildSource(); |
| 181 | + ChainFileListFilter<LsEntry> chainFileListFilter = new ChainFileListFilter<>(); |
| 182 | + SftpSystemMarkerFilePresentFileListFilter sftpSystemMarkerFilePresentFileListFilter = |
| 183 | + new SftpSystemMarkerFilePresentFileListFilter( |
| 184 | + new SftpSimplePatternFileListFilter("*"), ".trg"); |
| 185 | + SftpPersistentAcceptOnceFileListFilter sftpPersistentAcceptOnceFileListFilter = |
| 186 | + new SftpPersistentAcceptOnceFileListFilter(new SimpleMetadataStore(), "prefix"); |
| 187 | + chainFileListFilter.addFilter(sftpSystemMarkerFilePresentFileListFilter); |
| 188 | + chainFileListFilter.addFilter(sftpPersistentAcceptOnceFileListFilter); |
| 189 | + messageSource.setFilter(chainFileListFilter); |
| 190 | + messageSource.setMaxFetchSize(5); |
| 191 | + messageSource.afterPropertiesSet(); |
| 192 | + messageSource.start(); |
| 193 | + |
| 194 | + addFileAndTrigger("file001"); |
| 195 | + addFileAndTrigger("file002"); |
| 196 | + |
| 197 | + Message<InputStream> received = messageSource.receive(); |
| 198 | + assertThat(received).isNotNull(); |
| 199 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file001"); |
| 200 | + |
| 201 | + received = messageSource.receive(); |
| 202 | + assertThat(received).isNotNull(); |
| 203 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file002"); |
| 204 | + |
| 205 | + addFileAndTrigger("file003"); |
| 206 | + addFileAndTrigger("file004"); |
| 207 | + addFileAndTrigger("file005"); |
| 208 | + addFileAndTrigger("file006"); |
| 209 | + addFileAndTrigger("file007"); |
| 210 | + |
| 211 | + received = messageSource.receive(); |
| 212 | + assertThat(received).isNotNull(); |
| 213 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file003"); |
| 214 | + |
| 215 | + received = messageSource.receive(); |
| 216 | + assertThat(received).isNotNull(); |
| 217 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file004"); |
| 218 | + |
| 219 | + received = messageSource.receive(); |
| 220 | + assertThat(received).isNotNull(); |
| 221 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file005"); |
| 222 | + |
| 223 | + received = messageSource.receive(); |
| 224 | + assertThat(received).isNotNull(); |
| 225 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file006"); |
| 226 | + |
| 227 | + received = messageSource.receive(); |
| 228 | + assertThat(received).isNotNull(); |
| 229 | + assertThat(received.getHeaders().get(FileHeaders.REMOTE_FILE)).isEqualTo("file007"); |
| 230 | + } |
| 231 | + |
| 232 | + private void addFileAndTrigger(String filename) throws IOException { |
| 233 | + File file = new File(this.sourceRemoteDirectory, filename); |
| 234 | + FileUtils.writeStringToFile(file, "source1", StandardCharsets.UTF_8); |
| 235 | + |
| 236 | + file = new File(this.sourceRemoteDirectory, filename + ".trg"); |
| 237 | + file.createNewFile(); |
| 238 | + } |
| 239 | + |
170 | 240 | private SftpStreamingMessageSource buildSource() {
|
171 | 241 | SftpStreamingMessageSource messageSource =
|
172 | 242 | new SftpStreamingMessageSource(this.config.template(),
|
|
0 commit comments