/**
* Copyright 2005-2014 Restlet
*
* The contents of this file are subject to the terms of one of the following
* open source licenses: Apache 2.0 or or EPL 1.0 (the "Licenses"). You can
* select the license that you prefer but you may not use this file except in
* compliance with one of these Licenses.
*
* You can obtain a copy of the Apache 2.0 license at
* http://www.opensource.org/licenses/apache-2.0
*
* You can obtain a copy of the EPL 1.0 license at
* http://www.opensource.org/licenses/eclipse-1.0
*
* See the Licenses for the specific language governing permissions and
* limitations under the Licenses.
*
* Alternatively, you can obtain a royalty free commercial license with less
* limitations, transferable or non-transferable, directly at
* http://restlet.com/products/restlet-framework
*
* Restlet is a registered trademark of Restlet S.A.S.
*/
package org.restlet.representation;
import java.io.IOException;
import java.io.Reader;
import java.io.Writer;
import java.util.logging.Level;
import org.restlet.Context;
import org.restlet.data.MediaType;
import org.restlet.engine.io.IoUtils;
/**
* Transient representation based on a BIO characters reader.
*
* @author Jerome Louvel
*/
public class ReaderRepresentation extends CharacterRepresentation {
/** The representation's reader. */
private volatile Reader reader;
/**
* Constructor.
*
* @param reader
* The representation's stream.
*/
public ReaderRepresentation(Reader reader) {
this(reader, null);
}
/**
* Constructor.
*
* @param reader
* The representation's stream.
* @param mediaType
* The representation's media type.
*/
public ReaderRepresentation(Reader reader, MediaType mediaType) {
this(reader, mediaType, UNKNOWN_SIZE);
}
/**
* Constructor.
*
* @param reader
* The representation's stream.
* @param mediaType
* The representation's media type.
* @param expectedSize
* The expected reader size in bytes.
*/
public ReaderRepresentation(Reader reader, MediaType mediaType,
long expectedSize) {
super(mediaType);
setSize(expectedSize);
setTransient(true);
setReader(reader);
}
@Override
public Reader getReader() throws IOException {
final Reader result = this.reader;
setReader(null);
return result;
}
/**
* Note that this method relies on {@link #getStream()}. This stream is
* closed once fully read.
*/
@Override
public String getText() throws IOException {
return IoUtils.toString(getStream(), getCharacterSet());
}
/**
* Closes and releases the input stream.
*/
@Override
public void release() {
if (this.reader != null) {
try {
this.reader.close();
} catch (IOException e) {
Context.getCurrentLogger().log(Level.WARNING,
"Error while releasing the representation.", e);
}
this.reader = null;
}
super.release();
}
/**
* Sets the reader to use.
*
* @param reader
* The reader to use.
*/
public void setReader(Reader reader) {
this.reader = reader;
setAvailable(reader != null);
}
@Override
public void write(Writer writer) throws IOException {
IoUtils.copy(getReader(), writer);
}
}