001 /** 002 * Licensed to the Apache Software Foundation (ASF) under one or more 003 * contributor license agreements. See the NOTICE file distributed with 004 * this work for additional information regarding copyright ownership. 005 * The ASF licenses this file to You under the Apache License, Version 2.0 006 * (the "License"); you may not use this file except in compliance with 007 * the License. You may obtain a copy of the License at 008 * 009 * http://www.apache.org/licenses/LICENSE-2.0 010 * 011 * Unless required by applicable law or agreed to in writing, software 012 * distributed under the License is distributed on an "AS IS" BASIS, 013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 014 * See the License for the specific language governing permissions and 015 * limitations under the License. 016 */ 017 package org.apache.camel.component.file; 018 019 import java.io.File; 020 import java.io.FileOutputStream; 021 import java.io.InputStream; 022 import java.io.RandomAccessFile; 023 import java.nio.ByteBuffer; 024 import java.nio.channels.FileChannel; 025 026 import org.apache.camel.Exchange; 027 import org.apache.camel.Message; 028 import org.apache.camel.impl.DefaultProducer; 029 import org.apache.camel.util.ExchangeHelper; 030 import org.apache.camel.util.ObjectHelper; 031 import org.apache.camel.util.UuidGenerator; 032 import org.apache.commons.logging.Log; 033 import org.apache.commons.logging.LogFactory; 034 035 /** 036 * For producing files. 037 * 038 * @version $Revision: 42440 $ 039 */ 040 public class FileProducer extends DefaultProducer { 041 private static final transient Log LOG = LogFactory.getLog(FileProducer.class); 042 private FileEndpoint endpoint; 043 044 public FileProducer(FileEndpoint endpoint) { 045 super(endpoint); 046 this.endpoint = endpoint; 047 } 048 049 public FileEndpoint getEndpoint() { 050 return (FileEndpoint) super.getEndpoint(); 051 } 052 053 public void process(Exchange exchange) throws Exception { 054 FileExchange fileExchange = endpoint.createExchange(exchange); 055 process(fileExchange); 056 ExchangeHelper.copyResults(exchange, fileExchange); 057 } 058 059 public void process(FileExchange exchange) throws Exception { 060 if (ExchangeHelper.isOutCapable(exchange)) { 061 // lets poll the file 062 Message out = exchange.getOut(true); 063 endpoint.configureMessage(endpoint.getFile(), out); 064 return; 065 } 066 067 InputStream in = ExchangeHelper.getMandatoryInBody(exchange, InputStream.class); 068 File file = createFileName(exchange.getIn()); 069 buildDirectory(file); 070 071 if (LOG.isDebugEnabled()) { 072 LOG.debug("About to write to: " + file + " from exchange: " + exchange); 073 } 074 FileChannel fc = null; 075 try { 076 if (getEndpoint().isAppend()) { 077 fc = new RandomAccessFile(file, "rw").getChannel(); 078 fc.position(fc.size()); 079 } else { 080 fc = new FileOutputStream(file).getChannel(); 081 } 082 083 int size = getEndpoint().getBufferSize(); 084 byte[] buffer = new byte[size]; 085 ByteBuffer byteBuffer = ByteBuffer.wrap(buffer); 086 while (true) { 087 int count = in.read(buffer); 088 if (count <= 0) { 089 break; 090 } else if (count < size) { 091 byteBuffer = ByteBuffer.wrap(buffer, 0, count); 092 fc.write(byteBuffer); 093 break; 094 } else { 095 fc.write(byteBuffer); 096 byteBuffer.clear(); 097 } 098 } 099 } finally { 100 ObjectHelper.close(in, file.getName(), LOG); 101 ObjectHelper.close(fc, file.getName(), LOG); 102 } 103 } 104 105 protected File createFileName(Message message) { 106 File answer; 107 108 String name = null; 109 if (!endpoint.isIgnoreFileNameHeader()) { 110 name = message.getHeader(FileComponent.HEADER_FILE_NAME, String.class); 111 } 112 113 File endpointFile = endpoint.getFile(); 114 if (endpointFile.isDirectory()) { 115 if (name != null) { 116 answer = new File(endpointFile, name); 117 if (answer.isDirectory()) { 118 answer = new File(answer, endpoint.getGeneratedFileName(message)); 119 } 120 } else { 121 answer = new File(endpointFile, endpoint.getGeneratedFileName(message)); 122 } 123 } else { 124 if (name == null) { 125 answer = endpointFile; 126 } else { 127 answer = new File(endpointFile, name); 128 } 129 } 130 131 // TODO lets store the name in the header? 132 //message.setHeader(FileComponent.HEADER_FILE_NAME, answer.toString()); 133 134 return answer; 135 } 136 137 private void buildDirectory(File file) { 138 String dirName = file.getAbsolutePath(); 139 int index = dirName.lastIndexOf(File.separatorChar); 140 if (index > 0) { 141 dirName = dirName.substring(0, index); 142 File dir = new File(dirName); 143 dir.mkdirs(); 144 } 145 } 146 147 148 }