Java BinaryOutputArchive.writeRecord方法代码示例

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
private void send(DataOutputStream dout, byte[] response)
        throws IOException {
    QuorumAuthPacket authPacket;
    BufferedOutputStream bufferedOutput = new BufferedOutputStream(dout);
    BinaryOutputArchive boa = BinaryOutputArchive
            .getArchive(bufferedOutput);
    if (response != null && response.length < 0) {
        throw new IOException("Response length < 0");
    } else if (response == null) {
        authPacket = QuorumAuth.createPacket(
                QuorumAuth.Status.IN_PROGRESS, response);
    } else {
        authPacket = QuorumAuth.createPacket(
                QuorumAuth.Status.IN_PROGRESS, response);
    }

    boa.writeRecord(authPacket, QuorumAuth.QUORUM_AUTH_MESSAGE_TAG);
    bufferedOutput.flush();
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
private void send(DataOutputStream dout, byte[] challenge,
        QuorumAuth.Status s) throws IOException {
    BufferedOutputStream bufferedOutput = new BufferedOutputStream(dout);
    BinaryOutputArchive boa = BinaryOutputArchive
            .getArchive(bufferedOutput);
    QuorumAuthPacket authPacket;
    if (challenge != null && challenge.length < 0) {
        throw new IOException("Response length < 0");
    } else if (challenge == null && s != QuorumAuth.Status.SUCCESS) {
        authPacket = QuorumAuth.createPacket(
                QuorumAuth.Status.IN_PROGRESS, challenge);
    } else {
        authPacket = QuorumAuth.createPacket(s, challenge);
    }

    boa.writeRecord(authPacket, QuorumAuth.QUORUM_AUTH_MESSAGE_TAG);
    bufferedOutput.flush();
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
/**
 * SaslQuorumAuthServer throws exception on receiving an invalid quorum
 * auth packet.
 */
@Test(timeout = 30000)
public void testSaslQuorumAuthServerWithInvalidQuorumAuthPacket()
        throws Exception {
    Socket socket = getSocketPair();
    DataOutputStream dout = new DataOutputStream(socket.getOutputStream());
    BufferedOutputStream bufferedOutput = new BufferedOutputStream(dout);
    BinaryOutputArchive boa = BinaryOutputArchive
            .getArchive(bufferedOutput);
    QuorumAuthPacket authPacket = QuorumAuth
            .createPacket(QuorumAuth.Status.IN_PROGRESS, null);
    authPacket.setMagic(Long.MAX_VALUE); // invalid magic number
    boa.writeRecord(authPacket, null);
    bufferedOutput.flush();
    QuorumAuthServer authServer = new SaslQuorumAuthServer(true,
            "QuorumServer", authzHosts);
    BufferedInputStream is = new BufferedInputStream(
            socket.getInputStream());
    try {
        authServer.authenticate(socket, new DataInputStream(is));
        Assert.fail("Must throw exception as QuorumAuthPacket is invalid");
    } catch (SaslException e) {
        // expected
    }
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
/**
 * NullQuorumAuthServer should return true on receiving a valid quorum auth
 * packet.
 */
@Test(timeout = 30000)
public void testNullQuorumAuthServerWithValidQuorumAuthPacket()
        throws Exception {
    Socket socket = getSocketPair();
    DataOutputStream dout = new DataOutputStream(socket.getOutputStream());
    BufferedOutputStream bufferedOutput = new BufferedOutputStream(dout);
    BinaryOutputArchive boa = BinaryOutputArchive
            .getArchive(bufferedOutput);
    QuorumAuthPacket authPacket = QuorumAuth
            .createPacket(QuorumAuth.Status.IN_PROGRESS, null);
    boa.writeRecord(authPacket, null);
    bufferedOutput.flush();
    QuorumAuthServer authServer = new NullQuorumAuthServer();
    BufferedInputStream is = new BufferedInputStream(
            socket.getInputStream());
    // It will throw exception and fail the
    // test if any unexpected error. Not adding any extra assertion.
    authServer.authenticate(socket, new DataInputStream(is));
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
@Override
public void sendResponse(ReplyHeader h, Record r, String tag)
        throws IOException {
    if (!channel.isOpen()) {
        return;
    }
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    // Make space for length
    BinaryOutputArchive bos = BinaryOutputArchive.getArchive(baos);
    try {
        baos.write(fourBytes);
        bos.writeRecord(h, "header");
        if (r != null) {
            bos.writeRecord(r, tag);
        }
        baos.close();
    } catch (IOException e) {
        LOG.error("Error serializing response");
    }
    byte b[] = baos.toByteArray();
    ByteBuffer bb = ByteBuffer.wrap(b);
    bb.putInt(b.length - 4).rewind();
    sendBuffer(bb);
    if (h.getXid() > 0) {
        // zks cannot be null otherwise we would not have gotten here!
        if (!zkServer.shouldThrottle(outstandingCount.decrementAndGet())) {
            enableRecv();
        }
    }
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
@Override
public void sendResponse(ReplyHeader h, Record r, String tag)
        throws IOException {
    if (closingChannel || !channel.isOpen()) {
        return;
    }
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    // Make space for length
    BinaryOutputArchive bos = BinaryOutputArchive.getArchive(baos);
    try {
        baos.write(fourBytes);
        bos.writeRecord(h, "header");
        if (r != null) {
            bos.writeRecord(r, tag);
        }
        baos.close();
    } catch (IOException e) {
        LOG.error("Error serializing response");
    }
    byte b[] = baos.toByteArray();
    ByteBuffer bb = ByteBuffer.wrap(b);
    bb.putInt(b.length - 4).rewind();
    sendBuffer(bb);
    if (h.getXid() > 0) {
        // zks cannot be null otherwise we would not have gotten here!
        if (!zkServer.shouldThrottle(outstandingCount.decrementAndGet())) {
            enableRecv();
        }
    }
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
/**
   * Once connected to the leader, perform the handshake protocol to
   * establish a following / observing connection. 
   * @param pktType
   * @return the zxid the Leader sends for synchronization purposes.
   * @throws IOException
   */
  protected long registerWithLeader(int pktType) throws IOException{
      /*
       * Send follower info, including last zxid and sid
       */
  	long lastLoggedZxid = self.getLastLoggedZxid();
      QuorumPacket qp = new QuorumPacket();                
      qp.setType(pktType);
      qp.setZxid(ZxidUtils.makeZxid(self.getAcceptedEpoch(), 0));
      
      /*
       * Add sid to payload
       */
      LearnerInfo li = new LearnerInfo(self.getId(), 0x10000);
      ByteArrayOutputStream bsid = new ByteArrayOutputStream();
      BinaryOutputArchive boa = BinaryOutputArchive.getArchive(bsid);
      boa.writeRecord(li, "LearnerInfo");
      qp.setData(bsid.toByteArray());
      
      writePacket(qp, true);
      readPacket(qp);        
      final long newEpoch = ZxidUtils.getEpochFromZxid(qp.getZxid());
if (qp.getType() == Leader.LEADERINFO) {
      	// we are connected to a 1.0 server so accept the new epoch and read the next packet
      	leaderProtocolVersion = ByteBuffer.wrap(qp.getData()).getInt();
      	byte epochBytes[] = new byte[4];
      	final ByteBuffer wrappedEpochBytes = ByteBuffer.wrap(epochBytes);
      	if (newEpoch > self.getAcceptedEpoch()) {
      		wrappedEpochBytes.putInt((int)self.getCurrentEpoch());
      		self.setAcceptedEpoch(newEpoch);
      	} else if (newEpoch == self.getAcceptedEpoch()) {
      		// since we have already acked an epoch equal to the leaders, we cannot ack
      		// again, but we still need to send our lastZxid to the leader so that we can
      		// sync with it if it does assume leadership of the epoch.
      		// the -1 indicates that this reply should not count as an ack for the new epoch
              wrappedEpochBytes.putInt(-1);
      	} else {
      		throw new IOException("Leaders epoch, " + newEpoch + " is less than accepted epoch, " + self.getAcceptedEpoch());
      	}
      	QuorumPacket ackNewEpoch = new QuorumPacket(Leader.ACKEPOCH, lastLoggedZxid, epochBytes, null);
      	writePacket(ackNewEpoch, true);
          return ZxidUtils.makeZxid(newEpoch, 0);
      } else {
      	if (newEpoch > self.getAcceptedEpoch()) {
      		self.setAcceptedEpoch(newEpoch);
      	}
          if (qp.getType() != Leader.NEWLEADER) {
              LOG.error("First packet should have been NEWLEADER");
              throw new IOException("First packet should have been NEWLEADER");
          }
          return qp.getZxid();
      }
  }
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
@Test
public void syncTest() throws Exception {
	File tmpFile = File.createTempFile("test", ".dir", testData);
	tmpFile.delete();
	try {
		FileTxnSnapLog ftsl = new FileTxnSnapLog(tmpFile, tmpFile);
		SimpleLearner sl = new SimpleLearner(ftsl);
		long startZxid = sl.zk.getLastProcessedZxid();
		
		// Set up bogus streams
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		BinaryOutputArchive oa = BinaryOutputArchive.getArchive(baos);
		sl.leaderOs = BinaryOutputArchive.getArchive(new ByteArrayOutputStream());
		
		// make streams and socket do something innocuous
		sl.bufferedOutput = new BufferedOutputStream(System.out);
		sl.sock = new Socket();
		
		// fake messages from the server
		QuorumPacket qp = new QuorumPacket(Leader.SNAP, 0, null, null);
		oa.writeRecord(qp, null);
		sl.zk.getZKDatabase().serializeSnapshot(oa);
		oa.writeString("BenWasHere", "signature");
		TxnHeader hdr = new TxnHeader(0, 0, 0, 0, ZooDefs.OpCode.create);
		CreateTxn txn = new CreateTxn("/foo", new byte[0], new ArrayList<ACL>(), false, sl.zk.getZKDatabase().getNode("/").stat.getCversion());
        ByteArrayOutputStream tbaos = new ByteArrayOutputStream();
        BinaryOutputArchive boa = BinaryOutputArchive.getArchive(tbaos);
        hdr.serialize(boa, "hdr");
        txn.serialize(boa, "txn");
        tbaos.close();
		qp = new QuorumPacket(Leader.PROPOSAL, 1, tbaos.toByteArray(), null);
		oa.writeRecord(qp, null);

		// setup the messages to be streamed to follower
		sl.leaderIs = BinaryInputArchive.getArchive(new ByteArrayInputStream(baos.toByteArray()));
		
		try {
			sl.syncWithLeader(3);
		} catch(EOFException e) {}
		
		sl.zk.shutdown();
		sl = new SimpleLearner(ftsl);
		Assert.assertEquals(startZxid, sl.zk.getLastProcessedZxid());
	} finally {
		recursiveDelete(tmpFile);
	}
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
/**
   * Once connected to the leader, perform the handshake protocol to
   * establish a following / observing connection. 
   * @param pktType
   * @return the zxid the Leader sends for synchronization purposes.
   * @throws IOException
   */
  protected long registerWithLeader(int pktType) throws IOException{
      /*
       * Send follower info, including last zxid and sid
       */
  	long lastLoggedZxid = self.getLastLoggedZxid();
      QuorumPacket qp = new QuorumPacket();                
      qp.setType(pktType);
      qp.setZxid(ZxidUtils.makeZxid(self.getAcceptedEpoch(), 0));
      
      /*
       * Add sid to payload
       */
      LearnerInfo li = new LearnerInfo(self.getId(), 0x10000, self.getQuorumVerifier().getVersion());
      ByteArrayOutputStream bsid = new ByteArrayOutputStream();
      BinaryOutputArchive boa = BinaryOutputArchive.getArchive(bsid);
      boa.writeRecord(li, "LearnerInfo");
      qp.setData(bsid.toByteArray());
      
      writePacket(qp, true);
      readPacket(qp);        
      final long newEpoch = ZxidUtils.getEpochFromZxid(qp.getZxid());
if (qp.getType() == Leader.LEADERINFO) {
      	// we are connected to a 1.0 server so accept the new epoch and read the next packet
      	leaderProtocolVersion = ByteBuffer.wrap(qp.getData()).getInt();
      	byte epochBytes[] = new byte[4];
      	final ByteBuffer wrappedEpochBytes = ByteBuffer.wrap(epochBytes);
      	if (newEpoch > self.getAcceptedEpoch()) {
      		wrappedEpochBytes.putInt((int)self.getCurrentEpoch());
      		self.setAcceptedEpoch(newEpoch);
      	} else if (newEpoch == self.getAcceptedEpoch()) {
      		// since we have already acked an epoch equal to the leaders, we cannot ack
      		// again, but we still need to send our lastZxid to the leader so that we can
      		// sync with it if it does assume leadership of the epoch.
      		// the -1 indicates that this reply should not count as an ack for the new epoch
              wrappedEpochBytes.putInt(-1);
      	} else {
      		throw new IOException("Leaders epoch, " + newEpoch + " is less than accepted epoch, " + self.getAcceptedEpoch());
      	}
      	QuorumPacket ackNewEpoch = new QuorumPacket(Leader.ACKEPOCH, lastLoggedZxid, epochBytes, null);
      	writePacket(ackNewEpoch, true);
          return ZxidUtils.makeZxid(newEpoch, 0);
      } else {
      	if (newEpoch > self.getAcceptedEpoch()) {
      		self.setAcceptedEpoch(newEpoch);
      	}
          if (qp.getType() != Leader.NEWLEADER) {
              LOG.error("First packet should have been NEWLEADER");
              throw new IOException("First packet should have been NEWLEADER");
          }
          return qp.getZxid();
      }
  }
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
@Test
public void syncTest() throws Exception {
    File tmpFile = File.createTempFile("test", ".dir", testData);
    tmpFile.delete();
    try {
        FileTxnSnapLog ftsl = new FileTxnSnapLog(tmpFile, tmpFile);
        SimpleLearner sl = new SimpleLearner(ftsl);
        long startZxid = sl.zk.getLastProcessedZxid();

        // Set up bogus streams
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        BinaryOutputArchive oa = BinaryOutputArchive.getArchive(baos);
        sl.leaderOs = BinaryOutputArchive.getArchive(new ByteArrayOutputStream());

        // make streams and socket do something innocuous
        sl.bufferedOutput = new BufferedOutputStream(System.out);
        sl.sock = new Socket();

        // fake messages from the server
        QuorumPacket qp = new QuorumPacket(Leader.SNAP, 0, null, null);
        oa.writeRecord(qp, null);
        sl.zk.getZKDatabase().serializeSnapshot(oa);
        oa.writeString("BenWasHere", "signature");
        TxnHeader hdr = new TxnHeader(0, 0, 0, 0, ZooDefs.OpCode.create);
        CreateTxn txn = new CreateTxn("/foo", new byte[0], new ArrayList<ACL>(), false, sl.zk.getZKDatabase().getNode("/").stat.getCversion());
        ByteArrayOutputStream tbaos = new ByteArrayOutputStream();
        BinaryOutputArchive boa = BinaryOutputArchive.getArchive(tbaos);
        hdr.serialize(boa, "hdr");
        txn.serialize(boa, "txn");
        tbaos.close();
        qp = new QuorumPacket(Leader.PROPOSAL, 1, tbaos.toByteArray(), null);
        oa.writeRecord(qp, null);

        // setup the messages to be streamed to follower
        sl.leaderIs = BinaryInputArchive.getArchive(new ByteArrayInputStream(baos.toByteArray()));

        try {
            sl.syncWithLeader(3);
        } catch (EOFException e) {}

        sl.zk.shutdown();
        sl = new SimpleLearner(ftsl);
        Assert.assertEquals(startZxid, sl.zk.getLastProcessedZxid());
    } finally {
        TestUtils.deleteFileRecursively(tmpFile);
    }
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
@Test
public void syncTest() throws Exception {
	File tmpFile = File.createTempFile("test", ".dir");
	tmpFile.delete();
	try {
		FileTxnSnapLog ftsl = new FileTxnSnapLog(tmpFile, tmpFile);
		SimpleLearner sl = new SimpleLearner(ftsl);
		long startZxid = sl.zk.getLastProcessedZxid();
		
		// Set up bogus streams
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		BinaryOutputArchive oa = BinaryOutputArchive.getArchive(baos);
		sl.leaderOs = BinaryOutputArchive.getArchive(new ByteArrayOutputStream());
		
		// make streams and socket do something innocuous
		sl.bufferedOutput = new BufferedOutputStream(System.out);
		sl.sock = new Socket();
		
		// fake messages from the server
		QuorumPacket qp = new QuorumPacket(Leader.SNAP, 0, null, null);
		oa.writeRecord(qp, null);
		sl.zk.getZKDatabase().serializeSnapshot(oa);
		oa.writeString("BenWasHere", "signature");
		TxnHeader hdr = new TxnHeader(0, 0, 0, 0, ZooDefs.OpCode.create);
		CreateTxn txn = new CreateTxn("/foo", new byte[0], new ArrayList<ACL>(), false, sl.zk.getZKDatabase().getNode("/").stat.getCversion());
        ByteArrayOutputStream tbaos = new ByteArrayOutputStream();
        BinaryOutputArchive boa = BinaryOutputArchive.getArchive(tbaos);
        hdr.serialize(boa, "hdr");
        txn.serialize(boa, "txn");
        tbaos.close();
		qp = new QuorumPacket(Leader.PROPOSAL, 1, tbaos.toByteArray(), null);
		oa.writeRecord(qp, null);

		// setup the messages to be streamed to follower
		sl.leaderIs = BinaryInputArchive.getArchive(new ByteArrayInputStream(baos.toByteArray()));
		
		try {
			sl.syncWithLeader(3);
		} catch(EOFException e) {}
		
		sl.zk.shutdown();
		sl = new SimpleLearner(ftsl);
		Assert.assertEquals(startZxid, sl.zk.getLastProcessedZxid());
	} finally {
		recursiveDelete(tmpFile);
	}
}
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
/**
   * Once connected to the leader, perform the handshake protocol to
   * establish a following / observing connection. 
   * @param pktType
   * @return the zxid the Leader sends for synchronization purposes.
   * @throws IOException
   */
  protected long registerWithLeader(int pktType) throws IOException{
      /*
       * Send follower info, including last zxid and sid
       */
  	long lastLoggedZxid = self.getLastLoggedZxid();
      QuorumPacket qp = new QuorumPacket();                
      qp.setType(pktType);
      qp.setZxid(ZxidUtils.makeZxid(self.getAcceptedEpoch(), 0));
      
      /*
       * Add sid to payload
       */
      LearnerInfo li = new LearnerInfo(self.getId(), 0x10000);
      ByteArrayOutputStream bsid = new ByteArrayOutputStream();
      BinaryOutputArchive boa = BinaryOutputArchive.getArchive(bsid);
      boa.writeRecord(li, "LearnerInfo");
      qp.setData(bsid.toByteArray());
      
      writePacket(qp, true);
      readPacket(qp);        
      final long newEpoch = ZxidUtils.getEpochFromZxid(qp.getZxid());
if (qp.getType() == Leader.LEADERINFO) {
      	// we are connected to a 1.0 server so accept the new epoch and read the next packet
      	leaderProtocolVersion = ByteBuffer.wrap(qp.getData()).getInt();
      	byte epochBytes[] = new byte[4];
      	final ByteBuffer wrappedEpochBytes = ByteBuffer.wrap(epochBytes);
      	if (newEpoch > self.getAcceptedEpoch()) {
      		wrappedEpochBytes.putInt((int)self.getCurrentEpoch());
      		self.setAcceptedEpoch(newEpoch);
      	} else if (newEpoch == self.getAcceptedEpoch()) {
      		// since we have already acked an epoch equal to the leaders, we cannot ack
      		// again, but we still need to send our lastZxid to the leader so that we can
      		// sync with it if it does assume leadership of the epoch.
      		// the -1 indicates that this reply should not count as an ack for the new epoch
              wrappedEpochBytes.putInt(-1);
      	} else {
      		throw new IOException("Leaders epoch, " + newEpoch + " is less than accepted epoch, " + self.getAcceptedEpoch());
      	}
      	QuorumPacket ackNewEpoch = new QuorumPacket(Leader.ACKEPOCH, lastLoggedZxid, epochBytes, null);
      	writePacket(ackNewEpoch, true);
      	readPacket(qp);
      } else {
      	if (newEpoch > self.getAcceptedEpoch()) {
      		self.setAcceptedEpoch(newEpoch);
      	}
      }
      if (qp.getType() != Leader.NEWLEADER) {
          LOG.error("First packet should have been NEWLEADER");
          throw new IOException("First packet should have been NEWLEADER");
      }
      
      return qp.getZxid();
  }
 

import org.apache.jute.BinaryOutputArchive; //导入方法依赖的package包/类
@Test
public void syncTest() throws Exception {
    File tmpFile = File.createTempFile("test", ".dir", testData);
    tmpFile.delete();
    try {
        FileTxnSnapLog ftsl = new FileTxnSnapLog(tmpFile, tmpFile);
        SimpleLearner sl = new SimpleLearner(ftsl);
        long startZxid = sl.zk.getLastProcessedZxid();

        // Set up bogus streams
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        BinaryOutputArchive oa = BinaryOutputArchive.getArchive(baos);
        sl.leaderOs = BinaryOutputArchive.getArchive(new ByteArrayOutputStream());

        // make streams and socket do something innocuous
        sl.bufferedOutput = new BufferedOutputStream(System.out);
        sl.sock = new Socket();

        // fake messages from the server
        QuorumPacket qp = new QuorumPacket(Leader.SNAP, 0, null, null);
        oa.writeRecord(qp, null);
        sl.zk.getZKDatabase().serializeSnapshot(oa);
        oa.writeString("BenWasHere", "signature");
        TxnHeader hdr = new TxnHeader(0, 0, 0, 0, ZooDefs.OpCode.create);
        CreateTxn txn = new CreateTxn("/foo", new byte[0], new ArrayList<ACL>(), false, sl.zk.getZKDatabase().getNode("/").stat.getCversion());
        ByteArrayOutputStream tbaos = new ByteArrayOutputStream();
        BinaryOutputArchive boa = BinaryOutputArchive.getArchive(tbaos);
        hdr.serialize(boa, "hdr");
        txn.serialize(boa, "txn");
        tbaos.close();
        qp = new QuorumPacket(Leader.PROPOSAL, 1, tbaos.toByteArray(), null);
        oa.writeRecord(qp, null);

        // setup the messages to be streamed to follower
        sl.leaderIs = BinaryInputArchive.getArchive(new ByteArrayInputStream(baos.toByteArray()));

        try {
            sl.syncWithLeader(3);
        } catch (EOFException e) {}

        sl.zk.shutdown();
        sl = new SimpleLearner(ftsl);
        Assert.assertEquals(startZxid, sl.zk.getLastProcessedZxid());
    } finally {
        recursiveDelete(tmpFile);
    }
}