ProtocolResponseProcessor.java

package edu.uw.zookeeper.server;

import edu.uw.zookeeper.common.Processor;
import edu.uw.zookeeper.common.Processors;
import edu.uw.zookeeper.data.TxnOperation;
import edu.uw.zookeeper.protocol.Message;
import edu.uw.zookeeper.protocol.Operation;
import edu.uw.zookeeper.protocol.ProtocolResponseMessage;
import edu.uw.zookeeper.protocol.proto.Records;

public class ProtocolResponseProcessor implements Processor<TxnOperation.Request<?>, Message.ServerResponse<?>> {

    public static ProtocolResponseProcessor create(
            Processors.UncheckedProcessor<TxnOperation.Request<?>, Records.Response> processor) {
        return new ProtocolResponseProcessor(processor);
    }
    
    protected final Processors.UncheckedProcessor<TxnOperation.Request<?>, Records.Response> delegate;
    
    public ProtocolResponseProcessor(
            Processors.UncheckedProcessor<TxnOperation.Request<?>, Records.Response> delegate) {
        this.delegate = delegate;
    }
    
    @Override
    public Message.ServerResponse<Records.Response> apply(TxnOperation.Request<?> input) {
        Records.Response response = delegate.apply(input);

        int xid;
        if (response instanceof Operation.RequestId) {
            xid = ((Operation.RequestId) response).xid();
        } else {
            xid = input.xid();
        }
        long zxid = input.zxid();
        return ProtocolResponseMessage.of(xid, zxid, response);
    }
}