基于spring boot的博客网站

package com.wiblog.core.controller; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.wiblog.core.aop.AuthorizeCheck; import com.wiblog.core.aop.OpsRecord; import com.wiblog.core.aop.RequestRequire; import com.wiblog.core.common.BaseController; import com.wiblog.core.common.Constant; import com.wiblog.core.common.RoleEnum; import com.wiblog.core.common.ServerResponse; import com.wiblog.core.entity.Article; import com.wiblog.core.entity.EsArticle; import com.wiblog.core.entity.User; import com.wiblog.core.elastic.EsArticleRepository; import com.wiblog.core.service.IArticleService; import com.wiblog.core.utils.PinYinUtil; import com.wiblog.core.utils.WiblogUtil; import com.wiblog.core.utils.WordFilterUtil; import com.wiblog.core.vo.ArticleDetailVo; import com.wiblog.core.vo.ArticlePageVo; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.elasticsearch.action.search.SearchResponse; import org.elasticsearch.index.query.QueryBuilder; import org.elasticsearch.index.query.QueryBuilders; import org.elasticsearch.index.query.TermQueryBuilder; import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.domain.Page; import org.springframework.data.domain.PageRequest; import org.springframework.data.domain.Pageable; import org.springframework.data.elasticsearch.core.ElasticsearchOperations; import org.springframework.data.elasticsearch.core.ElasticsearchRestTemplate; import org.springframework.data.elasticsearch.core.SearchHit; import org.springframework.data.elasticsearch.core.SearchHits; import org.springframework.data.elasticsearch.core.aggregation.AggregatedPage; import org.springframework.data.elasticsearch.core.aggregation.impl.AggregatedPageImpl; import org.springframework.data.elasticsearch.core.mapping.IndexCoordinates; import org.springframework.data.elasticsearch.core.query.NativeSearchQuery; import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Map; import java.util.regex.Matcher; import java.util.regex.Pattern; /** * 文章控制层 * * @author pwm * @date 2019-06-12 */ @Slf4j @RestController @RequestMapping("/post") public class ArticleController extends BaseController { private static final Pattern PATTERN_HIGH_LIGHT = Pattern.compile(Constant.Regular.HIGH_LIGHT); private final IArticleService articleService; private final WordFilterUtil wordFilterUtil; private final PinYinUtil pinYinUtil; private final ElasticsearchRestTemplate elasticsearchTemplate; @Autowired public ArticleController(IArticleService articleService, WordFilterUtil wordFilterUtil, PinYinUtil pinYinUtil, ElasticsearchRestTemplate elasticsearchTemplate) { this.articleService = articleService; this.wordFilterUtil = wordFilterUtil; this.pinYinUtil = pinYinUtil; this.elasticsearchTemplate = elasticsearchTemplate; } /** * 首页文章列表 */ @PostMapping("/articles") public ServerResponse<IPage<ArticlePageVo>> articlePageList( @RequestParam(value = "pageNum", defaultValue = "1") Integer pageNum, @RequestParam(value = "pageSize", defaultValue = "10") Integer pageSize, Long categoryId) { return articleService.articlePageList(pageNum, pageSize, categoryId); } /** * 后台管理文章列表 */ @PostMapping("/articlesManage") @AuthorizeCheck(grade = RoleEnum.ADMIN) public ServerResponse<IPage<ArticlePageVo>> articlesManage( @RequestParam(value = "pageNum", defaultValue = "1") Integer pageNum, @RequestParam(value = "pageSize", defaultValue = "10") Integer pageSize) { return articleService.articlesManage(pageNum, pageSize); } /** * 获取所有文章标题列表 * 评论管理界面-按文章标题查找 * 管理员权限 * * @return ServerResponse */ @PostMapping("/allArticles") @AuthorizeCheck(grade = RoleEnum.ADMIN) public ServerResponse<?> articlePageList() { LambdaQueryWrapper<Article> queryWrapper = new QueryWrapper<Article>().lambda(); queryWrapper.select(Article::getTitle).eq(Article::getState, 1); List<Map<String, Object>> list = articleService.listMaps(queryWrapper); return ServerResponse.success(list); } /** * 文章详细信息 */ @GetMapping("/get/{id}") public ServerResponse<Article> getArticleById(@PathVariable Long id) { Article article = articleService.getById(id); return ServerResponse.success(article, "获取文章成功"); } /** * 通过url获取文章信息 */ @GetMapping("/getArticle") public ServerResponse<ArticleDetailVo> getArticle(HttpServletRequest request, String url) { User user = getLoginUser(request); return articleService.getArticle(url, user); } /** * 发表文章 管理员权限 * * @param article article * @return ServerResponse */ @PostMapping("/push") @AuthorizeCheck(grade = RoleEnum.ADMIN) @OpsRecord(msg = "发表了文章<<{0}>>") @RequestRequire(require = "title,content,tags,categoryId,imgUrl", parameter = Article.class) public ServerResponse<String> pushArticle(HttpServletRequest request, Article article) { // 分词 List<String> titles = wordFilterUtil.getParticiple(article.getTitle()); String title = pinYinUtil.getStringPinYin(titles); String articleUrl = "/post/" + title; Article sameUrlArticle = articleService.getOne(new QueryWrapper<Article>().eq("article_url", articleUrl)); if (sameUrlArticle != null) { return ServerResponse.error("文章发表失败，已存在相同标题", 30001); } Date date = new Date(); // 提取纯文本 String content = StringUtils.substring(article.getContent(), 0, 500); content = WiblogUtil.mdToHtml(content) .replaceAll(Constant.Regular.HTML, "") .replaceAll(Constant.Regular.WRAP, ""); content = StringUtils.substring(content, 0, 100); User user = getLoginUser(request); // 赋值 article.setUpdateTime(date) .setCreateTime(date) .setArticleUrl(articleUrl) .setUid(user.getUid()) .setAuthor(user.getUsername()) .setArticleSummary(content); boolean bool = articleService.save(article); if (!bool) { return ServerResponse.error("文章发表失败", 30001); } return ServerResponse.success(articleUrl, "文章发表成功", title); } /** * 修改文章 管理员权限 * * @param article article * @return ServerResponse */ @PostMapping("/update") @AuthorizeCheck(grade = RoleEnum.ADMIN) @OpsRecord(msg = "修改了文章<<{0}>>") @RequestRequire(require = "id,title,content,tags,categoryId", parameter = Article.class) public ServerResponse<String> updateArticle(Article article) { Date date = new Date(); article.setUpdateTime(date); // 提取纯文本 String content = WiblogUtil.mdToHtml(article.getContent()); content = content.replaceAll("<[^>]+>", ""); content = content.replaceAll("\\s*|\t|\r|\n", ""); article.setArticleSummary(content.substring(0, 100)); boolean bool = articleService.updateById(article); if (!bool) { return