提交 a4f62104 编写于 作者: Z zhenghu

[#1568] rocketmq need enhance stability when commitlog broken

 增加非空判断,防止空指针异常,导致循环事务 check,增加test覆盖,修复编码格式
上级 159d28d5
...@@ -110,14 +110,18 @@ public class TransactionalMessageBridgeTest { ...@@ -110,14 +110,18 @@ public class TransactionalMessageBridgeTest {
@Test @Test
public void testGetHalfMessage() { public void testGetHalfMessage() {
when(messageStore.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), ArgumentMatchers.nullable(MessageFilter.class))).thenReturn(createGetMessageResult(GetMessageStatus.NO_MESSAGE_IN_QUEUE)); when(messageStore
.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), ArgumentMatchers.nullable(MessageFilter.class)))
.thenReturn(createGetMessageResult(GetMessageStatus.NO_MESSAGE_IN_QUEUE));
PullResult result = transactionBridge.getHalfMessage(0, 0, 1); PullResult result = transactionBridge.getHalfMessage(0, 0, 1);
assertThat(result.getPullStatus()).isEqualTo(PullStatus.NO_NEW_MSG); assertThat(result.getPullStatus()).isEqualTo(PullStatus.NO_NEW_MSG);
} }
@Test @Test
public void testGetOpMessage() { public void testGetOpMessage() {
when(messageStore.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), ArgumentMatchers.nullable(MessageFilter.class))).thenReturn(createGetMessageResult(GetMessageStatus.NO_MESSAGE_IN_QUEUE)); when(messageStore
.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), ArgumentMatchers.nullable(MessageFilter.class)))
.thenReturn(createGetMessageResult(GetMessageStatus.NO_MESSAGE_IN_QUEUE));
PullResult result = transactionBridge.getOpMessage(0, 0, 1); PullResult result = transactionBridge.getOpMessage(0, 0, 1);
assertThat(result.getPullStatus()).isEqualTo(PullStatus.NO_NEW_MSG); assertThat(result.getPullStatus()).isEqualTo(PullStatus.NO_NEW_MSG);
} }
...@@ -143,9 +147,9 @@ public class TransactionalMessageBridgeTest { ...@@ -143,9 +147,9 @@ public class TransactionalMessageBridgeTest {
MessageExt messageExt = createMessageBrokerInner(); MessageExt messageExt = createMessageBrokerInner();
final String offset = "123456789"; final String offset = "123456789";
MessageExtBrokerInner msgInner = transactionBridge.renewImmunityHalfMessageInner(messageExt); MessageExtBrokerInner msgInner = transactionBridge.renewImmunityHalfMessageInner(messageExt);
MessageAccessor.putProperty(msgInner, MessageConst.PROPERTY_TRANSACTION_PREPARED_QUEUE_OFFSET,offset); MessageAccessor.putProperty(msgInner, MessageConst.PROPERTY_TRANSACTION_PREPARED_QUEUE_OFFSET, offset);
assertThat(msgInner).isNotNull(); assertThat(msgInner).isNotNull();
Map<String,String> properties = msgInner.getProperties(); Map<String, String> properties = msgInner.getProperties();
assertThat(properties).isNotNull(); assertThat(properties).isNotNull();
String resOffset = properties.get(MessageConst.PROPERTY_TRANSACTION_PREPARED_QUEUE_OFFSET); String resOffset = properties.get(MessageConst.PROPERTY_TRANSACTION_PREPARED_QUEUE_OFFSET);
assertThat(resOffset).isEqualTo(offset); assertThat(resOffset).isEqualTo(offset);
...@@ -157,11 +161,11 @@ public class TransactionalMessageBridgeTest { ...@@ -157,11 +161,11 @@ public class TransactionalMessageBridgeTest {
MessageExt messageExt = new MessageExt(); MessageExt messageExt = new MessageExt();
long bornTimeStamp = messageExt.getBornTimestamp(); long bornTimeStamp = messageExt.getBornTimestamp();
MessageExt messageExtRes = transactionBridge.renewHalfMessageInner(messageExt); MessageExt messageExtRes = transactionBridge.renewHalfMessageInner(messageExt);
assertThat( messageExtRes.getBornTimestamp()).isEqualTo(bornTimeStamp); assertThat(messageExtRes.getBornTimestamp()).isEqualTo(bornTimeStamp);
} }
@Test @Test
public void testLookMessageByOffset(){ public void testLookMessageByOffset() {
when(messageStore.lookMessageByOffset(anyLong())).thenReturn(new MessageExt()); when(messageStore.lookMessageByOffset(anyLong())).thenReturn(new MessageExt());
MessageExt messageExt = transactionBridge.lookMessageByOffset(123); MessageExt messageExt = transactionBridge.lookMessageByOffset(123);
assertThat(messageExt).isNotNull(); assertThat(messageExt).isNotNull();
...@@ -182,7 +186,7 @@ public class TransactionalMessageBridgeTest { ...@@ -182,7 +186,7 @@ public class TransactionalMessageBridgeTest {
.getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), ArgumentMatchers.nullable(MessageFilter.class))) .getMessage(anyString(), anyString(), anyInt(), anyLong(), anyInt(), ArgumentMatchers.nullable(MessageFilter.class)))
.thenReturn(null); .thenReturn(null);
PullResult result = transactionBridge.getHalfMessage(0, 0, 1); PullResult result = transactionBridge.getHalfMessage(0, 0, 1);
assertThat(result.getPullStatus()).isNull(); assertThat(result).isNull();
} }
private GetMessageResult createGetMessageResult(GetMessageStatus status) { private GetMessageResult createGetMessageResult(GetMessageStatus status) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册