/** 添加消息,为了保证添加顺序,这里不得不加锁 */
 @Override
 public void addMessage(
     final MessageStore store, final long msgId, final PutCommand putCmd, JournalLocation location)
     throws IOException {
   if (location == null) {
     // 非重放,添加put日志
     final AppendMessageCommand appendCmd =
         AppendMessageCommand.newBuilder()
             .setMessageId(msgId)
             .setPutCommand(ByteString.copyFrom(putCmd.encode().array()))
             .build();
     final TxCommand txCommand =
         TxCommand.newBuilder()
             .setCmdType(TxCommandType.APPEND_MSG)
             .setCmdContent(appendCmd.toByteString())
             .build();
     final Tx tx = this.getInflyTx(putCmd.getTransactionId());
     if (tx != null) {
       location = this.journalStore.write(txCommand, null, tx.location, false);
     } else {
       location = this.journalStore.write(txCommand, null, null, false);
     }
   }
   final Tx tx = this.getTx(putCmd.getTransactionId(), location);
   tx.add(store, msgId, putCmd);
 }
Ejemplo n.º 2
0
 public static final ByteBuffer makeMessageBuffer(
     final List<Long> msgIds, final List<PutCommand> reqs) {
   if (msgIds == null || reqs == null) {
     throw new IllegalArgumentException("Null id list or request list");
   }
   if (msgIds.size() != reqs.size()) {
     throw new IllegalArgumentException("id list is not adapte to request list");
   }
   int capacity = 0;
   for (final PutCommand req : reqs) {
     capacity += 4 + 4 + 8 + 4 + req.getData().length;
   }
   final ByteBuffer buffer = ByteBuffer.allocate(capacity);
   for (int i = 0; i < reqs.size(); i++) {
     final PutCommand req = reqs.get(i);
     final long msgId = msgIds.get(i);
     buffer.putInt(req.getData().length);
     buffer.putInt(CheckSum.crc32(req.getData()));
     buffer.putLong(msgId);
     buffer.putInt(req.getFlag());
     buffer.put(req.getData());
   }
   buffer.flip();
   return buffer;
 }
Ejemplo n.º 3
0
 /**
  * 创建消息buffer,实际存储在服务器的结构如下:
  *
  * <ul>
  *   <li>message length(4 bytes),including attribute and payload
  *   <li>checksum(4 bytes)
  *   <li>message id(8 bytes)
  *   <li>message flag(4 bytes)
  *   <li>attribute length(4 bytes) + attribute,optional
  *   <li>payload
  * </ul>
  *
  * @param req
  * @return
  */
 public static final ByteBuffer makeMessageBuffer(final long msgId, final PutCommand req) {
   // message length + checksum + id +flag + data
   final ByteBuffer buffer = ByteBuffer.allocate(4 + 4 + 8 + 4 + req.getData().length);
   buffer.putInt(req.getData().length);
   int checkSum = CheckSum.crc32(req.getData());
   // If client passes checksum,compare them
   if (req.getCheckSum() != -1) {
     if (checkSum != req.getCheckSum()) {
       throw new InvalidCheckSumException(
           "Checksum failure,message may be corrupted when transfering on networking.");
     }
   }
   buffer.putInt(checkSum);
   buffer.putLong(msgId);
   buffer.putInt(req.getFlag());
   buffer.put(req.getData());
   buffer.flip();
   return buffer;
 }