KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > apache > activemq > transport > reliable > UnreliableCommandDatagramSocket


1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements. See the NOTICE file distributed with
4  * this work for additional information regarding copyright ownership.
5  * The ASF licenses this file to You under the Apache License, Version 2.0
6  * (the "License"); you may not use this file except in compliance with
7  * the License. You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 package org.apache.activemq.transport.reliable;
18
19 import org.apache.activemq.openwire.OpenWireFormat;
20 import org.apache.activemq.transport.udp.CommandDatagramSocket;
21 import org.apache.activemq.transport.udp.DatagramHeaderMarshaller;
22 import org.apache.activemq.transport.udp.UdpTransport;
23 import org.apache.commons.logging.Log;
24 import org.apache.commons.logging.LogFactory;
25
26 import java.io.IOException JavaDoc;
27 import java.net.DatagramSocket JavaDoc;
28 import java.net.SocketAddress JavaDoc;
29
30 /**
31  *
32  * @version $Revision: $
33  */
34 public class UnreliableCommandDatagramSocket extends CommandDatagramSocket {
35     private static final Log log = LogFactory.getLog(UnreliableCommandDatagramSocket.class);
36
37     private DropCommandStrategy dropCommandStrategy;
38
39     public UnreliableCommandDatagramSocket(UdpTransport transport, OpenWireFormat wireFormat, int datagramSize,
40             SocketAddress JavaDoc targetAddress, DatagramHeaderMarshaller headerMarshaller, DatagramSocket channel,
41             DropCommandStrategy strategy) {
42         super(transport, wireFormat, datagramSize, targetAddress, headerMarshaller, channel);
43         this.dropCommandStrategy = strategy;
44     }
45
46     protected void sendWriteBuffer(int commandId, SocketAddress JavaDoc address, byte[] data, boolean redelivery)
47             throws IOException JavaDoc {
48         if (dropCommandStrategy.shouldDropCommand(commandId, address, redelivery)) {
49             log.info("Dropping datagram with command: " + commandId);
50
51             // lets still add it to the replay buffer though!
52 ReplayBuffer bufferCache = getReplayBuffer();
53             if (bufferCache != null && !redelivery) {
54                 bufferCache.addBuffer(commandId, data);
55             }
56         }
57         else {
58             super.sendWriteBuffer(commandId, address, data, redelivery);
59         }
60     }
61 }
62
Popular Tags