briar/components/net/sf/briar/protocol/SubscriptionUpdateReader.java

package net.sf.briar.protocol;

import java.io.IOException;
import java.util.Map;

import net.sf.briar.api.FormatException;
import net.sf.briar.api.protocol.Group;
import net.sf.briar.api.protocol.ProtocolConstants;
import net.sf.briar.api.protocol.SubscriptionUpdate;
import net.sf.briar.api.protocol.Types;
import net.sf.briar.api.serial.Consumer;
import net.sf.briar.api.serial.CountingConsumer;
import net.sf.briar.api.serial.ObjectReader;
import net.sf.briar.api.serial.Reader;

class SubscriptionUpdateReader implements ObjectReader<SubscriptionUpdate> {

	private final ObjectReader<Group> groupReader;
	private final SubscriptionUpdateFactory subscriptionFactory;

	SubscriptionUpdateReader(ObjectReader<Group> groupReader,
			SubscriptionUpdateFactory subscriptionFactory) {
		this.groupReader = groupReader;
		this.subscriptionFactory = subscriptionFactory;
	}

	public SubscriptionUpdate readObject(Reader r) throws IOException {
		// Initialise the consumer
		Consumer counting =
			new CountingConsumer(ProtocolConstants.MAX_PACKET_LENGTH);
		// Read the data
		r.addConsumer(counting);
		r.readStructId(Types.SUBSCRIPTION_UPDATE);
		r.addObjectReader(Types.GROUP, groupReader);
		Map<Group, Long> subs = r.readMap(Group.class, Long.class);
		r.removeObjectReader(Types.GROUP);
		long timestamp = r.readInt64();
		if(timestamp < 0L) throw new FormatException();
		r.removeConsumer(counting);
		// Build and return the subscription update
		return subscriptionFactory.createSubscriptions(subs, timestamp);
	}
}