Spring/spring-integration
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

DefaultFileNameGenerator now checks if the Message payload is a File. If so, and no 'filename' header has been set, it will reuse the same file name.

Browse Source
This commit is contained in:
Mark Fisher
2008-09-22 04:48:02 +00:00
parent 2aebfbd6ed
commit fdb704e086
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
org.springframework.integration.file/src/main/java/org/springframework/integration/file/DefaultFileNameGenerator.java
View File

@@ -16,12 +16,16 @@
package org.springframework.integration.file;
import java.io.File;
import org.springframework.integration.message.Message;
import org.springframework.util.StringUtils;
/**
* Default implementation of the filename generator strategy. It first checks
* for the "filename" Message header, then falls back to the Message ID.
* for the "filename" Message header. Next, it checks if the Message payload is
* a File instance, and if so, it uses the same name. Finally, it falls back to
* the Message ID and adds the suffix '.msg'.
*
* @author Mark Fisher
*/
@@ -29,8 +33,13 @@ public class DefaultFileNameGenerator implements FileNameGenerator {
public String generateFileName(Message<?> message) {
String filenameProperty = message.getHeaders().get(FILENAME_PROPERTY_KEY, String.class);
return StringUtils.hasText(filenameProperty) ?
filenameProperty : message.getHeaders().getId() + ".msg";
if (StringUtils.hasText(filenameProperty)) {
return filenameProperty;
}
if (message.getPayload() instanceof File) {
return ((File) message.getPayload()).getName();
}
return message.getHeaders().getId() + ".msg";
}
}