JAVA ДЛЯ ПОТОКОВ ИВБО / ПР-16 / VoteController

package com.example.demo.controller;

import com.example.demo.model.Vote;
import com.example.demo.repository.VoteRepository; // Ваш репозиторий для работы с данными
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;

@Controller
public class VoteController {

    @Autowired
    private VoteRepository voteRepository; // Инъекция зависимости репозитория

    // Отображение страницы голосования (GET-запрос)
    @GetMapping("/vote")
    public String showVotePage() {
        return "vote"; // Возвращает шаблон vote.html
    }

    // Обработка голосов (POST-запрос)
    @PostMapping("/vote")
    public String processVote(@RequestParam("option") String option, Model model) {
        // Проверяем правильность введённого варианта
        if ("Java".equals(option) || "Python".equals(option) || "C++".equals(option)) {
            voteRepository.save(new Vote(option)); // Сохраняем голос в репозитории
            model.addAttribute("message", "Спасибо за ваш голос!"); // Уведомление для пользователя
        } else {
            model.addAttribute("message", "Пожалуйста, выберите допустимый вариант!"); // Ошибка
        }
        return "vote"; // Возвращаемся на ту же страницу
    }

    // Вывод результатов голосования (GET-запрос)
    @GetMapping("/results")
    public String showResults(Model model) {
        // Считаем количество голосов для каждого варианта
        long javaVotes = voteRepository.countByOption("Java");
        long pythonVotes = voteRepository.countByOption("Python");
        long cppVotes = voteRepository.countByOption("C++");

        // Передаём данные в шаблон results.html
        model.addAttribute("javaVotes", javaVotes);
        model.addAttribute("pythonVotes", pythonVotes);
        model.addAttribute("cppVotes", cppVotes);

        return "results"; // Шаблон для отображения результатов голосования
    }
}