1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.transport.file; |
12 | |
|
13 | |
import org.mule.DefaultMuleMessage; |
14 | |
import org.mule.MuleServer; |
15 | |
import org.mule.api.DefaultMuleException; |
16 | |
import org.mule.api.MuleException; |
17 | |
import org.mule.api.MuleMessage; |
18 | |
import org.mule.api.endpoint.InboundEndpoint; |
19 | |
import org.mule.api.transport.MessageAdapter; |
20 | |
import org.mule.transport.AbstractMessageRequester; |
21 | |
import org.mule.transport.DefaultMessageAdapter; |
22 | |
import org.mule.transport.file.filters.FilenameWildcardFilter; |
23 | |
import org.mule.transport.file.i18n.FileMessages; |
24 | |
import org.mule.util.FileUtils; |
25 | |
|
26 | |
import java.io.File; |
27 | |
import java.io.FileNotFoundException; |
28 | |
import java.io.FilenameFilter; |
29 | |
import java.net.URLDecoder; |
30 | |
|
31 | |
|
32 | |
|
33 | |
|
34 | |
public class FileMessageRequester extends AbstractMessageRequester |
35 | |
{ |
36 | |
private final FileConnector connector; |
37 | |
|
38 | |
public FileMessageRequester(InboundEndpoint endpoint) |
39 | |
{ |
40 | 22 | super(endpoint); |
41 | 22 | this.connector = (FileConnector) endpoint.getConnector(); |
42 | 22 | } |
43 | |
|
44 | |
|
45 | |
|
46 | |
|
47 | |
|
48 | |
|
49 | |
public Object getDelegateSession() throws MuleException |
50 | |
{ |
51 | 0 | return null; |
52 | |
} |
53 | |
|
54 | |
|
55 | |
|
56 | |
|
57 | |
|
58 | |
|
59 | |
|
60 | |
|
61 | |
|
62 | |
|
63 | |
|
64 | |
|
65 | |
protected MuleMessage doRequest(long timeout) throws Exception |
66 | |
{ |
67 | 22 | File file = FileUtils.newFile(endpoint.getEndpointURI().getAddress()); |
68 | 22 | File result = null; |
69 | 22 | FilenameFilter filenameFilter = null; |
70 | 22 | String filter = (String) endpoint.getProperty("filter"); |
71 | 22 | if (filter != null) |
72 | |
{ |
73 | 0 | filter = URLDecoder.decode(filter, MuleServer.getMuleContext().getConfiguration().getDefaultEncoding()); |
74 | 0 | filenameFilter = new FilenameWildcardFilter(filter); |
75 | |
} |
76 | 22 | if (file.exists()) |
77 | |
{ |
78 | 22 | if (file.isFile()) |
79 | |
{ |
80 | 18 | result = file; |
81 | |
} |
82 | 4 | else if (file.isDirectory()) |
83 | |
{ |
84 | 4 | result = FileMessageDispatcher.getNextFile(endpoint.getEndpointURI().getAddress(), filenameFilter); |
85 | |
} |
86 | 22 | if (result != null) |
87 | |
{ |
88 | 22 | boolean checkFileAge = connector.getCheckFileAge(); |
89 | 22 | if (checkFileAge) |
90 | |
{ |
91 | 2 | long fileAge = connector.getFileAge(); |
92 | 2 | long lastMod = result.lastModified(); |
93 | 2 | long now = System.currentTimeMillis(); |
94 | 2 | long thisFileAge = now - lastMod; |
95 | 2 | if (thisFileAge < fileAge) |
96 | |
{ |
97 | 0 | if (logger.isDebugEnabled()) |
98 | |
{ |
99 | 0 | logger.debug("The file has not aged enough yet, will return nothing for: " |
100 | |
+ result.getCanonicalPath()); |
101 | |
} |
102 | 0 | return null; |
103 | |
} |
104 | |
} |
105 | |
|
106 | |
|
107 | 22 | String sourceFileOriginalName = result.getName(); |
108 | |
|
109 | |
|
110 | |
|
111 | |
|
112 | |
|
113 | 22 | DefaultMessageAdapter fileParserMsgAdaptor = new DefaultMessageAdapter(null); |
114 | 22 | fileParserMsgAdaptor.setProperty(FileConnector.PROPERTY_ORIGINAL_FILENAME, sourceFileOriginalName); |
115 | |
|
116 | |
|
117 | 22 | File destinationFile = null; |
118 | 22 | String movDir = connector.getMoveToDirectory(); |
119 | 22 | if (movDir != null) |
120 | |
{ |
121 | 8 | String destinationFileName = sourceFileOriginalName; |
122 | 8 | String moveToPattern = connector.getMoveToPattern(); |
123 | 8 | if (moveToPattern != null) |
124 | |
{ |
125 | 0 | destinationFileName = connector.getFilenameParser().getFilename( |
126 | |
fileParserMsgAdaptor, moveToPattern); |
127 | |
} |
128 | |
|
129 | 8 | destinationFile = FileUtils.newFile(movDir, destinationFileName); |
130 | |
} |
131 | |
|
132 | 22 | MessageAdapter msgAdapter = null; |
133 | |
try |
134 | |
{ |
135 | 22 | if (connector.isStreaming()) |
136 | |
{ |
137 | 14 | msgAdapter = connector.getMessageAdapter(new ReceiverFileInputStream(result, connector.isAutoDelete(), |
138 | |
destinationFile)); |
139 | |
} |
140 | |
else |
141 | |
{ |
142 | 8 | msgAdapter = connector.getMessageAdapter(result); |
143 | |
} |
144 | |
} |
145 | 0 | catch (FileNotFoundException e) |
146 | |
{ |
147 | |
|
148 | |
|
149 | 0 | logger.error("File being read disappeared!", e); |
150 | 0 | return null; |
151 | 22 | } |
152 | 22 | msgAdapter.setProperty(FileConnector.PROPERTY_ORIGINAL_FILENAME, sourceFileOriginalName); |
153 | |
|
154 | 22 | if (!connector.isStreaming()) |
155 | |
{ |
156 | 8 | moveOrDelete(result, destinationFile); |
157 | 8 | return new DefaultMuleMessage(msgAdapter); |
158 | |
} |
159 | |
else |
160 | |
{ |
161 | |
|
162 | |
|
163 | 14 | return new DefaultMuleMessage(msgAdapter); |
164 | |
} |
165 | |
} |
166 | |
} |
167 | 0 | return null; |
168 | |
} |
169 | |
|
170 | |
private void moveOrDelete(final File sourceFile, File destinationFile) throws DefaultMuleException |
171 | |
{ |
172 | |
|
173 | 8 | if (destinationFile != null) |
174 | |
{ |
175 | |
|
176 | 4 | if (!FileUtils.moveFile(sourceFile, destinationFile)) |
177 | |
{ |
178 | 0 | throw new DefaultMuleException(FileMessages.failedToMoveFile(sourceFile.getAbsolutePath(), |
179 | |
destinationFile.getAbsolutePath())); |
180 | |
} |
181 | |
} |
182 | 8 | if (connector.isAutoDelete()) |
183 | |
{ |
184 | |
|
185 | 4 | if (destinationFile == null) |
186 | |
{ |
187 | |
|
188 | 2 | if (!sourceFile.delete()) |
189 | |
{ |
190 | 0 | throw new DefaultMuleException(FileMessages.failedToDeleteFile(sourceFile.getAbsolutePath())); |
191 | |
} |
192 | |
} |
193 | |
else |
194 | |
{ |
195 | |
|
196 | |
|
197 | |
} |
198 | |
} |
199 | |
|
200 | 8 | } |
201 | |
|
202 | |
protected void doDispose() |
203 | |
{ |
204 | |
|
205 | 22 | } |
206 | |
|
207 | |
protected void doConnect() throws Exception |
208 | |
{ |
209 | |
|
210 | 22 | } |
211 | |
|
212 | |
protected void doDisconnect() throws Exception |
213 | |
{ |
214 | |
|
215 | 22 | } |
216 | |
|
217 | |
} |