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.InputStream;
24: import java.io.Reader;
25: import java.io.UnsupportedEncodingException;
26:
27: import de.schlund.pfixcore.webservice.ServiceRequest;
28: import de.schlund.pfixcore.webservice.ServiceRequestWrapper;
29:
30: /**
31: * @author mleidig@schlund.de
32: */
33: public class RecordingRequestWrapper extends ServiceRequestWrapper {
34:
35: String message;
36: RecordingInputStream recStream;
37: RecordingReader recReader;
38:
39: public RecordingRequestWrapper(ServiceRequest request) {
40: super (request);
41: }
42:
43: @Override
44: public String getMessage() throws IOException {
45: message = super .getMessage();
46: return message;
47: }
48:
49: @Override
50: public Reader getMessageReader() throws IOException {
51: recReader = new RecordingReader(super .getMessageReader());
52: return recReader;
53: }
54:
55: @Override
56: public InputStream getMessageStream() throws IOException {
57: recStream = new RecordingInputStream(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 (recReader != null) {
77: char[] chars = recReader.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: }
|