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 */
017package org.apache.camel.impl;
018
019import java.io.IOException;
020import java.io.InputStream;
021import java.io.OutputStream;
022
023import org.apache.camel.Exchange;
024import org.apache.camel.spi.DataFormat;
025import org.apache.camel.util.ExchangeHelper;
026
027/**
028 * The text based <a href="http://camel.apache.org/data-format.html">data format</a> supporting
029 * charset encoding.
030 *
031 * @version 
032 */
033public class StringDataFormat implements DataFormat {
034
035    private String charset;
036
037    public StringDataFormat() {
038    }
039
040    public StringDataFormat(String charset) {
041        this.charset = charset;
042    }
043
044    public String getCharset() {
045        return charset;
046    }
047
048    public void setCharset(String charset) {
049        this.charset = charset;
050    }
051
052    public void marshal(Exchange exchange, Object graph, OutputStream stream) throws IOException {
053        String text = ExchangeHelper.convertToType(exchange, String.class, graph);
054
055        byte[] bytes;
056        if (charset != null) {
057            bytes = text.getBytes(charset);
058        } else {
059            bytes = text.getBytes();
060        }
061        stream.write(bytes);
062    }
063
064    public Object unmarshal(Exchange exchange, InputStream stream) throws IOException {
065        byte[] bytes = ExchangeHelper.convertToType(exchange, byte[].class, stream);
066
067        String answer;
068        if (charset != null) {
069            answer = new String(bytes, charset);
070        } else {
071            answer = new String(bytes);
072        }
073
074        return answer;
075    }
076    
077}