ie.omk.smpp.message.DestinationTable Maven / Gradle / Ivy

Go to download
package ie.omk.smpp.message;

import ie.omk.smpp.Address;
import ie.omk.smpp.util.SMPPIO;

import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.ListIterator;

import org.apache.commons.logging.LogFactory;

public class DestinationTable implements Cloneable {
    private List dests;

    private int length;

    DestinationTable() {
        dests = new ArrayList();
    }

    synchronized void add(Address addr) {
        dests.add(addr);
        // Plus 1 for the dest type flag.
        length += addr.getLength() + 1;
    }

    synchronized void add(String distList) {
        dests.add(distList);
        // nul byte plus dest type flag
        length += distList.length() + 2;
    }

    public synchronized void remove(Address addr) {
        int i = dests.indexOf(addr);
        if (i > -1) {
            length -= ((Address) dests.remove(i)).getLength() + 1;
        }
    }

    public synchronized void remove(String distList) {
        int i = dests.indexOf(distList);
        if (i > -1) {
            length -= ((String) dests.remove(i)).length() + 2;
        }
    }

    public Iterator iterator() {
        return Collections.unmodifiableList(dests).iterator();
    }

    public ListIterator listIterator() {
        return Collections.unmodifiableList(dests).listIterator();
    }

    public synchronized int getLength() {
        return length;
    }

    public int size() {
        return dests.size();
    }

    public synchronized void writeTo(OutputStream out)
            throws java.io.IOException {
        Iterator i = dests.iterator();
        while (i.hasNext()) {
            Object o = i.next();
            if (o instanceof Address) {
                SMPPIO.writeInt(1, 1, out);
                ((Address) o).writeTo(out);
            } else {
                SMPPIO.writeInt(2, 1, out);
                SMPPIO.writeCString((String) o, out);
            }
        }
    }

    public synchronized void readFrom(byte[] table, int offset, int count) {
        for (int i = 0; i < count; i++) {
            int type = SMPPIO.bytesToInt(table, offset++, 1);
            if (type == 1) {
                // SME address..
                Address a = new Address();
                a.readFrom(table, offset);
                offset += a.getLength();
                dests.add(a);
            } else if (type == 2) {
                // Distribution list name
                String d = SMPPIO.readCString(table, offset);
                offset += d.length() + 1;
                dests.add(d);
            } else {
                LogFactory.getLog(DestinationTable.class).warn(
                        "Unidentified destination type on input.");
            }
        }

        calculateLength();
    }

    public Object clone() {
        try {
            return super.clone();
        } catch (CloneNotSupportedException x) {
            throw new RuntimeException("Clone not supported", x);
        }
    }
    
    private void calculateLength() {
        length = 0;

        Iterator i = dests.iterator();
        while (i.hasNext()) {
            Object o = i.next();
            // For the destination type flag
            length++;
            if (o instanceof Address) {
                length += ((Address) o).getLength();
            } else {
                length += ((String) o).length() + 1;
            }
        }
    }
}