count should not incremnet unless Insert is successful

strangelove-ventures · Sep 19, 2023 · 7bc4707 · 7bc4707
1 parent 3c341c4
commit 7bc4707
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/x/mailbox/keeper/msg_server.go b/x/mailbox/keeper/msg_server.go
@@ -99,10 +99,11 @@ func (k *Keeper) Dispatch(goCtx context.Context, msg *types.MsgDispatch) (*types
 
 	// Insert the message id into the tree
 	err := k.Tree.Insert(id)
-	k.ImtCount++
 	if err != nil {
 		return nil, err
 	}
+	k.ImtCount++
+
 	// Store that the leaf
 	store.Set(types.MailboxIMTKey(k.Tree.Count()-1), id)