[Refactor♻️]Refactor end_message_transaction function to accept mutable reference and improve message handling #3418

Which part of the code will be refactored?

No response

What problems does the refactor aim to solve?

No response

Do you have any specific design or implementation considerations?

No response

Is there a related issue or PR?

No response

Checklist

• Ensure the refactor does not introduce new bugs

• Update unit tests if applicable

• Ensure the refactor does not negatively impact performance

• Document any new patterns or architecture changes

• Test the refactor in different environments (e.g., staging, production)