CategoryItemController.java

package cn.ictgu.controller;

import cn.ictgu.config.security.AnyUser;
import cn.ictgu.serv.model.Category;
import cn.ictgu.serv.model.CategoryItem;
import cn.ictgu.serv.model.User;
import cn.ictgu.serv.service.CategoryItemService;
import cn.ictgu.serv.service.CategoryService;
import cn.ictgu.serv.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.annotation.AuthenticationPrincipal;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;

import java.util.List;

/**
 * CategoryItem Controller
 * Created by Silence on 2017/3/14.
 */
@Controller
public class CategoryItemController {

  @Autowired
  private UserService userService;

  @Autowired
  private CategoryItemService categoryItemService;

  @Autowired
  private CategoryService categoryService;

  @GetMapping("/user/item/{id}")
  public String list(@AuthenticationPrincipal AnyUser user, @PathVariable("id") Long id, Model model){
    List<CategoryItem> items = categoryItemService.list(id, user.getId());
    Category category = categoryService.getById(id, user.getId());
    model.addAttribute("user", user);
    model.addAttribute("category", category);
    model.addAttribute("items", items);
    return "items";
  }

  @GetMapping("/share/{md5}")
  public String share(@PathVariable("md5") String md5, Model model){
    Category category = categoryService.getByMd5(md5);
    if (category == null){
      return "error";
    }
    User user = userService.getById(category.getUserId());
    List<CategoryItem> items = categoryItemService.list(category.getId(), user.getId());
    model.addAttribute("user", user);
    model.addAttribute("category", category);
    model.addAttribute("items", items);
    return "share";
  }

}