Доработка чата который сделан Spring Boot и Vaadin

[Nastya2018]

Цена - договорная
Я создал чат с помощью Спринг бута и Ваадина 14. Он работает еще с базой данных MySQL. У меня есть 4 api -

1)api/save - сохраняет сообщения в базе данных
2)api/last - выводит последние 10 сообщений из базы данных, если новый пользователь войдет в чат.
3)api/unread - выводит непрочитанные сообщения новым пользователям которые вошли в чат
4)api/update - обновляет

У меня 1,2 api работают правильно. Однако 3/4 работают, но неправильно(говорят они не берут данные из Базы данных.
За место 3/4 нужно создать функцию "getLastUnreadMessage" - который будет выводить непрочитанные сообщения из базы данных.
Пример - каждый пользователь должен запоминать айди последнего сообщения в базе данных, это поможет показывать ему непрочитанные сообщения. Плюс надо создать класс "Timer" который по определенному времени(допустим 1 секунда) будет запускать какую-то функцию.

Здесь реализовано то чтобы он выводил непрочитанные сообщения и тд, этот код сказали пусть и работает но надо реализовать по другому

public class MessagesInfoManager {

    private final List<MessageInfo> messagesInfoQueue = new ArrayList<>();
    
    private final Map<Optional<UI>, MessageList> messageListMap = new HashMap<>();

    public void updateMessageUIInfo(MessageInfo messageInfo) {
    
        addMessageInfo(messageInfo);
        Message message = messageInfo.getMessage();

        for (Map.Entry<Optional<UI>, MessageList> entry : this.messageListMap.entrySet()) {
            final Optional<UI> uiOptional = entry.getKey();
            final MessageList messageList = entry.getValue();
            uiOptional.ifPresent(ui ->
                    
                    ui.access(() ->
                            messageList.add(new Paragraph(message.getFrom() + ": " + message.getMessage()))
                    ));
        }
    }

    private void addMessageInfo(MessageInfo messageInfo) {
        if (!messagesInfoQueue.contains(messageInfo)) {
            this.messagesInfoQueue.add(messageInfo);
            this.messageListMap.putIfAbsent(messageInfo.getUI(), messageInfo.getMessageList());
        }
    }

    public List<Message> getMessagesByUI(Optional<UI> uiOptional) {
        return messagesInfoQueue
                .stream()
                .filter(messageInfo -> !messageInfo.getUI().equals(uiOptional))
                .map(MessageInfo::getMessage)
                .collect(Collectors.toList());
    }
}

MainView

@StyleSheet("frontend://styles/styles.css")
@Route
@PWA(name = "Vaadin MessagesInfoManager", shortName = "Vaadin MessagesInfoManager")
@Push
public class MainView extends VerticalLayout {
    private final MessagesInfoManager messagesInfoManager;
    private final RestService restService;
    private String username;

    @Autowired
    public MainView(RestService restService) {
        this.messagesInfoManager = MessageConfigurator.getInstance().getChatMessagesInfoManager();
        addClassName("main-view");
        setSizeFull();
        setDefaultHorizontalComponentAlignment(Alignment.CENTER);

        H1 header = new H1("Vaadin Chat");
        header.getElement().getThemeList().add("dark");

        add(header);

        askUsername();
        this.restService = restService;
    }

    private void askUsername() {
        HorizontalLayout layout = new HorizontalLayout();
        TextField usernameField = new TextField();
        Button startButton = new Button("Start chat");

        layout.add(usernameField, startButton);

        startButton.addClickListener(click -> {
            username = usernameField.getValue();
            remove(layout);
            showChat(username);
        });

        add(layout);
    }

    private void showChat(String username) {
        MessageList messageList = new MessageList();

        List<Message> lasts = restService.getLast();
        for (Message message : lasts) {
            messageList.add(new Paragraph(message.getFrom() + ": " + message.getMessage()));
        }

        add(messageList, createInputLayout(username, messageList));
        expand(messageList);
    }

    private Component createInputLayout(String username, MessageList messageList) {
        HorizontalLayout layout = new HorizontalLayout();
        layout.setWidth("100%");

        TextField messageField = new TextField();
        messageField.addKeyDownListener(Key.ENTER, keyDownEvent -> sender(messageField, messageList));
        Button sendButton = new Button("Send");
        sendButton.addThemeVariants(ButtonVariant.LUMO_PRIMARY);

        layout.add(messageField, sendButton);
        layout.expand(messageField);
        
        messageField.addFocusListener(event -> {
            for (Message message : messagesInfoManager.getMessagesByUI(getUI())) {
                if (!message.getFrom().equals(username)) {
                    message.setUnread(false);
                    this.restService.updateMessage(message.getId(), message);
                }
            }
        });

        sendButton.addClickListener(click -> sender(messageField, messageList));
        messageField.focus();

        return layout;
    }

    private void sender(TextField textField, MessageList messageList) {
        Message message = new Message(username, textField.getValue());
        message = restService.saveMessage(message);
        messagesInfoManager.updateMessageUIInfo(new MessageInfo(messageList, message, this));
        textField.clear();
        textField.focus();
    }
}

RestService

@Service
public class RestService {
    private final RestTemplate restTemplate;

    public RestService(RestTemplateBuilder restTemplateBuilder) {
        this.restTemplate = restTemplateBuilder.build();
    }

    public Message saveMessage(Message message) {
        String url = "http://localhost:8080/api/save";

        return this.restTemplate.postForObject(url, message, Message.class);
    }

    public void updateMessage(long id, Message message) {
        String url = String.format("http://localhost:8080/api/update/%d", id);

        this.restTemplate.put(url, message);
    }

    public List<Message> getLast() {
        String url = "http://localhost:8080/api/last";

        String json = restTemplate.getForObject(url, String.class);
        return new Gson().fromJson(json, new TypeToken<List<Message>>(){}.getType());
    }
}

MessageServiceImple

mport java.util.stream.Collectors;

@Service
@Transactional
public class MessageServiceImpl implements MessageService {
    private final MessageRepository repository;
    private final PageRequest lastRequest;

    @Autowired
    public MessageServiceImpl(MessageRepository repository) {
        this.repository = repository;
        lastRequest = new PageRequest(0, 10, Sort.Direction.DESC, "id");
    }

    @Override
    public Message add(Message message) {
        message.setTime(new Timestamp(new Date().getTime()));
        return repository.saveAndFlush(message);
    }

    @Override
    public List<Message> getAllMessages() {
        return repository.findAll();
    }

    @Override
    public List<Message> getLast() {
        List<Message> result = repository.findAll(lastRequest).getContent();

        return result.stream()
                .sorted(Comparator.comparingLong(Message::getId))
                .collect(Collectors.toList());
    }

    @Override
    public List<Message> getUnreadMessages() {
        return repository.findAllByUnread(true);
    }

    @Override
    public void updateMessage(long id, Message message) {
        if (repository.findById(id).isPresent()) {
            message.setId(id);
            repository.save(message);
        }
    }
}