01: /*
02: * This file is part of PFIXCORE.
03: *
04: * PFIXCORE is free software; you can redistribute it and/or modify
05: * it under the terms of the GNU Lesser General Public License as published by
06: * the Free Software Foundation; either version 2 of the License, or
07: * (at your option) any later version.
08: *
09: * PFIXCORE is distributed in the hope that it will be useful,
10: * but WITHOUT ANY WARRANTY; without even the implied warranty of
11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12: * GNU Lesser General Public License for more details.
13: *
14: * You should have received a copy of the GNU Lesser General Public License
15: * along with PFIXCORE; if not, write to the Free Software
16: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17: *
18: */
19:
20: package de.schlund.pfixcore.webservice.utils;
21:
22: import java.io.IOException;
23: import java.io.OutputStream;
24: import java.io.UnsupportedEncodingException;
25: import java.io.Writer;
26:
27: import de.schlund.pfixcore.webservice.ServiceResponse;
28: import de.schlund.pfixcore.webservice.ServiceResponseWrapper;
29:
30: /**
31: * @author mleidig@schlund.de
32: */
33: public class RecordingResponseWrapper extends ServiceResponseWrapper {
34:
35: String message;
36: RecordingOutputStream recStream;
37: RecordingWriter recWriter;
38:
39: public RecordingResponseWrapper(ServiceResponse response) {
40: super (response);
41: }
42:
43: @Override
44: public void setMessage(String message) throws IOException {
45: this .message = message;
46: super .setMessage(message);
47: }
48:
49: @Override
50: public Writer getMessageWriter() throws IOException {
51: recWriter = new RecordingWriter(super .getMessageWriter());
52: return recWriter;
53: }
54:
55: @Override
56: public OutputStream getMessageStream() throws IOException {
57: recStream = new RecordingOutputStream(super .getMessageStream());
58: return recStream;
59: }
60:
61: public String getRecordedMessage() {
62: if (message != null)
63: return message;
64: if (recStream != null) {
65: byte[] bytes = recStream.getBytes();
66: String charset = super .getCharacterEncoding();
67: if (charset == null)
68: charset = "UTF-8";
69: try {
70: return new String(bytes, charset);
71: } catch (UnsupportedEncodingException x) {
72: throw new RuntimeException("Encoding '" + charset
73: + "' not supported.");
74: }
75: }
76: if (recWriter != null) {
77: char[] chars = recWriter.getCharacters();
78: return new String(chars);
79: }
80: return null;
81: }
82:
83: public void setRecordedMessage(String message) {
84: this.message = message;
85: }
86:
87: }
|