cakeMallSystem/src/main/java/space/anyi/controller/front/FavoriteController.java

package space.anyi.controller.front;

import space.anyi.entity.User;
import space.anyi.service.FavoriteService;
import space.anyi.util.Converter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.servlet.http.HttpSession;

@Controller
@RequestMapping("/favorite")
public class FavoriteController {

    @Autowired
    private FavoriteService favoriteService;

    @GetMapping
    public ModelAndView list(HttpSession session) {
        User user = (User) session.getAttribute("user");
        ModelAndView mv = new ModelAndView("favorite/list");
        mv.addObject("favorites", favoriteService.getFavoriteVOs(user.getId()));
        return mv;
    }

    @PostMapping("/add")
    public String add(@RequestParam Integer cakeId, HttpSession session) {
        User user = (User) session.getAttribute("user");
        favoriteService.add(user.getId(), cakeId);
        return "redirect:/favorite";
    }

    @PostMapping("/remove")
    public String remove(@RequestParam Integer cakeId, HttpSession session,
                         RedirectAttributes attr) {
        User user = (User) session.getAttribute("user");
        favoriteService.remove(user.getId(), cakeId);
        attr.addFlashAttribute("success", "已取消收藏");
        return "redirect:/favorite";
    }
}