org.apache.camel.converter.stream.StreamSourceCache Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.camel.converter.stream;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import javax.xml.transform.stream.StreamSource;
import org.apache.camel.Exchange;
import org.apache.camel.RuntimeCamelException;
import org.apache.camel.StreamCache;
import org.apache.camel.util.IOHelper;
/**
* {@link org.apache.camel.StreamCache} implementation for Cache the StreamSource {@link javax.xml.transform.stream.StreamSource}s
*/
public class StreamSourceCache extends StreamSource implements StreamCache {
private final InputStream stream;
private final StreamCache streamCache;
private final ReaderCache readCache;
public StreamSourceCache(StreamSource source, Exchange exchange) throws IOException {
if (source.getInputStream() != null) {
// set up CachedOutputStream with the properties
CachedOutputStream cos = new CachedOutputStream(exchange);
IOHelper.copyAndCloseInput(source.getInputStream(), cos);
streamCache = cos.getStreamCache();
readCache = null;
setSystemId(source.getSystemId());
stream = (InputStream) streamCache;
} else if (source.getReader() != null) {
String data = exchange.getContext().getTypeConverter().convertTo(String.class, source.getReader());
readCache = new ReaderCache(data);
streamCache = null;
setReader(readCache);
stream = new ByteArrayInputStream(data.getBytes());
} else {
streamCache = null;
readCache = null;
stream = null;
}
}
public void reset() {
if (streamCache != null) {
streamCache.reset();
}
if (readCache != null) {
readCache.reset();
}
if (stream != null) {
try {
stream.reset();
} catch (IOException e) {
throw new RuntimeCamelException(e);
}
}
}
public void writeTo(OutputStream os) throws IOException {
if (streamCache != null) {
streamCache.writeTo(os);
} else if (readCache != null) {
readCache.writeTo(os);
}
}
@Override
public InputStream getInputStream() {
return stream;
}
@Override
public void setInputStream(InputStream inputStream) {
// noop as the input stream is from stream or reader cache
}
}