|
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