SerializableEntity.java revision 069490a5ca2fd1988d29daf45d892f47ad665115 
1/*
2 * $HeadURL: http://svn.apache.org/repos/asf/httpcomponents/httpcore/trunk/module-main/src/main/java/org/apache/http/entity/SerializableEntity.java $
3 * $Revision: 647816 $
4 * $Date: 2008-04-14 07:37:13 -0700 (Mon, 14 Apr 2008) $
5 *
6 * ====================================================================
7 * Licensed to the Apache Software Foundation (ASF) under one
8 * or more contributor license agreements.  See the NOTICE file
9 * distributed with this work for additional information
10 * regarding copyright ownership.  The ASF licenses this file
11 * to you under the Apache License, Version 2.0 (the
12 * "License"); you may not use this file except in compliance
13 * with the License.  You may obtain a copy of the License at
14 *
15 *   http://www.apache.org/licenses/LICENSE-2.0
16 *
17 * Unless required by applicable law or agreed to in writing,
18 * software distributed under the License is distributed on an
19 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
20 * KIND, either express or implied.  See the License for the
21 * specific language governing permissions and limitations
22 * under the License.
23 * ====================================================================
24 *
25 * This software consists of voluntary contributions made by many
26 * individuals on behalf of the Apache Software Foundation.  For more
27 * information on the Apache Software Foundation, please see
28 * <http://www.apache.org/>.
29 *
30 */
31
32package org.apache.http.entity;
33
34import java.io.ByteArrayInputStream;
35import java.io.ByteArrayOutputStream;
36import java.io.IOException;
37import java.io.InputStream;
38import java.io.ObjectOutputStream;
39import java.io.OutputStream;
40import java.io.Serializable;
41
42public class SerializableEntity extends AbstractHttpEntity {
43
44    private byte[] objSer;
45
46    private Serializable objRef;
47
48    public SerializableEntity(Serializable ser, boolean bufferize) throws IOException {
49        super();
50        if (ser == null) {
51            throw new IllegalArgumentException("Source object may not be null");
52        }
53
54        if (bufferize) {
55            createBytes(ser);
56        } else {
57            this.objRef = ser;
58        }
59    }
60
61    private void createBytes(Serializable ser) throws IOException {
62        ByteArrayOutputStream baos = new ByteArrayOutputStream();
63        ObjectOutputStream out = new ObjectOutputStream(baos);
64        out.writeObject(ser);
65        out.flush();
66        this.objSer = baos.toByteArray();
67    }
68
69    public InputStream getContent() throws IOException, IllegalStateException {
70        if (this.objSer == null) {
71            createBytes(this.objRef);
72        }
73        return new ByteArrayInputStream(this.objSer);
74    }
75
76    public long getContentLength() {
77        if (this.objSer ==  null) {
78            return -1;
79        } else {
80            return this.objSer.length;
81        }
82    }
83
84    public boolean isRepeatable() {
85        return true;
86    }
87
88    public boolean isStreaming() {
89        return this.objSer == null;
90    }
91
92    public void writeTo(OutputStream outstream) throws IOException {
93        if (outstream == null) {
94            throw new IllegalArgumentException("Output stream may not be null");
95        }
96
97        if (this.objSer == null) {
98            ObjectOutputStream out = new ObjectOutputStream(outstream);
99            out.writeObject(this.objRef);
100            out.flush();
101        } else {
102            outstream.write(this.objSer);
103            outstream.flush();
104        }
105    }
106
107}
108