项目：个人博客系统 Part3 Controller层编写

个人博客系统 Part3

Controller层分为两个模块：user模块和admin模块，分别对应博客的前台页面和后台管理页面。

一、user模块

由于Controller传递到前台的变量较多，且重复率很高，所以可以定义全局变量来保持部分常用变量的变量名固定。

编写一个AttributeConstant类，定义MAIN_PAGE/USER/PAGER/ARTICLE/ERROR等变量：

public class AttributeConstant {
    public static final String ABOUT = "about";
    public static final String MAIN_PAGE = "mainPage";
    public static final String WEB_APP_DTO = "webAppDto";
    public static final String ARTICLES = "articles";
    public static final String ARTICLE ="article";
    public static final String PAGER = "pager";
    public static final String USER = "user";
    public static final String USERS = "users";
    public static final String CURRENT_USER = "currentUser";
    public static final String RETURN_INFO = "info";
    public static final Integer DAY_TIME = 1 * 60 * 60 * 24;
    public static final String CATEGORIES = "categories";
    public static final String CATEGORY = "category";
    public static final String ERROR ="error";
}

包括主页Controller、文章列表、文章详情Controller、分类列表与详情、login；

1.ListController

文章列表ListController，需要调用自定义工具类Pager类来完成分页功能：

@Controller("/list")
public class ListController {

    @Autowired
    private ArticleService articleService;

    @RequestMapping(method = RequestMethod.GET)
    public String list(@RequestParam(value = "pageIndex", defaultValue = "1") int pageIndex, ModelMap model, HttpServletRequest request) {
        Pager pager = new Pager(pageIndex, 4, articleService.count());
        model.addAttribute(AttributeConstant.MAIN_PAGE, "user/article/list.jsp");
        model.addAttribute(AttributeConstant.ARTICLES, articleService.getPageArticles(pager));
        model.addAttribute(AttributeConstant.PAGER, pager);
        return "index";
    }
}

其中index页面是公共页面，因为导航部分是所有页面都有的，所以可以共用：

<jsp:include page="common/nav.jsp"/>

<jsp:include page="${mainPage}"/>

2.ArticleController

显示文章详情页面，使用正则表达式[0-9]+匹配url，主要需要更新文章的点击量和获取上一篇、下一篇文章；

@Controller
public class ArticleController {

    @Autowired
    private ArticleService articleService;

    //显示 详细文章
    @RequestMapping("article/{articleId:[0-9]+}")
    public String showArticle(@PathVariable("articleId")Integer articleId, ModelMap model){
        ArticleDto articleDto = articleService.getArticle(articleId);
        model.addAttribute(AttributeConstant.MAIN_PAGE, "user/article/item.jsp");

        if(StringUtil.isNotEmpty(articleDto.getTitle())) {
            //点击量+1
            articleService.updateClicks(articleDto.getClicks() + 1, articleDto.getId());
            //更新一下用于显示
            articleDto.setClicks(articleDto.getClicks() + 1);
            //获取上一篇文章
            ArticleLiteDto preArticle = articleService.getPreArticle(articleDto.getId());
            //获取下一篇文章
            ArticleLiteDto nextArticle = articleService.getNextArticle(articleDto.getId());
            model.addAttribute(AttributeConstant.ARTICLE, articleDto);
            model.addAttribute("preArticle", preArticle);
            model.addAttribute("nextArticle", nextArticle);
        }else{
            model.addAttribute(AttributeConstant.ERROR,"没有此文章");
        }
        return "index";
    }
}

3.CategoryController

和Article类似，需要显示列表页面和详情页面，其中详情页面需要根据Category获取Articles：

@Controller
@RequestMapping()
public class CategoryController {
    @Autowired
    private CategoryService categoryService;
    @Autowired
    private ArticleService articleService;


    //全部list
    @RequestMapping("/category")
    public String list(ModelMap model) {
        List<CategoryDto> categories = categoryService.getCategories();
        model.addAttribute(AttributeConstant.MAIN_PAGE, "user/category/categoryList.jsp");
        model.addAttribute(AttributeConstant.CATEGORIES, categories);
        return "index";
    }

    //详情
    @RequestMapping("category/{categoryId:[0-9]+}")
    public String detail(@PathVariable("categoryId") Integer categoryId, ModelMap model) {

        Category category = categoryService.getCategory(categoryId);
        model.addAttribute(AttributeConstant.MAIN_PAGE, "user/category/detail.jsp");
        if (StringUtil.isNotEmpty(category.getName())) {
            List<ArticleLiteDto> articles = articleService.getArticlesByCategory(categoryId);
            if (articles.size() == 0) {
                articles = null;
            }
            model.addAttribute(AttributeConstant.CATEGORY, category);
            model.addAttribute(AttributeConstant.ARTICLES, articles);
        } else {
            model.addAttribute(AttributeConstant.ERROR, "找不到该分类");
        }
        return "index";
    }


}

4.LoginController

Login模块包括：访问login页面、提交login操作、生成验证码功能；

访问login页面包括从cookies读取username和password的缓存

    //登录页面
    @RequestMapping(value="/login",method = RequestMethod.GET)
    public String showLogin(HttpServletRequest request, Model model) {
        Cookie[] cookies = request.getCookies();
        if (cookies != null) {
            String username = "";
            String password = "";
            for (Cookie cookie : cookies) {
                if ("username".equals(cookie.getName())) {
                    username = cookie.getValue();
                    model.addAttribute("username", username);
                }
                if ("password".equals(cookie.getName())) {
                    password = cookie.getValue();
                    model.addAttribute("password", password);
                }
            }
        }
        return "user/login/login";
    }

login登录操作，判断用户名、密码、验证码；成功则转向manage后台管理页面，否则显示错误。

    //登陆操作
    @RequestMapping(value = "/login", method = RequestMethod.POST)
    public String login(String username, String password, @RequestParam(defaultValue = "no") String remember, String code, ModelMap model, HttpSession session, HttpServletResponse response) {
        User user = new User();
        user.setUsername(username);
        user.setPassword(MyMD5.MD5(password));
        UserDto currentUser = userService.login(user);
        if (code.toUpperCase().equals(session.getAttribute("code"))) {
            if (currentUser != null) {
                if (remember.equals("yes")) {
                    //记住我操作
                    rememberMe(username, password, response);
                }
                session.setAttribute(AttributeConstant.CURRENT_USER, currentUser);
                //跳转
                return "redirect:/manage";
            } else {
                model.addAttribute("error", "用户名或密码错误");
            }
        } else {
            model.addAttribute("error", "验证码错误");
        }
        return "user/login/login";
    }

二、admin模块

admin模块就是典型的后台管理模块了，主要是各个部分的增删改查功能；

1.ManageUserController

用户模块包括用户列表的显示以及用户增删改查：

@Controller
@RequestMapping("manage/user")
public class ManageUserController {
    @Autowired
    private UserService userService;

    //显示用户列表
    @RequestMapping(method = RequestMethod.GET)
    public String showUsers(ModelMap model, HttpSession session) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/user/editor.vm");
        model.addAttribute(AttributeConstant.USERS, userService.getUsers());
        return "admin/index";
    }

    //添加用户
    @RequestMapping(value = "create", method = RequestMethod.POST)
    public String createAction(User user, ModelMap model, HttpSession session) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/user/editor.vm");
        if (StringUtil.isNotEmpty(user.getUsername()) && StringUtil.isNotEmpty(user.getPassword()) && StringUtil.isNotEmpty(user.getNickname()) && StringUtil.isNotEmpty(user.getEmail())) {
            if (userService.userIsNotEmpty(user.getUsername())) {
                user.setPassword(MyMD5.MD5(user.getPassword()));
                user.setImagePath("/static/img/1.jpg");
                userService.saveUser(user);
                model.addAttribute(AttributeConstant.RETURN_INFO, "用户添加成功!");
            } else {
                model.addAttribute(AttributeConstant.ERROR, "用户已存在!");
            }
        } else {
            model.addAttribute(AttributeConstant.ERROR, "有未填写用户信息!");
        }
        model.addAttribute(AttributeConstant.USERS, userService.getUsers());
        return "admin/index";
    }

    //根据ID显示更新页面
    @RequestMapping(value = "update/{userId:[0-9]+}", method = RequestMethod.GET)
    public String showUpdate(@PathVariable("userId") Integer userId, ModelMap model, HttpSession session) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/user/editor.vm");
        User user = userService.getUser(userId);
        if (StringUtil.isNotEmpty(user.getUsername())) {
            model.addAttribute("editoruser", user);
        } else {
            model.addAttribute(AttributeConstant.ERROR, "用户修改失败!");
        }
        model.addAttribute(AttributeConstant.USERS, userService.getUsers());
        return "admin/index";
    }

    //更新用户操作
    @RequestMapping(value = "update", method = RequestMethod.POST)
    public String updateAction(User user, ModelMap model, HttpSession session) {
        UserDto userDto = (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER);
        model.addAttribute(AttributeConstant.USER, userDto);
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/user/editor.vm");
        if (StringUtil.isNotEmpty(user.getUsername()) && StringUtil.isNotEmpty(user.getPassword()) && StringUtil.isNotEmpty(user.getNickname()) && StringUtil.isNotEmpty(user.getEmail())) {
            //如果管理员没有编辑选择用户的密码,就不在对密码进行MD5加密
            if (!user.getPassword().equals(userService.getUser(user.getId()).getPassword())) {
                user.setPassword(MyMD5.MD5(user.getPassword()));
            }
            userService.updateUser(user);
            model.addAttribute(AttributeConstant.RETURN_INFO, "修改用户成功!");
            //判断更新用户 是否为 当前登陆用户 如果是 需要更新当前登陆用户的 显示信息.
            if (user.getId() == userDto.getId()) {
                UserDto currentUser = userService.login(user);
                session.setAttribute(AttributeConstant.CURRENT_USER, currentUser);
                model.addAttribute(AttributeConstant.USER, currentUser);
            }
        } else {
            model.addAttribute(AttributeConstant.ERROR, "有未填写用户信息!");
        }
        model.addAttribute(AttributeConstant.USERS, userService.getUsers());
        return "admin/index";
    }

    //根据ID删除用户
    @RequestMapping(value = "delete/{userId:[0-9]+}")
    public String deleteAction(ModelMap model, HttpSession session, @PathVariable("userId") Integer userId) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/user/editor.vm");
        User user = userService.getUser(userId);
        if (StringUtil.isNotEmpty(user.getUsername())) {
            userService.deleteUser(userId);
            model.addAttribute(AttributeConstant.RETURN_INFO, "删除用户成功!");
        } else {
            model.addAttribute(AttributeConstant.ERROR, "找不到该用户");
        }
        model.addAttribute(AttributeConstant.USERS, userService.getUsers());
        return "admin/index";
    }
}

2.ManageArticleController

文章管理模块

@Controller
@RequestMapping("manage/article")
public class ManageArticleController {
    @Autowired
    private CategoryService categoryService;
    @Autowired
    private ArticleService articleService;
    @Autowired
    private WebAppService webAppService;

    //显示创建页面
    @RequestMapping(value = "create", method = RequestMethod.GET)
    public String showCreatePage(ModelMap model, HttpSession session) {
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/editorArticle.vm");
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
        return "admin/index";
    }

    //创建操作
    @RequestMapping(value = "create", method = RequestMethod.POST)
    public String createAction(Article article, ModelMap model, HttpSession session) {
        String path;
        article.setClicks(0);
        SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        article.setPubDate(formatter.format(new Date()));
        if (StringUtil.isNotEmpty(article.getTitle()) && StringUtil.isNotEmpty(article.getMarkDown()) && StringUtil.isNotEmpty(article.getRemark())) {
            articleService.saveArticle(article);
            path = "redirect:/manage/article";
        } else {
            model.addAttribute(AttributeConstant.ERROR, "有未填选项,请核对后重新发布文章!");
            model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/editorArticle.vm");
            model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
            model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
            path = "admin/index";
        }
        return path;
    }

    //显示文章列表
    @RequestMapping(method = RequestMethod.GET)
    public String showListArticle(ModelMap model, @RequestParam(defaultValue = "1") Integer currentPage, HttpSession session) {
        Pager pager = new Pager(currentPage, webAppService.getWebAppDtos().get(0).getAdminPageArticleSize(), articleService.count());
        List<ArticleDto> articles = articleService.getPageArticles(pager);
        model.addAttribute(AttributeConstant.ARTICLES, articles);
        model.addAttribute(AttributeConstant.PAGER, pager);
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/listArticle.vm");
        return "admin/index";
    }

    //通过 ID 显示更新文章页面
    @RequestMapping(value = "update/{articleId:[0-9]+}", method = RequestMethod.GET)
    public String upDateArticle(ModelMap model, HttpSession session, @PathVariable("articleId") Integer articleId) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        ArticleDto articleDto = articleService.getArticle(articleId);
        if (StringUtil.isNotEmpty(articleDto.getTitle())) {
            model.addAttribute(AttributeConstant.ARTICLE, articleDto);
            model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/editorArticle.vm");
            model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
        } else {
            Pager pager = new Pager(1, 10, articleService.count());
            List<ArticleDto> articles = articleService.getPageArticles(pager);
            model.addAttribute(AttributeConstant.ERROR, "找不到该文章!");
            model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/listArticle.vm");
            model.addAttribute(AttributeConstant.PAGER, pager);
            model.addAttribute(AttributeConstant.ARTICLES, articles);
        }
        return "admin/index";
    }

    //通过ID更新文章 操作
    @RequestMapping(value = "update/{articleId:[0-9]+}", method = RequestMethod.POST)
    public String upDateArticleAction(Article article, @PathVariable("articleId") Integer articleId, ModelMap model, HttpSession session) {
        String path;
        if (StringUtil.isNotEmpty(article.getTitle()) && StringUtil.isNotEmpty(article.getMarkDown()) && StringUtil.isNotEmpty(article.getRemark())) {
            article.setClicks(articleService.getArticle(articleId).getClicks());
            article.setPubDate(articleService.getArticle(articleId).getPubDate());
            articleService.updateArticle(article);
            path = "redirect:/manage/article";
        } else {
            model.addAttribute(AttributeConstant.ERROR, "有未填选项,请核对后重新发布文章!");
            model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/editorArticle.vm");
            model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
            model.addAttribute(AttributeConstant.ARTICLE, articleService.getArticle(articleId));
            model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
            path = "admin/index";
        }
        return path;
    }

    //通过ID 删除文章
    @RequestMapping("delete/{articleId:[0-9]+}")
    public String deleteArticle(@PathVariable("articleId") Integer articleId, @RequestParam(value = "currentPage", defaultValue = "1") Integer currentPage) {
        String path = "redirect:/manage/article";
        articleService.deleteArticle(articleId);
        if (currentPage != 1) {
            path = "redirect:/manage/article/?currentPage=" + currentPage;
        }
        return path;
    }

    //搜索 文章
    @RequestMapping("search")
    public String search(String content, ModelMap model, HttpSession session) {
        Article article = new Article();
        article.setTitle(content);
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        List<ArticleDto> articleDtos = articleService.searchArticles(article);
        Pager pager = new Pager(1, 10, articleService.searchArticles(article).size());
        if (articleDtos.size() > 0) {
            model.addAttribute(AttributeConstant.ARTICLES, articleDtos);
        } else {
            model.addAttribute(AttributeConstant.ARTICLES, null);
        }
        model.addAttribute(AttributeConstant.PAGER, pager);

        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/article/listArticle.vm");
        return "admin/index";
    }
}

3.ManageCategoryController

分类管理模块

@Controller
@RequestMapping("/manage/category")
public class ManageCategoryController {
    @Autowired
    private CategoryService categoryService;
    //显示分类编辑页面
    @RequestMapping(method = RequestMethod.GET)
    public String showCategoriesPage(ModelMap model, HttpSession session) {

        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/category/editor.vm");
        return "admin/index";
    }
    //创建分类
    @RequestMapping(value = "/create", method = RequestMethod.POST)
    public String createAction(String categoryName, ModelMap model, HttpSession session) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/category/editor.vm");
        if (StringUtil.isNotEmpty(categoryName)) {
            Category category = new Category();
            category.setName(categoryName);
            categoryService.saveCategory(category);
        } else {
            model.addAttribute(AttributeConstant.ERROR, "分类名称未填写!");
        }
        model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
        return "admin/index";
    }
    //显示 更新分类页面
    @RequestMapping(value = "/update/{categoryId:[0-9]+}", method = RequestMethod.GET)
    public String upDatePage(@PathVariable("categoryId") Integer categoryId, ModelMap model, HttpSession session) {
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/category/editor.vm");
        Category category = categoryService.getCategory(categoryId);
        if (StringUtil.isNotEmpty(category.getName())) {
            model.addAttribute(AttributeConstant.CATEGORY, categoryService.getCategory(categoryId));
        }else{
            model.addAttribute(AttributeConstant.ERROR,"找不到该分类!");
        }
        model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
        return "admin/index";
    }
    //更新分类
    @RequestMapping(value = "/update",method = RequestMethod.POST)
    public String upDateAction(String categoryName,Integer categoryId,ModelMap model, HttpSession session){
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/category/editor.vm");
        Category category = new Category();
        category.setName(categoryName);
        category.setId(categoryId);
        categoryService.updateCategory(category);
        model.addAttribute(AttributeConstant.CATEGORIES, categoryService.getCategories());
        model.addAttribute(AttributeConstant.RETURN_INFO,"修改成功!");
        model.addAttribute(AttributeConstant.CATEGORY,category);
        return "admin/index";
    }

    //删除分类
    @RequestMapping(value = "/delete/{categoryId:[0-9]+}")
    public String deleteAction(@PathVariable("categoryId")Integer categoryId,ModelMap model,HttpSession session){
        model.addAttribute(AttributeConstant.USER, (UserDto) session.getAttribute(AttributeConstant.CURRENT_USER));
        model.addAttribute(AttributeConstant.MAIN_PAGE, "admin/category/editor.vm");
        Category category = categoryService.getCategory(categoryId);
        if(StringUtil.isNotEmpty(category.getName())){
            categoryService.deleteCategory(categoryId);
            model.addAttribute(AttributeConstant.RETURN_INFO,"删除成功!");
        }else{
            model.addAttribute(AttributeConstant.ERROR,"找不到该分类");
        }
        model.addAttribute(AttributeConstant.CATEGORIES,categoryService.getCategories());
        return "admin/index";
    }

}

后台管理的页面借用了AdminLTE模板，就不多介绍了。

以上就基本完成了个人博客系统的所有功能。