не знаю, что по этому тестовому заданию можно сказать, но могу дать комментарии, раз ты просишь.
1) в last_messages стейт это мапа. Мы так не делаем никогда, это делать плохо и заканчивается трудноуловимыми багами. Золотое правило: рекорд лучше мапы
2) all_msg не очень удачно, лучше передавать курсор с которого читать
Про первый понял, спасибо, почти всегда таплы использовал.
Про второй пункт можно подробности? Задачи хранить всю переписку нет, но чтобы последние х сообщений новому коннеку показывать на ходу придумал