项目：个人博客系统 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模板，就不多介绍了。

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