01: /**
02: * Licensed to the Apache Software Foundation (ASF) under one
03: * or more contributor license agreements. See the NOTICE file
04: * distributed with this work for additional information
05: * regarding copyright ownership. The ASF licenses this file
06: * to you under the Apache License, Version 2.0 (the
07: * "License"); you may not use this file except in compliance
08: * with the License. You may obtain a copy of the License at
09: *
10: * http://www.apache.org/licenses/LICENSE-2.0
11: *
12: * Unless required by applicable law or agreed to in writing,
13: * software distributed under the License is distributed on an
14: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15: * KIND, either express or implied. See the License for the
16: * specific language governing permissions and limitations
17: * under the License.
18: */package org.apache.cxf.ws.rm;
19:
20: import java.io.OutputStream;
21:
22: import org.apache.cxf.interceptor.Fault;
23: import org.apache.cxf.interceptor.StaxOutInterceptor;
24: import org.apache.cxf.io.WriteOnCloseOutputStream;
25: import org.apache.cxf.message.Message;
26: import org.apache.cxf.phase.AbstractPhaseInterceptor;
27: import org.apache.cxf.phase.Phase;
28:
29: /**
30: *
31: */
32: public class RetransmissionInterceptor extends AbstractPhaseInterceptor {
33:
34: RMManager manager;
35:
36: public RetransmissionInterceptor() {
37: super (Phase.PRE_STREAM);
38: addBefore(StaxOutInterceptor.class.getName());
39: }
40:
41: public RMManager getManager() {
42: return manager;
43: }
44:
45: public void setManager(RMManager manager) {
46: this .manager = manager;
47: }
48:
49: public void handleMessage(Message message) throws Fault {
50: handle(message, false);
51: }
52:
53: @Override
54: public void handleFault(Message message) {
55: handle(message, true);
56: }
57:
58: void handle(Message message, boolean isFault) {
59: if (null == getManager().getRetransmissionQueue()) {
60: return;
61: }
62:
63: OutputStream os = message.getContent(OutputStream.class);
64: if (null == os) {
65: return;
66: }
67:
68: WriteOnCloseOutputStream stream = RMUtils.createCachedStream(
69: message, os);
70: stream.registerCallback(new RetransmissionCallback(message,
71: getManager()));
72: }
73: }
|