UserController.java

package cn.ictgu.controller;

import cn.ictgu.config.security.AnyUser;
import cn.ictgu.serv.model.Category;
import cn.ictgu.serv.model.User;
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;

/**
 * Controller about user
 * Created by Silence on 2017/3/10.
 */
@Controller
public class UserController {

  @Autowired
  private UserService userService;

  @Autowired
  private CategoryService categoryService;

  @RequestMapping(value = "/validate/{token}", method = RequestMethod.GET)
  public String emailConfirm(@PathVariable("token") String token, Model model) {
    User user = userService.completeSignUp(token);
    if (user != null) {
      model.addAttribute("result", "注册成功，赶紧登陆体验吧！");
    } else {
      model.addAttribute("result", "链接已失效，请重新注册！");
    }
    return "login";
  }

  @GetMapping("/user")
  public String user(@AuthenticationPrincipal AnyUser user, Model model) {
    model.addAttribute("user", user);
    List<Category> categories = categoryService.getByUserId(user.getId());
    model.addAttribute("categories", categories);
    return "user";
  }

}