PacketDelivererImpl.java 2.87 KB
Newer Older
Matt Tucker's avatar
Matt Tucker committed
1 2 3 4 5
/**
 * $RCSfile$
 * $Revision$
 * $Date$
 *
Matt Tucker's avatar
Matt Tucker committed
6
 * Copyright (C) 2004 Jive Software. All rights reserved.
Matt Tucker's avatar
Matt Tucker committed
7
 *
Matt Tucker's avatar
Matt Tucker committed
8 9
 * This software is published under the terms of the GNU Public License (GPL),
 * a copy of which is included in this distribution.
Matt Tucker's avatar
Matt Tucker committed
10
 */
Matt Tucker's avatar
Matt Tucker committed
11

Matt Tucker's avatar
Matt Tucker committed
12 13 14
package org.jivesoftware.messenger.spi;

import org.jivesoftware.messenger.*;
15
import org.jivesoftware.messenger.component.InternalComponentManager;
Matt Tucker's avatar
Matt Tucker committed
16
import org.jivesoftware.messenger.auth.UnauthorizedException;
17
import org.jivesoftware.messenger.container.BasicModule;
Matt Tucker's avatar
Matt Tucker committed
18
import org.jivesoftware.messenger.net.SocketPacketWriteHandler;
19 20
import org.xmpp.component.Component;
import org.xmpp.packet.JID;
Matt Tucker's avatar
Matt Tucker committed
21 22
import org.xmpp.packet.Packet;

Matt Tucker's avatar
Matt Tucker committed
23 24 25 26 27 28 29 30 31 32
/**
 * In-memory implementation of the packet deliverer service
 *
 * @author Iain Shigeoka
 */
public class PacketDelivererImpl extends BasicModule implements PacketDeliverer {

    /**
     * The handler that does the actual delivery (could be a channel instead)
     */
33
    protected SocketPacketWriteHandler deliverHandler;
Matt Tucker's avatar
Matt Tucker committed
34

35 36
    private OfflineMessageStrategy messageStrategy;
    private SessionManager sessionManager;
Matt Tucker's avatar
Matt Tucker committed
37

38 39
    private InternalComponentManager componentManager;

Matt Tucker's avatar
Matt Tucker committed
40 41
    public PacketDelivererImpl() {
        super("Packet Delivery");
42
        componentManager = InternalComponentManager.getInstance();
Matt Tucker's avatar
Matt Tucker committed
43 44
    }

Derek DeMoro's avatar
Derek DeMoro committed
45
    public void deliver(Packet packet) throws UnauthorizedException, PacketException {
Matt Tucker's avatar
Matt Tucker committed
46 47 48 49
        if (packet == null) {
            throw new PacketException("Packet was null");
        }
        if (deliverHandler == null) {
Derek DeMoro's avatar
Derek DeMoro committed
50
            throw new PacketException("Could not send packet - no route" + packet.toString());
Matt Tucker's avatar
Matt Tucker committed
51
        }
52 53 54 55 56 57 58 59 60 61 62 63 64
        // Check if the target of the packet is the a component itself. If it does then just pass
        // the packet to the component for processing it
        JID recipient = packet.getTo();
        if (recipient != null && recipient.getNode() == null && recipient.getResource() == null) {
            // Check for registered components
            Component component = componentManager.getComponent(recipient);
            if (component != null) {
                component.processPacket(packet);
                return;
            }
        }
        // Let the SocketPacketWriteHandler process the packet. SocketPacketWriteHandler may send
        // it over the socket or store it when user is offline or drop it.
Matt Tucker's avatar
Matt Tucker committed
65 66 67
        deliverHandler.process(packet);
    }

68 69 70 71
    public void initialize(XMPPServer server) {
        super.initialize(server);
        messageStrategy = server.getOfflineMessageStrategy();
        sessionManager = server.getSessionManager();
Matt Tucker's avatar
Matt Tucker committed
72 73
    }

74 75
    public void start() throws IllegalStateException {
        super.start();
76 77 78
        deliverHandler =
                new SocketPacketWriteHandler(sessionManager,
                        XMPPServer.getInstance().getRoutingTable(), messageStrategy);
Matt Tucker's avatar
Matt Tucker committed
79 80
    }

81 82 83
    public void stop() {
        super.stop();
        deliverHandler = null;
Matt Tucker's avatar
Matt Tucker committed
84 85
    }
}