feat: 拆分为数据库地址

重命名为FileService 并修改方法为获取File Manager来处理文件

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,38 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behavior:
+1. Go to '...'
+2. Click on '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behavior**
+A clear and concise description of what you expected to happen.
+
+**Screenshots**
+If applicable, add screenshots to help explain your problem.
+
+**Desktop (please complete the following information):**
+ - OS: [e.g. iOS]
+ - Browser [e.g. chrome, safari]
+ - Version [e.g. 22]
+
+**Smartphone (please complete the following information):**
+ - Device: [e.g. iPhone6]
+ - OS: [e.g. iOS8.1]
+ - Browser [e.g. stock browser, safari]
+ - Version [e.g. 22]
+
+**Additional context**
+Add any other context about the problem here.

.github/workflows/build.yml

@@ -0,0 +1,28 @@
+# This workflow will build a package using Maven and then publish it to GitHub packages when a release is created
+# For more information see: https://github.com/actions/setup-java#apache-maven-with-a-settings-path
+
+name: Build
+
+on:
+  push:
+    branches:
+      - master
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+    env:
+      KEY: ${{ secrets.WEB_HOOK_ACCESS_KEY }}
+      QINIU_ACCESSKEY: ${{ secrets.QINIU_ACCESSKEY }}
+      QINIU_SECRETKEY: ${{ secrets.QINIU_SECRETKEY }}
+
+    steps:
+      - uses: actions/checkout@v2
+      - name: Set up JDK 1.8
+        uses: actions/setup-java@v1
+        with:
+          java-version: 1.8
+
+      - name: Deploy
+        run: mvn -B test --file pom.xml && curl http://bt.celess.cn:2271/hook?access_key=$KEY

.github/workflows/codeql-analysis.yml

@@ -0,0 +1,71 @@
+# For most projects, this workflow file will not need changing; you simply need
+# to commit it to your repository.
+#
+# You may wish to alter this file to override the set of languages analyzed,
+# or to provide custom queries or build logic.
+name: "CodeQL"
+
+on:
+  push:
+    branches: [master]
+  pull_request:
+    # The branches below must be a subset of the branches above
+    branches: [master]
+  schedule:
+    - cron: '0 14 * * 2'
+
+jobs:
+  analyze:
+    name: Analyze
+    runs-on: ubuntu-latest
+
+    strategy:
+      fail-fast: false
+      matrix:
+        # Override automatic language detection by changing the below list
+        # Supported options are ['csharp', 'cpp', 'go', 'java', 'javascript', 'python']
+        language: ['java']
+        # Learn more...
+        # https://docs.github.com/en/github/finding-security-vulnerabilities-and-errors-in-your-code/configuring-code-scanning#overriding-automatic-language-detection
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v2
+      with:
+        # We must fetch at least the immediate parents so that if this is
+        # a pull request then we can checkout the head.
+        fetch-depth: 2
+
+    # If this run was triggered by a pull request event, then checkout
+    # the head of the pull request instead of the merge commit.
+    - run: git checkout HEAD^2
+      if: ${{ github.event_name == 'pull_request' }}
+
+    # Initializes the CodeQL tools for scanning.
+    - name: Initialize CodeQL
+      uses: github/codeql-action/init@v1
+      with:
+        languages: ${{ matrix.language }}
+        # If you wish to specify custom queries, you can do so here or in a config file.
+        # By default, queries listed here will override any specified in a config file. 
+        # Prefix the list here with "+" to use these queries and those in the config file.
+        # queries: ./path/to/local/query, your-org/your-repo/queries@main
+
+    # Autobuild attempts to build any compiled languages  (C/C++, C#, or Java).
+    # If this step fails, then you should remove it and run the build manually (see below)
+    - name: Autobuild
+      uses: github/codeql-action/autobuild@v1
+
+    # ℹ️ Command-line programs to run using the OS shell.
+    # 📚 https://git.io/JvXDl
+
+    # ✏️ If the Autobuild fails above, remove it and uncomment the following three lines
+    #    and modify them (or add more) to build your code if your project
+    #    uses a compiled language
+
+    #- run: |
+    #   make bootstrap
+    #   make release
+
+    - name: Perform CodeQL Analysis
+      uses: github/codeql-action/analyze@v1

.github/workflows/deplay.yml

@@ -1,47 +0,0 @@
-# This workflow will build a package using Maven and then publish it to GitHub packages when a release is created
-# For more information see: https://github.com/actions/setup-java#apache-maven-with-a-settings-path
-
-name: Deplay
-
-on: 
-  push:
-    branches: 
-      - master 
-
-jobs:
-  build:
-
-    runs-on: ubuntu-latest
-    env:
-      APPLICATION_PROPERTIES_TEST: ${{ secrets.APPLICATION_PROPERTIES_TEST }}
-      APPLICATION_PROPERTIES_PROD: ${{ secrets.APPLICATION_PROPERTIES_PROD }}
-
-    steps:
-    - uses: actions/checkout@v2
-    - name: Set up JDK 1.8
-      uses: actions/setup-java@v1
-      with:
-        java-version: 1.8
-
-    - name: Build jar file
-      run: echo $APPLICATION_PROPERTIES_PROD|base64 -d> src/main/resources/application-prod.properties && mvn -B package --file pom.xml
-        
-    - name: SCP
-      uses: appleboy/scp-action@master
-      with:
-        host: ${{ secrets.SSH_HOST }}
-        username: ${{ secrets.SSH_USERNAME }}
-        password: ${{ secrets.SSH_PASSWORD }}
-        port: ${{ secrets.SSH_PORT }}
-        source: "target/blog-0.0.1-SNAPSHOT.jar"
-        target: "/www/wwwroot/api.celess.cn"
-
-    - name: Run SSH command
-      uses: garygrossgarten/github-action-ssh@v0.5.0
-      with:
-        command: cd /www/wwwroot/api.celess.cn && bash build.sh
-        host: ${{ secrets.SSH_HOST }}
-        username: ${{ secrets.SSH_USERNAME }}
-        password: ${{ secrets.SSH_PASSWORD }}
-        port: ${{ secrets.SSH_PORT }}
-    

.github/workflows/test.yml

@@ -1,7 +1,7 @@
 # This workflow will build a package using Maven and then publish it to GitHub packages when a release is created
 # For more information see: https://github.com/actions/setup-java#apache-maven-with-a-settings-path
 
-name: pr Test
+name: Test
 
 on:
   push:
@@ -13,17 +13,17 @@ jobs:
 
     runs-on: ubuntu-latest
     env:
-      APPLICATION_PROPERTIES_TEST: ${{ secrets.APPLICATION_PROPERTIES_TEST }}
-      APPLICATION_PROPERTIES_PROD: ${{ secrets.APPLICATION_PROPERTIES_PROD }}
+      QINIU_ACCESSKEY: ${{ secrets.QINIU_ACCESSKEY }}
+      QINIU_SECRETKEY: ${{ secrets.QINIU_SECRETKEY }}
 
     steps:
-    - uses: actions/checkout@v2
-    - name: Set up JDK 1.8
-      uses: actions/setup-java@v1
-      with:
-        java-version: 1.8
+      - uses: actions/checkout@v2
+      - name: Set up JDK 1.8
+        uses: actions/setup-java@v1
+        with:
+          java-version: 1.8
 
-    - name: Test
-      run: echo $APPLICATION_PROPERTIES_TEST|base64 -d > src/main/resources/application-test.properties && mvn -B test --file pom.xml
+      - name: Test
+        run: mvn -B test --file pom.xml
 
     

.gitignore

@@ -4,7 +4,5 @@
 target/
 
 # 本地项目的私有文件
-back-end/blog-dev.sql
 src/main/resources/application-dev.properties
 src/main/resources/application-prod.properties
-src/main/resources/application-test.properties

.mvn/wrapper/MavenWrapperDownloader.java

@@ -1,11 +1,11 @@
 /*
- * Copyright 2012-2019 the original author or authors.
+ * Copyright 2007-present the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
- *      https://www.apache.org/licenses/LICENSE-2.0
+ *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
@@ -13,7 +13,6 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-
 import java.net.*;
 import java.io.*;
 import java.nio.channels.*;
@@ -21,12 +20,12 @@ import java.util.Properties;
 
 public class MavenWrapperDownloader {
 
-    private static final String WRAPPER_VERSION = "0.5.5";
+    private static final String WRAPPER_VERSION = "0.5.6";
     /**
      * Default URL to download the maven-wrapper.jar from, if no 'downloadUrl' is provided.
      */
     private static final String DEFAULT_DOWNLOAD_URL = "https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/"
-            + WRAPPER_VERSION + "/maven-wrapper-" + WRAPPER_VERSION + ".jar";
+        + WRAPPER_VERSION + "/maven-wrapper-" + WRAPPER_VERSION + ".jar";
 
     /**
      * Path to the maven-wrapper.properties file, which might contain a downloadUrl property to
@@ -55,7 +54,7 @@ public class MavenWrapperDownloader {
         // wrapperUrl parameter.
         File mavenWrapperPropertyFile = new File(baseDirectory, MAVEN_WRAPPER_PROPERTIES_PATH);
         String url = DEFAULT_DOWNLOAD_URL;
-        if (mavenWrapperPropertyFile.exists()) {
+        if(mavenWrapperPropertyFile.exists()) {
             FileInputStream mavenWrapperPropertyFileInputStream = null;
             try {
                 mavenWrapperPropertyFileInputStream = new FileInputStream(mavenWrapperPropertyFile);
@@ -66,7 +65,7 @@ public class MavenWrapperDownloader {
                 System.out.println("- ERROR loading '" + MAVEN_WRAPPER_PROPERTIES_PATH + "'");
             } finally {
                 try {
-                    if (mavenWrapperPropertyFileInputStream != null) {
+                    if(mavenWrapperPropertyFileInputStream != null) {
                         mavenWrapperPropertyFileInputStream.close();
                     }
                 } catch (IOException e) {
@@ -77,8 +76,8 @@ public class MavenWrapperDownloader {
         System.out.println("- Downloading from: " + url);
 
         File outputFile = new File(baseDirectory.getAbsolutePath(), MAVEN_WRAPPER_JAR_PATH);
-        if (!outputFile.getParentFile().exists()) {
-            if (!outputFile.getParentFile().mkdirs()) {
+        if(!outputFile.getParentFile().exists()) {
+            if(!outputFile.getParentFile().mkdirs()) {
                 System.out.println(
                         "- ERROR creating output directory '" + outputFile.getParentFile().getAbsolutePath() + "'");
             }

.mvn/wrapper/maven-wrapper.properties

@@ -1,2 +1,2 @@
-distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.6.2/apache-maven-3.6.2-bin.zip
-wrapperUrl=https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.5/maven-wrapper-0.5.5.jar
+distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.6.3/apache-maven-3.6.3-bin.zip
+wrapperUrl=https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2019 小海
+Copyright (c) 2020 禾几海
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

README.md

@@ -1,11 +1,23 @@
- # 小海博客后端管理系统
-![Build Staus](https://github.com/xiaohai2271/blog-backEnd/workflows/Blog%20backEnd%20CI/badge.svg?branch=master&event=push) ![coverage report](https://gitlab.com/xiaohai2271/blog-backEnd/badges/master/coverage.svg) ![GitHub](https://img.shields.io/github/license/xiaohai2271/blog-backEnd) [![Website](https://img.shields.io/website?up_message=%E5%B0%8F%E6%B5%B7%E5%8D%9A%E5%AE%A2&url=https%3A%2F%2Fwww.celess.cn)](https://www.celess.cn)
-  ## 基于Springboot的后端博客管理系统
+<h1 align="center">
+小海博客后端管理系统
+</h1>
 
+<div align="center">
 
+基于Springboot的后端博客管理系统
 
+[![Build](https://github.com/xiaohai2271/blog-backEnd/workflows/Build/badge.svg)](https://github.com/xiaohai2271/blog-backEnd)
+[![Test](https://github.com/xiaohai2271/blog-backEnd/workflows/Test/badge.svg)](https://github.com/xiaohai2271/blog-backEnd)
+[![GitHub tag (latest SemVer)](https://img.shields.io/github/v/tag/xiaohai2271/blog-backEnd)](https://github.com/xiaohai2271/blog-backEnd)
+[![GitHub](https://img.shields.io/github/license/xiaohai2271/blog-backEnd)](https://github.com/xiaohai2271/blog-backEnd)
+[![GitHub top language](https://img.shields.io/github/languages/top/xiaohai2271/blog-backEnd)](https://github.com/xiaohai2271/blog-backEnd)
+[![GitHub commit activity](https://img.shields.io/github/commit-activity/m/xiaohai2271/blog-backEnd)](https://github.com/xiaohai2271/blog-backEnd)
+[![GitHub last commit](https://img.shields.io/github/last-commit/xiaohai2271/blog-backEnd)](https://github.com/xiaohai2271/blog-backEnd)
+[![Website](https://img.shields.io/website?up_message=%E5%B0%8F%E6%B5%B7%E5%8D%9A%E5%AE%A2&url=https%3A%2F%2Fwww.celess.cn)](https://www.celess.cn)
+</div>
 
-  ### 主要使用的技术
+> 请配合前端项目一起食用 [https://github.com/xiaohai2271/blog-frontEnd](https://github.com/xiaohai2271/blog-frontEnd)
+## 🎈主要使用的技术
 
 |    使用的技术    |    名称    |     版本      |
 | :--------------: | :--------: | :-----------: |
@@ -18,18 +30,16 @@
 |     接口文档     |  Swagger   |     2.6.1     |
 |   数据库连接池   |   druid    |    1.1.14     |
 | 缓存（线上环境） |   redis    |     3.0.6     |
-|数据库|mysql|5.7|
+|    数据库     |mysql          |5.7            |
 
-  ### 接口文档
+## 🔨如何构建
+详情参照[Build](./doc/Build.md)文档
 
-   项目采用swagger2，接口文档自动生成，具体为 http://ip:端口/swagger-ui.html
 
-  ###  📝TODO
+## 📒接口文档
+项目采用swagger2，接口文档自动生成，具体为 http://ip:port/doc.html
 
-  - [x] 密码重置
-  - [x] 信息修改
-  - [ ] 接入qq登录
+或者参照[离线API文档](./doc/API.md)
 
-  ### 📌FIXME
-
-  - [ ] `/write` 图片上传的跨域问题
+## ☀授权协议
+[MIT](./LICENSE)

build.sh

@@ -1,8 +0,0 @@
-#!/bin/sh
-echo "查询已存在进程"
-pgrep -af blog-0.0.1-SNAPSHOT.jar
-echo "结束已存在进程"
-pkill -f blog-0.0.1-SNAPSHOT.jar
-echo "开始运行小海博客"
-nohup java -jar -Dfile.encoding=UTF-8 blog-0.0.1-SNAPSHOT.jar >blog.log &
-echo "更新结束"

doc/Build.md

@@ -0,0 +1,33 @@
+# 如何构建
+### 1. 构建前准备
+- 安装jdk
+- 安装maven(也可使使用项目中的maven wrapper)
+- 安装mysql
+- 安装redis
+- 获取七牛云的AccessKey/SecretKey (个人中心-密钥管理) [七牛云官网](https://www.qiniu.com/)
+
+### 2. 拉取项目到本地
+``` shell script
+git clone https://github.com/xiaohai2271/blog-backEnd.git
+# 或
+git clone git@github.com:xiaohai2271/blog-backEnd.git
+```
+ 
+### 3. maven构建
+```shell script
+mvn package
+```
+
+### 4. 运行
+```shell script
+java -jar target/blog-0.0.1-SNAPSHOT.jar
+```
+
+### 5. 其他命令
+```shell script
+mvn clean # 清理项目资源
+mvn clean package # 清理项目资源并重新打包
+mvn clean package -DskipTests # 清理项目资源,重新打包并跳过测试
+mvn test # 运行测试 
+..... #待添加
+```

mvnw

@@ -8,7 +8,7 @@
 # "License"); you may not use this file except in compliance
 # with the License.  You may obtain a copy of the License at
 #
-#    https://www.apache.org/licenses/LICENSE-2.0
+#    http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
@@ -19,7 +19,7 @@
 # ----------------------------------------------------------------------------
 
 # ----------------------------------------------------------------------------
-# Maven2 Start Up Batch script
+# Maven Start Up Batch script
 #
 # Required ENV vars:
 # ------------------
@@ -34,135 +34,128 @@
 #   MAVEN_SKIP_RC - flag to disable loading of mavenrc files
 # ----------------------------------------------------------------------------
 
-if [ -z "$MAVEN_SKIP_RC" ]; then
+if [ -z "$MAVEN_SKIP_RC" ] ; then
 
-  if [ -f /etc/mavenrc ]; then
+  if [ -f /etc/mavenrc ] ; then
     . /etc/mavenrc
   fi
 
-  if [ -f "$HOME/.mavenrc" ]; then
+  if [ -f "$HOME/.mavenrc" ] ; then
     . "$HOME/.mavenrc"
   fi
 
 fi
 
 # OS specific support.  $var _must_ be set to either true or false.
-cygwin=false
-darwin=false
+cygwin=false;
+darwin=false;
 mingw=false
-case "$(uname)" in
-CYGWIN*) cygwin=true ;;
-MINGW*) mingw=true ;;
-Darwin*)
-  darwin=true
-  # Use /usr/libexec/java_home if available, otherwise fall back to /Library/Java/Home
-  # See https://developer.apple.com/library/mac/qa/qa1170/_index.html
-  if [ -z "$JAVA_HOME" ]; then
-    if [ -x "/usr/libexec/java_home" ]; then
-      export JAVA_HOME="$(/usr/libexec/java_home)"
-    else
-      export JAVA_HOME="/Library/Java/Home"
+case "`uname`" in
+  CYGWIN*) cygwin=true ;;
+  MINGW*) mingw=true;;
+  Darwin*) darwin=true
+    # Use /usr/libexec/java_home if available, otherwise fall back to /Library/Java/Home
+    # See https://developer.apple.com/library/mac/qa/qa1170/_index.html
+    if [ -z "$JAVA_HOME" ]; then
+      if [ -x "/usr/libexec/java_home" ]; then
+        export JAVA_HOME="`/usr/libexec/java_home`"
+      else
+        export JAVA_HOME="/Library/Java/Home"
+      fi
     fi
-  fi
-  ;;
+    ;;
 esac
 
-if [ -z "$JAVA_HOME" ]; then
-  if [ -r /etc/gentoo-release ]; then
-    JAVA_HOME=$(java-config --jre-home)
+if [ -z "$JAVA_HOME" ] ; then
+  if [ -r /etc/gentoo-release ] ; then
+    JAVA_HOME=`java-config --jre-home`
   fi
 fi
 
-if [ -z "$M2_HOME" ]; then
+if [ -z "$M2_HOME" ] ; then
   ## resolve links - $0 may be a link to maven's home
   PRG="$0"
 
   # need this for relative symlinks
-  while [ -h "$PRG" ]; do
-    ls=$(ls -ld "$PRG")
-    link=$(expr "$ls" : '.*-> \(.*\)$')
-    if expr "$link" : '/.*' >/dev/null; then
+  while [ -h "$PRG" ] ; do
+    ls=`ls -ld "$PRG"`
+    link=`expr "$ls" : '.*-> \(.*\)$'`
+    if expr "$link" : '/.*' > /dev/null; then
       PRG="$link"
     else
-      PRG="$(dirname "$PRG")/$link"
+      PRG="`dirname "$PRG"`/$link"
     fi
   done
 
-  saveddir=$(pwd)
+  saveddir=`pwd`
 
-  M2_HOME=$(dirname "$PRG")/..
+  M2_HOME=`dirname "$PRG"`/..
 
   # make it fully qualified
-  M2_HOME=$(cd "$M2_HOME" && pwd)
+  M2_HOME=`cd "$M2_HOME" && pwd`
 
   cd "$saveddir"
   # echo Using m2 at $M2_HOME
 fi
 
 # For Cygwin, ensure paths are in UNIX format before anything is touched
-if $cygwin; then
+if $cygwin ; then
   [ -n "$M2_HOME" ] &&
-    M2_HOME=$(cygpath --unix "$M2_HOME")
+    M2_HOME=`cygpath --unix "$M2_HOME"`
   [ -n "$JAVA_HOME" ] &&
-    JAVA_HOME=$(cygpath --unix "$JAVA_HOME")
+    JAVA_HOME=`cygpath --unix "$JAVA_HOME"`
   [ -n "$CLASSPATH" ] &&
-    CLASSPATH=$(cygpath --path --unix "$CLASSPATH")
+    CLASSPATH=`cygpath --path --unix "$CLASSPATH"`
 fi
 
 # For Mingw, ensure paths are in UNIX format before anything is touched
-if $mingw; then
+if $mingw ; then
   [ -n "$M2_HOME" ] &&
-    M2_HOME="$( (
-      cd "$M2_HOME"
-      pwd
-    ))"
+    M2_HOME="`(cd "$M2_HOME"; pwd)`"
   [ -n "$JAVA_HOME" ] &&
-    JAVA_HOME="$( (
-      cd "$JAVA_HOME"
-      pwd
-    ))"
+    JAVA_HOME="`(cd "$JAVA_HOME"; pwd)`"
 fi
 
 if [ -z "$JAVA_HOME" ]; then
-  javaExecutable="$(which javac)"
-  if [ -n "$javaExecutable" ] && ! [ "$(expr \"$javaExecutable\" : '\([^ ]*\)')" = "no" ]; then
+  javaExecutable="`which javac`"
+  if [ -n "$javaExecutable" ] && ! [ "`expr \"$javaExecutable\" : '\([^ ]*\)'`" = "no" ]; then
     # readlink(1) is not available as standard on Solaris 10.
-    readLink=$(which readlink)
-    if [ ! $(expr "$readLink" : '\([^ ]*\)') = "no" ]; then
-      if $darwin; then
-        javaHome="$(dirname \"$javaExecutable\")"
-        javaExecutable="$(cd \"$javaHome\" && pwd -P)/javac"
+    readLink=`which readlink`
+    if [ ! `expr "$readLink" : '\([^ ]*\)'` = "no" ]; then
+      if $darwin ; then
+        javaHome="`dirname \"$javaExecutable\"`"
+        javaExecutable="`cd \"$javaHome\" && pwd -P`/javac"
       else
-        javaExecutable="$(readlink -f \"$javaExecutable\")"
+        javaExecutable="`readlink -f \"$javaExecutable\"`"
       fi
-      javaHome="$(dirname \"$javaExecutable\")"
-      javaHome=$(expr "$javaHome" : '\(.*\)/bin')
+      javaHome="`dirname \"$javaExecutable\"`"
+      javaHome=`expr "$javaHome" : '\(.*\)/bin'`
       JAVA_HOME="$javaHome"
       export JAVA_HOME
     fi
   fi
 fi
 
-if [ -z "$JAVACMD" ]; then
-  if [ -n "$JAVA_HOME" ]; then
-    if [ -x "$JAVA_HOME/jre/sh/java" ]; then
+if [ -z "$JAVACMD" ] ; then
+  if [ -n "$JAVA_HOME"  ] ; then
+    if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
       # IBM's JDK on AIX uses strange locations for the executables
       JAVACMD="$JAVA_HOME/jre/sh/java"
     else
       JAVACMD="$JAVA_HOME/bin/java"
     fi
   else
-    JAVACMD="$(which java)"
+    JAVACMD="`which java`"
   fi
 fi
 
-if [ ! -x "$JAVACMD" ]; then
+if [ ! -x "$JAVACMD" ] ; then
   echo "Error: JAVA_HOME is not defined correctly." >&2
   echo "  We cannot execute $JAVACMD" >&2
   exit 1
 fi
 
-if [ -z "$JAVA_HOME" ]; then
+if [ -z "$JAVA_HOME" ] ; then
   echo "Warning: JAVA_HOME environment variable is not set."
 fi
 
@@ -172,24 +165,22 @@ CLASSWORLDS_LAUNCHER=org.codehaus.plexus.classworlds.launcher.Launcher
 # first directory with .mvn subdirectory is considered project base directory
 find_maven_basedir() {
 
-  if [ -z "$1" ]; then
+  if [ -z "$1" ]
+  then
     echo "Path not specified to find_maven_basedir"
     return 1
   fi
 
   basedir="$1"
   wdir="$1"
-  while [ "$wdir" != '/' ]; do
-    if [ -d "$wdir"/.mvn ]; then
+  while [ "$wdir" != '/' ] ; do
+    if [ -d "$wdir"/.mvn ] ; then
       basedir=$wdir
       break
     fi
     # workaround for JBEAP-8937 (on Solaris 10/Sparc)
     if [ -d "${wdir}" ]; then
-      wdir=$(
-        cd "$wdir/.."
-        pwd
-      )
+      wdir=`cd "$wdir/.."; pwd`
     fi
     # end of workaround
   done
@@ -199,13 +190,13 @@ find_maven_basedir() {
 # concatenates all lines of a file
 concat_lines() {
   if [ -f "$1" ]; then
-    echo "$(tr -s '\n' ' ' <"$1")"
+    echo "$(tr -s '\n' ' ' < "$1")"
   fi
 }
 
-BASE_DIR=$(find_maven_basedir "$(pwd)")
+BASE_DIR=`find_maven_basedir "$(pwd)"`
 if [ -z "$BASE_DIR" ]; then
-  exit 1
+  exit 1;
 fi
 
 ##########################################################################################
@@ -213,78 +204,75 @@ fi
 # This allows using the maven wrapper in projects that prohibit checking in binary data.
 ##########################################################################################
 if [ -r "$BASE_DIR/.mvn/wrapper/maven-wrapper.jar" ]; then
-  if [ "$MVNW_VERBOSE" = true ]; then
-    echo "Found .mvn/wrapper/maven-wrapper.jar"
-  fi
+    if [ "$MVNW_VERBOSE" = true ]; then
+      echo "Found .mvn/wrapper/maven-wrapper.jar"
+    fi
 else
-  if [ "$MVNW_VERBOSE" = true ]; then
-    echo "Couldn't find .mvn/wrapper/maven-wrapper.jar, downloading it ..."
-  fi
-  if [ -n "$MVNW_REPOURL" ]; then
-    jarUrl="$MVNW_REPOURL/io/takari/maven-wrapper/0.5.5/maven-wrapper-0.5.5.jar"
-  else
-    jarUrl="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.5/maven-wrapper-0.5.5.jar"
-  fi
-  while IFS="=" read key value; do
-    case "$key" in wrapperUrl)
-      jarUrl="$value"
-      break
-      ;;
-    esac
-  done <"$BASE_DIR/.mvn/wrapper/maven-wrapper.properties"
-  if [ "$MVNW_VERBOSE" = true ]; then
-    echo "Downloading from: $jarUrl"
-  fi
-  wrapperJarPath="$BASE_DIR/.mvn/wrapper/maven-wrapper.jar"
-  if $cygwin; then
-    wrapperJarPath=$(cygpath --path --windows "$wrapperJarPath")
-  fi
-
-  if command -v wget >/dev/null; then
     if [ "$MVNW_VERBOSE" = true ]; then
-      echo "Found wget ... using wget"
+      echo "Couldn't find .mvn/wrapper/maven-wrapper.jar, downloading it ..."
     fi
-    if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then
-      wget "$jarUrl" -O "$wrapperJarPath"
+    if [ -n "$MVNW_REPOURL" ]; then
+      jarUrl="$MVNW_REPOURL/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
     else
-      wget --http-user=$MVNW_USERNAME --http-password=$MVNW_PASSWORD "$jarUrl" -O "$wrapperJarPath"
+      jarUrl="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
     fi
-  elif command -v curl >/dev/null; then
+    while IFS="=" read key value; do
+      case "$key" in (wrapperUrl) jarUrl="$value"; break ;;
+      esac
+    done < "$BASE_DIR/.mvn/wrapper/maven-wrapper.properties"
     if [ "$MVNW_VERBOSE" = true ]; then
-      echo "Found curl ... using curl"
+      echo "Downloading from: $jarUrl"
     fi
-    if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then
-      curl -o "$wrapperJarPath" "$jarUrl" -f
-    else
-      curl --user $MVNW_USERNAME:$MVNW_PASSWORD -o "$wrapperJarPath" "$jarUrl" -f
-    fi
-
-  else
-    if [ "$MVNW_VERBOSE" = true ]; then
-      echo "Falling back to using Java to download"
-    fi
-    javaClass="$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.java"
-    # For Cygwin, switch paths to Windows format before running javac
+    wrapperJarPath="$BASE_DIR/.mvn/wrapper/maven-wrapper.jar"
     if $cygwin; then
-      javaClass=$(cygpath --path --windows "$javaClass")
+      wrapperJarPath=`cygpath --path --windows "$wrapperJarPath"`
     fi
-    if [ -e "$javaClass" ]; then
-      if [ ! -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then
+
+    if command -v wget > /dev/null; then
         if [ "$MVNW_VERBOSE" = true ]; then
-          echo " - Compiling MavenWrapperDownloader.java ..."
+          echo "Found wget ... using wget"
         fi
-        # Compiling the Java class
-        ("$JAVA_HOME/bin/javac" "$javaClass")
-      fi
-      if [ -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then
-        # Running the downloader
+        if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then
+            wget "$jarUrl" -O "$wrapperJarPath"
+        else
+            wget --http-user=$MVNW_USERNAME --http-password=$MVNW_PASSWORD "$jarUrl" -O "$wrapperJarPath"
+        fi
+    elif command -v curl > /dev/null; then
         if [ "$MVNW_VERBOSE" = true ]; then
-          echo " - Running MavenWrapperDownloader.java ..."
+          echo "Found curl ... using curl"
+        fi
+        if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then
+            curl -o "$wrapperJarPath" "$jarUrl" -f
+        else
+            curl --user $MVNW_USERNAME:$MVNW_PASSWORD -o "$wrapperJarPath" "$jarUrl" -f
+        fi
+
+    else
+        if [ "$MVNW_VERBOSE" = true ]; then
+          echo "Falling back to using Java to download"
+        fi
+        javaClass="$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.java"
+        # For Cygwin, switch paths to Windows format before running javac
+        if $cygwin; then
+          javaClass=`cygpath --path --windows "$javaClass"`
+        fi
+        if [ -e "$javaClass" ]; then
+            if [ ! -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then
+                if [ "$MVNW_VERBOSE" = true ]; then
+                  echo " - Compiling MavenWrapperDownloader.java ..."
+                fi
+                # Compiling the Java class
+                ("$JAVA_HOME/bin/javac" "$javaClass")
+            fi
+            if [ -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then
+                # Running the downloader
+                if [ "$MVNW_VERBOSE" = true ]; then
+                  echo " - Running MavenWrapperDownloader.java ..."
+                fi
+                ("$JAVA_HOME/bin/java" -cp .mvn/wrapper MavenWrapperDownloader "$MAVEN_PROJECTBASEDIR")
+            fi
         fi
-        ("$JAVA_HOME/bin/java" -cp .mvn/wrapper MavenWrapperDownloader "$MAVEN_PROJECTBASEDIR")
-      fi
     fi
-  fi
 fi
 ##########################################################################################
 # End of extension
@@ -299,13 +287,13 @@ MAVEN_OPTS="$(concat_lines "$MAVEN_PROJECTBASEDIR/.mvn/jvm.config") $MAVEN_OPTS"
 # For Cygwin, switch paths to Windows format before running java
 if $cygwin; then
   [ -n "$M2_HOME" ] &&
-    M2_HOME=$(cygpath --path --windows "$M2_HOME")
+    M2_HOME=`cygpath --path --windows "$M2_HOME"`
   [ -n "$JAVA_HOME" ] &&
-    JAVA_HOME=$(cygpath --path --windows "$JAVA_HOME")
+    JAVA_HOME=`cygpath --path --windows "$JAVA_HOME"`
   [ -n "$CLASSPATH" ] &&
-    CLASSPATH=$(cygpath --path --windows "$CLASSPATH")
+    CLASSPATH=`cygpath --path --windows "$CLASSPATH"`
   [ -n "$MAVEN_PROJECTBASEDIR" ] &&
-    MAVEN_PROJECTBASEDIR=$(cygpath --path --windows "$MAVEN_PROJECTBASEDIR")
+    MAVEN_PROJECTBASEDIR=`cygpath --path --windows "$MAVEN_PROJECTBASEDIR"`
 fi
 
 # Provide a "standardized" way to retrieve the CLI args that will

mvnw.cmd

@@ -7,7 +7,7 @@
 @REM "License"); you may not use this file except in compliance
 @REM with the License.  You may obtain a copy of the License at
 @REM
-@REM    https://www.apache.org/licenses/LICENSE-2.0
+@REM    http://www.apache.org/licenses/LICENSE-2.0
 @REM
 @REM Unless required by applicable law or agreed to in writing,
 @REM software distributed under the License is distributed on an
@@ -18,7 +18,7 @@
 @REM ----------------------------------------------------------------------------
 
 @REM ----------------------------------------------------------------------------
-@REM Maven2 Start Up Batch script
+@REM Maven Start Up Batch script
 @REM
 @REM Required ENV vars:
 @REM JAVA_HOME - location of a JDK home dir
@@ -26,7 +26,7 @@
 @REM Optional ENV vars
 @REM M2_HOME - location of maven2's installed home dir
 @REM MAVEN_BATCH_ECHO - set to 'on' to enable the echoing of the batch commands
-@REM MAVEN_BATCH_PAUSE - set to 'on' to wait for a key stroke before ending
+@REM MAVEN_BATCH_PAUSE - set to 'on' to wait for a keystroke before ending
 @REM MAVEN_OPTS - parameters passed to the Java VM when running Maven
 @REM     e.g. to debug Maven itself, use
 @REM set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000
@@ -120,7 +120,7 @@ SET MAVEN_JAVA_EXE="%JAVA_HOME%\bin\java.exe"
 set WRAPPER_JAR="%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.jar"
 set WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain
 
-set DOWNLOAD_URL="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.5/maven-wrapper-0.5.5.jar"
+set DOWNLOAD_URL="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
 
 FOR /F "tokens=1,2 delims==" %%A IN ("%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.properties") DO (
     IF "%%A"=="wrapperUrl" SET DOWNLOAD_URL=%%B
@@ -134,7 +134,7 @@ if exist %WRAPPER_JAR% (
     )
 ) else (
     if not "%MVNW_REPOURL%" == "" (
-        SET DOWNLOAD_URL="%MVNW_REPOURL%/io/takari/maven-wrapper/0.5.5/maven-wrapper-0.5.5.jar"
+        SET DOWNLOAD_URL="%MVNW_REPOURL%/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
     )
     if "%MVNW_VERBOSE%" == "true" (
         echo Couldn't find %WRAPPER_JAR%, downloading it ...

pom.xml

@@ -76,14 +76,6 @@
             <scope>compile</scope>
         </dependency>
 
-        <!--Json-->
-        <dependency>
-            <groupId>net.sf.json-lib</groupId>
-            <artifactId>json-lib</artifactId>
-            <version>2.4</version>
-            <classifier>jdk15</classifier>
-        </dependency>
-
         <!--    七牛云SDK    -->
         <dependency>
             <groupId>com.qiniu</groupId>
@@ -139,7 +131,7 @@
         <dependency>
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>
-            <version>4.12</version>
+            <version>4.13.1</version>
         </dependency>
 
         <!--    JJwt    -->
@@ -148,6 +140,69 @@
             <artifactId>jjwt</artifactId>
             <version>0.9.1</version>
         </dependency>
+
+        <!--    OkHttp  -->
+        <dependency>
+            <groupId>com.squareup.okhttp3</groupId>
+            <artifactId>okhttp</artifactId>
+            <version>4.8.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.jetbrains.kotlin</groupId>
+            <artifactId>kotlin-stdlib</artifactId>
+            <version>1.3.72</version>
+            <scope>compile</scope>
+        </dependency>
+
+
+        <dependency>
+            <groupId>net.sourceforge.htmlunit</groupId>
+            <artifactId>htmlunit</artifactId>
+            <version>2.42.0</version>
+        </dependency>
+
+
+        <dependency>
+            <groupId>com.h2database</groupId>
+            <artifactId>h2</artifactId>
+            <version>1.4.200</version>
+        </dependency>
+        <dependency>
+            <groupId>com.github.kstyrc</groupId>
+            <artifactId>embedded-redis</artifactId>
+            <version>0.6</version>
+            <scope>test</scope>
+        </dependency>
+
+
+        <dependency>
+            <groupId>javax.xml.bind</groupId>
+            <artifactId>jaxb-api</artifactId>
+            <version>2.3.0</version>
+        </dependency>
+        <dependency>
+            <groupId>com.sun.xml.bind</groupId>
+            <artifactId>jaxb-impl</artifactId>
+            <version>2.3.0</version>
+        </dependency>
+        <dependency>
+            <groupId>com.sun.xml.bind</groupId>
+            <artifactId>jaxb-core</artifactId>
+            <version>2.3.0</version>
+        </dependency>
+        <dependency>
+            <groupId>javax.activation</groupId>
+            <artifactId>activation</artifactId>
+            <version>1.1.1</version>
+        </dependency>
+
+
+        <dependency>
+            <groupId>org.lionsoul</groupId>
+            <artifactId>ip2region</artifactId>
+            <version>1.7.2</version>
+        </dependency>
+
     </dependencies>
 
     <build>
@@ -157,21 +212,15 @@
                 <artifactId>spring-boot-maven-plugin</artifactId>
             </plugin>
             <plugin>
-                <groupId>org.jacoco</groupId>
-                <artifactId>jacoco-maven-plugin</artifactId>
-                <version>0.7.9</version>
+                <groupId>org.jetbrains.kotlin</groupId>
+                <artifactId>kotlin-maven-plugin</artifactId>
+                <version>1.3.72</version>
                 <executions>
                     <execution>
-                        <id>pre-unit-test</id>
+                        <id>compile</id>
+                        <phase>process-sources</phase>
                         <goals>
-                            <goal>prepare-agent</goal>
-                        </goals>
-                    </execution>
-                    <execution>
-                        <id>post-unit-test</id>
-                        <phase>test</phase>
-                        <goals>
-                            <goal>report</goal>
+                            <goal>compile</goal>
                         </goals>
                     </execution>
                 </executions>

src/main/java/cn/celess/blog/BlogApplication.java

@@ -1,20 +1,34 @@
 package cn.celess.blog;
 
 import org.mybatis.spring.annotation.MapperScan;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
+import org.springframework.boot.ApplicationArguments;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.scheduling.annotation.EnableAsync;
 
+/**
+ * @author zheng
+ */
 @SpringBootApplication
 @EnableAsync
 @MapperScan("cn.celess.blog.mapper")
 public class BlogApplication {
-    public static final Logger logger = LoggerFactory.getLogger(BlogApplication.class);
+    public static ConfigurableApplicationContext context;
 
     public static void main(String[] args) {
-        SpringApplication.run(BlogApplication.class, args);
-        logger.info("启动完成！");
+        context = SpringApplication.run(BlogApplication.class, args);
+    }
+
+    public static void restart() {
+        ApplicationArguments args = context.getBean(ApplicationArguments.class);
+
+        Thread thread = new Thread(() -> {
+            context.close();
+            context = SpringApplication.run(BlogApplication.class, args.getSourceArgs());
+        });
+        thread.setDaemon(false);
+        thread.start();
     }
 }

src/main/java/cn/celess/blog/configuration/ApplicationListener.java

@@ -0,0 +1,59 @@
+package cn.celess.blog.configuration;
+
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.entity.Config;
+import cn.celess.blog.mapper.ConfigMapper;
+import cn.celess.blog.service.fileserviceimpl.LocalFileServiceImpl;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.ApplicationArguments;
+import org.springframework.boot.ApplicationRunner;
+import org.springframework.core.env.Environment;
+import org.springframework.stereotype.Component;
+
+import java.io.File;
+import java.util.List;
+import java.util.stream.Collectors;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/16 16:00
+ * @desc :
+ */
+@Component
+@Slf4j
+public class ApplicationListener implements ApplicationRunner {
+
+    @Autowired
+    ConfigMapper configMapper;
+    @Autowired
+    Environment env;
+
+    @Override
+    public void run(ApplicationArguments args) throws Exception {
+        log.info("博客启动！");
+        // 设置初始值
+        setProperty(ConfigKeyEnum.FILE_QINIU_SECRET_KEY);
+        setProperty(ConfigKeyEnum.FILE_QINIU_ACCESS_KEY);
+        setProperty(ConfigKeyEnum.FILE_QINIU_BUCKET);
+        setProperty(ConfigKeyEnum.BLOG_FILE_PATH);
+
+        List<Config> configurations = configMapper.getConfigurations()
+                .stream()
+                .filter(config -> config.getValue() != null && !"".equals(config.getValue()))
+                .collect(Collectors.toList());
+        configurations.forEach(config -> System.setProperty(config.getName(), config.getValue()));
+        log.debug("注入配置成功 {}", configurations.stream().map(Config::getName).collect(Collectors.toList()));
+        File path = new File(LocalFileServiceImpl.getPath(System.getProperty(ConfigKeyEnum.BLOG_FILE_PATH.getKey())));
+        if (!path.exists() && !path.mkdirs()) {
+            throw new IllegalAccessException("创建数据目录失败==>" + path.getAbsolutePath());
+        }
+    }
+
+    private void setProperty(ConfigKeyEnum e) {
+        String property = env.getProperty(e.getKey());
+        if (property != null) {
+            System.setProperty(e.getKey(), property);
+        }
+    }
+}

src/main/java/cn/celess/blog/configuration/DruidConfig.java

@@ -1,41 +1,81 @@
 package cn.celess.blog.configuration;
 
 import com.alibaba.druid.pool.DruidDataSource;
-import org.springframework.beans.factory.annotation.Value;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.core.env.Environment;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Properties;
 
 /**
  * @author : xiaohai
  * @date : 2019/03/28 14:26
  */
+@Slf4j
 @Configuration
 public class DruidConfig {
-    @Value("${spring.datasource.url}")
-    private String dbUrl;
 
-    @Value("${spring.datasource.username}")
-    private String username;
+    @Autowired
+    Environment env;
 
-    @Value("${spring.datasource.password}")
-    private String password;
-
-    @Value("${spring.datasource.driver-class-name}")
-    private String driverClassName;
+    public static final String DB_CONFIG_PATH = System.getProperty("user.home") + "/blog/application.properties";
+    public static final String DB_CONFIG_URL_PREFIX = "spring.datasource.url";
+    public static final String DB_CONFIG_USERNAME_PREFIX = "spring.datasource.username";
+    public static final String DB_CONFIG_PASSWORD_PREFIX = "spring.datasource.password";
+    public static final String DB_CONFIG_DRIVER_CLASS_NAME_PREFIX = "spring.datasource.driver-class-name";
+    public static final String TEST_PROFILES = "test";
 
     @Bean
-    public DruidDataSource druidDataSource() {
-        DruidDataSource dataSource = new DruidDataSource();
-        dataSource.setDriverClassName("com.mysql.cj.jdbc.Driver");
-        // 数据库基本信息
-        dataSource.setUrl(dbUrl);
-        dataSource.setUsername(username);
-        dataSource.setPassword(password);
-
-        // 数据库连接池配置
+    public DruidDataSource initDataSource() throws IOException {
+        DruidDataSource dataSource;
+        File file = new File(DB_CONFIG_PATH);
+        if (file.exists() && !Arrays.asList(env.getActiveProfiles()).contains(TEST_PROFILES)) {
+            log.debug("从文件中获取数据库配置");
+            dataSource = readConfigFromFile(file);
+        } else {
+            log.debug("默认数据库配置");
+            dataSource = defaultDruidSource();
+        }
         dataSource.setInitialSize(10);
         dataSource.setMinIdle(10);
         dataSource.setMaxActive(100);
+        dataSource.setValidationQuery("select 1");
+        return dataSource;
+    }
+
+    private DruidDataSource readConfigFromFile(File file) throws IOException {
+        Properties properties = new Properties();
+        FileInputStream fis = new FileInputStream(file);
+        properties.load(fis);
+        fis.close();
+        String url = properties.getProperty(DB_CONFIG_URL_PREFIX, null);
+        String username = properties.getProperty(DB_CONFIG_USERNAME_PREFIX, null);
+        String password = properties.getProperty(DB_CONFIG_PASSWORD_PREFIX, null);
+        String className = properties.getProperty(DB_CONFIG_DRIVER_CLASS_NAME_PREFIX, null);
+        if (url == null || username == null || password == null || className == null) {
+            return defaultDruidSource();
+        }
+        DruidDataSource dataSource = new DruidDataSource();
+        dataSource.setDriverClassName(className);
+        dataSource.setUrl(url);
+        dataSource.setUsername(username);
+        dataSource.setPassword(password);
+        return dataSource;
+    }
+
+
+    private DruidDataSource defaultDruidSource() {
+        DruidDataSource dataSource = new DruidDataSource();
+        dataSource.setDriverClassName(env.getProperty(DruidConfig.DB_CONFIG_DRIVER_CLASS_NAME_PREFIX));
+        dataSource.setUrl(env.getProperty(DruidConfig.DB_CONFIG_URL_PREFIX));
+        dataSource.setUsername(env.getProperty(DruidConfig.DB_CONFIG_USERNAME_PREFIX));
+        dataSource.setPassword(env.getProperty(DruidConfig.DB_CONFIG_PASSWORD_PREFIX));
         return dataSource;
     }
 }

src/main/java/cn/celess/blog/configuration/RedisConfig.java

@@ -31,19 +31,16 @@ public class RedisConfig extends CachingConfigurerSupport {
     @Override
     @Bean
     public KeyGenerator keyGenerator() {
-        return new KeyGenerator() {
-            @Override
-            public Object generate(Object target, Method method, Object... params) {
-                StringBuilder sb = new StringBuilder();
-                String name = target.getClass().getName();
-                sb.append(name.substring(name.lastIndexOf(".") + 1));
-                sb.append(":");
-                sb.append(method.getName());
-                for (Object obj : params) {
-                    sb.append("-").append(obj.toString());
-                }
-                return sb.toString();
+        return (target, method, params) -> {
+            StringBuilder sb = new StringBuilder();
+            String name = target.getClass().getName();
+            sb.append(name.substring(name.lastIndexOf(".") + 1));
+            sb.append(":");
+            sb.append(method.getName());
+            for (Object obj : params) {
+                sb.append("-").append(obj.toString());
             }
+            return sb.toString();
         };
     }
 

src/main/java/cn/celess/blog/configuration/filter/AuthenticationFilter.java

@@ -5,7 +5,7 @@ import cn.celess.blog.entity.Response;
 import cn.celess.blog.service.UserService;
 import cn.celess.blog.util.JwtUtil;
 import cn.celess.blog.util.RedisUtil;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -84,7 +84,7 @@ public class AuthenticationFilter implements HandlerInterceptor {
         response.setHeader("Content-Type", "application/json;charset=UTF-8");
         try {
             logger.info("鉴权失败，[code:{},msg:{},path:{}]", e.getCode(), e.getMsg(), request.getRequestURI() + "?" + request.getQueryString());
-            response.getWriter().println(JSONObject.fromObject(Response.response(e, null)));
+            response.getWriter().println(new ObjectMapper().writeValueAsString(Response.response(e, null)));
         } catch (IOException ex) {
             ex.printStackTrace();
         }

src/main/java/cn/celess/blog/configuration/listener/SessionListener.java

@@ -25,7 +25,6 @@ public class SessionListener implements HttpSessionListener {
 
     @Override
     public void sessionCreated(HttpSessionEvent se) {
-        // TODO : can move 'visit' api  to here
         se.getSession().setAttribute("visitDetail", new HashMap<String, Integer>());
         // 10s for debug
         // se.getSession().setMaxInactiveInterval(10);
@@ -44,9 +43,7 @@ public class SessionListener implements HttpSessionListener {
         sb.append("\t登录情况 => ");
         String email = (String) se.getSession().getAttribute("email");
         sb.append(email == null ? "游客访问" : email);
-        visitDetail.forEach((s, integer) -> {
-            sb.append("\n").append("Method:[").append(s.split(":")[1]).append("]\tTimes:[").append(integer).append("]\tPath:[").append(s.split(":")[0]).append("]");
-        });
+        visitDetail.forEach((s, integer) -> sb.append("\n").append("Method:[").append(s.split(":")[1]).append("]\tTimes:[").append(integer).append("]\tPath:[").append(s.split(":")[0]).append("]"));
         log.info(sb.toString());
     }
 }

src/main/java/cn/celess/blog/controller/ArticleController.java

@@ -3,17 +3,15 @@ package cn.celess.blog.controller;
 import cn.celess.blog.enmu.ResponseEnum;
 import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.model.ArticleModel;
-import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.request.ArticleReq;
 import cn.celess.blog.service.ArticleService;
 import cn.celess.blog.util.RedisUserUtil;
 import cn.celess.blog.util.SitemapGenerateUtil;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletRequest;
-import java.util.List;
-import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -27,6 +25,8 @@ public class ArticleController {
     SitemapGenerateUtil sitemapGenerateUtil;
     @Autowired
     RedisUserUtil redisUserUtil;
+    @Value("${spring.profiles.active}")
+    private String activeModel;
 
     /**
      * 新建一篇文章
@@ -37,7 +37,9 @@ public class ArticleController {
     @PostMapping("/admin/article/create")
     public Response create(@RequestBody ArticleReq body) {
         ArticleModel articleModel = articleService.create(body);
-        sitemapGenerateUtil.createSitemap();
+        if ("prod".equals(activeModel)) {
+            sitemapGenerateUtil.createSitemap();
+        }
         return Response.success(articleModel);
     }
 
@@ -50,7 +52,9 @@ public class ArticleController {
     @DeleteMapping("/admin/article/del")
     public Response delete(@RequestParam("articleID") long articleId) {
         boolean delete = articleService.delete(articleId);
-        sitemapGenerateUtil.createSitemap();
+        if ("prod".equals(activeModel)) {
+            sitemapGenerateUtil.createSitemap();
+        }
         return Response.success(delete);
     }
 
@@ -63,7 +67,9 @@ public class ArticleController {
     @PutMapping("/admin/article/update")
     public Response update(@RequestBody ArticleReq body) {
         ArticleModel update = articleService.update(body);
-        sitemapGenerateUtil.createSitemap();
+        if ("prod".equals(activeModel)) {
+            sitemapGenerateUtil.createSitemap();
+        }
         return Response.success(update);
     }
 
@@ -114,7 +120,7 @@ public class ArticleController {
     @GetMapping("/admin/articles")
     public Response adminArticles(@RequestParam(name = "page", defaultValue = "1") int page,
                                   @RequestParam(name = "count", defaultValue = "10") int count,
-                                  @RequestParam(name = "deleted", defaultValue = "false") boolean deleted) {
+                                  @RequestParam(name = "deleted", required = false) Boolean deleted) {
         return Response.success(articleService.adminArticles(count, page, deleted));
     }
 

src/main/java/cn/celess/blog/controller/CommonController.java

@@ -2,15 +2,16 @@ package cn.celess.blog.controller;
 
 import cn.celess.blog.enmu.ResponseEnum;
 import cn.celess.blog.entity.Response;
-import cn.celess.blog.entity.model.QiniuResponse;
+import cn.celess.blog.entity.model.FileResponse;
 import cn.celess.blog.exception.MyException;
 import cn.celess.blog.service.CountService;
-import cn.celess.blog.service.QiniuService;
+import cn.celess.blog.service.FileService;
 import cn.celess.blog.util.HttpUtil;
+import cn.celess.blog.util.RedisUserUtil;
 import cn.celess.blog.util.RedisUtil;
 import cn.celess.blog.util.VeriCodeUtil;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -27,9 +28,7 @@ import javax.servlet.http.HttpServletResponse;
 import java.awt.image.BufferedImage;
 import java.io.IOException;
 import java.io.OutputStream;
-import java.util.Enumeration;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
 import java.util.concurrent.TimeUnit;
 
 /**
@@ -43,11 +42,11 @@ public class CommonController {
     @Autowired
     CountService countService;
     @Autowired
-    QiniuService qiniuService;
+    FileService fileService;
     @Autowired
     RedisUtil redisUtil;
     @Autowired
-    HttpServletRequest request;
+    RedisUserUtil redisUserUtil;
 
 
     @GetMapping("/counts")
@@ -89,7 +88,7 @@ public class CommonController {
      * @throws IOException IOException
      */
     @GetMapping(value = "/imgCode", produces = MediaType.IMAGE_PNG_VALUE)
-    public void getImg(HttpServletResponse response) throws IOException {
+    public void getImg(HttpServletRequest request, HttpServletResponse response) throws IOException {
         Object[] obj = VeriCodeUtil.createImage();
         request.getSession().setAttribute("code", obj[0]);
         //将图片输出给浏览器
@@ -133,14 +132,14 @@ public class CommonController {
      * FIXME :: 单张图片多次上传的问题
      * editor.md图片上传的接口
      * FUCK !!!
-     *
+     * !! 推荐使用 fileUpload(/fileUpload) 接口
      * @param file 文件
-     * @return
-     * @throws IOException
      */
+    @Deprecated
     @PostMapping("/imgUpload")
     public void upload(HttpServletRequest request, HttpServletResponse response, @RequestParam("editormd-image-file") MultipartFile file) throws IOException {
-        JSONObject jsonObject = new JSONObject();
+        Map<String, Object> map = new HashMap<>();
+        ObjectMapper mapper = new ObjectMapper();
         String uploadTimesStr = redisUtil.get(request.getRemoteAddr() + "-ImgUploadTimes");
         int uploadTimes = 0;
         if (uploadTimesStr != null) {
@@ -150,41 +149,74 @@ public class CommonController {
             throw new MyException(ResponseEnum.FAILURE.getCode(), "上传次数已达10次，请2小时后在上传");
         }
         request.setCharacterEncoding("utf-8");
-        response.setContentType("text/html");
+        response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
         if (file.isEmpty()) {
-            jsonObject.put("success", 0);
-            jsonObject.put("message", "上传失败，请选择文件");
-            response.getWriter().println(jsonObject.toString());
+            map.put("success", 0);
+            map.put("message", "上传失败，请选择文件");
+            response.getWriter().println(mapper.writeValueAsString(map));
             return;
         }
         String fileName = file.getOriginalFilename();
+        assert fileName != null;
         String mime = fileName.substring(fileName.lastIndexOf("."));
         if (".png".equals(mime.toLowerCase()) || ".jpg".equals(mime.toLowerCase()) ||
                 ".jpeg".equals(mime.toLowerCase()) || ".bmp".equals(mime.toLowerCase())) {
-            QiniuResponse qiniuResponse = qiniuService.uploadFile(file.getInputStream(), "img_" + System.currentTimeMillis() + mime);
-            jsonObject.put("success", 1);
-            jsonObject.put("message", "上传成功");
-            jsonObject.put("url", "http://cdn.celess.cn/" + qiniuResponse.key);
-            response.getWriter().println(jsonObject.toString());
+            FileResponse fileResponse = fileService.getFileManager().uploadFile(file.getInputStream(), "img_" + System.currentTimeMillis() + mime);
+            map.put("success", 1);
+            map.put("message", "上传成功");
+            map.put("url", "http://cdn.celess.cn/" + fileResponse.key);
+            response.getWriter().println(mapper.writeValueAsString(map));
             redisUtil.setEx(request.getRemoteAddr() + "-ImgUploadTimes", uploadTimes + 1 + "", 2, TimeUnit.HOURS);
             return;
         }
-        jsonObject.put("success", 0);
-        jsonObject.put("message", "上传失败，请上传图片文件");
-        response.getWriter().println(jsonObject.toString());
+        map.put("success", 0);
+        map.put("message", "上传失败，请上传图片文件");
+        response.getWriter().println(mapper.writeValueAsString(map));
     }
 
     @GetMapping("/bingPic")
     public Response bingPic() {
-
-        JSONObject imageObj;
+        ObjectMapper mapper = new ObjectMapper();
+        JsonNode root;
         try {
-            imageObj = JSONObject.fromObject(HttpUtil.get("https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=zh-CN"));
+            root = mapper.readTree(HttpUtil.get("https://cn.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=zh-CN"));
         } catch (IOException e) {
             return Response.failure(null);
         }
-        JSONArray jsonArray = imageObj.getJSONArray("images");
-        String imageName = jsonArray.getJSONObject(0).getString("url");
-        return Response.success("https://cn.bing.com" + imageName);
+        JsonNode images = root.get("images").elements().next();
+        return Response.success("https://cn.bing.com" + images.get("url").asText());
+    }
+
+    @PostMapping("/fileUpload")
+    public Response<List<Map<String, Object>>> fileUpload(@RequestParam("file[]") MultipartFile[] files) throws IOException {
+        List<Map<String, Object>> result = new ArrayList<>();
+        String uploadTimesStr = redisUtil.get(redisUserUtil.get().getEmail() + "-fileUploadTimes");
+        int uploadTimes = 0;
+        if (uploadTimesStr != null) {
+            uploadTimes = Integer.parseInt(uploadTimesStr);
+        }
+        if (uploadTimes == 10) {
+            throw new MyException(ResponseEnum.FAILURE.getCode(), "上传次数已达10次，请2小时后在上传");
+        }
+        if (files.length == 0) {
+            throw new MyException(ResponseEnum.NO_FILE);
+        }
+        for (MultipartFile file : files) {
+            Map<String, Object> resp = new HashMap<>(4);
+
+            String fileName = file.getOriginalFilename();
+            assert fileName != null;
+            String mime = fileName.substring(fileName.lastIndexOf("."));
+            String name = fileName.replace(mime, "").replaceAll(" ", "");
+            FileResponse fileResponse = fileService.getFileManager().uploadFile(file.getInputStream(), "file_" + name + '_' + System.currentTimeMillis() + mime);
+            resp.put("originalFilename", fileName);
+            resp.put("success", fileResponse != null);
+            if (fileResponse != null) {
+                resp.put("host", "http://cdn.celess.cn/");
+                resp.put("path", fileResponse.key);
+            }
+            result.add(resp);
+        }
+        return Response.success(result);
     }
 }

src/main/java/cn/celess/blog/controller/ConfigController.java

@@ -0,0 +1,39 @@
+package cn.celess.blog.controller;
+
+import cn.celess.blog.entity.Config;
+import cn.celess.blog.entity.Response;
+import cn.celess.blog.mapper.ConfigMapper;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/16 20:56
+ * @desc : 配置管理接口
+ */
+@RestController
+public class ConfigController {
+    @Autowired
+    ConfigMapper configMapper;
+
+    @GetMapping("/admin/config")
+    public Response<List<Config>> getConfiguration() {
+        return Response.success(configMapper.getConfigurations());
+    }
+
+    @PutMapping("/admin/config")
+    public Response<List<Config>> updateConfiguration(@RequestBody List<Config> configs) {
+        configs.forEach(config -> configMapper.updateConfiguration(config));
+        configs.forEach(config -> System.setProperty(config.getName(), config.getValue()));
+        return Response.success(configMapper.getConfigurations());
+    }
+
+    @PostMapping("/admin/config")
+    public Response<List<Config>> addConfiguration(@RequestBody List<Config> configs) {
+        configs.forEach(config -> configMapper.addConfiguration(config));
+        configs.forEach(config -> System.setProperty(config.getName(), config.getValue()));
+        return Response.success(configMapper.getConfigurations());
+    }
+}

src/main/java/cn/celess/blog/controller/InstallController.java

@@ -0,0 +1,78 @@
+package cn.celess.blog.controller;
+
+import cn.celess.blog.BlogApplication;
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.enmu.ResponseEnum;
+import cn.celess.blog.entity.Config;
+import cn.celess.blog.entity.InstallParam;
+import cn.celess.blog.entity.Response;
+import cn.celess.blog.exception.MyException;
+import cn.celess.blog.mapper.ConfigMapper;
+import cn.celess.blog.service.InstallService;
+import cn.celess.blog.util.RegexUtil;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Controller;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.ResponseBody;
+
+import javax.validation.Valid;
+import java.util.Map;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/18 15:36
+ * @desc :
+ */
+@Slf4j
+@Controller
+@Validated
+public class InstallController {
+
+    @Autowired
+    InstallService installService;
+    @Autowired
+    ConfigMapper configMapper;
+
+
+    @GetMapping("/is_install")
+    @ResponseBody
+    public Response<Map<String, Object>> isInstall() {
+        Map<String, Object> install = installService.isInstall();
+        return Response.success(install);
+    }
+
+
+    @PostMapping("/install")
+    @ResponseBody
+    public Response<Object> install(@Valid @RequestBody InstallParam installParam) {
+        if (!RegexUtil.emailMatch(installParam.getEmail())) {
+            throw new MyException(ResponseEnum.PARAMETERS_EMAIL_ERROR);
+        }
+        if (!RegexUtil.pwdMatch(installParam.getPassword())) {
+            throw new MyException(ResponseEnum.PARAMETERS_PWD_ERROR);
+        }
+        return Response.success(installService.install(installParam));
+    }
+
+
+    @GetMapping("/default_config")
+    @ResponseBody
+    public Response<String> defaultConfig() {
+        return null;
+    }
+
+
+    @GetMapping("/install")
+    public String install() {
+        Config configuration = configMapper.getConfiguration(ConfigKeyEnum.BLOG_INSTALLED.getKey());
+        if (Boolean.parseBoolean(configuration.getValue())) {
+            return "index.html";
+        }
+        log.info("博客第一次运行，还未安装");
+        return "install.html";
+    }
+}

src/main/java/cn/celess/blog/controller/LinksController.java

@@ -1,23 +1,18 @@
 package cn.celess.blog.controller;
 
-import cn.celess.blog.enmu.ResponseEnum;
 import cn.celess.blog.entity.PartnerSite;
 import cn.celess.blog.entity.Response;
+import cn.celess.blog.entity.request.LinkApplyReq;
 import cn.celess.blog.entity.request.LinkReq;
-import cn.celess.blog.exception.MyException;
 import cn.celess.blog.service.MailService;
 import cn.celess.blog.service.PartnerSiteService;
-import cn.celess.blog.util.DateFormatUtil;
 import cn.celess.blog.util.RedisUtil;
-import cn.celess.blog.util.RegexUtil;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.mail.SimpleMailMessage;
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletRequest;
-import java.util.ArrayList;
 import java.util.List;
-import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -51,47 +46,24 @@ public class LinksController {
 
     @GetMapping("/links")
     public Response allForOpen() {
-        List<PartnerSite> sites = new ArrayList<>();
-        for (PartnerSite p : partnerSiteService.findAll()) {
-            if (p.getOpen()) {
-                //隐藏open字段
-                p.setOpen(null);
-                sites.add(p);
-            }
-        }
+        List<PartnerSite> sites = partnerSiteService.findAll().stream().peek(partnerSite -> partnerSite.setOpen(null)).collect(Collectors.toList());
         return Response.success(sites);
     }
 
     @GetMapping("/admin/links")
     public Response all(@RequestParam("page") int page,
-                        @RequestParam("count") int count) {
-        return Response.success(partnerSiteService.partnerSitePages(page, count));
+                        @RequestParam("count") int count,
+                        @RequestParam(value = "deleted", required = false) Boolean deleted) {
+        return Response.success(partnerSiteService.partnerSitePages(page, count, deleted));
     }
 
     @PostMapping("/apply")
-    public Response apply(@RequestParam("name") String name,
-                          @RequestParam("url") String url) {
-        // TODO :: 弃用发送邮件的方式。
-        if (name == null || name.replaceAll(" ", "").isEmpty()) {
-            return Response.response(ResponseEnum.PARAMETERS_ERROR, null);
-        }
-        if (!RegexUtil.urlMatch(url)) {
-            return Response.response(ResponseEnum.PARAMETERS_URL_ERROR, null);
-        }
-        String applyTimeStr = redisUtil.get(request.getRemoteAddr() + "-Apply");
-        int applyTime = 0;
-        if (applyTimeStr != null) {
-            applyTime = Integer.parseInt(applyTimeStr);
-        }
-        if (applyTime == 10) {
-            throw new MyException(ResponseEnum.FAILURE.getCode(), "申请次数已达10次，请2小时后重试");
-        }
-        SimpleMailMessage message = new SimpleMailMessage();
-        message.setSubject("友链申请：" + name);
-        message.setTo("a@celess.cn");
-        message.setText("name:" + name + "\nurl:" + url + "\n" + DateFormatUtil.getNow());
-        Boolean send = mailService.send(message);
-        redisUtil.setEx(request.getRemoteAddr() + "-Apply", applyTime + 1 + "", 2, TimeUnit.HOURS);
-        return send ? Response.success("") : Response.failure("");
+    public Response apply(@RequestBody() LinkApplyReq linkApplyReq) {
+        return Response.success(partnerSiteService.apply(linkApplyReq));
+    }
+
+    @PostMapping("/reapply")
+    public Response reapply(@RequestParam("key") String key) {
+        return Response.success(partnerSiteService.reapply(key));
     }
 }

src/main/java/cn/celess/blog/controller/TagController.java

@@ -3,12 +3,13 @@ package cn.celess.blog.controller;
 import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.model.TagModel;
 import cn.celess.blog.service.TagService;
-import net.sf.json.JSONObject;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author : xiaohai
@@ -44,13 +45,13 @@ public class TagController {
 
     @GetMapping("/tags/nac")
     public Response getTagNameAndCount() {
-        List<JSONObject> nameAndCount = new ArrayList<>();
+        List<Map<String, Object>> nameAndCount = new ArrayList<>();
         List<TagModel> all = tagService.findAll();
         for (TagModel t : all) {
-            JSONObject jsonObject = new JSONObject();
-            jsonObject.put("name", t.getName());
-            jsonObject.put("size", t.getArticles().size());
-            nameAndCount.add(jsonObject);
+            Map<String, Object> map = new HashMap<>(2);
+            map.put("name", t.getName());
+            map.put("size", t.getArticles().size());
+            nameAndCount.add(map);
         }
         return Response.success(nameAndCount);
     }

src/main/java/cn/celess/blog/controller/UserController.java

@@ -116,8 +116,8 @@ public class UserController {
     }
 
     @GetMapping("/admin/users")
-    public Response getAllUser(@RequestParam("page") int pageNum, @RequestParam("count") int count) {
-        return Response.success(userService.getUserList(pageNum, count));
+    public Response getAllUser(@RequestParam("page") int pageNum, @RequestParam("count") int count, @RequestParam(name = "status", required = false) Integer status) {
+        return Response.success(userService.getUserList(pageNum, count, status));
     }
 
     @GetMapping("/emailStatus/{email}")

src/main/java/cn/celess/blog/controller/VisitorController.java

@@ -27,7 +27,7 @@ public class VisitorController {
     @GetMapping("/admin/visitor/page")
     public Response page(@RequestParam(value = "count", required = false, defaultValue = "10") int count,
                          @RequestParam(value = "page", required = false, defaultValue = "1") int page,
-                         @RequestParam(value = "showLocation", required = false, defaultValue = "false") boolean showLocation) {
+                         @RequestParam(value = "showLocation", required = false, defaultValue = "true") boolean showLocation) {
         return Response.success(visitorService.visitorPage(page, count, showLocation));
     }
 

src/main/java/cn/celess/blog/enmu/ConfigKeyEnum.java

@@ -0,0 +1,37 @@
+package cn.celess.blog.enmu;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/16 16:41
+ * @desc :
+ */
+public enum ConfigKeyEnum {
+
+    /**
+     * 枚举
+     */
+    FILE_TYPE("file.type"),
+    FILE_QINIU_ACCESS_KEY("file.qiniu.accessKey"),
+    FILE_QINIU_SECRET_KEY("file.qiniu.secretKey"),
+    FILE_QINIU_BUCKET("file.qiniu.bucket"),
+    FILE_LOCAL_DIRECTORY_PATH("file.local.directoryPath"),
+    BLOG_FILE_PATH("blog.file.path"),
+    BLOG_INSTALLED("blog.installed"),
+    DB_TYPE("db.type"),
+    DB_URL("spring.datasource.url"),
+    DB_USERNAME("spring.datasource.username"),
+    DB_PASSWORD("spring.datasource.password"),
+    DB_DRIVER_CLASS_NAME("spring.datasource.driver-class-name"),
+    BLOG_DB_PATH("blog.db.path");
+
+
+    private final String key;
+
+    ConfigKeyEnum(String key) {
+        this.key = key;
+    }
+
+    public String getKey() {
+        return key;
+    }
+}

src/main/java/cn/celess/blog/enmu/ResponseEnum.java

@@ -34,6 +34,7 @@ public enum ResponseEnum {
     PWD_NOT_SAME(3602, "新密码与原密码不相同"),
     LOGIN_EXPIRED(3700, "登陆过期"),
     LOGOUT(3710, "账户已注销"),
+    CAN_NOT_USE(3711, "账户不可用"),
     PWD_WRONG(3800, "密码不正确"),
 
     JWT_EXPIRED(3810, "Token过期"),
@@ -58,6 +59,11 @@ public enum ResponseEnum {
     DATA_HAS_EXIST(7020, "数据已存在"),
 
     //其他
+    APPLY_LINK_NO_ADD_THIS_SITE(7200, "暂未在您的网站中抓取到本站链接"),
+    DATA_EXPIRED(7300, "数据过期"),
+    CANNOT_GET_DATA(7400, "暂无法获取到数据"),
+    NO_FILE(7500, "未选择文件，请重新选择"),
+
 
     //提交更新之前，没有获取数据/,
     DID_NOT_GET_THE_DATA(8020, "非法访问"),

src/main/java/cn/celess/blog/enmu/UserAccountStatusEnum.java

@@ -1,5 +1,11 @@
 package cn.celess.blog.enmu;
 
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonValue;
+
+import java.util.HashMap;
+import java.util.Map;
+
 /**
  * @Author: 小海
  * @Date: 2020-05-22 21:32
@@ -36,4 +42,31 @@ public enum UserAccountStatusEnum {
     public String getDesc() {
         return desc;
     }
+
+    public static UserAccountStatusEnum get(int code) {
+        for (UserAccountStatusEnum value : UserAccountStatusEnum.values()) {
+            if (value.code == code) {
+                return value;
+            }
+        }
+        return null;
+    }
+
+    @JsonCreator
+    public static UserAccountStatusEnum get(Map<String, Object> map) {
+        for (UserAccountStatusEnum value : UserAccountStatusEnum.values()) {
+            if (value.code == (int) map.get("code") && value.desc.equals(map.get("desc"))) {
+                return value;
+            }
+        }
+        return null;
+    }
+
+    @JsonValue
+    public Map<String, Object> toJson() {
+        Map<String, Object> map = new HashMap<>(2);
+        map.put("code", code);
+        map.put("desc", desc);
+        return map;
+    }
 }

src/main/java/cn/celess/blog/entity/Config.java

@@ -0,0 +1,26 @@
+package cn.celess.blog.entity;
+
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/16 15:24
+ * @desc :
+ */
+@Data
+@NoArgsConstructor
+public class Config {
+    private Integer id;
+    private String name;
+    private String value;
+
+    public Config(String name) {
+        this.name = name;
+    }
+
+    public Config(ConfigKeyEnum e) {
+        this.name = e.getKey();
+    }
+}

src/main/java/cn/celess/blog/entity/InstallParam.java

@@ -0,0 +1,38 @@
+package cn.celess.blog.entity;
+
+import lombok.Data;
+
+import javax.validation.constraints.NotBlank;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/18 14:52
+ * @desc :
+ */
+@Data
+public class InstallParam {
+    @NotBlank(message = "数据库类型不可为空")
+    private String dbType;
+
+    @NotBlank(message = "数据库主机不可为空")
+    private String dbHost;
+
+    @NotBlank(message = "数据库名称不可为空")
+    private String dbName;
+
+    @NotBlank(message = "数据库用户名不可为空")
+    private String dbUsername;
+
+    @NotBlank(message = "数据库密码不可为空")
+    private String dbPassword;
+
+
+    /**
+     * user 相关
+     */
+    @NotBlank(message = "用户邮箱地址不可为空")
+    private String email;
+
+    @NotBlank(message = "用户密码不可为空")
+    private String password;
+}

src/main/java/cn/celess/blog/entity/PartnerSite.java

@@ -25,6 +25,10 @@ public class PartnerSite {
 
     private Boolean delete = false;
 
+    private String email;
+
+    private Boolean notification = true;
+
     public PartnerSite() {
     }
 

src/main/java/cn/celess/blog/entity/Response.java

@@ -1,8 +1,9 @@
 package cn.celess.blog.entity;
 
 import cn.celess.blog.enmu.ResponseEnum;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.Data;
-import net.sf.json.JSONObject;
+import lombok.SneakyThrows;
 
 import java.io.Serializable;
 
@@ -31,8 +32,8 @@ public class Response<T> implements Serializable {
      * @param result 结果
      * @return Response
      */
-    public static Response success(Object result) {
-        return new Response(ResponseEnum.SUCCESS.getCode(), ResponseEnum.SUCCESS.getMsg(), result);
+    public static <T> Response<T> success(T result) {
+        return new Response<T>(ResponseEnum.SUCCESS.getCode(), ResponseEnum.SUCCESS.getMsg(), result);
     }
 
     /**
@@ -41,8 +42,8 @@ public class Response<T> implements Serializable {
      * @param result 结果
      * @return Response
      */
-    public static Response failure(String result) {
-        return new Response(ResponseEnum.FAILURE.getCode(), ResponseEnum.FAILURE.getMsg(), result);
+    public static Response<String> failure(String result) {
+        return new Response<String>(ResponseEnum.FAILURE.getCode(), ResponseEnum.FAILURE.getMsg(), result);
     }
 
     /**
@@ -52,13 +53,13 @@ public class Response<T> implements Serializable {
      * @param result 结果
      * @return Response
      */
-    public static Response response(ResponseEnum r, String result) {
-        return new Response(r.getCode(), r.getMsg(), result);
+    public static <T> Response<T> response(ResponseEnum r, T result) {
+        return new Response<T>(r.getCode(), r.getMsg(), result);
     }
 
+    @SneakyThrows
     @Override
     public String toString() {
-        JSONObject jsonObject = JSONObject.fromObject(this);
-        return jsonObject.toString();
+        return new ObjectMapper().writeValueAsString(this);
     }
 }

src/main/java/cn/celess/blog/entity/model/FileInfo.java

@@ -0,0 +1,25 @@
+package cn.celess.blog.entity.model;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/15 22:16
+ * @desc :
+ */
+public class FileInfo {
+    /**
+     * 文件名
+     */
+    public String key;
+    /**
+     * 文件hash值
+     */
+    public String hash;
+    /**
+     * 文件大小，单位：字节
+     */
+    public long size;
+    /**
+     * 文件上传时间，单位为：100纳秒
+     */
+    public long uploadTime;
+}

src/main/java/cn/celess/blog/entity/model/FileResponse.java

@@ -5,9 +5,9 @@ package cn.celess.blog.entity.model;
  * @author : xiaohai
  * @date : 2019/04/21 22:43
  */
-public class QiniuResponse {
+public class FileResponse {
     public String key;
     public String hash;
-    public String bucket;
-    public long fsize;
+    public String type;
+    public long size;
 }

src/main/java/cn/celess/blog/entity/model/UserModel.java

@@ -1,5 +1,6 @@
 package cn.celess.blog.entity.model;
 
+import cn.celess.blog.enmu.UserAccountStatusEnum;
 import lombok.Getter;
 import lombok.Setter;
 
@@ -38,5 +39,5 @@ public class UserModel {
 
     private String token;
 
-    private int status;
+    private UserAccountStatusEnum status;
 }

src/main/java/cn/celess/blog/entity/request/LinkApplyReq.java

@@ -0,0 +1,17 @@
+package cn.celess.blog.entity.request;
+
+import lombok.Data;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/07/31 20:50
+ */
+@Data
+public class LinkApplyReq {
+    private String name;
+    private String email;
+    private String url;
+    private String linkUrl;
+    private String desc;
+    private String iconPath;
+}

src/main/java/cn/celess/blog/entity/request/LoginReq.java

@@ -1,12 +1,16 @@
 package cn.celess.blog.entity.request;
 
+import lombok.AllArgsConstructor;
 import lombok.Data;
+import lombok.NoArgsConstructor;
 
 /**
  * @author : xiaohai
  * @date : 2019/06/01 22:47
  */
 @Data
+@NoArgsConstructor
+@AllArgsConstructor
 public class LoginReq {
     private String email;
     private String password;

src/main/java/cn/celess/blog/exception/ExceptionHandle.java

@@ -10,7 +10,9 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.mail.SimpleMailMessage;
 import org.springframework.validation.BindException;
+import org.springframework.validation.BindingResult;
 import org.springframework.web.HttpRequestMethodNotSupportedException;
+import org.springframework.web.bind.MethodArgumentNotValidException;
 import org.springframework.web.bind.MissingServletRequestParameterException;
 import org.springframework.web.bind.annotation.ControllerAdvice;
 import org.springframework.web.bind.annotation.ExceptionHandler;
@@ -18,6 +20,10 @@ import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;
 
 import javax.servlet.http.HttpServletRequest;
+import javax.validation.ConstraintViolation;
+import javax.validation.ConstraintViolationException;
+import javax.validation.ValidationException;
+import java.util.Set;
 
 /**
  * @author : xiaohai
@@ -39,28 +45,34 @@ public class ExceptionHandle {
     public Response handle(Exception e) {
         //自定义错误
         if (e instanceof MyException) {
-            logger.debug("返回了自定义的exception,[code={},msg={}]", ((MyException) e).getCode(), e.getMessage());
-            return new Response(((MyException) e).getCode(), e.getMessage(), null);
+            MyException exception = (MyException) e;
+            logger.debug("返回了自定义的exception,[code={},msg={},result={}]", exception.getCode(), e.getMessage(), exception.getResult());
+            return new Response<>(exception.getCode(), e.getMessage(), exception.getResult());
         }
         //请求路径不支持该方法
         if (e instanceof HttpRequestMethodNotSupportedException) {
             logger.debug("遇到请求路径与请求方法不匹配的请求，[msg={}，path:{},method:{}]", e.getMessage(), request.getRequestURL(), request.getMethod());
-            return new Response(ResponseEnum.ERROR.getCode(), e.getMessage(), null);
+            return new Response<>(ResponseEnum.ERROR.getCode(), e.getMessage(), null);
         }
         //数据输入类型不匹配
         if (e instanceof MethodArgumentTypeMismatchException) {
             logger.debug("输入类型不匹配,[msg={}]", e.getMessage());
-            return new Response(ResponseEnum.PARAMETERS_ERROR.getCode(), "数据输入有问题，请修改后再访问", null);
+            return new Response<>(ResponseEnum.PARAMETERS_ERROR.getCode(), "数据输入有问题，请修改后再访问", null);
         }
         //数据验证失败
         if (e instanceof BindException) {
             logger.debug("数据验证失败,[msg={}]", e.getMessage());
-            return new Response(ResponseEnum.PARAMETERS_ERROR.getCode(), "数据输入有问题，请修改", null);
+            return new Response<>(ResponseEnum.PARAMETERS_ERROR.getCode(), "数据输入有问题，请修改", null);
         }
         //数据输入不完整
         if (e instanceof MissingServletRequestParameterException) {
             logger.debug("数据输入不完整,[msg={}]", e.getMessage());
-            return new Response(ResponseEnum.PARAMETERS_ERROR.getCode(), "数据输入不完整,请检查", null);
+            return new Response<>(ResponseEnum.PARAMETERS_ERROR.getCode(), "数据输入不完整,请检查", null);
+        }
+        if (e instanceof MethodArgumentNotValidException) {
+            logger.debug("数据验证失败，[msg: {}]", e.getMessage());
+            BindingResult violations = ((MethodArgumentNotValidException) e).getBindingResult();
+            return new Response<>(ResponseEnum.PARAMETERS_ERROR.getCode(), violations.getAllErrors().get(0).getDefaultMessage(), null);
         }
 
         // 发送错误信息到邮箱
@@ -69,7 +81,7 @@ public class ExceptionHandle {
             sendMessage(e);
         }
         e.printStackTrace();
-        return new Response(ResponseEnum.ERROR.getCode(), "服务器出现错误，已记录", null);
+        return new Response<>(ResponseEnum.ERROR.getCode(), "服务器出现错误，已记录", null);
     }
 
     /**
@@ -82,7 +94,12 @@ public class ExceptionHandle {
         simpleMailMessage.setTo("a@celess.cn");
         simpleMailMessage.setSubject("服务器出现了错误");
         StringBuilder msg = new StringBuilder();
-        msg.append("requirePath:\n").append(request.getRequestURL().toString()).append("?").append(request.getQueryString()).append("\n\n\n");
+        String queryString = request.getQueryString();
+        msg.append("requirePath:\n").append(request.getRequestURL().toString());
+        if (queryString != null) {
+            msg.append("?").append(queryString);
+        }
+        msg.append("\n\n\n");
         msg.append("msg:\n").append(e.getMessage()).append("\n\n\n");
         msg.append("date:\n").append(DateFormatUtil.getNow()).append("\n\n\n");
         msg.append("from:\n").append(request.getHeader("User-Agent")).append("\n\n\n");

src/main/java/cn/celess/blog/exception/MyException.java

@@ -1,13 +1,16 @@
 package cn.celess.blog.exception;
 
 import cn.celess.blog.enmu.ResponseEnum;
+import lombok.Data;
 
 /**
  * @author : xiaohai
  * @date : 2019/03/28 16:56
  */
+@Data
 public class MyException extends RuntimeException {
     private int code;
+    private Object result;
 
     public MyException(int code, String msg) {
         super(msg);
@@ -19,16 +22,20 @@ public class MyException extends RuntimeException {
         this.code = e.getCode();
     }
 
+    public MyException(ResponseEnum e, Object result) {
+        super(e.getMsg());
+        this.code = e.getCode();
+        this.result = result;
+    }
+
     public MyException(ResponseEnum e, String msg) {
         super(msg + e.getMsg());
         this.code = e.getCode();
     }
 
-    public int getCode() {
-        return code;
-    }
-
-    public void setCode(int code) {
-        this.code = code;
+    public MyException(ResponseEnum e, String msg, Object result) {
+        super(e.getMsg());
+        this.code = e.getCode();
+        this.result = result;
     }
 }

src/main/java/cn/celess/blog/mapper/ConfigMapper.java

@@ -0,0 +1,55 @@
+package cn.celess.blog.mapper;
+
+import cn.celess.blog.entity.Config;
+import org.apache.ibatis.annotations.Mapper;
+import org.springframework.stereotype.Repository;
+
+import java.util.List;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/16 15:24
+ * @desc :
+ */
+@Mapper
+@Repository
+public interface ConfigMapper {
+    /**
+     * 获取单个配置
+     *
+     * @param key 配置名
+     * @return 配置
+     */
+    Config getConfiguration(String key);
+
+    /**
+     * 更新配置
+     *
+     * @param c 配置
+     * @return 改变数据行数
+     */
+    int updateConfiguration(Config c);
+
+    /**
+     * 获取所有配置
+     *
+     * @return 所有配置
+     */
+    List<Config> getConfigurations();
+
+    /**
+     * 新增配置
+     *
+     * @param c 配置
+     * @return 改变行数
+     */
+    int addConfiguration(Config c);
+
+    /**
+     * 删除配置
+     *
+     * @param id 主键id
+     * @return 改变行数
+     */
+    int deleteConfiguration(int id);
+}

src/main/java/cn/celess/blog/mapper/PartnerMapper.java

@@ -36,5 +36,6 @@ public interface PartnerMapper {
 
     List<PartnerSite> findAll();
 
+    List<PartnerSite> findAll(Boolean deleted);
 
 }

src/main/java/cn/celess/blog/mapper/UserMapper.java

@@ -53,6 +53,8 @@ public interface UserMapper {
 
     int setUserRole(Long id, String role);
 
+    List<User> findAll(Integer status);
+
     List<User> findAll();
 
     int update(User user);

src/main/java/cn/celess/blog/service/ArticleService.java

@@ -52,7 +52,7 @@ public interface ArticleService {
      * @param page  数据页
      * @return 分页数据
      */
-    PageData<ArticleModel> adminArticles(int count, int page, boolean deleted);
+    PageData<ArticleModel> adminArticles(int count, int page, Boolean deleted);
 
     /**
      * 获取文章状态为开放的文章

src/main/java/cn/celess/blog/service/FileManager.java

@@ -0,0 +1,40 @@
+package cn.celess.blog.service;
+
+import cn.celess.blog.entity.model.FileInfo;
+import cn.celess.blog.entity.model.FileResponse;
+import org.springframework.stereotype.Service;
+
+import java.io.InputStream;
+import java.util.List;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/15 18:19
+ * @desc : 文件管理器 定义操作文件的方法
+ */
+@Service
+public interface FileManager {
+    /**
+     * 上传文件到文件存储容器中
+     *
+     * @param is       文件流
+     * @param fileName 文件名
+     * @return FileResponse
+     */
+    FileResponse uploadFile(InputStream is, String fileName);
+
+    /**
+     * 获取文件列表
+     *
+     * @return 文件信息
+     */
+    List<FileInfo> getFileList();
+
+    /**
+     * 删除文件
+     *
+     * @param fileName 文件名
+     * @return 是否删除成功
+     */
+    boolean deleteFile(String fileName);
+}

src/main/java/cn/celess/blog/service/FileService.java

@@ -0,0 +1,18 @@
+package cn.celess.blog.service;
+
+import org.springframework.stereotype.Service;
+
+/**
+ * @author : xiaohai
+ * @date : 2019/04/25 18:15
+ */
+@Service("fileService")
+@FunctionalInterface
+public interface FileService {
+    /**
+     * 获取文件管理器
+     *
+     * @return 文件管理器 可以操作文件
+     */
+    FileManager getFileManager();
+}

src/main/java/cn/celess/blog/service/InstallService.java

@@ -0,0 +1,20 @@
+package cn.celess.blog.service;
+
+import cn.celess.blog.entity.InstallParam;
+import org.springframework.stereotype.Service;
+
+import javax.validation.constraints.NotNull;
+import java.util.Map;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/23 16:22
+ * @desc :
+ */
+@Service
+public interface InstallService {
+    Map<String, Object> isInstall();
+
+    Map<String, Object> install(@NotNull InstallParam installParam);
+
+}

src/main/java/cn/celess/blog/service/PartnerSiteService.java

@@ -2,6 +2,7 @@ package cn.celess.blog.service;
 
 import cn.celess.blog.entity.PartnerSite;
 import cn.celess.blog.entity.model.PageData;
+import cn.celess.blog.entity.request.LinkApplyReq;
 import cn.celess.blog.entity.request.LinkReq;
 import org.springframework.stereotype.Service;
 
@@ -44,7 +45,7 @@ public interface PartnerSiteService {
      * @param page  数据页
      * @return 分页数据
      */
-    PageData<PartnerSite> partnerSitePages(int page, int count);
+    PageData<PartnerSite> partnerSitePages(int page, int count, Boolean deleted);
 
     /**
      * 获取全部数据
@@ -53,4 +54,19 @@ public interface PartnerSiteService {
      */
     List<PartnerSite> findAll();
 
+    /**
+     * 申请友链
+     *
+     * @param linkApplyReq linkApplyReq
+     * @return linkApplyReq
+     */
+    PartnerSite apply(LinkApplyReq linkApplyReq);
+
+    /**
+     * 重写申请友链
+     *
+     * @param key key
+     * @return msg
+     */
+    String reapply(String key);
 }

src/main/java/cn/celess/blog/service/QiniuService.java

@@ -1,31 +0,0 @@
-package cn.celess.blog.service;
-
-import cn.celess.blog.entity.model.QiniuResponse;
-import com.qiniu.storage.model.FileInfo;
-import org.springframework.stereotype.Service;
-
-import java.io.InputStream;
-
-/**
- * @author : xiaohai
- * @date : 2019/04/25 18:15
- */
-@Service
-public interface QiniuService {
-    /**
-     * 上传文件
-     *
-     * @param is       InputStream流
-     * @param fileName 文件名
-     * @return 响应数据
-     */
-    QiniuResponse uploadFile(InputStream is, String fileName);
-
-    /**
-     * 获取文件列表
-     *
-     * @return 文件列表
-     */
-    FileInfo[] getFileList();
-
-}

src/main/java/cn/celess/blog/service/UserService.java

@@ -129,7 +129,7 @@ public interface UserService {
      * @param page  数据页
      * @return 分页数据
      */
-    PageData<UserModel> getUserList(Integer page, Integer count);
+    PageData<UserModel> getUserList(Integer page, Integer count, Integer status);
 
     /**
      * 更改用户信息

src/main/java/cn/celess/blog/service/WebUpdateInfoService.java

@@ -2,10 +2,10 @@ package cn.celess.blog.service;
 
 import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.model.WebUpdateModel;
-import net.sf.json.JSONObject;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author : xiaohai
@@ -59,5 +59,5 @@ public interface WebUpdateInfoService {
      *
      * @return
      */
-    JSONObject getLastestUpdateTime();
+    Map<String, Object> getLastestUpdateTime();
 }

src/main/java/cn/celess/blog/service/fileserviceimpl/FileServiceImpl.java

@@ -0,0 +1,36 @@
+package cn.celess.blog.service.fileserviceimpl;
+
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.service.FileManager;
+import cn.celess.blog.service.FileService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/15 18:52
+ * @desc : 提供文件管理器的服务
+ */
+@Slf4j
+@Service("fileServiceImpl")
+public class FileServiceImpl implements FileService {
+    @Resource(name = "qiniuFileServiceImpl")
+    FileManager qiniuFileManager;
+
+    @Resource(name = "localFileServiceImpl")
+    FileManager localFileServiceImpl;
+
+    @Override
+    public FileManager getFileManager() {
+        String property = System.getProperty(ConfigKeyEnum.FILE_TYPE.getKey());
+        log.info("获取到{}:{}", ConfigKeyEnum.FILE_TYPE.getKey(), property);
+        if ("qiniu".equals(property)){
+            return qiniuFileManager;
+        }else if ("local".equals(property)){
+            return localFileServiceImpl;
+        }
+        return localFileServiceImpl;
+    }
+}

src/main/java/cn/celess/blog/service/fileserviceimpl/LocalFileServiceImpl.java

@@ -0,0 +1,104 @@
+package cn.celess.blog.service.fileserviceimpl;
+
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.entity.model.FileInfo;
+import cn.celess.blog.entity.model.FileResponse;
+import cn.celess.blog.service.FileManager;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.InputStream;
+import java.net.URLEncoder;
+import java.nio.file.Files;
+import java.nio.file.attribute.BasicFileAttributes;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/16 14:39
+ * @desc :
+ */
+@Slf4j
+@Service("localFileServiceImpl")
+public class LocalFileServiceImpl implements FileManager {
+    private static String path = null;
+
+    @SneakyThrows
+    @Override
+    public FileResponse uploadFile(InputStream is, String fileName) {
+        if (path == null) {
+            path = System.getProperty(ConfigKeyEnum.FILE_LOCAL_DIRECTORY_PATH.getKey());
+        }
+        // 判断文件夹是否存在
+        File directory = new File(getPath(path));
+        if (!directory.exists() && directory.mkdirs()) {
+            log.info("不存在文件夹，创建文件夹=>{}", directory.getAbsolutePath());
+        }
+        log.info("上传文件 {}", fileName);
+        // 存储文件
+        File file1 = new File(getPath(path) + File.separator + fileName);
+        FileOutputStream fos = new FileOutputStream(file1);
+        byte[] cache = new byte[1024 * 100];
+        int len = 0;
+        FileResponse fileResponse = new FileResponse();
+        while ((len = is.read(cache)) != -1) {
+            fileResponse.size += len;
+            fos.write(cache, 0, len);
+        }
+        fos.flush();
+        fos.close();
+        is.close();
+        fileResponse.key = URLEncoder.encode(fileName, "UTF-8");
+        fileResponse.type = "local";
+        return fileResponse;
+    }
+
+    @SneakyThrows
+    @Override
+    public List<FileInfo> getFileList() {
+        if (path == null) {
+            path = System.getProperty(ConfigKeyEnum.FILE_LOCAL_DIRECTORY_PATH.getKey());
+        }
+        File file = new File(getPath(path));
+        File[] files = file.listFiles();
+        List<FileInfo> fileInfoList = new ArrayList<>();
+        if (files == null) {
+            return null;
+        }
+        for (File file1 : files) {
+            FileInfo fileInfo = new FileInfo();
+            fileInfo.key = URLEncoder.encode(file1.getName(), "UTF-8");
+            fileInfo.size = file1.length();
+            BasicFileAttributes basicFileAttributes = Files.readAttributes(file1.toPath(), BasicFileAttributes.class);
+            fileInfo.uploadTime = basicFileAttributes.creationTime().toMillis();
+            fileInfoList.add(fileInfo);
+        }
+        return fileInfoList;
+    }
+
+    @Override
+    public boolean deleteFile(String fileName) {
+        if (path == null) {
+            path = System.getProperty(ConfigKeyEnum.FILE_LOCAL_DIRECTORY_PATH.getKey());
+        }
+        File file = new File(getPath(path) + File.separator + fileName);
+        return file.delete();
+    }
+
+    public static String getPath(String path) {
+        if (path == null) {
+            return "";
+        }
+        String pathCop = path.replaceAll("//", File.separator);
+        if (path.startsWith("~")) {
+            // 家目录
+            pathCop = path.replaceFirst("~", "");
+            pathCop = System.getProperty("user.home") + pathCop;
+        }
+        return pathCop;
+    }
+}

src/main/java/cn/celess/blog/service/fileserviceimpl/QiniuFileServiceImpl.java

@@ -0,0 +1,120 @@
+package cn.celess.blog.service.fileserviceimpl;
+
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.entity.model.FileInfo;
+import cn.celess.blog.entity.model.FileResponse;
+import cn.celess.blog.service.FileManager;
+import com.qiniu.common.QiniuException;
+import com.qiniu.common.Zone;
+import com.qiniu.http.Response;
+import com.qiniu.storage.BucketManager;
+import com.qiniu.storage.Configuration;
+import com.qiniu.storage.UploadManager;
+import com.qiniu.util.Auth;
+import com.qiniu.util.StringMap;
+import lombok.SneakyThrows;
+import org.springframework.stereotype.Service;
+
+import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.stream.Collectors;
+
+/**
+ * @author : xiaohai
+ * @date : 2019/04/25 18:15
+ */
+@Service("qiniuFileServiceImpl")
+public class QiniuFileServiceImpl implements FileManager {
+    private static final Configuration cfg = new Configuration(Zone.zone2());
+    private static UploadManager uploadManager;
+    private static BucketManager bucketManager;
+    private static Auth auth;
+
+    private String bucket;
+
+    /**
+     * todo :: 添加reload 方法 配置修改重新创建对象
+     */
+    private void init() {
+        String accessKey = System.getProperty(ConfigKeyEnum.FILE_QINIU_ACCESS_KEY.getKey());
+        String secretKey = System.getProperty(ConfigKeyEnum.FILE_QINIU_SECRET_KEY.getKey());
+        this.bucket = System.getProperty(ConfigKeyEnum.FILE_QINIU_BUCKET.getKey());
+        if (auth == null || uploadManager == null || bucketManager == null) {
+            auth = Auth.create(accessKey, secretKey);
+            uploadManager = new UploadManager(cfg);
+            bucketManager = new BucketManager(auth, cfg);
+        }
+    }
+
+    @Override
+    public FileResponse uploadFile(InputStream is, String fileName) {
+        init();
+        //文件存在则删除文件
+        if (continueFile(fileName)) {
+            try {
+                System.out.println(bucketManager.delete(bucket, fileName).toString());
+            } catch (QiniuException e) {
+                e.printStackTrace();
+            }
+        }
+        //上传
+        try {
+            Response response = uploadManager.put(is, fileName, auth.uploadToken(bucket), null, null);
+            FileResponse fileResponse = new FileResponse();
+            StringMap stringMap = response.jsonToMap();
+
+            fileResponse.key = (String) stringMap.get("key");
+            fileResponse.type = "qiniu";
+            fileResponse.size = 0;
+            fileResponse.hash = (String) stringMap.get("hash");
+
+            return fileResponse;
+        } catch (QiniuException e) {
+            Response r = e.response;
+            System.err.println(r.toString());
+            return null;
+        }
+    }
+
+    @Override
+    public List<FileInfo> getFileList() {
+        init();
+        List<FileInfo> infoList = null;
+        BucketManager.FileListIterator fileListIterator = bucketManager.createFileListIterator(bucket, "", 1000, "");
+        while (fileListIterator.hasNext()) {
+            //处理获取的file list结果
+            infoList = new ArrayList<com.qiniu.storage.model.FileInfo>(Arrays.asList(fileListIterator.next()))
+                    .stream()
+                    .map(item -> {
+                        FileInfo fileInfo = new FileInfo();
+                        fileInfo.key = item.key;
+                        fileInfo.hash = item.hash;
+                        fileInfo.size = item.fsize;
+                        fileInfo.uploadTime = item.putTime;
+                        return fileInfo;
+                    })
+                    .collect(Collectors.toList());
+        }
+        return infoList;
+    }
+
+    @SneakyThrows
+    @Override
+    public boolean deleteFile(String fileName) {
+        init();
+        Response response = bucketManager.delete(bucket, fileName);
+        return "".equals(response.bodyString());
+    }
+
+    private boolean continueFile(String key) {
+        List<FileInfo> fileList = getFileList();
+        for (FileInfo fileInfo : fileList) {
+            if (key.equals(fileInfo.key)) {
+                return true;
+            }
+        }
+        return false;
+    }
+}

src/main/java/cn/celess/blog/service/serviceimpl/ArticleServiceImpl.java

@@ -22,7 +22,6 @@ import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
-import sun.security.krb5.internal.PAData;
 
 import javax.servlet.http.HttpServletRequest;
 import java.util.ArrayList;
@@ -279,11 +278,16 @@ public class ArticleServiceImpl implements ArticleService {
      * @return PageInfo
      */
     @Override
-    public PageData<ArticleModel> adminArticles(int count, int page, boolean deleted) {
+    public PageData<ArticleModel> adminArticles(int count, int page, Boolean deleted) {
         List<Article> articleList = articleMapper.findAll();
 
         PageData<ArticleModel> pageData = new PageData<>(null, 0, count, page);
-        List<Article> collect = articleList.stream().filter(article -> article.isDeleted() == deleted).collect(Collectors.toList());
+        List<Article> collect;
+        if (deleted != null) {
+            collect = articleList.stream().filter(article -> article.isDeleted() == deleted).collect(Collectors.toList());
+        } else {
+            collect = articleList;
+        }
         pageData.setTotal(collect.size());
         List<ArticleModel> articleModels = collect.stream()
                 .peek(article -> article.setMdContent(null))
@@ -300,16 +304,12 @@ public class ArticleServiceImpl implements ArticleService {
     public PageData<ArticleModel> retrievePageForOpen(int count, int page) {
         PageHelper.startPage(page, count);
         List<Article> articleList = articleMapper.findAllByOpen(true);
-        PageData<ArticleModel> pageData = new PageData<>(new PageInfo<Article>(articleList));
-
-        List<ArticleModel> articleModelList = new ArrayList<>();
-
-        articleList.forEach(article -> {
-            ArticleModel model = ModalTrans.article(article, true);
-            setPreAndNextArticle(model);
-            articleModelList.add(model);
-        });
+        PageData<ArticleModel> pageData = new PageData<>(new PageInfo<>(articleList));
 
+        List<ArticleModel> articleModelList = articleList
+                .stream()
+                .map(article -> setPreAndNextArticle(ModalTrans.article(article, true)))
+                .collect(Collectors.toList());
         pageData.setList(articleModelList);
         return pageData;
     }
@@ -323,17 +323,14 @@ public class ArticleServiceImpl implements ArticleService {
         PageHelper.startPage(page, count);
         List<Article> open = articleMapper.findAllByCategoryIdAndOpen(category.getId());
 
-        List<ArticleModel> modelList = new ArrayList<>();
-
-        open.forEach(article -> {
-            ArticleModel model = ModalTrans.article(article, true);
-            model.setTags(null);
-            //            setPreAndNextArticle(model);
-            model.setNextArticle(null);
-            model.setPreArticle(null);
-            modelList.add(model);
-        });
-        return new PageData<ArticleModel>(new PageInfo<Article>(open), modelList);
+        List<ArticleModel> modelList = open.stream()
+                .map(article -> ModalTrans.article(article, true))
+                .peek(articleModel -> {
+                    articleModel.setNextArticle(null);
+                    articleModel.setPreArticle(null);
+                })
+                .collect(Collectors.toList());
+        return new PageData<>(new PageInfo<>(open), modelList);
     }
 
     @Override
@@ -344,21 +341,22 @@ public class ArticleServiceImpl implements ArticleService {
         }
         PageHelper.startPage(page, count);
         List<ArticleTag> articleByTag = articleTagMapper.findArticleByTagAndOpen(tag.getId());
-        List<ArticleModel> modelList = new ArrayList<>();
-        articleByTag.forEach(articleTag -> {
-            ArticleModel model = ModalTrans.article(articleTag.getArticle(), true);
-            model.setNextArticle(null);
-            model.setPreArticle(null);
-            modelList.add(model);
-        });
-        return new PageData<ArticleModel>(new PageInfo<ArticleTag>(articleByTag), modelList);
+        List<ArticleModel> modelList = articleByTag
+                .stream()
+                .map(articleTag -> ModalTrans.article(articleTag.getArticle(), true))
+                .peek(articleModel -> {
+                    articleModel.setNextArticle(null);
+                    articleModel.setPreArticle(null);
+                }).collect(Collectors.toList());
+        return new PageData<>(new PageInfo<>(articleByTag), modelList);
     }
 
-    private void setPreAndNextArticle(ArticleModel articleModel) {
+    private ArticleModel setPreAndNextArticle(ArticleModel articleModel) {
         if (articleModel == null) {
-            return;
+            return null;
         }
         articleModel.setPreArticle(ModalTrans.article(articleMapper.getPreArticle(articleModel.getId()), true));
         articleModel.setNextArticle(ModalTrans.article(articleMapper.getNextArticle(articleModel.getId()), true));
+        return articleModel;
     }
 }

src/main/java/cn/celess/blog/service/serviceimpl/CategoryServiceImpl.java

@@ -17,8 +17,8 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import javax.servlet.http.HttpServletRequest;
-import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -68,22 +68,25 @@ public class CategoryServiceImpl implements CategoryService {
     public PageData<CategoryModel> retrievePage(int page, int count) {
         PageHelper.startPage(page, count);
         List<Category> all = categoryMapper.findAll();
-        List<CategoryModel> modelList = new ArrayList<>();
-        all.forEach(e -> {
-            CategoryModel model = ModalTrans.category(e);
-            List<Article> allByCategoryId = articleMapper.findAllByCategoryId(e.getId());
-            List<ArticleModel> articleModelList = new ArrayList<>();
-            allByCategoryId.forEach(article -> {
-                ArticleModel articleModel = ModalTrans.article(article, true);
-                articleModel.setPreArticle(null);
-                articleModel.setNextArticle(null);
-                articleModel.setTags(null);
-                articleModelList.add(articleModel);
-            });
-            model.setArticles(articleModelList);
-            modelList.add(model);
-        });
-
-        return new PageData<CategoryModel>(new PageInfo<Category>(all), modelList);
+        // 遍历没一个category
+        List<CategoryModel> modelList = all
+                .stream()
+                .map(ModalTrans::category)
+                .peek(categoryModel -> {
+                    // 根据category去查article，并赋值给categoryModel
+                    List<Article> allByCategoryId = articleMapper.findAllByCategoryId(categoryModel.getId());
+                    List<ArticleModel> articleModelList = allByCategoryId
+                            .stream()
+                            .map(article -> ModalTrans.article(article, true))
+                            .peek(articleModel -> {
+                                // 去除不必要的字段
+                                articleModel.setPreArticle(null);
+                                articleModel.setNextArticle(null);
+                                articleModel.setTags(null);
+                            })
+                            .collect(Collectors.toList());
+                    categoryModel.setArticles(articleModelList);
+                }).collect(Collectors.toList());
+        return new PageData<>(new PageInfo<>(all), modelList);
     }
 }

src/main/java/cn/celess/blog/service/serviceimpl/CommentServiceImpl.java

@@ -23,6 +23,7 @@ import org.springframework.stereotype.Service;
 import javax.servlet.http.HttpServletRequest;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -59,7 +60,7 @@ public class CommentServiceImpl implements CommentService {
         Comment comment = new Comment();
         comment.setFromUser(user);
         User userTo = new User();
-        userTo.setId(-1L);
+        userTo.setId(null);
         if (reqBody.getToUserId() != -1) {
             userTo = userMapper.findById(reqBody.getToUserId());
             comment.setToUser(userTo);
@@ -107,14 +108,11 @@ public class CommentServiceImpl implements CommentService {
     @Override
     public List<CommentModel> retrievePageByPid(long pid) {
         List<Comment> allByPagePath = commentMapper.findAllByPid(pid);
-        List<CommentModel> commentModels = new ArrayList<>();
-        allByPagePath.forEach(comment -> {
-            if (comment.getStatus() != CommentStatusEnum.DELETED.getCode()) {
-                commentModels.add(ModalTrans.comment(comment));
-            }
-        });
-
-        return commentModels;
+        return allByPagePath
+                .stream()
+                .filter(comment -> comment.getStatus() != CommentStatusEnum.DELETED.getCode())
+                .map(ModalTrans::comment)
+                .collect(Collectors.toList());
     }
 
     @Override
@@ -145,14 +143,11 @@ public class CommentServiceImpl implements CommentService {
 
     private PageData<CommentModel> pageTrans(List<Comment> commentList, boolean noResponseList) {
         PageInfo<Comment> p = PageInfo.of(commentList);
-        List<CommentModel> modelList = new ArrayList<>();
-        commentList.forEach(l -> {
-            CommentModel model = ModalTrans.comment(l);
-            if (!noResponseList) {
-                model.setRespComment(this.retrievePageByPid(model.getId()));
-            }
-            modelList.add(model);
-        });
-        return new PageData<CommentModel>(p, modelList);
+        List<CommentModel> modelList = commentList
+                .stream()
+                .map(ModalTrans::comment)
+                .peek(commentModel -> commentModel.setRespComment(this.retrievePageByPid(commentModel.getId())))
+                .collect(Collectors.toList());
+        return new PageData<>(p, modelList);
     }
 }

src/main/java/cn/celess/blog/service/serviceimpl/InstallServiceImpl.java

@@ -0,0 +1,118 @@
+package cn.celess.blog.service.serviceimpl;
+
+import cn.celess.blog.BlogApplication;
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.enmu.ResponseEnum;
+import cn.celess.blog.entity.Config;
+import cn.celess.blog.entity.InstallParam;
+import cn.celess.blog.entity.User;
+import cn.celess.blog.exception.MyException;
+import cn.celess.blog.mapper.ConfigMapper;
+import cn.celess.blog.mapper.UserMapper;
+import cn.celess.blog.service.InstallService;
+import cn.celess.blog.service.fileserviceimpl.LocalFileServiceImpl;
+import cn.celess.blog.util.MD5Util;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import javax.validation.constraints.NotNull;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Properties;
+import java.util.stream.Collectors;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/10/23 16:23
+ * @desc :
+ */
+@Slf4j
+@Service
+public class InstallServiceImpl implements InstallService {
+
+    public static final String MYSQL_DRIVER_CLASS_NAME = "com.mysql.cj.jdbc.Driver";
+    public static final String H2_DRIVER_CLASS_NAME = "org.h2.Driver";
+
+    @Autowired
+    ConfigMapper configMapper;
+    @Autowired
+    UserMapper userMapper;
+
+    @Override
+    public Map<String, Object> isInstall() {
+        Config installed = configMapper.getConfiguration(ConfigKeyEnum.BLOG_INSTALLED.getKey());
+        Boolean isInstall = Boolean.valueOf(installed.getValue());
+        Map<String, Object> result = new HashMap<>(3);
+        result.put("is_install", isInstall);
+        if (isInstall) {
+            Map<String, String> configMap = configMapper.getConfigurations()
+                    .stream()
+                    .filter(config -> config.getValue() != null)
+                    .collect(Collectors.toMap(Config::getName, Config::getValue));
+            result.put(ConfigKeyEnum.FILE_TYPE.getKey(), configMap.get(ConfigKeyEnum.FILE_TYPE.getKey()));
+            result.put(ConfigKeyEnum.DB_TYPE.getKey(), configMap.get(ConfigKeyEnum.DB_TYPE.getKey()));
+        }
+        return result;
+    }
+
+    @SneakyThrows
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public Map<String, Object> install(@NotNull InstallParam installParam) {
+        User user = new User(installParam.getEmail(), MD5Util.getMD5(installParam.getPassword()));
+        userMapper.addUser(user);
+        userMapper.setUserRole(user.getId(), "admin");
+
+
+        Config install = configMapper.getConfiguration(ConfigKeyEnum.BLOG_INSTALLED.getKey());
+        boolean isInstall = Boolean.parseBoolean(install.getValue());
+        if (isInstall) {
+            throw new MyException(ResponseEnum.FAILURE, "已经安装过了");
+        }
+
+        Config config = new Config(ConfigKeyEnum.DB_TYPE);
+        config.setValue(installParam.getDbType());
+        Properties properties = new Properties();
+        StringBuilder urlSb = new StringBuilder();
+        if ("h2".equals(installParam.getDbType())) {
+            properties.setProperty(ConfigKeyEnum.DB_DRIVER_CLASS_NAME.getKey(), H2_DRIVER_CLASS_NAME);
+            urlSb.append("jdbc:h2:");
+        } else {
+            properties.setProperty(ConfigKeyEnum.DB_DRIVER_CLASS_NAME.getKey(), MYSQL_DRIVER_CLASS_NAME);
+            urlSb.append("jdbc:mysql:");
+        }
+        // TODO ::
+        urlSb.append(installParam.getDbHost()).append('/').append(installParam.getDbName());
+        properties.setProperty(ConfigKeyEnum.DB_URL.getKey(), urlSb.toString());
+        properties.setProperty(ConfigKeyEnum.DB_USERNAME.getKey(), installParam.getDbUsername());
+        properties.setProperty(ConfigKeyEnum.DB_PASSWORD.getKey(), installParam.getDbPassword());
+
+        Config configuration = configMapper.getConfiguration(ConfigKeyEnum.BLOG_DB_PATH.getKey());
+
+        File file = new File(LocalFileServiceImpl.getPath(configuration.getValue()));
+        if (!file.exists() && file.createNewFile()) {
+            log.info("创建数据库配置文件: {}", file.getAbsolutePath());
+        }
+        FileInputStream fis = new FileInputStream(file);
+        FileOutputStream fos = new FileOutputStream(file);
+        properties.load(fis);
+
+        configMapper.addConfiguration(config);
+        properties.store(fos, "DB CONFIG");
+
+        install.setValue(Boolean.valueOf(true).toString());
+        configMapper.updateConfiguration(install);
+
+        log.info("重启...");
+
+        // 重启
+        BlogApplication.restart();
+        return null;
+    }
+}

src/main/java/cn/celess/blog/service/serviceimpl/PartnerSiteServiceImpl.java

@@ -3,18 +3,29 @@ package cn.celess.blog.service.serviceimpl;
 import cn.celess.blog.enmu.ResponseEnum;
 import cn.celess.blog.entity.PartnerSite;
 import cn.celess.blog.entity.model.PageData;
+import cn.celess.blog.entity.request.LinkApplyReq;
 import cn.celess.blog.entity.request.LinkReq;
 import cn.celess.blog.exception.MyException;
 import cn.celess.blog.mapper.PartnerMapper;
+import cn.celess.blog.service.MailService;
 import cn.celess.blog.service.PartnerSiteService;
+import cn.celess.blog.util.HttpUtil;
+import cn.celess.blog.util.RedisUtil;
 import cn.celess.blog.util.RegexUtil;
+import com.alibaba.druid.util.StringUtils;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
+import lombok.SneakyThrows;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.mail.SimpleMailMessage;
 import org.springframework.stereotype.Service;
 
+import java.util.Date;
 import java.util.List;
+import java.util.UUID;
+import java.util.concurrent.TimeUnit;
 
 /**
  * @author : xiaohai
@@ -24,6 +35,13 @@ import java.util.List;
 public class PartnerSiteServiceImpl implements PartnerSiteService {
     @Autowired
     PartnerMapper partnerMapper;
+    @Autowired
+    MailService mailService;
+    @Autowired
+    RedisUtil redisUtil;
+    private static final String SITE_NAME = "小海博客";
+    private static final String SITE_URL = "celess.cn";
+    private static final String SITE_EMAIL = "a@celess.cn";
 
     @Override
     public PartnerSite create(LinkReq reqBody) {
@@ -86,17 +104,28 @@ public class PartnerSiteServiceImpl implements PartnerSiteService {
         if (!reqBody.getUrl().contains("http://") && !reqBody.getUrl().contains("https://")) {
             reqBody.setUrl("http://" + reqBody.getUrl());
         }
+        if (reqBody.isOpen() != partnerSite.getOpen() && !partnerSite.getNotification() && !StringUtils.isEmpty(partnerSite.getEmail())) {
+            SimpleMailMessage smm = new SimpleMailMessage();
+            smm.setTo(partnerSite.getEmail());
+            smm.setText("您的友链申请，已通过");
+            smm.setSubject("友链申请通过");
+            smm.setSentDate(new Date());
+            mailService.send(smm);
+            partnerSite.setNotification(true);
+        }
+        BeanUtils.copyProperties(reqBody, partnerSite);
+        partnerMapper.update(partnerSite);
         partnerSite.setName(reqBody.getName());
         partnerSite.setUrl(reqBody.getUrl());
         partnerSite.setOpen(reqBody.isOpen());
-        partnerMapper.update(partnerSite);
+
         return partnerSite;
     }
 
     @Override
-    public PageData<PartnerSite> partnerSitePages(int page, int count) {
+    public PageData<PartnerSite> partnerSitePages(int page, int count, Boolean deleted) {
         PageHelper.startPage(page, count);
-        List<PartnerSite> sitePage = partnerMapper.findAll();
+        List<PartnerSite> sitePage = partnerMapper.findAll(deleted);
         PageInfo<PartnerSite> pageInfo = new PageInfo<PartnerSite>(sitePage);
         return new PageData<>(pageInfo, sitePage);
     }
@@ -108,5 +137,92 @@ public class PartnerSiteServiceImpl implements PartnerSiteService {
         return all;
     }
 
+    @SneakyThrows
+    @Override
+    public PartnerSite apply(LinkApplyReq linkApplyReq) {
+        // 空值字段
+        if (StringUtils.isEmpty(linkApplyReq.getName())
+                || StringUtils.isEmpty(linkApplyReq.getUrl())
+                || StringUtils.isEmpty(linkApplyReq.getEmail())
+                || StringUtils.isEmpty(linkApplyReq.getLinkUrl())) {
+            throw new MyException(ResponseEnum.PARAMETERS_ERROR);
+        }
+        // 链接不合法
+        if (!RegexUtil.emailMatch(linkApplyReq.getEmail())) {
+            throw new MyException(ResponseEnum.PARAMETERS_EMAIL_ERROR);
+        }
+        if (!RegexUtil.urlMatch(linkApplyReq.getLinkUrl()) || !RegexUtil.urlMatch(linkApplyReq.getUrl())) {
+            throw new MyException(ResponseEnum.PARAMETERS_URL_ERROR);
+        }
+        if (!StringUtils.isEmpty(linkApplyReq.getIconPath()) && !RegexUtil.urlMatch(linkApplyReq.getIconPath())) {
+            throw new MyException(ResponseEnum.PARAMETERS_URL_ERROR);
+        }
+        // 非强制字段 设置空
+        if (StringUtils.isEmpty(linkApplyReq.getIconPath())) {
+            linkApplyReq.setIconPath("");
+        }
+        // 抓取页面
+        String resp = HttpUtil.getAfterRendering(linkApplyReq.getLinkUrl());
+        if (resp == null) {
+            throw new MyException(ResponseEnum.CANNOT_GET_DATA);
+        }
+        PartnerSite ps = new PartnerSite();
+        if (resp.contains(SITE_URL)) {
+            //包含站点
+            BeanUtils.copyProperties(linkApplyReq, ps);
+            ps.setNotification(false);
+            ps.setOpen(false);
+            boolean exists = partnerMapper.existsByUrl(linkApplyReq.getUrl());
+            if (!exists) {
+                partnerMapper.insert(ps);
+            } else {
+                ps.setId(partnerMapper.findByUrl(linkApplyReq.getUrl()).getId());
+            }
+            SimpleMailMessage smm = new SimpleMailMessage();
+            smm.setSubject("友链申请");
+            smm.setText("有一条友链申请" + (exists ? "，已存在的友链链接" : "") + "，[\n" + linkApplyReq.toString() + "\n]");
+            smm.setTo(SITE_EMAIL);
+            smm.setSentDate(new Date());
+            mailService.send(smm);
+        } else {
+            //  不包含站点
+            String uuid;
+            ObjectMapper mapper = new ObjectMapper();
+            if (redisUtil.hasKey(linkApplyReq.getUrl())) {
+                uuid = redisUtil.get(linkApplyReq.getUrl());
+                if (!redisUtil.hasKey(uuid)) {
+                    redisUtil.setEx(uuid, mapper.writeValueAsString(linkApplyReq), 10, TimeUnit.MINUTES);
+                }
+            } else {
+                uuid = UUID.randomUUID().toString().replaceAll("-", "");
+                redisUtil.setEx(uuid, mapper.writeValueAsString(linkApplyReq), 10, TimeUnit.MINUTES);
+                redisUtil.setEx(linkApplyReq.getUrl(), uuid, 10, TimeUnit.MINUTES);
+            }
+            throw new MyException(ResponseEnum.APPLY_LINK_NO_ADD_THIS_SITE, null, uuid);
+        }
+        return ps;
+    }
 
+    @SneakyThrows
+    @Override
+    public String reapply(String key) {
+        if (!redisUtil.hasKey(key)) {
+            throw new MyException(ResponseEnum.DATA_EXPIRED);
+        }
+        String s = redisUtil.get(key);
+        ObjectMapper mapper = new ObjectMapper();
+        LinkApplyReq linkApplyReq = mapper.readValue(s, LinkApplyReq.class);
+        if (linkApplyReq == null) {
+            throw new MyException(ResponseEnum.DATA_NOT_EXIST);
+        }
+        SimpleMailMessage smm = new SimpleMailMessage();
+        smm.setSubject("友链申请");
+        smm.setText("有一条未抓取到信息的友链申请，[\n" + linkApplyReq.toString() + "\n]");
+        smm.setTo(SITE_EMAIL);
+        smm.setSentDate(new Date());
+        mailService.send(smm);
+        redisUtil.delete(key);
+        redisUtil.delete(linkApplyReq.getUrl());
+        return "success";
+    }
 }

src/main/java/cn/celess/blog/service/serviceimpl/QiniuServiceImpl.java

@@ -1,87 +0,0 @@
-package cn.celess.blog.service.serviceimpl;
-
-import cn.celess.blog.entity.model.QiniuResponse;
-import cn.celess.blog.service.QiniuService;
-import com.qiniu.common.QiniuException;
-import com.qiniu.common.Zone;
-import com.qiniu.http.Response;
-import com.qiniu.storage.BucketManager;
-import com.qiniu.storage.Configuration;
-import com.qiniu.storage.UploadManager;
-import com.qiniu.storage.model.FileInfo;
-import com.qiniu.util.Auth;
-import org.springframework.beans.factory.annotation.Value;
-import org.springframework.stereotype.Service;
-
-import java.io.InputStream;
-
-/**
- * @author : xiaohai
- * @date : 2019/04/25 18:15
- */
-@Service
-public class QiniuServiceImpl implements QiniuService {
-    private static final Configuration cfg = new Configuration(Zone.zone2());
-    private static UploadManager uploadManager;
-    private static BucketManager bucketManager;
-    private static Auth auth;
-
-    @Value("${qiniu.accessKey}")
-    private String accessKey;
-    @Value("${qiniu.secretKey}")
-    private String secretKey;
-    @Value("${qiniu.bucket}")
-    private String bucket;
-
-    private void init() {
-        if (auth == null || uploadManager == null || bucketManager == null) {
-            auth = Auth.create(accessKey, secretKey);
-            uploadManager = new UploadManager(cfg);
-            bucketManager = new BucketManager(auth, cfg);
-        }
-    }
-
-    @Override
-    public QiniuResponse uploadFile(InputStream is, String fileName) {
-        init();
-        //文件存在则删除文件
-        if (continueFile(fileName)) {
-            try {
-                System.out.println(bucketManager.delete(bucket, fileName).toString());
-            } catch (QiniuException e) {
-                e.printStackTrace();
-            }
-        }
-        //上传
-        try {
-            Response response = uploadManager.put(is, fileName, auth.uploadToken(bucket), null, null);
-            return response.jsonToObject(QiniuResponse.class);
-        } catch (QiniuException e) {
-            Response r = e.response;
-            System.err.println(r.toString());
-            return null;
-        }
-    }
-
-    @Override
-    public FileInfo[] getFileList() {
-        init();
-        BucketManager.FileListIterator fileListIterator = bucketManager.createFileListIterator(bucket, "", 1000, "");
-        FileInfo[] items = null;
-        while (fileListIterator.hasNext()) {
-            //处理获取的file list结果
-            items = fileListIterator.next();
-        }
-        return items;
-    }
-
-    private boolean continueFile(String key) {
-        FileInfo[] allFile = getFileList();
-        for (FileInfo fileInfo : allFile) {
-            if (key.equals(fileInfo.key)) {
-                return true;
-            }
-        }
-        return false;
-    }
-}

src/main/java/cn/celess/blog/service/serviceimpl/TagServiceImpl.java

@@ -21,6 +21,7 @@ import org.springframework.transaction.annotation.Transactional;
 import javax.servlet.http.HttpServletRequest;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -81,20 +82,22 @@ public class TagServiceImpl implements TagService {
         List<Tag> tagList = tagMapper.findAll();
         List<TagModel> modelList = new ArrayList<>();
         tagList.forEach(tag -> modelList.add(ModalTrans.tag(tag)));
-        return new PageData<TagModel>(new PageInfo<Tag>(tagList), modelList);
+        return new PageData<>(new PageInfo<>(tagList), modelList);
     }
 
     @Override
     public List<TagModel> findAll() {
-        List<TagModel> list = new ArrayList<>();
-        tagMapper.findAll().forEach(e -> {
-            TagModel model = ModalTrans.tag(e);
-            List<ArticleTag> articleByTagAndOpen = articleTagMapper.findArticleByTagAndOpen(e.getId());
-            List<ArticleModel> articleModelList = new ArrayList<>();
-            articleByTagAndOpen.forEach(articleTag -> articleModelList.add(ModalTrans.article(articleTag.getArticle(), true)));
-            model.setArticles(articleModelList);
-            list.add(model);
-        });
-        return list;
+        return tagMapper.findAll().stream()
+                .map(ModalTrans::tag)
+                .peek(tagModel -> {
+                    List<ArticleTag> articleByTagAndOpen = articleTagMapper.findArticleByTagAndOpen(tagModel.getId());
+                    tagModel.setArticles(
+                            articleByTagAndOpen
+                                    .stream()
+                                    .map(articleTag -> ModalTrans.article(articleTag.getArticle(), true))
+                                    .collect(Collectors.toList())
+                    );
+                })
+                .collect(Collectors.toList());
     }
 }

src/main/java/cn/celess/blog/service/serviceimpl/UserServiceImpl.java

@@ -2,36 +2,35 @@ package cn.celess.blog.service.serviceimpl;
 
 import cn.celess.blog.enmu.ResponseEnum;
 import cn.celess.blog.enmu.RoleEnum;
+import cn.celess.blog.enmu.UserAccountStatusEnum;
 import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.User;
+import cn.celess.blog.entity.model.FileResponse;
 import cn.celess.blog.entity.model.PageData;
-import cn.celess.blog.entity.model.QiniuResponse;
 import cn.celess.blog.entity.model.UserModel;
 import cn.celess.blog.entity.request.LoginReq;
 import cn.celess.blog.entity.request.UserReq;
 import cn.celess.blog.exception.MyException;
 import cn.celess.blog.mapper.UserMapper;
+import cn.celess.blog.service.FileService;
 import cn.celess.blog.service.MailService;
-import cn.celess.blog.service.QiniuService;
 import cn.celess.blog.service.UserService;
 import cn.celess.blog.util.*;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.mail.SimpleMailMessage;
 import org.springframework.stereotype.Service;
 
+import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.beans.Transient;
 import java.io.InputStream;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.UUID;
+import java.util.*;
 import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -47,8 +46,8 @@ public class UserServiceImpl implements UserService {
     HttpServletRequest request;
     @Autowired
     MailService mailService;
-    @Autowired
-    QiniuService qiniuService;
+    @Resource(name = "fileServiceImpl")
+    FileService fileService;
     @Autowired
     RedisUtil redisUtil;
     @Autowired
@@ -102,6 +101,16 @@ public class UserServiceImpl implements UserService {
         if (!RegexUtil.pwdMatch(loginReq.getPassword())) {
             throw new MyException(ResponseEnum.PARAMETERS_PWD_ERROR);
         }
+
+        User user = userMapper.findByEmail(loginReq.getEmail());
+        if (user == null) {
+            // 用户不存在
+            throw new MyException(ResponseEnum.USER_NOT_EXIST);
+        }
+        if (user.getStatus() != UserAccountStatusEnum.NORMAL.getCode()) {
+            throw new MyException(ResponseEnum.CAN_NOT_USE, UserAccountStatusEnum.get(user.getStatus()));
+        }
+
         //获取redis缓存中登录失败次数
         String s = redisUtil.get(loginReq.getEmail() + "-passwordWrongTime");
         if (s != null) {
@@ -109,14 +118,9 @@ public class UserServiceImpl implements UserService {
                 throw new MyException(ResponseEnum.LOGIN_LATER, loginReq.getEmail());
             }
         }
-        User user = null;
-        user = userMapper.findByEmail(loginReq.getEmail());
-        String token = null;
+
+        String token;
         // 密码比对
-        if (user == null) {
-            // 用户不存在
-            throw new MyException(ResponseEnum.USER_NOT_EXIST);
-        }
         if (user.getPwd().equals(MD5Util.getMD5(loginReq.getPassword()))) {
             logger.info("====> {}  进行权限认证  状态：登录成功 <====", loginReq.getEmail());
             userMapper.updateLoginTime(loginReq.getEmail());
@@ -182,7 +186,7 @@ public class UserServiceImpl implements UserService {
     @Override
     public Object updateUserAavatarImg(InputStream is, String mime) {
         User user = redisUserUtil.get();
-        QiniuResponse upload = qiniuService.uploadFile(is, user.getEmail() + "_" + user.getId() + mime.toLowerCase());
+        FileResponse upload = fileService.getFileManager().uploadFile(is, user.getEmail() + "_" + user.getId() + mime.toLowerCase());
         user.setAvatarImgUrl(upload.key);
         userMapper.updateAvatarImgUrl(upload.key, user.getId());
         redisUserUtil.set(user);
@@ -312,14 +316,14 @@ public class UserServiceImpl implements UserService {
 
     @Override
     public Object deleteUser(Integer[] id) {
-        JSONArray status = new JSONArray();
+        List<Map<String, Object>> status = new ArrayList<>();
         if (id == null || id.length == 0) {
             return null;
         }
         for (Integer integer : id) {
             String role = userMapper.getRoleById(integer);
             int deleteResult = 0;
-            JSONObject deleteStatus = new JSONObject();
+            Map<String, Object> deleteStatus = new HashMap<>(3);
             deleteStatus.put("id", integer);
             // 管理员账户不可删
             if ("admin".equals(role)) {
@@ -342,12 +346,11 @@ public class UserServiceImpl implements UserService {
     }
 
     @Override
-    public PageData<UserModel> getUserList(Integer page, Integer count) {
+    public PageData<UserModel> getUserList(Integer page, Integer count, Integer status) {
         PageHelper.startPage(page, count);
-        List<User> all = userMapper.findAll();
-        List<UserModel> modelList = new ArrayList<>();
-        all.forEach(user -> modelList.add(ModalTrans.userFullInfo(user)));
-        return new PageData<UserModel>(PageInfo.of(all), modelList);
+        List<User> all = userMapper.findAll(status);
+        List<UserModel> modelList = all.stream().map(ModalTrans::userFullInfo).collect(Collectors.toList());
+        return new PageData<>(PageInfo.of(all), modelList);
     }
 
     @Override

src/main/java/cn/celess/blog/service/serviceimpl/VisitorServiceImpl.java

@@ -7,6 +7,7 @@ import cn.celess.blog.entity.model.VisitorModel;
 import cn.celess.blog.exception.MyException;
 import cn.celess.blog.mapper.VisitorMapper;
 import cn.celess.blog.service.VisitorService;
+import cn.celess.blog.util.AddressUtil;
 import cn.celess.blog.util.DateFormatUtil;
 import cn.celess.blog.util.RedisUtil;
 import com.github.pagehelper.PageHelper;
@@ -15,7 +16,6 @@ import eu.bitwalker.useragentutils.Browser;
 import eu.bitwalker.useragentutils.OperatingSystem;
 import eu.bitwalker.useragentutils.UserAgent;
 import eu.bitwalker.useragentutils.Version;
-import org.apache.commons.lang.time.DateUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.json.JsonParserFactory;
 import org.springframework.stereotype.Service;
@@ -26,6 +26,8 @@ import java.io.InputStream;
 import java.io.InputStreamReader;
 import java.net.HttpURLConnection;
 import java.net.URL;
+import java.time.LocalDateTime;
+import java.time.temporal.ChronoUnit;
 import java.util.*;
 import java.util.concurrent.TimeUnit;
 
@@ -49,7 +51,7 @@ public class VisitorServiceImpl implements VisitorService {
     public PageData<VisitorModel> visitorPage(int page, int count, boolean showLocation) {
         PageHelper.startPage(page, count);
         List<Visitor> visitorList = visitorMapper.findAll();
-        return new PageData<VisitorModel>(new PageInfo<Visitor>(visitorList), list2List(visitorList, showLocation));
+        return new PageData<>(new PageInfo<>(visitorList), list2List(visitorList, showLocation));
     }
 
     @Override
@@ -67,8 +69,9 @@ public class VisitorServiceImpl implements VisitorService {
         visitor.setUa(request.getHeader("User-Agent"));
         //记录当日的访问
         String dayVisitCount = redisUtil.get("dayVisitCount");
-        long secondsLeftToday = 86400 - DateUtils.getFragmentInSeconds(Calendar.getInstance(), Calendar.DATE);
-        Date date = new Date(Calendar.YEAR);
+
+        LocalDateTime midnight = LocalDateTime.now().plusDays(1).withHour(0).withMinute(0).withSecond(0).withNano(0);
+        long secondsLeftToday = ChronoUnit.SECONDS.between(LocalDateTime.now(), midnight);
         if (dayVisitCount == null) {
             redisUtil.setEx("dayVisitCount", "1", secondsLeftToday, TimeUnit.SECONDS);
         } else {
@@ -149,60 +152,7 @@ public class VisitorServiceImpl implements VisitorService {
      * @return
      */
     private String getLocation(String ip) {
-        StringBuilder result = new StringBuilder();
-        URL url;
-        HttpURLConnection conn = null;
-        InputStream inputStream = null;
-        InputStreamReader inputStreamReader = null;
-        BufferedReader bufferedReader = null;
-        try {
-            url = new URL("http://ip.taobao.com/service/getIpInfo.php?ip=" + ip);
-            conn = (HttpURLConnection) url.openConnection();
-            conn.setConnectTimeout(3000);
-            conn.setDoInput(true);
-            conn.setRequestMethod("GET");
-            inputStream = conn.getInputStream();
-            inputStreamReader = new InputStreamReader(inputStream);
-            bufferedReader = new BufferedReader(inputStreamReader);
-            String tmp;
-            while ((tmp = bufferedReader.readLine()) != null) {
-                result.append(tmp);
-            }
-        } catch (Exception e) {
-            // ignore
-        } finally {
-            try {
-                if (conn != null) {
-                    conn.disconnect();
-                }
-                if (inputStream != null) {
-                    inputStream.close();
-                }
-                if (inputStreamReader != null) {
-                    inputStreamReader.close();
-                }
-                if (bufferedReader != null) {
-                    bufferedReader.close();
-                }
-            } catch (Exception e) {
-                // ignore
-            }
-        }
-        StringBuilder sb = new StringBuilder();
-        if ("".equals(result.toString())) {
-            throw new MyException(ResponseEnum.FAILURE);
-        }
-        Map<String, Object> stringObjectMap = JsonParserFactory.getJsonParser().parseMap(result.toString());
-        if ((Integer) stringObjectMap.get("code") == 0) {
-            LinkedHashMap data = (LinkedHashMap) stringObjectMap.get("data");
-            sb.append(data.get("country"))
-                    .append("-")
-                    .append(data.get("region"))
-                    .append("-")
-                    .append(data.get("city"));
-        }
-        return sb.toString();
-
+        return AddressUtil.getCityInfo(ip);
     }
 
 

src/main/java/cn/celess/blog/service/serviceimpl/WebUpdateInfoServiceImpl.java

@@ -10,20 +10,19 @@ import cn.celess.blog.service.WebUpdateInfoService;
 import cn.celess.blog.util.DateFormatUtil;
 import cn.celess.blog.util.HttpUtil;
 import cn.celess.blog.util.ModalTrans;
+import com.alibaba.druid.util.StringUtils;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
 import lombok.extern.slf4j.Slf4j;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.io.IOException;
-import java.text.SimpleDateFormat;
 import java.time.Instant;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * @author : xiaohai
@@ -74,7 +73,7 @@ public class WebUpdateInfoServiceImpl implements WebUpdateInfoService {
     public PageData<WebUpdateModel> pages(int count, int page) {
         PageHelper.startPage(page, count);
         List<WebUpdate> updateList = webUpdateInfoMapper.findAll();
-        return new PageData<WebUpdateModel>(new PageInfo<WebUpdate>(updateList), list2List(updateList));
+        return new PageData<>(new PageInfo<>(updateList), list2List(updateList));
     }
 
     @Override
@@ -84,29 +83,32 @@ public class WebUpdateInfoServiceImpl implements WebUpdateInfoService {
     }
 
     @Override
-    public JSONObject getLastestUpdateTime() {
-        JSONObject jsonObject = new JSONObject();
-        jsonObject.put("lastUpdateTime", DateFormatUtil.get(webUpdateInfoMapper.getLastestOne().getUpdateTime()));
-        jsonObject.put("lastUpdateInfo", webUpdateInfoMapper.getLastestOne().getUpdateInfo());
+
+    public Map<String, Object> getLastestUpdateTime() {
+        Map<String, Object> map = new HashMap<>();
+        map.put("lastUpdateTime", DateFormatUtil.get(webUpdateInfoMapper.getLastestOne().getUpdateTime()));
+        map.put("lastUpdateInfo", webUpdateInfoMapper.getLastestOne().getUpdateInfo());
         try {
-            JSONArray array = JSONArray.fromObject(HttpUtil.get("https://api.github.com/repos/xiaohai2271/blog-frontEnd/commits?page=1&per_page=1"));
-            JSONObject object = array.getJSONObject(0);
-            JSONObject commit = object.getJSONObject("commit");
-            jsonObject.put("lastCommit", commit.getString("message"));
-            jsonObject.put("committerAuthor", commit.getJSONObject("committer").getString("name"));
-            SimpleDateFormat sdf = new SimpleDateFormat();
-            Instant parse = Instant.parse(commit.getJSONObject("committer").getString("date"));
-            jsonObject.put("committerDate", DateFormatUtil.get(Date.from(parse)));
-            jsonObject.put("commitUrl", "https://github.com/xiaohai2271/blog-frontEnd/tree/" + object.getString("sha"));
+            ObjectMapper mapper = new ObjectMapper();
+            String respStr = HttpUtil.get("https://api.github.com/repos/xiaohai2271/blog-frontEnd/commits?page=1&per_page=1");
+            if (!StringUtils.isEmpty(respStr)) {
+                JsonNode root = mapper.readTree(respStr);
+                Iterator<JsonNode> elements = root.elements();
+                JsonNode next = elements.next();
+                JsonNode commit = next.get("commit");
+                map.put("lastCommit", commit.get("message"));
+                map.put("committerAuthor", commit.get("committer").get("name"));
+                Instant parse = Instant.parse(commit.get("committer").get("date").asText());
+                map.put("committerDate", DateFormatUtil.get(Date.from(parse)));
+                map.put("commitUrl", "https://github.com/xiaohai2271/blog-frontEnd/tree/" + next.get("sha").asText());
+            }
         } catch (IOException e) {
             log.info("网络请求失败{}", e.getMessage());
         }
-        return jsonObject;
+        return map;
     }
 
     private List<WebUpdateModel> list2List(List<WebUpdate> webUpdates) {
-        List<WebUpdateModel> webUpdateModels = new ArrayList<>();
-        webUpdates.forEach(update -> webUpdateModels.add(ModalTrans.webUpdate(update)));
-        return webUpdateModels;
+        return webUpdates.stream().map(ModalTrans::webUpdate).collect(Collectors.toList());
     }
 }

src/main/java/cn/celess/blog/util/AddressUtil.java

@@ -0,0 +1,71 @@
+package cn.celess.blog.util;
+
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.io.FileUtils;
+import org.lionsoul.ip2region.DataBlock;
+import org.lionsoul.ip2region.DbConfig;
+import org.lionsoul.ip2region.DbSearcher;
+import org.lionsoul.ip2region.Util;
+
+import java.io.File;
+import java.lang.reflect.Method;
+import java.util.Objects;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/09/04 9:36
+ */
+
+@Slf4j
+public class AddressUtil {
+
+    public static String getCityInfo(String ip) {
+        File file;
+        try {
+            //db
+            String dbPath = AddressUtil.class.getResource("/ip2region/ip2region.db").getPath();
+            file = new File(dbPath);
+            if (!file.exists()) {
+                String tmpDir = System.getProperties().getProperty("java.io.tmpdir");
+                dbPath = tmpDir + "ip.db";
+                file = new File(dbPath);
+                FileUtils.copyInputStreamToFile(Objects.requireNonNull(AddressUtil.class.getClassLoader().getResourceAsStream("classpath:ip2region/ip2region.db")), file);
+            }
+            //查询算法
+            //B-tree
+            int algorithm = DbSearcher.BTREE_ALGORITHM;
+            try {
+                DbConfig config = new DbConfig();
+                DbSearcher searcher = new DbSearcher(config, dbPath);
+                Method method = null;
+                switch (algorithm) {
+                    case DbSearcher.BTREE_ALGORITHM:
+                        method = searcher.getClass().getMethod("btreeSearch", String.class);
+                        break;
+                    case DbSearcher.BINARY_ALGORITHM:
+                        method = searcher.getClass().getMethod("binarySearch", String.class);
+                        break;
+                    case DbSearcher.MEMORY_ALGORITYM:
+                        method = searcher.getClass().getMethod("memorySearch", String.class);
+                        break;
+                }
+
+                DataBlock dataBlock;
+                if (!Util.isIpAddress(ip)) {
+                    System.out.println("Error: Invalid ip address");
+                }
+
+                dataBlock = (DataBlock) method.invoke(searcher, ip);
+
+                return dataBlock.getRegion();
+
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+
+        return null;
+    }
+}

src/main/java/cn/celess/blog/util/HttpUtil.java

@@ -1,12 +1,15 @@
 package cn.celess.blog.util;
 
-import java.io.BufferedReader;
+import com.gargoylesoftware.htmlunit.BrowserVersion;
+import com.gargoylesoftware.htmlunit.NicelyResynchronizingAjaxController;
+import com.gargoylesoftware.htmlunit.WebClient;
+import com.gargoylesoftware.htmlunit.html.HtmlPage;
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.Response;
+
 import java.io.IOException;
-import java.io.InputStream;
-import java.io.InputStreamReader;
-import java.net.HttpURLConnection;
-import java.net.URL;
-import java.nio.charset.StandardCharsets;
+import java.util.Objects;
 
 /**
  * @Author: 小海
@@ -14,38 +17,34 @@ import java.nio.charset.StandardCharsets;
  * @Desc:
  */
 public class HttpUtil {
+    private static final OkHttpClient CLIENT = new OkHttpClient();
 
-    public static String get(String urlStr) throws IOException {
-
-        StringBuffer sb = new StringBuffer();
-
-        HttpURLConnection urlConnection = null;
-        try {
-            URL url = new URL(urlStr);
-
-            //打开http
-            urlConnection = (HttpURLConnection) url.openConnection();
-            urlConnection.setDoInput(true);
-            urlConnection.setRequestMethod("GET");
-            urlConnection.connect();
-
-            try (
-                    InputStream inputStream = urlConnection.getInputStream();
-                    InputStreamReader inputStreamReader = new InputStreamReader(inputStream, StandardCharsets.UTF_8);
-                    BufferedReader bufferedReader = new BufferedReader(inputStreamReader)
-            ) {
-                //将bufferReader的值给放到buffer里
-                String str = null;
-                while ((str = bufferedReader.readLine()) != null) {
-                    sb.append(str);
-                }
-            }
-        } finally {
-            //断开连接
-            if (urlConnection != null) {
-                urlConnection.disconnect();
-            }
+    public static String get(String urlStr) {
+        Request request = new Request.Builder()
+                .url(urlStr)
+                .get()
+                .build();
+        try (Response response = CLIENT.newCall(request).execute()) {
+            return Objects.requireNonNull(response.body()).string();
+        } catch (IOException e) {
+            return null;
+        }
+    }
+
+
+    public static String getAfterRendering(String url) {
+        try (final WebClient webClient = new WebClient(BrowserVersion.CHROME)) {
+            webClient.getOptions().setCssEnabled(false);
+            webClient.getOptions().setJavaScriptEnabled(true);
+            webClient.getOptions().setThrowExceptionOnFailingStatusCode(false);
+            webClient.getOptions().setThrowExceptionOnScriptError(false);
+            webClient.getOptions().setDownloadImages(false);
+            webClient.getOptions().setActiveXNative(false);
+            webClient.setAjaxController(new NicelyResynchronizingAjaxController());
+            final HtmlPage page = webClient.getPage(url);
+            return page.asXml();
+        } catch (IOException e) {
+            return null;
         }
-        return sb.toString();
     }
 }

src/main/java/cn/celess/blog/util/ModalTrans.java

@@ -1,5 +1,6 @@
 package cn.celess.blog.util;
 
+import cn.celess.blog.enmu.UserAccountStatusEnum;
 import cn.celess.blog.entity.*;
 import cn.celess.blog.entity.model.*;
 import org.springframework.beans.BeanUtils;
@@ -46,6 +47,7 @@ public class ModalTrans {
         }
         UserModel userModel = new UserModel();
         BeanUtils.copyProperties(user, userModel);
+        userModel.setStatus(UserAccountStatusEnum.get(user.getStatus()));
         userModel.setAvatarImgUrl(user.getAvatarImgUrl() == null || user.getAvatarImgUrl().length() == 0 ?
                 null :
                 "http://cdn.celess.cn/" + user.getAvatarImgUrl());

src/main/java/cn/celess/blog/util/RedisUserUtil.java

@@ -3,7 +3,8 @@ package cn.celess.blog.util;
 import cn.celess.blog.enmu.ResponseEnum;
 import cn.celess.blog.entity.User;
 import cn.celess.blog.exception.MyException;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.SneakyThrows;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
@@ -31,24 +32,27 @@ public class RedisUserUtil {
         return user;
     }
 
+    @SneakyThrows
     public User getWithOutExc() {
         String token = request.getHeader("Authorization");
         if (token == null || token.isEmpty()) {
             return null;
         }
         String email = jwtUtil.getUsernameFromToken(token);
-        return (User) JSONObject.toBean(JSONObject.fromObject(redisUtil.get(email + "-login")), User.class);
+        return new ObjectMapper().readValue(redisUtil.get(email + "-login"), User.class);
     }
 
+    @SneakyThrows
     public User set(User user) {
         Long expire = redisUtil.getExpire(user.getEmail() + "-login");
-        redisUtil.setEx(user.getEmail() + "-login", JSONObject.fromObject(user).toString(),
+        redisUtil.setEx(user.getEmail() + "-login", new ObjectMapper().writeValueAsString(user),
                 expire > 0 ? expire : JwtUtil.EXPIRATION_SHORT_TIME, TimeUnit.MILLISECONDS);
         return user;
     }
 
+    @SneakyThrows
     public User set(User user, boolean isRemember) {
-        redisUtil.setEx(user.getEmail() + "-login", JSONObject.fromObject(user).toString(),
+        redisUtil.setEx(user.getEmail() + "-login", new ObjectMapper().writeValueAsString(user),
                 isRemember ? JwtUtil.EXPIRATION_LONG_TIME : JwtUtil.EXPIRATION_SHORT_TIME, TimeUnit.MILLISECONDS);
         request.getSession().setAttribute("email", user.getEmail());
         return user;

src/main/resources/application-easyDeploy.properties

@@ -0,0 +1,65 @@
+spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
+spring.datasource.url=jdbc:h2:~/blog/db
+spring.datasource.driver-class-name=org.h2.Driver
+spring.datasource.username=sa
+spring.datasource.password=sa
+
+spring.datasource.platform=h2
+spring.datasource.sql-script-encoding=utf-8
+spring.datasource.initialization-mode=never
+spring.datasource.schema=classpath:sql/schema-h2.sql
+spring.datasource.data=classpath:sql/data.sql
+
+spring.h2.console.path=/h2-console
+spring.h2.console.enabled=true
+
+# 生成JWT时候的密钥
+jwt.secret=11111222223333444455556667778888
+# sitemap 存放地址
+sitemap.path=/www/wwwroot/celess.cn/sitemap.xml
+
+
+###############  email  ##############
+spring.mail.host=smtp.163.com
+spring.mail.username=
+spring.mail.password=
+spring.mail.properties.mail.smtp.auth=true
+spring.mail.properties.mail.smtp.starttls.enable=true
+spring.mail.properties.mail.smtp.starttls.required=true
+spring.mail.default-encoding=UTF-8
+spring.mail.port=465
+spring.mail.properties.mail.smtp.socketFactory.port=465
+spring.mail.properties.mail.smtp.socketFactory.class=javax.net.ssl.SSLSocketFactory
+spring.mail.properties.mail.smtp.socketFactory.fallback=false
+
+##################  mybatis  ##################
+mybatis.mapper-locations=classpath:mapper/*.xml
+mybatis.type-aliases-package=cn.celess.blog.entity
+
+
+pagehelper.helper-dialect=mysql
+pagehelper.reasonable=true
+pagehelper.support-methods-arguments=true
+pagehelper.params=count=countSql
+
+############### redis ##############
+# REDIS (RedisProperties)
+# Redis数据库索引（默认为0）
+spring.redis.database=1
+# Redis服务器地址
+spring.redis.host=127.0.0.1
+# Redis服务器连接端口 解决端口冲突  防止使用本地的redis
+spring.redis.port=6379
+# Redis服务器连接密码（默认为空）
+spring.redis.password=
+# 连接池最大连接数（使用负值表示没有限制）
+spring.redis.jedis.pool.max-active=-1
+# 连接池最大阻塞等待时间（使用负值表示没有限制）
+spring.redis.jedis.pool.max-wait=-1
+# 连接池中的最大空闲连接
+spring.redis.jedis.pool.max-idle=8
+# 连接池中的最小空闲连接
+spring.redis.jedis.pool.min-idle=0
+# 连接超时时间（毫秒）
+spring.redis.timeout=5000
+

src/main/resources/application-openSource.properties

@@ -1,9 +1,9 @@
 server.port=8081
 
 # 七牛的密钥配置
-qiniu.accessKey=
-qiniu.secretKey=
-qiniu.bucket=
+file.qiniu.accessKey=
+file.qiniu.secretKey=
+file.qiniu.bucket=
 # sitemap 存放地址
 sitemap.path=
 # 生成JWT时候的密钥
@@ -26,6 +26,12 @@ spring.datasource.url=
 spring.datasource.username=
 spring.datasource.password=
 spring.datasource.driver-class-name=com.mysql.jdbc.Driver
+spring.datasource.platform=mysql
+# never / always / embedded
+spring.datasource.initialization-mode=never
+spring.datasource.sql-script-encoding=utf-8
+spring.datasource.schema=classpath:sql/schema.sql
+spring.datasource.data=classpath:sql/data.sql
 
 
 ##################  mybatis  ##################
@@ -56,12 +62,6 @@ spring.mail.properties.mail.smtp.socketFactory.class=javax.net.ssl.SSLSocketFact
 spring.mail.properties.mail.smtp.socketFactory.fallback=false
 
 
-#### 用于nginx的代理 获取真实ip
-server.use-forward-headers = true
-server.tomcat.remote-ip-header = X-Real-IP
-server.tomcat.protocol-header = X-Forwarded-Proto
-
-
 ############### redis ##############
 # REDIS (RedisProperties)
 # Redis数据库索引（默认为0）

src/main/resources/application-test.properties

@@ -0,0 +1,88 @@
+server.port=8081
+
+# 七牛的密钥配置
+file.qiniu.accessKey=${QINIU_ACCESSKEY:null}
+file.qiniu.secretKey=${QINIU_SECRETKEY:null}
+file.qiniu.bucket=xiaohai
+# sitemap 存放地址
+sitemap.path=
+# 生成JWT时候的密钥
+jwt.secret=sdaniod213k123123ipoeqowekqwe
+
+##spring.jpa.show-sql=false
+##spring.jpa.hibernate.ddl-auto=update
+
+mybatis.type-handlers-package=cn.celess.blog.mapper.typehandler
+logging.level.cn.celess.blog.mapper=debug
+#      上传单个文件的大小
+spring.servlet.multipart.max-file-size=10MB
+#      上传文件的总大小
+spring.servlet.multipart.max-request-size=10MB
+
+spring.jackson.default-property-inclusion=non_null
+
+
+#################  数据库  ##################
+spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
+
+#h2
+spring.datasource.driver-class-name=org.h2.Driver
+spring.datasource.url=jdbc:h2:mem:testdb;mode=mysql;DB_CLOSE_DELAY=-1;DATABASE_TO_UPPER=false
+spring.datasource.username=sa
+spring.datasource.password=
+
+
+spring.datasource.sql-script-encoding=utf-8
+spring.datasource.initialization-mode=ALWAYS
+spring.datasource.schema=classpath:sql/schema-h2.sql
+spring.datasource.data=classpath:sql/data.sql
+
+
+##################  mybatis  ##################
+mybatis.mapper-locations=classpath:mapper/*.xml
+mybatis.type-aliases-package=cn.celess.blog.entity
+
+
+pagehelper.helper-dialect=mysql
+pagehelper.reasonable=true
+pagehelper.support-methods-arguments=true
+pagehelper.params=count=countSql
+
+
+###############  email  ##############
+spring.mail.host=smtp.163.com
+spring.mail.username=
+spring.mail.password=
+spring.mail.properties.mail.smtp.auth=true
+spring.mail.properties.mail.smtp.starttls.enable=true
+spring.mail.properties.mail.smtp.starttls.required=true
+spring.mail.default-encoding=UTF-8
+spring.mail.port=465
+spring.mail.properties.mail.smtp.socketFactory.port=465
+spring.mail.properties.mail.smtp.socketFactory.class=javax.net.ssl.SSLSocketFactory
+spring.mail.properties.mail.smtp.socketFactory.fallback=false
+
+
+
+############### redis ##############
+
+# REDIS (RedisProperties)
+# Redis数据库索引（默认为0）
+spring.redis.database=1
+# Redis服务器地址
+spring.redis.host=127.0.0.1
+# Redis服务器连接端口 解决端口冲突  防止使用本地的redis
+spring.redis.port=6380
+# Redis服务器连接密码（默认为空）
+spring.redis.password=
+# 连接池最大连接数（使用负值表示没有限制）
+spring.redis.jedis.pool.max-active=-1
+# 连接池最大阻塞等待时间（使用负值表示没有限制）
+spring.redis.jedis.pool.max-wait=-1
+# 连接池中的最大空闲连接
+spring.redis.jedis.pool.max-idle=8
+# 连接池中的最小空闲连接
+spring.redis.jedis.pool.min-idle=0
+# 连接超时时间（毫秒）
+spring.redis.timeout=5000
+

src/main/resources/application.properties

@@ -1,7 +1,10 @@
-spring.profiles.active=prod
-####七牛的配置
-####cn.celess.blog.service.serviceimpl.QiniuServiceImpl
+spring.profiles.active=easyDeploy
 logging.level.cn.celess.blog=debug
 logging.level.cn.celess.blog.mapper=info
 
+#### 用于nginx的代理 获取真实ip
+server.use-forward-headers = true
+server.tomcat.remote-ip-header = X-Real-IP
+server.tomcat.protocol-header = X-Forwarded-Proto
+
 ## 修改openSource 添加-test 文件用于测试   -prod文件用于线上发布

src/main/resources/mapper/ConfigMapper.xml

@@ -0,0 +1,50 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="cn.celess.blog.mapper.ConfigMapper">
+
+    <resultMap type="cn.celess.blog.entity.Config" id="ConfigMap">
+        <result property="id" column="conf_id" jdbcType="INTEGER"/>
+        <result property="name" column="conf_name" jdbcType="VARCHAR"/>
+        <result property="value" column="conf_value" jdbcType="VARCHAR"/>
+    </resultMap>
+
+    <!--查询单个-->
+    <select id="getConfiguration" resultMap="ConfigMap">
+        select conf_id,
+               conf_name,
+               conf_value
+        from config
+        where conf_name = #{key}
+    </select>
+
+    <!--查询指定行数据-->
+    <select id="getConfigurations" resultMap="ConfigMap">
+        select conf_id,
+               conf_name,
+               conf_value
+        from config
+    </select>
+
+
+    <!--新增所有列-->
+    <insert id="addConfiguration" keyProperty="id" useGeneratedKeys="true">
+        insert into config(conf_name, conf_value)
+        values (#{name}, #{value})
+    </insert>
+
+
+    <!--通过主键修改数据-->
+    <update id="updateConfiguration">
+        update config
+        set conf_value = #{value}
+        where conf_id = #{id}
+    </update>
+
+    <!--通过主键删除-->
+    <delete id="deleteConfiguration">
+        delete
+        from config
+        where conf_id = #{id}
+    </delete>
+
+</mapper>

src/main/resources/mapper/PartnerSiteMapper.xml

@@ -9,20 +9,26 @@
         <result column="l_icon_path" property="iconPath"/>
         <result column="l_desc" property="desc"/>
         <result column="is_delete" property="delete"/>
+        <result column="l_email" property="email"/>
+        <result column="l_notification" property="notification"/>
     </resultMap>
 
     <insert id="insert" parameterType="cn.celess.blog.entity.PartnerSite" useGeneratedKeys="true" keyProperty="id">
-        insert into links (l_name, l_is_open, l_url, l_icon_path, l_desc, is_delete)
-        values (#{name}, #{open}, #{url}, #{iconPath}, #{desc}, false)
+        insert into links (l_name, l_is_open, l_url, l_icon_path, l_desc, l_email, l_notification, is_delete)
+        values (#{name}, #{open}, #{url}, #{iconPath}, #{desc}, #{email}, #{notification}, false)
     </insert>
 
     <update id="update" parameterType="cn.celess.blog.entity.PartnerSite">
-        update links set
-        <if test="open!=null">l_is_open=#{open},</if>
-        <if test="iconPath!=null">l_icon_path=#{iconPath},</if>
-        <if test="desc!=null">l_desc=#{desc},</if>
-        <if test="url!=null">l_url=#{url},</if>
-        <if test="name!=null">l_name=#{name}</if>
+        update links
+        <trim prefix="set" suffixOverrides=",">
+            <if test="open!=null">l_is_open=#{open},</if>
+            <if test="iconPath!=null">l_icon_path=#{iconPath},</if>
+            <if test="desc!=null">l_desc=#{desc},</if>
+            <if test="url!=null">l_url=#{url},</if>
+            <if test="name!=null">l_name=#{name},</if>
+            <if test="notification!=null">l_notification=#{notification},</if>
+            <if test="email!=null">l_email=#{email}</if>
+        </trim>
         where l_id=#{id}
     </update>
 
@@ -72,6 +78,9 @@
     <select id="findAll" resultMap="partnerSiteResultMap">
         select *
         from links
+        <if test="_parameter != null">
+            where is_delete=#{deleted}
+        </if>
     </select>
 
 

src/main/resources/mapper/UserMapper.xml

@@ -130,5 +130,8 @@
     <select id="findAll" resultMap="userResultMap">
         select *
         from user
+        <if test="_parameter != null">
+            where status=#{status}
+        </if>
     </select>
 </mapper>

src/main/resources/public/css/install.css

@@ -0,0 +1,110 @@
+* {
+    margin: 0;
+    padding: 0;
+}
+
+html, body {
+    width: 100%;
+    height: 100%;
+}
+
+.box {
+    width: 600px;
+    height: 45vh;
+    padding: 8px 10px;
+    border: 1px solid rgba(50, 50, 50, .1);
+    border-radius: 3px;
+    position: relative;
+    top: 50%;
+    left: 50%;
+    transform: translate(-50%, -50%);
+}
+
+.box-inner {
+    height: 100%;
+    width: 100%;
+    overflow: hidden;
+}
+
+.box-inner select, input, button {
+    width: 90%;
+    margin: 12px 5%;
+    height: 40px;
+    padding: 5px 10px;
+    border-radius: 3px;
+    border: 1px solid rgba(50, 50, 50, .1);
+    box-sizing: border-box;
+}
+
+input:focus {
+    border: 1px solid rgba(0, 50, 50, .6);
+    outline: none;
+}
+
+.box-inner h3 {
+    text-align: center;
+    margin-bottom: 20px;
+}
+
+.err-msg {
+    display: none;
+    color: red;
+    width: 100%;
+    padding-left: 30px;
+    font-size: small;
+    font-weight: lighter;
+}
+
+
+.show-err-tip {
+    display: block;
+}
+
+#first-page, #second-page {
+    height: 100%;
+    margin: 10px 0;
+}
+
+.hidden {
+    height: 0 !important;
+    visibility: hidden;
+    position: absolute;
+}
+
+.show {
+    width: 100%;
+    position: relative;
+    height: 100%;
+    visibility: visible;
+}
+
+
+#next-step, .button-group {
+    position: absolute;
+    left: 0;
+    right: 0;
+    bottom: 10px;
+}
+
+.button-group button {
+    width: 80px;
+}
+
+.button-group #install {
+    position: absolute;
+    bottom: 0;
+    right: 20px;
+    background: #1890ff;
+    color: white;
+}
+
+.loading-show {
+    z-index: 10000;
+    display: flex !important;
+    position: fixed;
+    left: 0;
+    right: 0;
+    bottom: 0;
+    top: 0;
+}
+

src/main/resources/public/css/loading.css

@@ -0,0 +1,171 @@
+.pacman {
+    position: relative;
+}
+
+
+.loading {
+    box-sizing: border-box;
+    flex: 0 1 auto;
+    display: none;
+    flex-direction: column;
+    flex-grow: 1;
+    flex-shrink: 0;
+    flex-basis: 100%;
+    max-width: 100%;
+    /*height: 200px;*/
+    align-items: center;
+    justify-content: center;
+    background: #ececec;
+}
+
+
+.pacman > div:nth-child(2) {
+    -webkit-animation: pacman-balls 1s 0s infinite linear;
+    animation: pacman-balls 1s 0s infinite linear;
+}
+
+.pacman > div:nth-child(3) {
+    -webkit-animation: pacman-balls 1s 0.33s infinite linear;
+    animation: pacman-balls 1s 0.33s infinite linear;
+}
+
+.pacman > div:nth-child(4) {
+    -webkit-animation: pacman-balls 1s 0.66s infinite linear;
+    animation: pacman-balls 1s 0.66s infinite linear;
+}
+
+.pacman > div:nth-child(5) {
+    -webkit-animation: pacman-balls 1s 0.99s infinite linear;
+    animation: pacman-balls 1s 0.99s infinite linear;
+}
+
+.pacman > div:first-of-type {
+    width: 0px;
+    height: 0px;
+    border-right: 25px solid transparent;
+    border-top: 25px solid #ed5565;
+    border-left: 25px solid #ed5565;
+    border-bottom: 25px solid #ed5565;
+    border-radius: 25px;
+    -webkit-animation: rotate_pacman_half_up 0.5s 0s infinite;
+    animation: rotate_pacman_half_up 0.5s 0s infinite;
+}
+
+.pacman > div:nth-child(2) {
+    width: 0px;
+    height: 0px;
+    border-right: 25px solid transparent;
+    border-top: 25px solid #ed5565;
+    border-left: 25px solid #ed5565;
+    border-bottom: 25px solid #ed5565;
+    border-radius: 25px;
+    -webkit-animation: rotate_pacman_half_down 0.5s 0s infinite;
+    animation: rotate_pacman_half_down 0.5s 0s infinite;
+    margin-top: -50px;
+}
+
+.pacman > div:nth-child(3), .pacman > div:nth-child(4), .pacman > div:nth-child(5), .pacman > div:nth-child(6) {
+    background-color: #ed5565;
+    width: 15px;
+    height: 15px;
+    border-radius: 100%;
+    margin: 2px;
+    width: 10px;
+    height: 10px;
+    position: absolute;
+    -webkit-transform: translate(0, -6.25px);
+    -ms-transform: translate(0, -6.25px);
+    transform: translate(0, -6.25px);
+    top: 25px;
+    left: 100px;
+}
+
+@-webkit-keyframes rotate_pacman_half_up {
+    0% {
+        -webkit-transform: rotate(270deg);
+        transform: rotate(270deg);
+    }
+
+    50% {
+        -webkit-transform: rotate(360deg);
+        transform: rotate(360deg);
+    }
+
+    100% {
+        -webkit-transform: rotate(270deg);
+        transform: rotate(270deg);
+    }
+}
+
+@keyframes rotate_pacman_half_up {
+    0% {
+        -webkit-transform: rotate(270deg);
+        transform: rotate(270deg);
+    }
+
+    50% {
+        -webkit-transform: rotate(360deg);
+        transform: rotate(360deg);
+    }
+
+    100% {
+        -webkit-transform: rotate(270deg);
+        transform: rotate(270deg);
+    }
+}
+
+@-webkit-keyframes rotate_pacman_half_down {
+    0% {
+        -webkit-transform: rotate(90deg);
+        transform: rotate(90deg);
+    }
+
+    50% {
+        -webkit-transform: rotate(0deg);
+        transform: rotate(0deg);
+    }
+
+    100% {
+        -webkit-transform: rotate(90deg);
+        transform: rotate(90deg);
+    }
+}
+
+@keyframes rotate_pacman_half_down {
+    0% {
+        -webkit-transform: rotate(90deg);
+        transform: rotate(90deg);
+    }
+
+    50% {
+        -webkit-transform: rotate(0deg);
+        transform: rotate(0deg);
+    }
+
+    100% {
+        -webkit-transform: rotate(90deg);
+        transform: rotate(90deg);
+    }
+}
+
+@-webkit-keyframes pacman-balls {
+    75% {
+        opacity: 0.7;
+    }
+
+    100% {
+        -webkit-transform: translate(-100px, -6.25px);
+        transform: translate(-100px, -6.25px);
+    }
+}
+
+@keyframes pacman-balls {
+    75% {
+        opacity: 0.7;
+    }
+
+    100% {
+        -webkit-transform: translate(-100px, -6.25px);
+        transform: translate(-100px, -6.25px);
+    }
+}

src/main/resources/public/install.html

@@ -0,0 +1,164 @@
+<!DOCTYPE html>
+<html lang="zh">
+<head>
+    <meta charset="UTF-8">
+    <title>博客安装页面</title>
+    <link rel="stylesheet" href="css/install.css">
+    <link rel="stylesheet" href="css/loading.css">
+    <script src="//cdn.bootcdn.net/ajax/libs/jquery/1.10.0/jquery.min.js"></script>
+</head>
+<body>
+<div class="loading">
+    <div class="loader-inner pacman">
+        <div></div>
+        <div></div>
+        <div></div>
+        <div></div>
+        <div></div>
+    </div>
+    <p style="margin-top: 20px;margin-left: 10px;">
+        加载中....
+    </p>
+</div>
+<div class="box">
+    <div class="box-inner">
+        <div id="first-page" class="show">
+            <h3>数据库配置</h3>
+            <!-- 数据库-->
+            <select id="db-type">
+                <option value="h2">H2数据库</option>
+                <option value="mysql">Mysql数据库</option>
+            </select>
+
+            <input id="db-host" type="text" placeholder="请输入数据库的主机地址">
+            <span class="err-msg" id="err-tip-db-host">数据库的主机地址不可为空</span>
+
+            <input id="db-name" type="text" placeholder="请输入数据库的名称">
+            <span class="err-msg" id="err-tip-db-name">数据库的地址名称不可为空</span>
+
+            <input id="db-username" type="text" placeholder="请输入数据库用户名">
+            <span class="err-msg" id="err-tip-db-username">数据库用户名不可为空</span>
+
+            <input id="db-password" type="password" placeholder="请输入数据库密码">
+            <span class="err-msg" id="err-tip-db-password">数据库密码不可为空</span>
+
+            <button onclick="nextPage()" id="next-step">下一步</button>
+        </div>
+        <div id="second-page" class="hidden">
+            <h3>新建管理员用户</h3>
+            <input id="email" type="email" placeholder="请输入用户名">
+            <span class="err-msg" id="err-tip-email">用户名不可为空</span>
+            <input id="password" type="password" placeholder="请输入密码">
+            <span class="err-msg" id="err-tip-password">密码不可为空</span>
+            <div class="button-group">
+                <button onclick="preStep()" id="pre-step">上一步</button>
+                <button onclick="postInstallEvent()" id="install"> 安装</button>
+            </div>
+        </div>
+    </div>
+</div>
+<script type="text/javascript">
+    let requestData;
+    let interValId;
+
+    function postInstallEvent() {
+        requestData.email = $('#email').val()
+        requestData.password = $('#password').val()
+        $('#err-tip-email').removeClass("show-err-tip");
+        $('#err-tip-password').removeClass("show-err-tip");
+        !requestData.password && checkInput($('#password'), $('#err-tip-password'))
+        !requestData.email && checkInput($('#email'), $('#err-tip-email'))
+        if (!requestData.email || !requestData.password) return;
+        $('.loading').addClass('loading-show');
+        $.ajax('/install', {
+            method: "POST",
+            contentType: 'application/json',
+            data: JSON.stringify(requestData),
+            withCredentials: true,
+            dataType: 'json',
+            success: function (data) {
+                console.log("data", data);
+                console.log("requestData", requestData);
+                interValId = setInterval(checkConnection, 500)
+                checkConnection();
+            }
+        })
+    }
+
+    function checkConnection() {
+        $.ajax('/is_install', {
+            method: "GET",
+            dataType: 'json',
+            success: function (data) {
+                if (data.result.is_install) {
+                    console.log("安装成功");
+                    window.location.href = '/';
+                } else {
+                    console.log("安装失败");
+                }
+                $('.loading').removeClass('loading-show');
+                clearInterval(interValId);
+            },
+            error: function (err) {
+                console.log(err);
+            }
+        })
+    }
+
+    /**
+     * 切换下一页
+     */
+    function nextPage() {
+        requestData = {
+            dbPassword: $('#db-password').val(),
+            dbType: $('#db-type').val(),
+            dbHost: $('#db-host').val(),
+            dbName: $('#db-name').val(),
+            dbUsername: $('#db-username').val(),
+            email: '',
+            password: '',
+        }
+        $('#err-tip-db-type').removeClass("show-err-tip");
+        $('#err-tip-db-host').removeClass("show-err-tip");
+        $('#err-tip-db-name').removeClass("show-err-tip");
+        $('#err-tip-db-username').removeClass("show-err-tip");
+        $('#err-tip-db-password').removeClass("show-err-tip");
+        !requestData.dbType && checkInput($('#db-type'), $('#err-tip-db-type'))
+        !requestData.dbHost && checkInput($('#db-host'), $('#err-tip-db-host'))
+        !requestData.dbName && checkInput($('#db-name'), $('#err-tip-db-name'))
+        !requestData.dbUsername && checkInput($('#db-username'), $('#err-tip-db-username'))
+        !requestData.dbPassword && checkInput($('#db-password'), $('#err-tip-db-password'))
+        if (!requestData.dbType || !requestData.dbUrl || !requestData.dbUsername || !requestData.dbPassword) return
+        $('#first-page').addClass('hidden');
+        $('#first-page').removeClass('show');
+        $('#second-page').addClass('show');
+        $('#second-page').removeClass('hidden');
+    }
+
+    function preStep() {
+        $('#second-page').addClass('hidden');
+        $('#second-page').removeClass('show');
+        $('#first-page').addClass('show');
+        $('#first-page').removeClass('hidden');
+
+        $('#err-tip-email').removeClass("show-err-tip");
+        $('#err-tip-password').removeClass("show-err-tip");
+    }
+
+    /**
+     * 检查组件的状态
+     * @param component 输入框
+     * @param errTipComponent 错误信息提示框
+     * @returns {boolean} 是否有错误
+     */
+    function checkInput(component, errTipComponent) {
+        if (!component.val()) {
+            errTipComponent.addClass("show-err-tip");
+            component.focus();
+            return true;
+        }
+        return false;
+    }
+</script>
+</body>
+</html>

src/main/resources/sql/schema-h2.sql

@@ -0,0 +1,201 @@
+drop view if exists articleView;
+drop view if exists commentView;
+drop table if exists article_tag;
+drop table if exists comment;
+drop table if exists article;
+drop table if exists user;
+drop table if exists tag_category;
+drop table if exists links;
+drop table if exists visitor;
+drop table if exists web_update;
+drop table if exists config;
+
+--  用户表
+CREATE TABLE `user`
+(
+    `u_id`                   int         not null primary key auto_increment,
+    `u_email`                varchar(50) not null,
+    `u_pwd`                  varchar(40) not null comment '密码',
+    `u_email_status`         boolean              default false comment '邮箱验证状态',
+    `u_avatar`               varchar(255)         default null comment '用户头像',
+    `u_desc`                 tinytext             default null comment '用户的描述',
+    `u_recently_landed_time` datetime             default null comment '最近的登录时间',
+    `u_display_name`         varchar(30)          default null comment '展示的昵称',
+    `u_role`                 varchar(40) not null default 'user' comment '权限组',
+    `status`                 tinyint(1)  not null default 0 comment '账户状态',
+    unique key `uni_user_id` (`u_id`),
+    unique key `uni_user_email` (`u_email`)
+);
+
+--  标签和分类表
+CREATE TABLE `tag_category`
+(
+    `t_id`        bigint(20) primary key auto_increment,
+    `t_name`      varchar(255) not null,
+    `is_category` boolean      not null default true,
+    `is_delete`   boolean      not null default false comment '该数据是否被删除'
+);
+
+--  文章表
+CREATE TABLE `article`
+(
+    `a_id`             bigint(20) primary key auto_increment,
+    `a_title`          varchar(255) not null unique comment '文章标题',
+    `a_summary`        varchar(255) not null comment '文章摘要',
+    `a_md_content`     longtext     not null comment '文章Markdown内容',
+    `a_url`            tinytext              default null comment '转载文章的原文链接',
+    `a_author_id`      int          not null comment '作者id',
+    `a_is_original`    boolean               default true comment '文章是否原创',
+    `a_reading_number` int                   default 0 comment '文章阅读数',
+    `a_like`           int                   default 0 comment '文章点赞数',
+    `a_dislike`        int                   default 0 comment '文章不喜欢数',
+    `a_category_id`    bigint       not null comment '文章分类id',
+    `a_publish_date`   datetime              default CURRENT_TIMESTAMP comment '文章发布时间',
+    `a_update_date`    datetime              default null comment '文章的更新时间',
+    `a_is_open`        boolean               default true comment '文章是否可见',
+    `is_delete`        boolean      not null default false comment '该数据是否被删除',
+    foreign key (a_category_id) references tag_category (t_id),
+    foreign key (a_author_id) references user (u_id)
+);
+
+-- 文章标签表
+CREATE TABLE `article_tag`
+(
+    `at_id` bigint(20) primary key auto_increment,
+    `a_id`  bigint(20) not null comment '文章id',
+    `t_id`  bigint     not null comment 'tag/category 的id',
+    foreign key (a_id) references article (a_id),
+    foreign key (t_id) references tag_category (t_id)
+);
+
+-- 评论/留言表
+CREATE TABLE `comment`
+(
+    `co_id`             bigint(20) primary key auto_increment,
+    `co_page_path`      varchar(255) not null comment '评论/留言的页面',
+    `co_content`        text         not null comment '评论/留言内容',
+    `co_date`           datetime     not null comment '评论/留言的日期',
+    `co_status`         tinyint      not null default 0 comment '评论的状态',
+    `co_pid`            bigint       not null default -1 comment '评论/留言的父id',
+    `co_from_author_id` int          not null comment '留言者id',
+    `co_to_author_id`   int                   default null comment '父评论的作者id',
+    foreign key (co_from_author_id) references user (u_id),
+    foreign key (co_to_author_id) references user (u_id)
+);
+
+--  友站表
+CREATE TABLE `links`
+(
+    `l_id`           bigint(20) primary key auto_increment,
+    `l_name`         varchar(255)        not null comment '友站名称',
+    `l_is_open`      boolean                      default true comment '是否公开',
+    `l_url`          varchar(255) unique not null comment '首页地址',
+    `l_icon_path`    varchar(255)        not null comment '友链的icon地址',
+    `l_desc`         varchar(255)        not null comment '友链的说明描述',
+    `is_delete`      boolean             not null default false comment '该数据是否被删除',
+    `l_email`        varchar(255) comment '网站管理员的邮箱',
+    `l_notification` boolean                      default false comment '是否通知了'
+);
+
+--  访客表
+CREATE TABLE `visitor`
+(
+    `v_id`         bigint(20) primary key auto_increment,
+    `v_date`       datetime     not null comment '访问时间',
+    `v_ip`         varchar(255) not null comment '访客ip',
+    `v_user_agent` text comment '访客ua',
+    `is_delete`    boolean      not null default false comment '该数据是否被删除'
+);
+
+-- 更新内容表
+CREATE TABLE `web_update`
+(
+    `wu_id`     int primary key auto_increment,
+    `wu_info`   varchar(255) not null comment '更新内容',
+    `wu_time`   datetime     not null comment '更新时间',
+    `is_delete` boolean      not null default false comment '该数据是否被删除'
+);
+
+CREATE TABLE `config`
+(
+    `conf_id`    int primary key auto_increment,
+    `conf_name`  varchar(255) unique not null comment '配置名',
+    `conf_value` varchar(255) default null comment '配置值'
+);
+
+CREATE VIEW articleView
+            (articleId, title, summary, mdContent, url, isOriginal, readingCount, likeCount, dislikeCount,
+             publishDate, updateDate, isOpen,
+             categoryId, categoryName, tagId, tagName,
+             authorId, userEmail, userAvatar, userDisplayName, isDelete)
+as
+select article.a_id             as articleId,
+       article.a_title          as title,
+       article.a_summary        as summary,
+       article.a_md_content     as mdContent,
+       article.a_url            as url,
+       article.a_is_original    as isOriginal,
+       article.a_reading_number as readingCount,
+       article.a_like           as likeCount,
+       article.a_dislike        as dislikeCount,
+       article.a_publish_date   as publishDate,
+       article.a_update_date    as updateDate,
+       article.a_is_open        as isOpen,
+       category.t_id            as categoryId,
+       category.t_name          as categoryName,
+       tag.t_id                 as tagId,
+       tag.t_name               as tagName,
+       user.u_id                as authorId,
+       user.u_email             as userEmail,
+       user.u_avatar            as userAvatar,
+       user.u_display_name      as userDisplayName,
+       article.is_delete        as isDelete
+from article,
+     tag_category as tag,
+     tag_category as category,
+     article_tag,
+     user
+where article.a_id = article_tag.a_id
+  and article_tag.t_id = tag.t_id
+  and tag.is_category = false
+  and article.a_category_id = category.t_id
+  and category.is_category = true
+  and article.a_author_id = user.u_id;
+
+
+CREATE VIEW commentView
+            (commentId, pagePath, content, date, status, pid, toAuthorId, toAuthorEmail, toAuthorDisplayName,
+             toAuthorAvatar, fromAuthorId, fromAuthorEmail, fromAuthorDisplayName,
+             fromAuthorAvatar)
+as
+select cuT.co_id               as commentId,
+       cuT.co_page_path        as pagePath,
+       cuT.co_content          as content,
+       cuT.co_date             as date,
+       cuT.co_status           as status,
+       cuT.co_pid              as pid,
+       cuT.co_to_author_id     as toAuthorId,
+       cuT.toEmail             as toAuthorEmail,
+       cuT.toDisplayName       as toAuthorDisplayName,
+       cuT.toAvatar            as toAuthorAvatar,
+       userFrom.u_id           as fromAuthorId,
+       userFrom.u_email        as fromAuthorEmail,
+       userFrom.u_display_name as fromAuthorDisplayName,
+       userFrom.u_avatar       as fromAuthorAvatar
+from (select comment.co_id,
+             comment.co_page_path,
+             comment.co_content,
+             comment.co_date,
+             comment.co_status,
+             comment.co_pid,
+             comment.co_from_author_id,
+             comment.co_to_author_id,
+             userTo.u_email        as toEmail,
+             userTo.u_display_name as toDisplayName,
+             userTo.u_avatar       as toAvatar
+      from comment
+               left join user userTo on (comment.co_to_author_id = userTo.u_id)
+     ) as cuT,
+     user as userFrom
+where cuT.co_from_author_id = userFrom.u_id;
+

src/main/resources/sql/schema.sql

@@ -1,6 +1,15 @@
-CREATE DATABASE if not exists `blog`;
+drop view if exists articleView;
+drop view if exists commentView;
+drop table if exists article_tag;
+drop table if exists comment;
+drop table if exists article;
+drop table if exists user;
+drop table if exists tag_category;
+drop table if exists links;
+drop table if exists visitor;
+drop table if exists web_update;
+drop table if exists config;
 
-USE blog;
 
 CREATE TABLE `user`
 (
@@ -66,19 +75,23 @@ CREATE TABLE `comment`
     `co_status`         tinyint                         not null default 0 comment '评论的状态',
     `co_pid`            bigint                          not null default -1 comment '评论/留言的父id',
     `co_from_author_id` int                             not null comment '留言者id',
-    `co_to_author_id`   int                                      default null comment '父评论的作者id'
+    `co_to_author_id`   int                                      default null comment '父评论的作者id',
+    foreign key (co_from_author_id) references user (u_id),
+    foreign key (co_to_author_id) references user (u_id)
 ) DEFAULT CHARSET = utf8mb4
   COLLATE utf8mb4_general_ci,comment '评论/留言表';
 
 CREATE TABLE `links`
 (
-    `l_id`        bigint(20) primary key auto_increment,
-    `l_name`      varchar(255) COLLATE utf8mb4_unicode_ci not null comment '友站名称',
-    `l_is_open`   boolean                                          default true comment '是否公开',
-    `l_url`       varchar(255)                            not null comment '首页地址',
-    `l_icon_path` varchar(255)                            not null comment '友链的icon地址',
-    `l_desc`      varchar(255) COLLATE utf8mb4_unicode_ci not null comment '友链的说明描述',
-    `is_delete`   boolean                                 not null default false comment '该数据是否被删除'
+    `l_id`           bigint(20) primary key auto_increment,
+    `l_name`         varchar(255) COLLATE utf8mb4_unicode_ci not null comment '友站名称',
+    `l_is_open`      boolean                                          default true comment '是否公开',
+    `l_url`          varchar(255) unique                     not null comment '首页地址',
+    `l_icon_path`    varchar(255)                            not null comment '友链的icon地址',
+    `l_desc`         varchar(255) COLLATE utf8mb4_unicode_ci not null comment '友链的说明描述',
+    `is_delete`      boolean                                 not null default false comment '该数据是否被删除',
+    `l_email`        varchar(255) comment '网站管理员的邮箱',
+    `l_notification` boolean                                          default false comment '是否通知了'
 ) comment '友站表';
 
 CREATE TABLE `visitor`
@@ -98,6 +111,13 @@ CREATE TABLE `web_update`
     `is_delete` boolean                                 not null default false comment '该数据是否被删除'
 ) comment '更新内容表';
 
+CREATE TABLE `config`
+(
+    `conf_id`    int primary key auto_increment,
+    `conf_name`  varchar(255) unique not null comment '配置名',
+    `conf_value` varchar(255) default null comment '配置值'
+);
+
 CREATE VIEW articleView
             (articleId, title, summary, mdContent, url, isOriginal, readingCount, likeCount, dislikeCount,
              publishDate, updateDate, isOpen,

src/test/java/cn/celess/blog/BaseTest.java

@@ -2,29 +2,44 @@ package cn.celess.blog;
 
 
 import cn.celess.blog.entity.Response;
+import cn.celess.blog.entity.model.FileInfo;
+import cn.celess.blog.entity.model.FileResponse;
 import cn.celess.blog.entity.model.UserModel;
 import cn.celess.blog.entity.request.LoginReq;
+import cn.celess.blog.service.MailService;
+import cn.celess.blog.service.FileManager;
+import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.extern.slf4j.Slf4j;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.http.MediaType;
+import org.springframework.mail.SimpleMailMessage;
 import org.springframework.mock.web.MockHttpSession;
 import org.springframework.test.context.ActiveProfiles;
 import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.context.web.WebAppConfiguration;
 import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.MvcResult;
 import org.springframework.test.web.servlet.ResultActions;
 import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
-import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 import org.springframework.web.context.WebApplicationContext;
 
-
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.UnsupportedEncodingException;
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
 import java.util.UUID;
 
 import static cn.celess.blog.enmu.ResponseEnum.SUCCESS;
@@ -44,15 +59,27 @@ import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.
 @ActiveProfiles("test")
 public class BaseTest {
 
+
+    private final Logger logger = LoggerFactory.getLogger(this.getClass());
+
     protected MockMvc mockMvc;
     protected final static String Code = "code";
     protected final static String Result = "result";
-    private static String userToken = null;
-    private static String adminToken = null;
+    protected final static String USERE_MAIL = "zh56462271@qq.com";
+    protected final static String ADMIN_EMAIL = "a@celess.cn";
+
     /**
      * jackson 序列化/反序列化Json
      */
     protected final ObjectMapper mapper = new ObjectMapper();
+    protected static final TypeReference<?> BOOLEAN_TYPE = new TypeReference<Response<Boolean>>() {
+    };
+    protected static final TypeReference<?> STRING_TYPE = new TypeReference<Response<String>>() {
+    };
+    protected static final TypeReference<?> OBJECT_TYPE = new TypeReference<Response<Object>>() {
+    };
+    protected static final TypeReference<?> MAP_OBJECT_TYPE = new TypeReference<Response<Map<String, Object>>>() {
+    };
 
     @Autowired
     private WebApplicationContext wac;
@@ -76,14 +103,13 @@ public class BaseTest {
      * @return token
      */
     protected String adminLogin() {
-        if (adminToken != null) return adminToken;
         LoginReq req = new LoginReq();
-        req.setEmail("a@celess.cn");
+        req.setEmail(ADMIN_EMAIL);
         req.setPassword("123456789");
-        req.setIsRememberMe(false);
-        adminToken = login(req);
-        assertNotNull(adminToken);
-        return adminToken;
+        req.setIsRememberMe(true);
+        String token = login(req);
+        assertNotNull(token);
+        return token;
     }
 
     /**
@@ -92,14 +118,13 @@ public class BaseTest {
      * @return token
      */
     protected String userLogin() {
-        if (userToken != null) return userToken;
         LoginReq req = new LoginReq();
-        req.setEmail("zh56462271@qq.com");
+        req.setEmail(USERE_MAIL);
         req.setPassword("123456789");
-        req.setIsRememberMe(false);
-        userToken = login(req);
-        assertNotNull(userToken);
-        return userToken;
+        req.setIsRememberMe(true);
+        String token = login(req);
+        assertNotNull(token);
+        return token;
     }
 
     /**
@@ -108,7 +133,7 @@ public class BaseTest {
      * @param req 用户信息
      * @return token | null
      */
-    private String login(LoginReq req) {
+    protected String login(LoginReq req) {
         String str = null;
         try {
             str = getMockData(post("/login"), null, req)
@@ -120,8 +145,10 @@ public class BaseTest {
             assertNotNull(token);
             return token;
         } catch (Exception e) {
+            logger.error("测试登录错误");
             e.printStackTrace();
         }
+        assertNotNull(str);
         return null;
     }
 
@@ -130,9 +157,11 @@ public class BaseTest {
         // 测试登录
         assertNotNull(userLogin());
         assertNotNull(adminLogin());
+        assertNotEquals(userLogin(), adminLogin());
         try {
             // 测试getMockData方法
             assertNotNull(getMockData(get("/headerInfo")));
+            getMockData((get("/headerInfo"))).andDo(result -> assertNotNull(getResponse(result, OBJECT_TYPE)));
         } catch (Exception e) {
             e.printStackTrace();
         }
@@ -141,13 +170,21 @@ public class BaseTest {
     /**
      * 产生指定长度的随机字符
      *
-     * @param len
-     * @return
+     * @param len len
+     * @return str
      */
     protected String randomStr(int len) {
         return UUID.randomUUID().toString().replaceAll("-", "").substring(0, len);
     }
 
+    /**
+     * 产生指定长度的随机字符
+     *
+     * @return str
+     */
+    protected String randomStr() {
+        return UUID.randomUUID().toString();
+    }
 
     /**
      * 抽离的mock请求方法
@@ -188,7 +225,131 @@ public class BaseTest {
         }
         if (content != null) {
             builder.content(mapper.writeValueAsString(content)).contentType(MediaType.APPLICATION_JSON);
+            logger.debug("param::json->{}", mapper.writeValueAsString(content));
         }
         return mockMvc.perform(builder).andExpect(status().isOk());
     }
+
+
+    protected <T> Response<T> getResponse(String json) {
+        return getResponse(json, OBJECT_TYPE);
+    }
+
+    protected <T> Response<T> getResponse(MvcResult result) {
+        return getResponse(result, OBJECT_TYPE);
+    }
+
+    /**
+     * 根据json 信息反序列化成Response对象
+     *
+     * @param json json
+     * @param <T>  泛型
+     * @return Response对象
+     */
+    protected <T> Response<T> getResponse(String json, TypeReference<?> responseType) {
+        Response<T> response = null;
+        try {
+            response = mapper.readValue(json, responseType);
+        } catch (IOException e) {
+            logger.error("解析json Response对象错误，json:[{}]", json);
+            e.printStackTrace();
+        }
+        assertNotNull(response);
+        return response;
+    }
+
+    /**
+     * 根据json 信息反序列化成Response对象
+     *
+     * @param result MvcResult
+     * @param <T>    泛型
+     * @return Response对象
+     */
+    protected <T> Response<T> getResponse(MvcResult result, TypeReference<?> responseType) {
+        try {
+            return getResponse(result.getResponse().getContentAsString(), responseType);
+        } catch (UnsupportedEncodingException e) {
+            logger.error("解析json Response对象错误，result:[{}]", result);
+            e.printStackTrace();
+        }
+        return null;
+    }
+
+
+    /**
+     * 修改 mailService 的实现类
+     *
+     * @param service       service 类
+     * @param mailFiledName service 中自动注入的mailService字段名
+     */
+    protected void mockInjectInstance(Object service, String mailFiledName, Object impl) {
+        Field field;
+        try {
+            assertNotNull(service);
+            field = service.getClass().getDeclaredField(mailFiledName);
+            field.setAccessible(true);
+            field.set(service, impl);
+        } catch (NoSuchFieldException | IllegalAccessException e) {
+            e.printStackTrace();
+        }
+    }
+
+
+    @Slf4j
+    public static class TestMailServiceImpl implements MailService {
+
+        @Override
+        public Boolean AsyncSend(SimpleMailMessage message) {
+            log.debug("异步邮件请求,SimpleMailMessage:[{}]", getJson(message));
+            return true;
+        }
+
+        @Override
+        public Boolean send(SimpleMailMessage message) {
+            log.debug("邮件请求,SimpleMailMessage:[{}]", getJson(message));
+            return true;
+        }
+
+        /**
+         * 转json
+         *
+         * @param o
+         * @return
+         */
+        private String getJson(Object o) {
+            ObjectMapper mapper = new ObjectMapper();
+            try {
+                return mapper.writeValueAsString(o);
+            } catch (JsonProcessingException e) {
+                return null;
+            }
+        }
+    }
+
+    @Slf4j
+    public static class TestFileManager implements FileManager {
+        @Override
+        public FileResponse uploadFile(InputStream is, String fileName) {
+            FileResponse response = new FileResponse();
+            log.debug("上传文件请求，[fileName:{}]", fileName);
+
+            response.key = "key";
+            response.type = "test";
+            response.hash = "hash";
+            response.size = 1;
+            return response;
+        }
+
+        @Override
+        public List<FileInfo> getFileList() {
+            log.debug("获取文件列表请求");
+            return new ArrayList<>();
+        }
+
+        @Override
+        public boolean deleteFile(String fileName) {
+            log.debug("删除[{}]成功", fileName);
+            return true;
+        }
+    }
 }

src/test/java/cn/celess/blog/RedisServerMock.java

@@ -0,0 +1,37 @@
+package cn.celess.blog;
+
+import org.springframework.stereotype.Component;
+import redis.embedded.RedisServer;
+
+import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
+import java.io.IOException;
+
+/**
+ * @author : xiaohai
+ * @date : 2020/08/14 16:20
+ */
+@Component
+public class RedisServerMock {
+
+    private RedisServer redisServer;
+
+    /**
+     * 构造方法之后执行.
+     *
+     * @throws IOException e
+     */
+    @PostConstruct
+    public void startRedis() throws IOException {
+        redisServer = new RedisServer(6380);
+        redisServer.start();
+    }
+
+    /**
+     * 析构方法之后执行.
+     */
+    @PreDestroy
+    public void stopRedis() {
+        redisServer.stop();
+    }
+}

src/test/java/cn/celess/blog/configuration/DruidConfigTest.java

@@ -0,0 +1,131 @@
+package cn.celess.blog.configuration;
+
+import cn.celess.blog.BaseTest;
+import cn.celess.blog.enmu.ConfigKeyEnum;
+import cn.celess.blog.service.fileserviceimpl.LocalFileServiceImpl;
+import com.alibaba.druid.pool.DruidDataSource;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.core.env.Environment;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Properties;
+
+import static org.junit.Assert.*;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class DruidConfigTest extends BaseTest {
+
+    @Mock
+    Environment env;
+
+    @Autowired
+    Environment globalEnvironment;
+
+
+    private File configFile;
+    private File bakConfigFile;
+
+    @Test
+    public void initDataSource() throws IOException {
+
+        DruidConfig druidConfig = new DruidConfig();
+
+        druidConfig.env = env;
+
+        System.out.println(Arrays.toString(env.getActiveProfiles()));
+
+        // 无配置文件
+        assertTrue(!configFile.exists() || configFile.delete());
+        DruidDataSource druidDataSource = druidConfig.initDataSource();
+
+        // 加载初始化时候配置文件的数据库连接
+        assertEquals(env.getProperty(DruidConfig.DB_CONFIG_URL_PREFIX), druidDataSource.getUrl());
+        assertEquals(env.getProperty(DruidConfig.DB_CONFIG_DRIVER_CLASS_NAME_PREFIX), druidDataSource.getDriverClassName());
+        assertEquals(env.getProperty(DruidConfig.DB_CONFIG_USERNAME_PREFIX), druidDataSource.getUsername());
+        assertEquals(env.getProperty(DruidConfig.DB_CONFIG_PASSWORD_PREFIX), druidDataSource.getPassword());
+        assertTrue(configFile.createNewFile());
+
+        // 有配置文件的测试
+        Properties properties = new Properties();
+        FileInputStream fileInputStream = new FileInputStream(configFile);
+        properties.load(fileInputStream);
+        fileInputStream.close();
+        properties.setProperty(DruidConfig.DB_CONFIG_URL_PREFIX, "jdbc:mysql://localhost:3306/blog");
+        properties.setProperty(DruidConfig.DB_CONFIG_DRIVER_CLASS_NAME_PREFIX, "com.mysql.cj.jdbc.Driver");
+        properties.setProperty(DruidConfig.DB_CONFIG_USERNAME_PREFIX, "username");
+        properties.setProperty(DruidConfig.DB_CONFIG_PASSWORD_PREFIX, "password");
+        // 保存到文件
+        FileOutputStream fileOutputStream = new FileOutputStream(configFile);
+        properties.store(fileOutputStream, "数据库配置");
+        fileOutputStream.close();
+        druidDataSource = druidConfig.initDataSource();
+        assertEquals(properties.getProperty(DruidConfig.DB_CONFIG_URL_PREFIX), druidDataSource.getUrl());
+        assertEquals(properties.getProperty(DruidConfig.DB_CONFIG_DRIVER_CLASS_NAME_PREFIX), druidDataSource.getDriverClassName());
+        assertEquals(properties.getProperty(DruidConfig.DB_CONFIG_USERNAME_PREFIX), druidDataSource.getUsername());
+        assertEquals(properties.getProperty(DruidConfig.DB_CONFIG_PASSWORD_PREFIX), druidDataSource.getPassword());
+    }
+
+    @After
+    public void setConfigBack() {
+        if (bakConfigFile.exists()) {
+            System.out.println("恢复配置文件成功");
+            copyFile(bakConfigFile, configFile);
+            assertTrue(bakConfigFile.delete());
+        } else {
+            configFile.deleteOnExit();
+        }
+    }
+
+    @Before
+    public void recordConfig() {
+        File path = new File(LocalFileServiceImpl.getPath(System.getProperty(ConfigKeyEnum.BLOG_FILE_PATH.getKey())));
+        if (!path.exists() && !path.mkdirs()) {
+            fail("创建失败");
+        }
+        this.bakConfigFile = new File(DruidConfig.DB_CONFIG_PATH + ".bak");
+        this.configFile = new File(DruidConfig.DB_CONFIG_PATH);
+        if (configFile.exists()) {
+            System.out.println("备份文件成功");
+            copyFile(configFile, bakConfigFile);
+        }
+        when(this.env.getActiveProfiles()).thenReturn(new String[]{"dev"});
+        when(this.env.getProperty(DruidConfig.DB_CONFIG_PASSWORD_PREFIX)).thenReturn(globalEnvironment.getProperty(DruidConfig.DB_CONFIG_PASSWORD_PREFIX));
+        when(this.env.getProperty(DruidConfig.DB_CONFIG_URL_PREFIX)).thenReturn(globalEnvironment.getProperty(DruidConfig.DB_CONFIG_URL_PREFIX));
+        when(this.env.getProperty(DruidConfig.DB_CONFIG_USERNAME_PREFIX)).thenReturn(globalEnvironment.getProperty(DruidConfig.DB_CONFIG_USERNAME_PREFIX));
+        when(this.env.getProperty(DruidConfig.DB_CONFIG_DRIVER_CLASS_NAME_PREFIX)).thenReturn(globalEnvironment.getProperty(DruidConfig.DB_CONFIG_DRIVER_CLASS_NAME_PREFIX));
+    }
+
+    /**
+     * 复制文件
+     *
+     * @param src
+     * @param dist
+     * @return
+     */
+    private boolean copyFile(File src, File dist) {
+        try {
+            // 复制备份文件
+            FileOutputStream fos = new FileOutputStream(dist);
+            FileInputStream fis = new FileInputStream(src);
+            int len = 0;
+            byte[] bytes = new byte[1024];
+            while ((len = fis.read(bytes)) != -1) {
+                fos.write(bytes, 0, len);
+            }
+            fos.close();
+            fis.close();
+            return true;
+        } catch (IOException e) {
+            return false;
+        }
+    }
+}

src/test/java/cn/celess/blog/controller/ArticleControllerTest.java

@@ -9,29 +9,30 @@ import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.request.ArticleReq;
 import cn.celess.blog.mapper.ArticleMapper;
 import com.fasterxml.jackson.core.type.TypeReference;
-import com.fasterxml.jackson.databind.ObjectMapper;
-import net.sf.json.JSONObject;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
 import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
 
+import java.util.List;
 
-import java.util.*;
-
+import static cn.celess.blog.enmu.ResponseEnum.*;
 import static org.junit.Assert.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
-import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
-import static cn.celess.blog.enmu.ResponseEnum.*;
 
 public class ArticleControllerTest extends BaseTest {
     @Autowired
     ArticleMapper articleMapper;
+    private static final TypeReference<?> ARTICLE_MODEL_TYPE = new TypeReference<Response<ArticleModel>>() {
+    };
+    private static final TypeReference<?> ARTICLE_MODEL_PAGE_TYPE = new TypeReference<Response<PageData<ArticleModel>>>() {
+    };
 
     @Test
     public void create() {
         ArticleReq articleReq = new ArticleReq();
         // 应该正常通过
-        articleReq.setTitle("test-" + UUID.randomUUID().toString());
+        articleReq.setTitle("test-" + randomStr());
         articleReq.setMdContent("# test title");
         articleReq.setCategory("随笔");
         String[] tagList = {"tag", "category"};
@@ -39,58 +40,29 @@ public class ArticleControllerTest extends BaseTest {
         articleReq.setOpen(true);
         articleReq.setType(true);
         articleReq.setUrl("http://xxxx.com");
-        JSONObject jsonObject = JSONObject.fromObject(articleReq);
+        MockHttpServletRequestBuilder post = post("/admin/article/create");
 
         try {
-            // 未登录
-            mockMvc.perform(post("/admin/article/create")
-                    .content(jsonObject.toString())
-                    .contentType("application/json"))
-                    .andExpect(status().isOk())
-                    .andDo(result -> {
-                        assertEquals(HAVE_NOT_LOG_IN.getCode(),
-                                JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)
-                        );
-                    });
-            // User权限
-            String token = userLogin();
-            mockMvc.perform(post("/admin/article/create")
-                    .content(jsonObject.toString())
-                    .contentType("application/json")
-                    .header("Authorization", token))
-                    .andExpect(status().isOk())
-                    .andDo(result -> {
-                        assertEquals(PERMISSION_ERROR.getCode(),
-                                JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)
-                        );
-                    });
-
-            // Admin权限
-            token = adminLogin();
-            mockMvc.perform(post("/admin/article/create")
-                    .content(jsonObject.toString())
-                    .contentType("application/json")
-                    .header("Authorization", token))
-                    .andExpect(status().isOk())
-                    .andDo(result -> {
-                        JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                        ArticleModel articleModel = (ArticleModel) JSONObject.toBean(object.getJSONObject(Result), ArticleModel.class);
-                        assertNotNull(articleModel.getId());
-                        assertNotNull(articleModel.getTitle());
-                        assertNotNull(articleModel.getSummary());
-                        assertNotNull(articleModel.getOriginal());
-                        assertNotNull(articleModel.getTags());
-                        assertNotNull(articleModel.getCategory());
-                        assertNotNull(articleModel.getPublishDateFormat());
-                        assertNotNull(articleModel.getMdContent());
-                        assertNotNull(articleModel.getPreArticle());
-                        assertNull(articleModel.getNextArticle());
-                        assertNotNull(articleModel.getOpen());
-                        assertNotNull(articleModel.getReadingNumber());
-                        assertNotNull(articleModel.getAuthor());
-                        assertNotNull(articleModel.getUrl());
-                    });
+            getMockData(post, adminLogin(), articleReq).andDo(result -> {
+                Response<ArticleModel> response = getResponse(result, ARTICLE_MODEL_TYPE);
+                assertEquals(SUCCESS.getCode(), response.getCode());
+                assertNotNull(response.getResult());
+                ArticleModel articleModel = response.getResult();
+                assertNotNull(articleModel.getId());
+                assertNotNull(articleModel.getTitle());
+                assertNotNull(articleModel.getSummary());
+                assertNotNull(articleModel.getOriginal());
+                assertNotNull(articleModel.getTags());
+                assertNotNull(articleModel.getCategory());
+                assertNotNull(articleModel.getPublishDateFormat());
+                assertNotNull(articleModel.getMdContent());
+                assertNotNull(articleModel.getPreArticle());
+                assertNull(articleModel.getNextArticle());
+                assertNotNull(articleModel.getOpen());
+                assertNotNull(articleModel.getReadingNumber());
+                assertNotNull(articleModel.getAuthor());
+                assertNotNull(articleModel.getUrl());
+            });
         } catch (Exception e) {
             e.printStackTrace();
         }
@@ -98,33 +70,22 @@ public class ArticleControllerTest extends BaseTest {
 
     @Test
     public void delete() {
-        Article article = articleMapper.getLastestArticle();
-
+        Article article;
+        do {
+            article = articleMapper.getLastestArticle();
+            create();
+        } while (article.isDeleted());
+        assertFalse(article.isDeleted());
+        MockHttpServletRequestBuilder delete = MockMvcRequestBuilders.delete("/admin/article/del?articleID=" + article.getId());
         try {
-            // 未登录删除文章
-            mockMvc.perform(MockMvcRequestBuilders.delete("/admin/article/del?articleID=" + article.getId())
-            ).andDo(result -> {
-                assertEquals(HAVE_NOT_LOG_IN.getCode(),
-                        JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)
-                );
+            Article finalArticle = article;
+            getMockData(delete, adminLogin()).andDo(result -> {
+                Response<Boolean> response = getResponse(result, BOOLEAN_TYPE);
+                assertEquals(SUCCESS.getCode(), response.getCode());
+                // 断言删除成功
+                assertTrue(response.getResult());
+                assertTrue(articleMapper.isDeletedById(finalArticle.getId()));
             });
-            // user 权限删除文章
-            String token = userLogin();
-            mockMvc.perform(MockMvcRequestBuilders.delete("/admin/article/del?articleID=" + article.getId())
-                    .header("Authorization", token))
-                    .andDo(result -> assertEquals(PERMISSION_ERROR.getCode(),
-                            JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
-                    );
-            // admin 权限删除文章
-            token = adminLogin();
-            mockMvc.perform(MockMvcRequestBuilders.delete("/admin/article/del?articleID=" + article.getId())
-                    .header("Authorization", token))
-                    .andDo(result -> {
-                        JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                        // 断言删除成功
-                        assertTrue(object.getBoolean(Result));
-                    });
         } catch (Exception e) {
             e.printStackTrace();
         }
@@ -146,27 +107,21 @@ public class ArticleControllerTest extends BaseTest {
         articleReq.setTags(tagList);
         articleReq.setTitle("test-" + article.getTitle());
         try {
-            // Admin 权限
-            mockMvc.perform(put("/admin/article/update")
-                    .content(JSONObject.fromObject(articleReq).toString())
-                    .contentType("application/json")
-                    .header("Authorization", adminLogin()))
-                    .andExpect(status().isOk())
-                    .andDo(result -> {
-                        JSONObject jsonObject = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        assertEquals(SUCCESS.getCode(), jsonObject.getInt(Code));
-                        ArticleModel a = (ArticleModel) JSONObject.toBean(jsonObject.getJSONObject(Result), ArticleModel.class);
-                        assertEquals(articleReq.getCategory(), a.getCategory());
-                        assertEquals(articleReq.getUrl(), a.getUrl());
-                        assertEquals(articleReq.getMdContent(), a.getMdContent());
-                        assertEquals(articleReq.getTitle(), a.getTitle());
-                        assertEquals(articleReq.getType(), a.getOriginal());
-                        // Tag
-                        List<Tag> asList = a.getTags();
-                        assertEquals(3, asList.size());
-                        assertEquals(articleReq.getOpen(), a.getOpen());
-                        assertEquals(articleReq.getId(), a.getId());
-                    });
+            getMockData(put("/admin/article/update"), adminLogin(), articleReq).andDo(result -> {
+                Response<ArticleModel> response = getResponse(result, ARTICLE_MODEL_TYPE);
+                assertEquals(SUCCESS.getCode(), response.getCode());
+                ArticleModel a = response.getResult();
+                assertEquals(articleReq.getCategory(), a.getCategory());
+                assertEquals(articleReq.getUrl(), a.getUrl());
+                assertEquals(articleReq.getMdContent(), a.getMdContent());
+                assertEquals(articleReq.getTitle(), a.getTitle());
+                assertEquals(articleReq.getType(), a.getOriginal());
+                // Tag
+                List<Tag> asList = a.getTags();
+                assertEquals(3, asList.size());
+                assertEquals(articleReq.getOpen(), a.getOpen());
+                assertEquals(articleReq.getId(), a.getId());
+            });
         } catch (Exception e) {
             e.printStackTrace();
         }
@@ -176,43 +131,35 @@ public class ArticleControllerTest extends BaseTest {
     public void retrieveOneById() {
         try {
             long articleID = 3;
-            mockMvc.perform(MockMvcRequestBuilders.get("/article/articleID/" + articleID))
-                    .andExpect(status().is(200));
-            mockMvc.perform(MockMvcRequestBuilders.get("/article/articleID/" + articleID + "?update=true"))
-                    .andExpect(status().is(200));
+            getMockData(MockMvcRequestBuilders.get("/article/articleID/" + articleID));
+            getMockData(MockMvcRequestBuilders.get("/article/articleID/" + articleID + "?update=true"));
 
             // 文章不存在
-            mockMvc.perform(MockMvcRequestBuilders.get("/article/articleID/-1"))
-                    .andExpect(status().is(200))
-                    .andDo(result -> {
-                        JSONObject jsonObject = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        assertEquals(ARTICLE_NOT_EXIST.getCode(), jsonObject.getInt(Code));
-                    });
+            getMockData(MockMvcRequestBuilders.get("/article/articleID/-1"))
+                    .andDo(result -> assertEquals(ARTICLE_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode()));
 
             // 正常情况
-            mockMvc.perform(MockMvcRequestBuilders.get("/article/articleID/" + articleID + "?update=false"))
-                    .andExpect(status().is(200))
-                    .andDo(result -> {
-                        JSONObject articleJson = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        // 断言获取数据成功
-                        assertEquals(SUCCESS.getCode(), articleJson.getInt(Code));
-                        assertNotNull(articleJson.getJSONObject(Result));
+            getMockData(MockMvcRequestBuilders.get("/article/articleID/" + articleID + "?update=false")).andDo(result -> {
+                Response<ArticleModel> response = getResponse(result, ARTICLE_MODEL_TYPE);
+                // 断言获取数据成功
+                assertEquals(SUCCESS.getCode(), response.getCode());
+                assertNotNull(response.getResult());
 
-                        ArticleModel a = (ArticleModel) JSONObject.toBean(articleJson.getJSONObject(Result), ArticleModel.class);
-                        assertNotNull(a.getTitle());
-                        assertNotNull(a.getId());
-                        assertNotNull(a.getSummary());
-                        assertNotNull(a.getMdContent());
-                        assertNotNull(a.getUrl());
-                        assertNotNull(a.getUpdateDateFormat());
-                        assertTrue(a.getPreArticle() != null || a.getNextArticle() != null);
-                        assertNotNull(a.getReadingNumber());
-                        assertNotNull(a.getOriginal());
-                        assertNotNull(a.getPublishDateFormat());
-                        assertNotNull(a.getCategory());
-                        assertNotNull(a.getTags());
-                        assertNotNull(a.getAuthor());
-                    });
+                ArticleModel a = response.getResult();
+                assertNotNull(a.getTitle());
+                assertNotNull(a.getId());
+                assertNotNull(a.getSummary());
+                assertNotNull(a.getMdContent());
+                assertNotNull(a.getUrl());
+                assertNotNull(a.getUpdateDateFormat());
+                assertTrue(a.getPreArticle() != null || a.getNextArticle() != null);
+                assertNotNull(a.getReadingNumber());
+                assertNotNull(a.getOriginal());
+                assertNotNull(a.getPublishDateFormat());
+                assertNotNull(a.getCategory());
+                assertNotNull(a.getTags());
+                assertNotNull(a.getAuthor());
+            });
         } catch (Exception e) {
             e.printStackTrace();
         }
@@ -222,39 +169,32 @@ public class ArticleControllerTest extends BaseTest {
     public void articles() {
         try {
             // 测试不带参数访问
-            mockMvc.perform(MockMvcRequestBuilders.get("/articles"))
-                    .andExpect(status().is(200));
-
-            mockMvc.perform(MockMvcRequestBuilders.get("/articles?page=1&count=5"))
-                    .andExpect(status().is(200))
-                    .andDo(result -> {
-                        JSONObject articlesJSON = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        Response response = (Response) JSONObject.toBean(articlesJSON, Response.class);
-                        // 断言获取数据成功
-                        assertEquals(SUCCESS.getCode(), response.getCode());
-                        // 结果集非空
-                        assertNotNull(response.getResult());
-                        // 判断pageInfo是否包装完全
-                        JSONObject resultJson = JSONObject.fromObject(response.getResult());
-                        PageData<ArticleModel> pageData = (PageData<ArticleModel>) JSONObject.toBean(resultJson, PageData.class);
-                        assertNotEquals(0, pageData.getTotal());
-                        assertEquals(1, pageData.getPageNum());
-                        assertEquals(5, pageData.getPageSize());
-                        // 内容完整
-                        for (Object arc : pageData.getList()) {
-                            ArticleModel a = (ArticleModel) JSONObject.toBean(JSONObject.fromObject(arc), ArticleModel.class);
-                            assertNotNull(a.getTitle());
-                            assertNotNull(a.getId());
-                            assertNotNull(a.getSummary());
-                            assertNotNull(a.getOriginal());
-                            assertNotNull(a.getPublishDateFormat());
-                            assertNotNull(a.getCategory());
-                            assertNotNull(a.getTags());
-                            assertNotNull(a.getAuthor());
-                            assertNull(a.getOpen());
-                            assertNull(a.getMdContent());
-                        }
-                    });
+            getMockData(MockMvcRequestBuilders.get("/articles"));
+            getMockData(MockMvcRequestBuilders.get("/articles?page=1&count=5")).andDo(result -> {
+                Response<PageData<ArticleModel>> response = getResponse(result, ARTICLE_MODEL_PAGE_TYPE);
+                // 断言获取数据成功
+                assertEquals(SUCCESS.getCode(), response.getCode());
+                // 结果集非空
+                assertNotNull(response.getResult());
+                // 判断pageInfo是否包装完全
+                PageData<ArticleModel> pageData = response.getResult();
+                assertNotEquals(0, pageData.getTotal());
+                assertEquals(1, pageData.getPageNum());
+                assertEquals(5, pageData.getPageSize());
+                // 内容完整
+                for (ArticleModel a : pageData.getList()) {
+                    assertNotNull(a.getTitle());
+                    assertNotNull(a.getId());
+                    assertNotNull(a.getSummary());
+                    assertNotNull(a.getOriginal());
+                    assertNotNull(a.getPublishDateFormat());
+                    assertNotNull(a.getCategory());
+                    assertNotNull(a.getTags());
+                    assertNotNull(a.getAuthor());
+                    assertNull(a.getOpen());
+                    assertNull(a.getMdContent());
+                }
+            });
         } catch (Exception e) {
             e.printStackTrace();
         }
@@ -264,19 +204,18 @@ public class ArticleControllerTest extends BaseTest {
     public void adminArticles() {
         try {
             getMockData(get("/admin/articles?page=1&count=10")).andExpect(result ->
-                    assertEquals(HAVE_NOT_LOG_IN.getCode(), mapper.readValue(result.getResponse().getContentAsString(), Response.class).getCode())
+                    assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result, STRING_TYPE).getCode())
             );
 
             // User权限登陆
             getMockData(get("/admin/articles?page=1&count=10"), userLogin()).andDo(result ->
-                    assertEquals(PERMISSION_ERROR.getCode(), mapper.readValue(result.getResponse().getContentAsString(), Response.class).getCode())
+                    assertEquals(PERMISSION_ERROR.getCode(), getResponse(result, STRING_TYPE).getCode())
             );
             for (int i = 0; i < 2; i++) {
                 // admin权限登陆
                 int finalI = i;
                 getMockData(get("/admin/articles?page=1&count=10&deleted=" + (i == 1)), adminLogin()).andDo(result -> {
-                    Response<PageData<ArticleModel>> response = mapper.readValue(result.getResponse().getContentAsString(), new TypeReference<Response<PageData<ArticleModel>>>() {
-                    });
+                    Response<PageData<ArticleModel>> response = getResponse(result, ARTICLE_MODEL_PAGE_TYPE);
                     assertEquals(SUCCESS.getCode(), response.getCode());
                     assertNotNull(response.getResult());
                     // 判断pageInfo是否包装完全
@@ -309,27 +248,22 @@ public class ArticleControllerTest extends BaseTest {
         try {
             // 分类不存在
             String categoryName = "NoSuchCategory";
-            mockMvc.perform(MockMvcRequestBuilders.get("/articles/category/" + categoryName + "?page=1&count=10"))
-                    .andExpect(status().is(200))
-                    .andDo(result -> {
-                        assertEquals(CATEGORY_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                    });
+            getMockData(MockMvcRequestBuilders.get("/articles/category/" + categoryName + "?page=1&count=10"))
+                    .andDo(result -> assertEquals(CATEGORY_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode()));
             // 正常查询
             categoryName = "linux";
-            mockMvc.perform(MockMvcRequestBuilders.get("/articles/category/" + categoryName + "?page=1&count=10"))
-                    .andExpect(status().is(200))
+            getMockData(MockMvcRequestBuilders.get("/articles/category/" + categoryName + "?page=1&count=10"))
                     .andDo(result -> {
-                        JSONObject jsonObject = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        assertEquals(SUCCESS.getCode(), jsonObject.getInt(Code));
-                        PageData<ArticleModel> pageData = (PageData<ArticleModel>) JSONObject.toBean(jsonObject.getJSONObject(Result), PageData.class);
+                        Response<PageData<ArticleModel>> response = getResponse(result, ARTICLE_MODEL_PAGE_TYPE);
+                        assertEquals(SUCCESS.getCode(), response.getCode());
+                        PageData<ArticleModel> pageData = response.getResult();
                         assertNotEquals(0, pageData.getTotal());
                         assertEquals(1, pageData.getPageNum());
                         assertEquals(10, pageData.getPageSize());
-                        for (Object arc : pageData.getList()) {
-                            JSONObject jsonObject1 = JSONObject.fromObject(arc);
-                            assertNotEquals(0, jsonObject1.getInt("id"));
-                            assertNotNull(jsonObject1.getString("title"));
-                            assertNotNull(jsonObject1.getString("summary"));
+                        for (ArticleModel arc : pageData.getList()) {
+                            assertNotEquals(0, arc.getId().longValue());
+                            assertNotNull(arc.getTitle());
+                            assertNotNull(arc.getSummary());
                         }
                     });
         } catch (Exception e) {
@@ -342,28 +276,23 @@ public class ArticleControllerTest extends BaseTest {
         try {
             // 分类不存在
             String tagName = "NoSuchTag";
-            mockMvc.perform(MockMvcRequestBuilders.get("/articles/tag/" + tagName + "?page=1&count=10"))
-                    .andExpect(status().is(200))
-                    .andDo(result -> {
-                        assertEquals(TAG_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                    });
+            getMockData(MockMvcRequestBuilders.get("/articles/tag/" + tagName + "?page=1&count=10"))
+                    .andDo(result -> assertEquals(TAG_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode()));
             // 正常查询
             tagName = "linux";
-            mockMvc.perform(MockMvcRequestBuilders.get("/articles/tag/" + tagName + "?page=1&count=10"))
-                    .andExpect(status().is(200))
+            getMockData(MockMvcRequestBuilders.get("/articles/tag/" + tagName + "?page=1&count=10"))
                     .andDo(result -> {
-                        JSONObject jsonObject = JSONObject.fromObject(result.getResponse().getContentAsString());
-                        assertEquals(SUCCESS.getCode(), jsonObject.getInt(Code));
-                        PageData<ArticleModel> pageData = (PageData<ArticleModel>) JSONObject.toBean(jsonObject.getJSONObject(Result), PageData.class);
+                        Response<PageData<ArticleModel>> response = getResponse(result, ARTICLE_MODEL_PAGE_TYPE);
+                        assertEquals(SUCCESS.getCode(), response.getCode());
+                        PageData<ArticleModel> pageData = response.getResult();
                         assertNotEquals(0, pageData.getTotal());
                         assertEquals(1, pageData.getPageNum());
                         assertEquals(10, pageData.getPageSize());
 
-                        for (Object arc : pageData.getList()) {
-                            JSONObject jsonObject1 = JSONObject.fromObject(arc);
-                            assertNotEquals(0, jsonObject1.getInt("id"));
-                            assertNotNull(jsonObject1.getString("title"));
-                            assertNotNull(jsonObject1.getString("summary"));
+                        for (ArticleModel arc : pageData.getList()) {
+                            assertNotEquals(0, arc.getId().longValue());
+                            assertNotNull(arc.getTitle());
+                            assertNotNull(arc.getSummary());
                         }
                     });
         } catch (Exception e) {

src/test/java/cn/celess/blog/controller/CategoryControllerTest.java

@@ -2,132 +2,77 @@ package cn.celess.blog.controller;
 
 import cn.celess.blog.BaseTest;
 import cn.celess.blog.entity.Category;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.model.CategoryModel;
+import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.mapper.CategoryMapper;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.core.type.TypeReference;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 
-import java.util.UUID;
-
+import static cn.celess.blog.enmu.ResponseEnum.SUCCESS;
 import static org.junit.Assert.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
-import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
-import static cn.celess.blog.enmu.ResponseEnum.*;
 
 public class CategoryControllerTest extends BaseTest {
 
     @Autowired
     CategoryMapper categoryMapper;
+    private static final TypeReference<?> CATEGORY_MODEL_TYPE = new TypeReference<Response<CategoryModel>>() {
+    };
+    private static final TypeReference<?> CATEGORY_MODEL_PAGE_TYPE = new TypeReference<Response<PageData<CategoryModel>>>() {
+    };
 
     @Test
     public void addOne() throws Exception {
         String categoryName = randomStr(4);
-        System.out.println("categoryName: ==> " + categoryName);
-        // 未登录
-        mockMvc.perform(post("/admin/category/create?name=" + categoryName)).andExpect(status().isOk())
-                .andDo(result -> {
-                    assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                });
-        // User权限
-        String token = userLogin();
-        mockMvc.perform(post("/admin/category/create?name=" + categoryName)
-                .header("Authorization", token))
-                .andExpect(status().isOk())
-                .andDo(result -> {
-                    assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                });
-        // Admin权限
-        token = adminLogin();
-        mockMvc.perform(post("/admin/category/create?name=" + categoryName)
-                .header("Authorization", token))
-                .andExpect(status().isOk())
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    CategoryModel category = (CategoryModel) JSONObject.toBean(object.getJSONObject(Result), CategoryModel.class);
-                    assertEquals(categoryName, category.getName());
-                    assertNotNull(category.getId());
-                    assertNotEquals(0, category.getArticles());
-                });
+        getMockData(post("/admin/category/create?name=" + categoryName), adminLogin()).andDo(result -> {
+            Response<CategoryModel> response = getResponse(result, CATEGORY_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            CategoryModel category = response.getResult();
+            assertEquals(categoryName, category.getName());
+            assertNotNull(category.getId());
+            assertNull(category.getArticles());
+        });
     }
 
     @Test
     public void deleteOne() throws Exception {
         Category category = categoryMapper.getLastestCategory();
-        // 未登录
-        mockMvc.perform(delete("/admin/category/del?id=" + category.getId())).andExpect(status().isOk())
-                .andDo(result -> {
-                    assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                });
-        // User权限
-        String token = userLogin();
-        mockMvc.perform(delete("/admin/category/del?id=" + category.getId())
-                .header("Authorization", token))
-                .andExpect(status().isOk())
-                .andDo(result -> {
-                    assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                });
-        // Admin权限
-        token = adminLogin();
-        mockMvc.perform(delete("/admin/category/del?id=" + category.getId())
-                .header("Authorization", token))
-                .andExpect(status().isOk())
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    assertTrue(object.getBoolean(Result));
-                });
+        getMockData(delete("/admin/category/del?id=" + category.getId()), adminLogin()).andDo(result -> {
+            Response<Boolean> response = getResponse(result, BOOLEAN_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertTrue(response.getResult());
+        });
     }
 
     @Test
     public void updateOne() throws Exception {
         Category category = categoryMapper.getLastestCategory();
         String name = randomStr(4);
-        // 未登录
-        mockMvc.perform(put("/admin/category/update?id=" + category.getId() + "&name=" + name)).andExpect(status().isOk())
-                .andDo(result -> {
-                    assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                });
-        // User权限
-        String token = userLogin();
-        mockMvc.perform(put("/admin/category/update?id=" + category.getId() + "&name=" + name)
-                .header("Authorization", token))
-                .andExpect(status().isOk())
-                .andDo(result -> {
-                    assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-                });
-        // Admin权限
-        token = adminLogin();
-        mockMvc.perform(put("/admin/category/update?id=" + category.getId() + "&name=" + name)
-                .header("Authorization", token))
-                .andExpect(status().isOk())
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    CategoryModel c = (CategoryModel) JSONObject.toBean(object.getJSONObject(Result), CategoryModel.class);
-                    assertEquals(name, c.getName());
-                    assertNotEquals(0, c.getArticles());
-                    assertNotNull(c.getId());
-                });
+        getMockData(put("/admin/category/update?id=" + category.getId() + "&name=" + name), adminLogin()).andDo(result -> {
+            //            Response<CategoryModel> response = mapper.readValue(result.getResponse().getContentAsString(), new ResponseType<Response<CategoryModel>>());
+            Response<CategoryModel> response = getResponse(result, CATEGORY_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            CategoryModel c = response.getResult();
+            assertEquals(name, c.getName());
+            assertNull(c.getArticles());
+            assertNotNull(c.getId());
+        });
     }
 
     @Test
     public void getPage() throws Exception {
-        mockMvc.perform(get("/categories")).andExpect(status().isOk())
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    JSONArray jsonArray = object.getJSONObject(Result).getJSONArray("list");
-                    assertNotNull(jsonArray);
-                    jsonArray.forEach(o -> {
-                        CategoryModel c = (CategoryModel) JSONObject.toBean(JSONObject.fromObject(o), CategoryModel.class);
-                        assertNotNull(c.getName());
-                        assertNotNull(c.getId());
-                        assertNotEquals(0, c.getArticles());
-                    });
-                });
+        getMockData(get("/categories")).andDo(result -> {
+            Response<PageData<CategoryModel>> response = getResponse(result, CATEGORY_MODEL_PAGE_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
+            response.getResult().getList().forEach(c -> {
+                assertNotNull(c.getName());
+                assertNotNull(c.getId());
+                assertNotNull(c.getArticles());
+            });
+        });
 
     }
 }

src/test/java/cn/celess/blog/controller/CommentControllerTest.java

@@ -3,21 +3,22 @@ package cn.celess.blog.controller;
 import cn.celess.blog.BaseTest;
 import cn.celess.blog.entity.Article;
 import cn.celess.blog.entity.Comment;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.User;
 import cn.celess.blog.entity.model.CommentModel;
 import cn.celess.blog.entity.request.CommentReq;
 import cn.celess.blog.mapper.ArticleMapper;
 import cn.celess.blog.mapper.CommentMapper;
 import cn.celess.blog.mapper.UserMapper;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.core.type.TypeReference;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.http.MediaType;
 
-import java.util.UUID;
+import java.util.List;
 
+import static cn.celess.blog.enmu.ResponseEnum.DATA_IS_DELETED;
+import static cn.celess.blog.enmu.ResponseEnum.SUCCESS;
 import static org.junit.Assert.*;
-import static cn.celess.blog.enmu.ResponseEnum.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
 
 public class CommentControllerTest extends BaseTest {
@@ -27,46 +28,40 @@ public class CommentControllerTest extends BaseTest {
     UserMapper userMapper;
     @Autowired
     CommentMapper commentMapper;
+    private static final TypeReference<?> COMMENT_MODEL_TYPE = new TypeReference<Response<CommentModel>>() {
+    };
 
     @Test
     public void addOne() throws Exception {
         Article article = articleMapper.getLastestArticle();
         CommentReq commentReq = new CommentReq();
         commentReq.setPagePath("/article/" + article.getId());
-        commentReq.setContent(UUID.randomUUID().toString());
-        commentReq.setPid(-1L);
-        commentReq.setToUserId(-1L);
-        String token = userLogin();
-        mockMvc.perform(post("/user/comment/create")
-                .contentType(MediaType.APPLICATION_JSON_UTF8)
-                .content(JSONObject.fromObject(commentReq).toString())
-                .header("Authorization", token)
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            CommentModel model = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
+        commentReq.setContent(randomStr());
+        List<User> all = userMapper.findAll();
+        commentReq.setPid(1L);
+        commentReq.setToUserId(2l);
+        getMockData(post("/user/comment/create"), userLogin(), commentReq).andDo(result -> {
+            Response<CommentModel> response = getResponse(result, COMMENT_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            CommentModel model = response.getResult();
             assertNotEquals(0, model.getId());
             assertEquals(commentReq.getPid(), model.getPid().longValue());
-            assertEquals(-1, model.getPid().longValue());
+            assertEquals(1, model.getPid().longValue());
             assertEquals(commentReq.getContent(), model.getContent());
             assertNotNull(model.getDate());
             assertNotNull(model.getFromUser());
-            assertNull(model.getToUser());
+            assertNotNull(model.getToUser());
         });
 
 
         commentReq.setPagePath("/article/" + article.getId());
-        commentReq.setContent(UUID.randomUUID().toString());
+        commentReq.setContent(randomStr());
         commentReq.setPid(-1L);
         commentReq.setToUserId(2);
-        mockMvc.perform(post("/user/comment/create")
-                .contentType(MediaType.APPLICATION_JSON_UTF8)
-                .content(JSONObject.fromObject(commentReq).toString())
-                .header("Authorization", token)
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            CommentModel model = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
+        getMockData(post("/user/comment/create"), userLogin(), commentReq).andDo(result -> {
+            Response<CommentModel> response = getResponse(result, COMMENT_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            CommentModel model = response.getResult();
             // 响应数据的完整性
             assertNotEquals(0, model.getId());
             assertEquals(commentReq.getPid(), model.getPid().longValue());
@@ -79,20 +74,16 @@ public class CommentControllerTest extends BaseTest {
         });
 
         // 测试二级回复
-        Comment lastestComment = commentMapper.getLastestComment();
+        Comment latestComment = commentMapper.getLastestComment();
         commentReq.setPagePath("/article/" + article.getId());
-        commentReq.setContent(UUID.randomUUID().toString());
-        commentReq.setPid(lastestComment.getId());
-        mockMvc.perform(post("/user/comment/create")
-                .contentType(MediaType.APPLICATION_JSON_UTF8)
-                .content(JSONObject.fromObject(commentReq).toString())
-                .header("Authorization", token)
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            CommentModel model = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
+        commentReq.setContent(randomStr());
+        commentReq.setPid(latestComment.getId());
+        getMockData(post("/user/comment/create"), userLogin(), commentReq).andDo(result -> {
+            Response<CommentModel> response = getResponse(result, COMMENT_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            CommentModel model = response.getResult();
             // 重新获取父评论信息
-            Comment pCommon = commentMapper.findCommentById(lastestComment.getId());
+            Comment pCommon = commentMapper.findCommentById(latestComment.getId());
             assertEquals(pCommon.getId(), model.getPid());
         });
     }
@@ -116,16 +107,13 @@ public class CommentControllerTest extends BaseTest {
         // 接口测试
         long id = comment.getId();
         assertNotEquals(0, id);
-        String token = userLogin();
-        mockMvc.perform(delete("/user/comment/del?id=" + id).header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertTrue(object.getBoolean(Result));
-        });
-        mockMvc.perform(delete("/user/comment/del?id=" + id).header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(DATA_IS_DELETED.getCode(), object.getInt(Code));
+        getMockData(delete("/user/comment/del?id=" + id), userLogin()).andDo(result -> {
+            Response<Boolean> response = getResponse(result, BOOLEAN_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertTrue(response.getResult());
         });
+        getMockData(delete("/user/comment/del?id=" + id), userLogin())
+                .andDo(result -> assertEquals(DATA_IS_DELETED.getCode(), getResponse(result, COMMENT_MODEL_TYPE).getCode()));
     }
 
     @Test
@@ -133,16 +121,12 @@ public class CommentControllerTest extends BaseTest {
         Comment comment = commentMapper.getLastestComment();
         CommentReq commentReq = new CommentReq();
         commentReq.setId(comment.getId());
-        commentReq.setContent(UUID.randomUUID().toString());
+        commentReq.setContent(randomStr());
         // 不合法数据 setResponseId
-        mockMvc.perform(put("/user/comment/update")
-                .content(JSONObject.fromObject(commentReq).toString())
-                .contentType(MediaType.APPLICATION_JSON_UTF8)
-                .header("Authorization", userLogin())
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            CommentModel c = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
+        getMockData(put("/user/comment/update"), userLogin(), commentReq).andDo(result -> {
+            Response<CommentModel> response = getResponse(result, COMMENT_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            CommentModel c = response.getResult();
             assertEquals(commentReq.getContent(), c.getContent());
         });
     }

src/test/java/cn/celess/blog/controller/LinksControllerTest.java

@@ -2,42 +2,49 @@ package cn.celess.blog.controller;
 
 import cn.celess.blog.BaseTest;
 import cn.celess.blog.entity.PartnerSite;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.model.PageData;
+import cn.celess.blog.entity.request.LinkApplyReq;
 import cn.celess.blog.entity.request.LinkReq;
+import cn.celess.blog.exception.MyException;
 import cn.celess.blog.mapper.PartnerMapper;
-import com.github.pagehelper.PageInfo;
-import net.sf.json.JSONObject;
+import cn.celess.blog.service.PartnerSiteService;
+import com.fasterxml.jackson.core.type.TypeReference;
+import lombok.extern.slf4j.Slf4j;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.http.MediaType;
 
-import java.util.UUID;
+import java.util.List;
 
 import static cn.celess.blog.enmu.ResponseEnum.*;
 import static org.junit.Assert.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
 
+@Slf4j
 public class LinksControllerTest extends BaseTest {
 
     @Autowired
     PartnerMapper mapper;
+    private static final TypeReference<?> LINK_MODEL_TYPE = new TypeReference<Response<PartnerSite>>() {
+    };
+    private static final TypeReference<?> LINK_MODEL_LIST_TYPE = new TypeReference<Response<List<PartnerSite>>>() {
+    };
+    private static final TypeReference<?> LINK_MODEL_PAGE_TYPE = new TypeReference<Response<PageData<PartnerSite>>>() {
+    };
+    @Autowired
+    PartnerSiteService partnerSiteService;
+
 
     @Test
     public void create() throws Exception {
         LinkReq linkReq = new LinkReq();
-        linkReq.setName(UUID.randomUUID().toString().substring(0, 4));
+        linkReq.setName(randomStr(4));
         linkReq.setOpen(false);
-        linkReq.setUrl("https://example.com");
-        String token = adminLogin();
-        mockMvc.perform(
-                post("/admin/links/create")
-                        .content(JSONObject.fromObject(linkReq).toString())
-                        .header("Authorization", token)
-                        .contentType(MediaType.APPLICATION_JSON)
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            PartnerSite site = (PartnerSite) JSONObject.toBean(object.getJSONObject(Result), PartnerSite.class);
+        linkReq.setUrl("https://" + randomStr(4) + "example.com");
+        getMockData(post("/admin/links/create"), adminLogin(), linkReq).andDo(result -> {
+            Response<PartnerSite> response = getResponse(result, LINK_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            PartnerSite site = response.getResult();
             assertNotNull(site.getId());
             assertEquals(linkReq.getName(), site.getName());
             assertEquals(linkReq.getUrl(), site.getUrl());
@@ -45,54 +52,45 @@ public class LinksControllerTest extends BaseTest {
         });
 
         // https/http
-        linkReq.setName(UUID.randomUUID().toString().substring(0, 4));
+        linkReq.setName(randomStr(4));
         linkReq.setOpen(false);
-        linkReq.setUrl("example.com");
-        mockMvc.perform(
-                post("/admin/links/create")
-                        .content(JSONObject.fromObject(linkReq).toString())
-                        .header("Authorization", token)
-                        .contentType("application/json")
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            PartnerSite site = (PartnerSite) JSONObject.toBean(object.getJSONObject(Result), PartnerSite.class);
-            assertEquals("http://example.com", site.getUrl());
+        linkReq.setUrl(randomStr(4) + ".example.com");
+        getMockData(post("/admin/links/create"), adminLogin(), linkReq).andDo(result -> {
+            Response<PartnerSite> response = getResponse(result, LINK_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            PartnerSite site = response.getResult();
+            assertEquals("http://" + linkReq.getUrl(), site.getUrl());
         });
 
         // 测试已存在的数据
-        mockMvc.perform(
-                post("/admin/links/create")
-                        .content(JSONObject.fromObject(linkReq).toString())
-                        .header("Authorization", token)
-                        .contentType("application/json")
-        ).andDo(result -> assertEquals(DATA_HAS_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
+        getMockData(post("/admin/links/create"), adminLogin(), linkReq).andDo(result ->
+                assertEquals(DATA_HAS_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode())
+        );
     }
 
     @Test
     public void del() throws Exception {
         PartnerSite partnerSite = new PartnerSite();
-        partnerSite.setName(UUID.randomUUID().toString().substring(0, 4));
+        partnerSite.setName(randomStr(4));
         partnerSite.setOpen(true);
         partnerSite.setDesc("");
         partnerSite.setIconPath("");
-        partnerSite.setUrl("https://www.celess.cn");
+        partnerSite.setUrl("https://" + randomStr(4) + ".celess.cn");
         mapper.insert(partnerSite);
-        PartnerSite lastest = mapper.getLastest();
-        assertNotNull(lastest.getId());
-        String token = adminLogin();
-        mockMvc.perform(delete("/admin/links/del/" + lastest.getId()).header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertTrue(object.getBoolean(Result));
+        PartnerSite latest = mapper.getLastest();
+        assertNotNull(latest.getId());
+        getMockData(delete("/admin/links/del/" + latest.getId()), adminLogin()).andDo(result -> {
+            Response<Boolean> response = getResponse(result, BOOLEAN_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertTrue(response.getResult());
         });
-        long id = lastest.getId();
+        long id = latest.getId();
         do {
             id += 1;
         } while (mapper.existsById(id));
         System.out.println("删除ID=" + id + "的数据");
-        mockMvc.perform(delete("/admin/links/del/" + id).header("Authorization", token)).andDo(result ->
-                assertEquals(DATA_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
+        getMockData(delete("/admin/links/del/" + id), adminLogin()).andDo(result ->
+                assertEquals(DATA_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode())
         );
     }
 
@@ -100,32 +98,27 @@ public class LinksControllerTest extends BaseTest {
     public void update() throws Exception {
         // 增数据
         PartnerSite partnerSite = new PartnerSite();
-        partnerSite.setName(UUID.randomUUID().toString().substring(0, 4));
+        partnerSite.setName(randomStr(4));
         partnerSite.setOpen(true);
         partnerSite.setDesc("");
         partnerSite.setIconPath("");
         partnerSite.setDelete(false);
-        partnerSite.setUrl("https://www.celess.cn");
+        partnerSite.setUrl("https://" + randomStr(5) + ".celess.cn");
         mapper.insert(partnerSite);
         // 查数据
-        PartnerSite lastest = mapper.getLastest();
-        assertNotNull(lastest.getId());
-        String token = adminLogin();
+        PartnerSite latest = mapper.getLastest();
+        assertNotNull(latest.getId());
         // 构建请求
         LinkReq linkReq = new LinkReq();
-        linkReq.setUrl(lastest.getUrl());
-        linkReq.setOpen(!lastest.getOpen());
-        linkReq.setName(UUID.randomUUID().toString().substring(0, 4));
-        linkReq.setId(lastest.getId());
-        mockMvc.perform(
-                put("/admin/links/update")
-                        .content(JSONObject.fromObject(linkReq).toString())
-                        .contentType(MediaType.APPLICATION_JSON)
-                        .header("Authorization", token)
-        ).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            PartnerSite site = (PartnerSite) JSONObject.toBean(object.getJSONObject(Result), PartnerSite.class);
+        linkReq.setUrl(latest.getUrl());
+        linkReq.setOpen(!latest.getOpen());
+        linkReq.setName(randomStr(4));
+        linkReq.setId(latest.getId());
+
+        getMockData(put("/admin/links/update"), adminLogin(), linkReq).andDo(result -> {
+            Response<PartnerSite> response = getResponse(result, LINK_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            PartnerSite site = response.getResult();
             assertNotNull(site.getId());
             assertEquals(linkReq.getId(), site.getId().longValue());
             assertEquals(linkReq.getUrl(), site.getUrl());
@@ -136,11 +129,10 @@ public class LinksControllerTest extends BaseTest {
 
     @Test
     public void allForOpen() throws Exception {
-        mockMvc.perform(get("/links")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            object.getJSONArray(Result).forEach(o -> {
-                PartnerSite site = (PartnerSite) JSONObject.toBean(JSONObject.fromObject(o), PartnerSite.class);
+        getMockData(get("/links")).andDo(result -> {
+            Response<List<PartnerSite>> response = getResponse(result, LINK_MODEL_LIST_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            response.getResult().forEach(site -> {
                 assertNotNull(site.getUrl());
                 assertNull(site.getOpen());
                 assertNotNull(site.getName());
@@ -150,14 +142,13 @@ public class LinksControllerTest extends BaseTest {
 
     @Test
     public void all() throws Exception {
-        mockMvc.perform(get("/admin/links?page=1&count=10").header("Authorization", adminLogin())).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            PageData<PartnerSite> pageData = (PageData<PartnerSite>) JSONObject.toBean(object.getJSONObject(Result), PageData.class);
+        getMockData(get("/admin/links?page=1&count=10"), adminLogin()).andDo(result -> {
+            Response<PageData<PartnerSite>> response = getResponse(result, LINK_MODEL_PAGE_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            PageData<PartnerSite> pageData = response.getResult();
             assertEquals(1, pageData.getPageNum());
             assertEquals(10, pageData.getPageSize());
-            for (Object o : pageData.getList()) {
-                PartnerSite site = (PartnerSite) JSONObject.toBean(JSONObject.fromObject(o), PartnerSite.class);
+            for (PartnerSite site : pageData.getList()) {
                 assertNotNull(site.getUrl());
                 assertNotNull(site.getName());
                 assertNotNull(site.getOpen());
@@ -165,19 +156,83 @@ public class LinksControllerTest extends BaseTest {
         });
     }
 
-    // 手动测试
-    // @Test
-    public void apply() throws Exception {
-        long l = System.currentTimeMillis();
-        String url = "https://www.example.com";
-        mockMvc.perform(post("/apply?name=小海博客Api测试，请忽略&url=" + url)).andDo(result -> {
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-        });
-        System.out.println("耗时：" + (System.currentTimeMillis() - l) / 1000 + "s");
-        url = "xxxxxxxxxm";
-        mockMvc.perform(post("/apply?name=小海博客Api测试，请忽略&url=" + url)).andDo(result -> {
-            assertEquals(PARAMETERS_URL_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-        });
+    @Test
+    public void apply() {
+        // 做service 层的测试
+        //        mockEmailServiceInstance(partnerSiteService, "mailService");
+        mockInjectInstance(partnerSiteService, "mailService", new TestMailServiceImpl());
+        LinkApplyReq req = new LinkApplyReq();
+        req.setName(randomStr(4));
+        req.setUrl("https://" + randomStr(4) + ".celess.cn");
+        req.setIconPath("https://www.celess.cn/example.png");
+        req.setDesc("desc :" + randomStr());
+        req.setEmail(randomStr(4) + "@celess.cn");
+        req.setLinkUrl(req.getUrl() + "/links");
+        try {
+            // 抓取不到数据的链接
+            partnerSiteService.apply(req);
+        } catch (MyException e) {
+            log.debug("测试抓取不到数据");
+            assertEquals(CANNOT_GET_DATA.getCode(), e.getCode());
+        }
 
+        req.setLinkUrl("https://bing.com");
+        req.setUrl(req.getLinkUrl());
+        try {
+            partnerSiteService.apply(req);
+        } catch (MyException e) {
+            log.debug("测试未添加本站链接的友链申请");
+            assertEquals(APPLY_LINK_NO_ADD_THIS_SITE.getCode(), e.getCode());
+            assertNotNull(e.getResult());
+            try {
+                // 测试uuid一致性
+                log.debug("测试uuid一致性");
+                partnerSiteService.apply(req);
+            } catch (MyException e2) {
+                assertEquals(e.getResult(), e2.getResult());
+            }
+        }
+        log.debug("测试正常申请");
+        req.setLinkUrl("https://www.celess.cn");
+        req.setUrl(req.getLinkUrl());
+        PartnerSite apply = partnerSiteService.apply(req);
+        assertNotNull(apply);
+        assertNotNull(apply.getId());
+    }
+
+    @Test
+    public void reapply() {
+        //mockEmailServiceInstance(partnerSiteService, "mailService");
+        try {
+            partnerSiteService.reapply(randomStr());
+            throw new AssertionError();
+        } catch (MyException e) {
+            assertEquals(DATA_EXPIRED.getCode(), e.getCode());
+        }
+
+        LinkApplyReq req = new LinkApplyReq();
+        req.setName(randomStr(4));
+        req.setIconPath("https://www.celess.cn/example.png");
+        req.setDesc("desc :" + randomStr());
+        req.setEmail(randomStr(4) + "@celess.cn");
+        req.setLinkUrl("https://bing.com");
+        req.setUrl(req.getLinkUrl());
+        String uuid;
+        try {
+            partnerSiteService.apply(req);
+            // err here
+            throw new AssertionError();
+        } catch (MyException e) {
+            uuid = (String) e.getResult();
+            String reapply = partnerSiteService.reapply(uuid);
+            assertEquals(reapply, "success");
+        }
+
+        try {
+            partnerSiteService.reapply(uuid);
+            throw new AssertionError();
+        } catch (MyException e) {
+            assertEquals(DATA_EXPIRED.getCode(), e.getCode());
+        }
     }
 }

src/test/java/cn/celess/blog/controller/TagControllerTest.java

@@ -1,36 +1,42 @@
 package cn.celess.blog.controller;
 
 import cn.celess.blog.BaseTest;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.Tag;
 import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.model.TagModel;
 import cn.celess.blog.mapper.TagMapper;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.core.type.TypeReference;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 
-import java.util.UUID;
+import java.util.List;
+import java.util.Map;
 
+import static cn.celess.blog.enmu.ResponseEnum.*;
 import static org.junit.Assert.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
-import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
-import static cn.celess.blog.enmu.ResponseEnum.*;
 
 public class TagControllerTest extends BaseTest {
     @Autowired
     TagMapper tagMapper;
+    private static final TypeReference<?> TAG_MODEL_TYPE = new TypeReference<Response<TagModel>>() {
+
+    };
+    private static final TypeReference<?> TAG_MODEL_PAGE_TYPE = new TypeReference<Response<PageData<TagModel>>>() {
+    };
+    private static final TypeReference<?> TAG_NAC_LIST_TYPE = new TypeReference<Response<List<Map<String, Object>>>>() {
+    };
 
     @Test
     public void addOne() throws Exception {
         String name = randomStr(4);
-        mockMvc.perform(post("/admin/tag/create?name=" + name)).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(post("/admin/tag/create?name=" + name).header("authorization", userLogin())).andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(post("/admin/tag/create?name=" + name).header("authorization", adminLogin())).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            JSONObject resJson = object.getJSONObject(Result);
-            TagModel tag = (TagModel) JSONObject.toBean(resJson, TagModel.class);
+        getMockData(post("/admin/tag/create?name=" + name)).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result, STRING_TYPE).getCode()));
+        getMockData(post("/admin/tag/create?name=" + name), userLogin()).andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), getResponse(result, STRING_TYPE).getCode()));
+        getMockData(post("/admin/tag/create?name=" + name), adminLogin()).andDo(result -> {
+            Response<TagModel> response = getResponse(result, TAG_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            TagModel tag = response.getResult();
             assertNotNull(tag.getId());
             assertEquals(name, tag.getName());
         });
@@ -42,16 +48,14 @@ public class TagControllerTest extends BaseTest {
     public void delOne() throws Exception {
         Tag lastestTag = tagMapper.getLastestTag();
         assertNotNull(lastestTag.getId());
-        String token = adminLogin();
-        mockMvc.perform(delete("/admin/tag/del?id=" + lastestTag.getId()).header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertTrue(object.getBoolean(Result));
+        getMockData(delete("/admin/tag/del?id=" + lastestTag.getId()), adminLogin()).andDo(result -> {
+            Response<Boolean> response = getResponse(result, BOOLEAN_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertTrue(response.getResult());
         });
         long id = lastestTag.getId() * 2;
-        mockMvc.perform(delete("/admin/tag/del?id=" + id).header("Authorization", token)).andDo(result ->
-                assertEquals(TAG_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
-        );
+        getMockData(delete("/admin/tag/del?id=" + id), adminLogin())
+                .andDo(result -> assertEquals(TAG_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode()));
 
     }
 
@@ -59,12 +63,12 @@ public class TagControllerTest extends BaseTest {
     public void updateOne() throws Exception {
         Tag tag = tagMapper.getLastestTag();
         assertNotNull(tag.getId());
-        String name = UUID.randomUUID().toString().substring(0, 4);
-        mockMvc.perform(put("/admin/tag/update?id=" + tag.getId() + "&name=" + name).header("Authorization", adminLogin())).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertNotNull(object.getJSONObject(Result));
-            TagModel t = (TagModel) JSONObject.toBean(object.getJSONObject(Result), TagModel.class);
+        String name = randomStr(4);
+        getMockData(put("/admin/tag/update?id=" + tag.getId() + "&name=" + name), adminLogin()).andDo(result -> {
+            Response<TagModel> response = getResponse(result, TAG_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
+            TagModel t = response.getResult();
             assertEquals(name, t.getName());
             assertEquals(tag.getId(), t.getId());
         });
@@ -73,40 +77,33 @@ public class TagControllerTest extends BaseTest {
 
     @Test
     public void getPage() throws Exception {
-        mockMvc.perform(get("/tags?page=1&count=5"))
-                .andExpect(status().is(200))
-                .andDo(result -> {
-                    JSONObject articlesJSON = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    // 断言获取数据成功
-                    assertEquals(SUCCESS.getCode(), articlesJSON.getInt(Code));
-                    // 结果集非空
-                    assertNotNull(articlesJSON.getJSONObject(Result));
-                    // 判断pageInfo是否包装完全
-                    JSONObject resultJson = JSONObject.fromObject(articlesJSON.getJSONObject(Result));
-                    PageData<TagModel> pageData = (PageData<TagModel>) JSONObject.toBean(resultJson, PageData.class);
-                    assertNotEquals(0, pageData.getTotal());
-                    assertEquals(1, pageData.getPageNum());
-                    assertEquals(5, pageData.getPageSize());
-                    // 内容完整
-                    for (Object tag : pageData.getList()) {
-                        TagModel t = (TagModel) JSONObject.toBean(JSONObject.fromObject(tag), TagModel.class);
-                        assertNotNull(t.getId());
-                        assertNotNull(t.getName());
-                    }
-                });
+        getMockData(get("/tags?page=1&count=5")).andDo(result -> {
+            Response<PageData<TagModel>> response = getResponse(result, TAG_MODEL_PAGE_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            // 结果集非空
+            assertNotNull(response.getResult());
+            // 判断pageInfo是否包装完全
+            PageData<TagModel> pageData = response.getResult();
+            assertNotEquals(0, pageData.getTotal());
+            assertEquals(1, pageData.getPageNum());
+            assertEquals(5, pageData.getPageSize());
+            // 内容完整
+            for (TagModel t : pageData.getList()) {
+                assertNotNull(t.getId());
+                assertNotNull(t.getName());
+            }
+        });
     }
 
     @Test
     public void getTagNameAndCount() throws Exception {
-        mockMvc.perform(get("/tags/nac")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            JSONArray jsonArray = object.getJSONArray(Result);
-            assertNotNull(jsonArray);
-            jsonArray.forEach(o -> {
-                JSONObject json = JSONObject.fromObject(o);
-                assertTrue(json.containsKey("size"));
-                assertTrue(json.containsKey("name"));
+        getMockData(get("/tags/nac")).andDo(result -> {
+            Response<List<Map<String, Object>>> response = getResponse(result, TAG_NAC_LIST_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
+            response.getResult().forEach(o -> {
+                assertNotNull(o.get("name"));
+                assertNotNull(o.get("size"));
             });
         });
     }

src/test/java/cn/celess/blog/controller/UserControllerTest.java

@@ -1,36 +1,57 @@
 package cn.celess.blog.controller;
 
 import cn.celess.blog.BaseTest;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.User;
 import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.model.UserModel;
 import cn.celess.blog.entity.request.LoginReq;
 import cn.celess.blog.entity.request.UserReq;
 import cn.celess.blog.mapper.UserMapper;
+import cn.celess.blog.service.UserService;
+import cn.celess.blog.service.FileService;
 import cn.celess.blog.util.MD5Util;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
+import cn.celess.blog.util.RedisUtil;
+import com.fasterxml.jackson.core.type.TypeReference;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.MediaType;
 import org.springframework.mock.web.MockMultipartFile;
+import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
 import org.springframework.util.LinkedMultiValueMap;
 import org.springframework.util.MultiValueMap;
 
 import java.io.InputStream;
 import java.net.HttpURLConnection;
 import java.net.URL;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Random;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
+import static cn.celess.blog.enmu.ResponseEnum.*;
 import static org.junit.Assert.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
-import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
-import static cn.celess.blog.enmu.ResponseEnum.*;
+
 
 public class UserControllerTest extends BaseTest {
 
     @Autowired
     UserMapper userMapper;
+    @Autowired
+
+    RedisUtil redisUtil;
+    private static final TypeReference<?> USER_MODEL_TYPE = new TypeReference<Response<UserModel>>() {
+    };
+    private static final TypeReference<?> USER_MODEL_PAGE_TYPE = new TypeReference<Response<PageData<UserModel>>>() {
+    };
+    private static final TypeReference<?> USER_MODEL_LIST_TYPE = new TypeReference<Response<List<Map<String, Object>>>>() {
+    };
+    @Autowired
+    UserService userService;
+
 
     @Test
     public void login() throws Exception {
@@ -41,11 +62,7 @@ public class UserControllerTest extends BaseTest {
         req.setEmail("zh@celess.cn");
         req.setPassword("123456789");
         req.setIsRememberMe(false);
-        JSONObject loginReq = JSONObject.fromObject(req);
-        mockMvc.perform(post("/login").content(loginReq.toString()).contentType("application/json"))
-                .andDo(result ->
-                        assertEquals(USER_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
-                );
+        getMockData(post("/login"), null, req).andDo(result -> assertEquals(USER_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode()));
     }
 
     @Test
@@ -55,41 +72,39 @@ public class UserControllerTest extends BaseTest {
 
     @Test
     public void logout() throws Exception {
-        mockMvc.perform(get("/logout")).andDo(result -> assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(get("/logout").header("Authorization", userLogin())).andDo(result -> assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
+        getMockData(get("/logout")).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result, STRING_TYPE).getCode()));
+        getMockData(get("/logout"), adminLogin()).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
     }
 
     @Test
     public void updateInfo() throws Exception {
-        String desc = UUID.randomUUID().toString().substring(0, 4);
-        String disPlayName = UUID.randomUUID().toString().substring(0, 4);
-        mockMvc.perform(put("/user/userInfo/update?desc=" + desc + "&displayName=" + disPlayName).header("Authorization", userLogin()))
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    UserModel u = (UserModel) JSONObject.toBean(object.getJSONObject(Result), UserModel.class);
-                    assertEquals(desc, u.getDesc());
-                    assertEquals(disPlayName, u.getDisplayName());
-                    assertNotNull(u.getId());
-                });
+        String desc = randomStr(4);
+        String disPlayName = randomStr(4);
+        getMockData(put("/user/userInfo/update?desc=" + desc + "&displayName=" + disPlayName), userLogin()).andDo(result -> {
+            Response<UserModel> response = getResponse(result, USER_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            UserModel u = response.getResult();
+            assertEquals(desc, u.getDesc());
+            assertEquals(disPlayName, u.getDisplayName());
+            assertNotNull(u.getId());
+        });
     }
 
     @Test
     public void getUserInfo() throws Exception {
-        mockMvc.perform(get("/user/userInfo").header("Authorization", userLogin()))
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    UserModel u = (UserModel) JSONObject.toBean(object.getJSONObject(Result), UserModel.class);
-                    assertNotNull(u.getId());
-                    assertNotNull(u.getEmail());
-                    assertNotNull(u.getDisplayName());
-                    assertNotNull(u.getEmailStatus());
-                    assertNotNull(u.getAvatarImgUrl());
-                    assertNotNull(u.getDesc());
-                    assertNotNull(u.getRecentlyLandedDate());
-                    assertNotNull(u.getRole());
-                });
+        getMockData(get("/user/userInfo"), adminLogin()).andDo(result -> {
+            Response<UserModel> response = getResponse(result, USER_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            UserModel u = response.getResult();
+            assertNotNull(u.getId());
+            assertNotNull(u.getEmail());
+            assertNotNull(u.getDisplayName());
+            assertNotNull(u.getEmailStatus());
+            assertNotNull(u.getAvatarImgUrl());
+            assertNotNull(u.getDesc());
+            assertNotNull(u.getRecentlyLandedDate());
+            assertNotNull(u.getRole());
+        });
     }
 
     @Test
@@ -99,12 +114,15 @@ public class UserControllerTest extends BaseTest {
         connection.setRequestMethod("GET");
         InputStream inputStream = connection.getInputStream();
         assertNotNull(inputStream);
+
+        // mock 实现类
+        mockInjectInstance(userService, "fileService", (FileService) TestFileManager::new);
+
         MockMultipartFile file = new MockMultipartFile("file", "logo.png", MediaType.IMAGE_PNG_VALUE, inputStream);
-        mockMvc.perform(multipart("/user/imgUpload").file(file)).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(multipart("/user/imgUpload").file(file).header("Authorization", userLogin())).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertNotNull(object.getString(Result));
+        getMockData(multipart("/user/imgUpload").file(file), userLogin()).andDo(result -> {
+            Response<Object> response = getResponse(result, OBJECT_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
         });
     }
 
@@ -119,20 +137,41 @@ public class UserControllerTest extends BaseTest {
     }
 
     @Test
-    public void emailVerify() {
-        // ignore
+    public void emailVerify() throws Exception {
+        String email = randomStr(4) + "@celess.cn";
+        String pwd = MD5Util.getMD5("123456789");
+        userMapper.addUser(new User(email, pwd));
+        String verifyId = randomStr();
+        LoginReq req = new LoginReq(email, "123456789", true);
+        redisUtil.setEx(email + "-verify", verifyId, 2, TimeUnit.DAYS);
+        getMockData(post("/emailVerify").param("verifyId", verifyId).param("email", email), login(req)).andDo(result ->
+                assertEquals(SUCCESS.getCode(), getResponse(result, OBJECT_TYPE).getCode())
+        );
     }
 
     @Test
-    public void resetPwd() {
-        // ignore
+    public void resetPwd() throws Exception {
+        String email = randomStr(4) + "@celess.cn";
+        String pwd = MD5Util.getMD5("1234567890");
+        userMapper.addUser(new User(email, pwd));
+        LoginReq req = new LoginReq(email, "1234567890", true);
+        String verifyId = randomStr();
+        // 设置验证id
+        redisUtil.setEx(email + "-resetPwd", verifyId, 2, TimeUnit.DAYS);
+        MockHttpServletRequestBuilder resetPwd = post("/resetPwd").param("verifyId", verifyId).param("email", email).param("pwd", "123456789");
+        //  未验证
+        getMockData(resetPwd, login(req)).andDo(result -> assertEquals(USEREMAIL_NOT_VERIFY.getCode(), getResponse(result, OBJECT_TYPE).getCode()));
+        // 设置未验证
+        userMapper.updateEmailStatus(email, true);
+        // 正常
+        getMockData(resetPwd, login(req)).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result, OBJECT_TYPE).getCode()));
     }
 
     @Test
     public void multipleDelete() throws Exception {
         List<User> userList = new ArrayList<>();
         for (int i = 0; i < 10; i++) {
-            String s = UUID.randomUUID().toString();
+            String s = randomStr();
             String email = s.substring(s.length() - 4) + "@celess.cn";
             String pwd = MD5Util.getMD5("123456789");
             User user = new User(email, pwd);
@@ -146,130 +185,93 @@ public class UserControllerTest extends BaseTest {
                 userMapper.setUserRole(userMapper.findByEmail(email).getId(), "admin");
             }
         }
-        List<Long> idList = new ArrayList<>();
-        userList.forEach(user -> idList.add(user.getId()));
-        System.out.println("id :: == > " + idList.toString());
-        mockMvc.perform(delete("/admin/user/delete").content(idList.toString()).contentType("application/json"))
-                .andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(delete("/admin/user/delete").content(idList.toString()).contentType("application/json").header("Authorization", userLogin()))
-                .andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(delete("/admin/user/delete").content(idList.toString()).contentType("application/json").header("Authorization", adminLogin()))
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    JSONArray jsonArray = object.getJSONArray(Result);
-                    jsonArray.forEach(o -> {
-                        JSONObject json = JSONObject.fromObject(o);
-                        // 判断响应数据中是否包含输入的id
-                        assertTrue(idList.contains((long) json.getInt("id")));
-                        // 判断处理状态
-                        boolean status = json.getBoolean("status");
-                        if (json.containsKey("msg"))
-                            assertFalse(status);
-                        else
-                            assertTrue(status);
-                    });
-                });
+        List<Integer> idList = userList.stream().map(user -> user.getId().intValue()).collect(Collectors.toList());
+        getMockData(delete("/admin/user/delete"), adminLogin(), idList).andDo(result -> {
+            Response<List<Map<String, Object>>> response = getResponse(result, USER_MODEL_LIST_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            response.getResult().forEach(o -> {
+                // 判断响应数据中是否包含输入的id
+                assertTrue(idList.contains((int) o.get("id")));
+                // 判断处理状态
+                boolean status = (boolean) o.get("status");
+                if (o.containsKey("msg"))
+                    assertFalse(status);
+                else
+                    assertTrue(status);
+            });
+        });
 
     }
 
     @Test
     public void updateInfoByAdmin() throws Exception {
         UserReq userReq = new UserReq();
-        String email = UUID.randomUUID().toString().substring(0, 4) + "@celess.cn";
+        String email = randomStr(4) + "@celess.cn";
         User user = new User(email, MD5Util.getMD5("123456789"));
         userMapper.addUser(user);
         User userByDb = userMapper.findByEmail(email);
         userReq.setId(userByDb.getId());
-        userReq.setPwd(UUID.randomUUID().toString().replaceAll("-", "").substring(0, 10));
-        userReq.setDesc(UUID.randomUUID().toString());
+        userReq.setPwd(randomStr().substring(0, 10));
+        userReq.setDesc(randomStr());
         userReq.setEmailStatus(new Random().nextBoolean());
         userReq.setRole("admin");
-        userReq.setDisplayName(UUID.randomUUID().toString().substring(0, 4));
-        userReq.setEmail(UUID.randomUUID().toString().substring(0, 5) + "@celess.cn");
-        mockMvc.perform(put("/admin/user").contentType("application/json").content(JSONObject.fromObject(userReq).toString()))
-                .andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(put("/admin/user").contentType("application/json").header("Authorization", userLogin()).content(JSONObject.fromObject(userReq).toString()))
-                .andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(put("/admin/user").contentType("application/json").header("Authorization", adminLogin()).content(JSONObject.fromObject(userReq).toString()))
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    UserModel userModel = (UserModel) JSONObject.toBean(object.getJSONObject(Result), UserModel.class);
-                    assertEquals(userReq.getId(), userModel.getId());
-                    assertEquals(userReq.getRole(), userModel.getRole());
-                    assertEquals(userReq.getEmail(), userModel.getEmail());
-                    assertEquals(userReq.getDesc(), userModel.getDesc());
-                    assertEquals(userReq.getDisplayName(), userModel.getDisplayName());
-                });
+        userReq.setDisplayName(randomStr(4));
+        userReq.setEmail(randomStr(5) + "@celess.cn");
+        getMockData(put("/admin/user"), adminLogin(), userReq).andDo(result -> {
+            Response<UserModel> response = getResponse(result, USER_MODEL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            UserModel userModel = response.getResult();
+            assertEquals(userReq.getId(), userModel.getId());
+            assertEquals(userReq.getRole(), userModel.getRole());
+            assertEquals(userReq.getEmail(), userModel.getEmail());
+            assertEquals(userReq.getDesc(), userModel.getDesc());
+            assertEquals(userReq.getDisplayName(), userModel.getDisplayName());
+        });
     }
 
     @Test
     public void getAllUser() throws Exception {
-        mockMvc.perform(get("/admin/users?page=1&count=10"))
-                .andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(get("/admin/users?page=1&count=10").header("authorization", userLogin()))
-                .andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
-        mockMvc.perform(get("/admin/users?page=1&count=10").header("Authorization", adminLogin()))
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    // 结果集非空
-                    assertNotNull(object.getJSONObject(Result));
-                    // 判断pageInfo是否包装完全
-                    JSONObject resultJson = JSONObject.fromObject(object.getJSONObject(Result));
-                    PageData<UserModel> pageData = (PageData<UserModel>) JSONObject.toBean(resultJson, PageData.class);
-                    assertNotEquals(0, pageData.getTotal());
-                    assertEquals(1, pageData.getPageNum());
-                    assertEquals(10, pageData.getPageSize());
-                    // 内容完整
-                    for (Object user : pageData.getList()) {
-                        UserModel u = (UserModel) JSONObject.toBean(JSONObject.fromObject(user), UserModel.class);
-                        assertNotNull(u.getId());
-                        assertNotNull(u.getEmail());
-                        assertNotNull(u.getRole());
-                        assertNotNull(u.getEmailStatus());
-                        assertNotNull(u.getDisplayName());
-                    }
-                });
+        getMockData(get("/admin/users?page=1&count=10"), adminLogin()).andDo(result -> {
+            Response<PageData<UserModel>> response = getResponse(result, USER_MODEL_PAGE_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            // 结果集非空
+            assertNotNull(response.getResult());
+            // 判断pageInfo是否包装完全
+            PageData<UserModel> pageData = response.getResult();
+            assertNotEquals(0, pageData.getTotal());
+            assertEquals(1, pageData.getPageNum());
+            assertEquals(10, pageData.getPageSize());
+            // 内容完整
+            for (UserModel u : pageData.getList()) {
+                assertNotNull(u.getId());
+                assertNotNull(u.getEmail());
+                assertNotNull(u.getRole());
+                assertNotNull(u.getEmailStatus());
+                assertNotNull(u.getDisplayName());
+            }
+        });
     }
 
     @Test
     public void getEmailStatus() throws Exception {
-        String email = UUID.randomUUID().toString().substring(0, 4) + "@celess.cn";
-        mockMvc.perform(get("/emailStatus/" + email)).andDo(result -> {
-            String content = result.getResponse().getContentAsString();
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(content).getInt(Code));
-            assertFalse(JSONObject.fromObject(content).getBoolean(Result));
-        });
-        email = "a@celess.cn";
-        mockMvc.perform(get("/emailStatus/" + email)).andDo(result -> {
-            String content = result.getResponse().getContentAsString();
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(content).getInt(Code));
-            assertTrue(JSONObject.fromObject(content).getBoolean(Result));
-        });
+        String email = randomStr(4) + "@celess.cn";
+        getMockData(get("/emailStatus/" + email)).andDo(result -> assertFalse((Boolean) getResponse(result, BOOLEAN_TYPE).getResult()));
+        getMockData(get("/emailStatus/" + ADMIN_EMAIL)).andDo(result -> assertTrue((Boolean) getResponse(result, BOOLEAN_TYPE).getResult()));
     }
 
     @Test
     public void setPwd() throws Exception {
-        String email = UUID.randomUUID().toString().substring(0, 4) + "@celess.cn";
+        String email = randomStr(4) + "@celess.cn";
         assertEquals(1, userMapper.addUser(new User(email, MD5Util.getMD5("1234567890"))));
-        LoginReq req = new LoginReq();
-        req.setEmail(email);
-        req.setPassword("1234567890");
-        req.setIsRememberMe(false);
-        JSONObject loginReq = JSONObject.fromObject(req);
-        String contentAsString = mockMvc.perform(post("/login").content(loginReq.toString()).contentType("application/json")).andReturn().getResponse().getContentAsString();
-        assertNotNull(contentAsString);
-        String token = JSONObject.fromObject(contentAsString).getJSONObject(Result).getString("token");
+        LoginReq req = new LoginReq(email, "1234567890", false);
+        String token = login(req);
         assertNotNull(token);
         MultiValueMap<String, String> param = new LinkedMultiValueMap<String, String>();
         param.add("pwd", "1234567890");
         param.add("newPwd", "aaabbbccc");
         param.add("confirmPwd", "aaabbbccc");
-        mockMvc.perform(post("/user/setPwd").header("Authorization", token).params(param)).andDo(result -> {
-            String content = result.getResponse().getContentAsString();
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(content).getInt(Code));
+        getMockData(post("/user/setPwd").params(param), token).andDo(result -> {
+            assertEquals(SUCCESS.getCode(), getResponse(result).getCode());
             assertEquals(MD5Util.getMD5("aaabbbccc"), userMapper.getPwd(email));
         });
     }

src/test/java/cn/celess/blog/controller/VisitorControllerTest.java

@@ -1,25 +1,29 @@
 package cn.celess.blog.controller;
 
 import cn.celess.blog.BaseTest;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.model.VisitorModel;
-import com.github.pagehelper.PageInfo;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.core.type.TypeReference;
 import org.junit.Test;
-import org.springframework.test.web.servlet.result.MockMvcResultHandlers;
 
+import static cn.celess.blog.enmu.ResponseEnum.SUCCESS;
 import static org.junit.Assert.*;
-import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
-import static cn.celess.blog.enmu.ResponseEnum.*;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
 
 public class VisitorControllerTest extends BaseTest {
+    private final TypeReference<?> VISITOR_PAGE_TYPE = new TypeReference<Response<PageData<VisitorModel>>>() {
+    };
+    private final TypeReference<?> VISITOR_TYPE = new TypeReference<Response<VisitorModel>>() {
+    };
 
     @Test
     public void getVisitorCount() throws Exception {
-        mockMvc.perform(get("/visitor/count")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertTrue(object.containsKey(Result));
+        getMockData(get("/visitor/count")).andDo(result -> {
+            Response<Object> response = getResponse(result);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
         });
     }
 
@@ -27,29 +31,28 @@ public class VisitorControllerTest extends BaseTest {
     public void page() throws Exception {
         int count = 10;
         int page = 1;
-        mockMvc.perform(get("/admin/visitor/page?count=" + count + "&page=" + page).header("Authorization", adminLogin()))
-                .andDo(result -> {
-                    JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-                    assertEquals(SUCCESS.getCode(), object.getInt(Code));
-                    JSONObject resultJson = JSONObject.fromObject(object.getJSONObject(Result));
-                    PageData<VisitorModel> pageData = (PageData<VisitorModel>) JSONObject.toBean(resultJson, PageData.class);
-                    assertNotEquals(0, pageData.getTotal());
-                    assertEquals(1, pageData.getPageNum());
-                    assertEquals(10, pageData.getPageSize());
-                    for (Object ver : pageData.getList()) {
-                        VisitorModel v = (VisitorModel) JSONObject.toBean(JSONObject.fromObject(ver), VisitorModel.class);
-                        assertNotEquals(0, v.getId());
-                        assertNotNull(v.getDate());
-                    }
-                });
+        // 默认显示location
+        getMockData(get("/admin/visitor/page?count=" + count + "&page=" + page), adminLogin()).andDo(result -> {
+            Response<PageData<VisitorModel>> response = getResponse(result, VISITOR_PAGE_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            PageData<VisitorModel> pageData = response.getResult();
+            assertNotEquals(0, pageData.getTotal());
+            assertEquals(1, pageData.getPageNum());
+            assertEquals(10, pageData.getPageSize());
+            for (VisitorModel v : pageData.getList()) {
+                assertNotEquals(0, v.getId());
+                assertNotNull(v.getDate());
+                assertNotNull(v.getLocation());
+            }
+        });
     }
 
     @Test
     public void add() throws Exception {
-        mockMvc.perform(post("/visit")).andDo(MockMvcResultHandlers.print()).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            VisitorModel visitorModel = (VisitorModel) JSONObject.toBean(object.getJSONObject(Result), VisitorModel.class);
+        getMockData(post("/visit")).andDo(result -> {
+            Response<VisitorModel> response = getResponse(result, VISITOR_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            VisitorModel visitorModel = response.getResult();
             assertNotEquals(0, visitorModel.getId());
             assertNotNull(visitorModel.getIp());
         });
@@ -57,26 +60,26 @@ public class VisitorControllerTest extends BaseTest {
 
     @Test
     public void dayVisitCount() throws Exception {
-        mockMvc.perform(get("/dayVisitCount")).andDo(MockMvcResultHandlers.print()).andDo(result ->
-                assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
-        );
+        getMockData(get("/dayVisitCount")).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
     }
 
     // 手动测试
     // @Test
     public void ipLocation() throws Exception {
         String ip = "127.0.0.1";
-        mockMvc.perform(get("/ip/" + ip)).andDo(MockMvcResultHandlers.print()).andDo(result -> {
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-            assertTrue(JSONObject.fromObject(result.getResponse().getContentAsString()).containsKey(Result));
+        getMockData(get("/ip/" + ip)).andDo(result -> {
+            Response<Object> response = getResponse(result);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
         });
     }
 
     @Test
     public void getIp() throws Exception {
-        mockMvc.perform(get("/ip")).andDo(MockMvcResultHandlers.print()).andDo(result -> {
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-            assertEquals("127.0.0.1", JSONObject.fromObject(result.getResponse().getContentAsString()).getString(Result));
+        getMockData(get("/ip")).andDo(result -> {
+            Response<String> response = getResponse(result, STRING_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertEquals("127.0.0.1", response.getResult());
         });
     }
 }

src/test/java/cn/celess/blog/controller/WebUpdateInfoControllerTest.java

@@ -1,35 +1,46 @@
 package cn.celess.blog.controller;
 
 import cn.celess.blog.BaseTest;
+import cn.celess.blog.entity.Response;
 import cn.celess.blog.entity.WebUpdate;
 import cn.celess.blog.entity.model.PageData;
 import cn.celess.blog.entity.model.WebUpdateModel;
 import cn.celess.blog.mapper.WebUpdateInfoMapper;
-import net.sf.json.JSONArray;
-import net.sf.json.JSONObject;
+import com.fasterxml.jackson.core.type.TypeReference;
+import lombok.extern.slf4j.Slf4j;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 
 import java.util.Date;
 import java.util.List;
-import java.util.UUID;
 
-import static cn.celess.blog.enmu.ResponseEnum.*;
+import static cn.celess.blog.enmu.ResponseEnum.DATA_NOT_EXIST;
+import static cn.celess.blog.enmu.ResponseEnum.SUCCESS;
 import static org.junit.Assert.*;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
 
+@Slf4j
 public class WebUpdateInfoControllerTest extends BaseTest {
+
+    private final TypeReference<?> MODAL_TYPE = new TypeReference<Response<WebUpdateModel>>() {
+    };
+    private final TypeReference<?> MODAL_LIST_TYPE = new TypeReference<Response<List<WebUpdateModel>>>() {
+    };
+    private final TypeReference<?> MODAL_PAGE_TYPE = new TypeReference<Response<PageData<WebUpdateModel>>>() {
+    };
+
+
     @Autowired
     WebUpdateInfoMapper mapper;
 
     @Test
     public void create() throws Exception {
-        String info = UUID.randomUUID().toString();
-        mockMvc.perform(post("/admin/webUpdate/create?info=" + info).header("Authorization", adminLogin())).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertTrue(object.containsKey(Result));
-            WebUpdateModel webUpdateModel = (WebUpdateModel) JSONObject.toBean(object.getJSONObject(Result), WebUpdateModel.class);
+        String info = randomStr();
+        getMockData(post("/admin/webUpdate/create?info=" + info), adminLogin()).andDo(result -> {
+            Response<WebUpdateModel> response = getResponse(result, MODAL_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
+            WebUpdateModel webUpdateModel = response.getResult();
             assertEquals(info, webUpdateModel.getInfo());
             assertNotNull(webUpdateModel.getTime());
             assertNotEquals(0, webUpdateModel.getId());
@@ -40,7 +51,7 @@ public class WebUpdateInfoControllerTest extends BaseTest {
     public void del() throws Exception {
         // 新增数据
         WebUpdate webUpdate = new WebUpdate();
-        webUpdate.setUpdateInfo(UUID.randomUUID().toString());
+        webUpdate.setUpdateInfo(randomStr());
         webUpdate.setUpdateTime(new Date());
         mapper.insert(webUpdate);
         // 接口测试
@@ -49,32 +60,31 @@ public class WebUpdateInfoControllerTest extends BaseTest {
         assertNotEquals(0, update.getId());
 
         long id = update.getId();
-        mockMvc.perform(delete("/admin/webUpdate/del/" + id).header("Authorization", adminLogin())).andDo(result -> {
-            assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code));
-            assertTrue(JSONObject.fromObject(result.getResponse().getContentAsString()).getBoolean(Result));
+        getMockData(delete("/admin/webUpdate/del/" + id), adminLogin()).andDo(result -> {
+            Response<Object> response = getResponse(result);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
         });
         do {
             id += 2;
         } while (mapper.existsById(id));
-        System.out.println("准备删除ID=" + id + "的不存在记录");
-        mockMvc.perform(delete("/admin/webUpdate/del/" + id).header("Authorization", adminLogin())).andDo(result ->
-                assertEquals(DATA_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
-        );
+        log.debug("准备删除ID={}的不存在记录", id);
+        getMockData(delete("/admin/webUpdate/del/" + id), adminLogin()).andDo(result -> assertEquals(DATA_NOT_EXIST.getCode(), getResponse(result).getCode()));
     }
 
     @Test
     public void update() throws Exception {
         // 新增数据
         WebUpdate webUpdate = new WebUpdate();
-        webUpdate.setUpdateInfo(UUID.randomUUID().toString());
+        webUpdate.setUpdateInfo(randomStr());
         webUpdate.setUpdateTime(new Date());
         mapper.insert(webUpdate);
         List<WebUpdate> all = mapper.findAll();
         WebUpdate update = all.get(all.size() - 1);
         assertNotEquals(0, update.getId());
         assertNotNull(update.getUpdateInfo());
-        String info = UUID.randomUUID().toString();
-        mockMvc.perform(put("/admin/webUpdate/update?id=" + update.getId() + "&info=" + info).header("Authorization", adminLogin())).andDo(result -> {
+        String info = randomStr();
+        getMockData(put("/admin/webUpdate/update?id=" + update.getId() + "&info=" + info), adminLogin()).andDo(result -> {
             List<WebUpdate> list = mapper.findAll();
             WebUpdate up = list.get(list.size() - 1);
             assertEquals(update.getId(), up.getId());
@@ -85,12 +95,12 @@ public class WebUpdateInfoControllerTest extends BaseTest {
 
     @Test
     public void findAll() throws Exception {
-        mockMvc.perform(get("/webUpdate")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            JSONArray jsonArray = object.getJSONArray(Result);
-            jsonArray.forEach(o -> {
-                WebUpdateModel webUpdate = (WebUpdateModel) JSONObject.toBean(JSONObject.fromObject(o), WebUpdateModel.class);
+        getMockData(get("/webUpdate")).andDo(result -> {
+            Response<List<WebUpdateModel>> response = getResponse(result, MODAL_LIST_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
+            assertNotEquals(0, response.getResult());
+            response.getResult().forEach(webUpdate -> {
                 assertNotEquals(0, webUpdate.getId());
                 assertNotNull(webUpdate.getTime());
                 assertNotNull(webUpdate.getInfo());
@@ -100,15 +110,14 @@ public class WebUpdateInfoControllerTest extends BaseTest {
 
     @Test
     public void page() throws Exception {
-        mockMvc.perform(get("/webUpdate/pages?page=1&count=10")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-            assertNotNull(object.getJSONObject(Result));
-            PageData<WebUpdateModel> pageData = (PageData<WebUpdateModel>) JSONObject.toBean(object.getJSONObject(Result), PageData.class);
+        getMockData(get("/webUpdate/pages?page=1&count=10")).andDo(result -> {
+            Response<PageData<WebUpdateModel>> response = getResponse(result, MODAL_PAGE_TYPE);
+            assertEquals(SUCCESS.getCode(), response.getCode());
+            assertNotNull(response.getResult());
+            PageData<WebUpdateModel> pageData = response.getResult();
             assertEquals(1, pageData.getPageNum());
             assertEquals(10, pageData.getPageSize());
-            for (Object o : pageData.getList()) {
-                WebUpdateModel model = (WebUpdateModel) JSONObject.toBean(JSONObject.fromObject(o), WebUpdateModel.class);
+            for (WebUpdateModel model : pageData.getList()) {
                 assertNotEquals(0, model.getId());
                 assertNotNull(model.getTime());
                 assertNotNull(model.getInfo());
@@ -118,7 +127,6 @@ public class WebUpdateInfoControllerTest extends BaseTest {
 
     @Test
     public void lastestUpdateTime() throws Exception {
-        mockMvc.perform(get("/lastestUpdate")).andDo(result ->
-                assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
+        getMockData(get("/lastestUpdate")).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
     }
 }

src/test/java/cn/celess/blog/enmu/UserAccountStatusEnumTest.java

@@ -0,0 +1,36 @@
+package cn.celess.blog.enmu;
+
+import cn.celess.blog.BaseTest;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.junit.Test;
+
+import java.io.IOException;
+
+import static org.junit.Assert.*;
+
+public class UserAccountStatusEnumTest extends BaseTest {
+
+    @Test
+    public void get() {
+        assertEquals(UserAccountStatusEnum.NORMAL, UserAccountStatusEnum.get(0));
+        assertEquals(UserAccountStatusEnum.LOCKED, UserAccountStatusEnum.get(1));
+        assertEquals(UserAccountStatusEnum.DELETED, UserAccountStatusEnum.get(2));
+    }
+
+    @Test
+    public void toJson() throws JsonProcessingException {
+        // 序列化
+        ObjectMapper objectMapper = new ObjectMapper();
+        assertEquals("{\"code\":0,\"desc\":\"正常\"}", objectMapper.writeValueAsString(UserAccountStatusEnum.NORMAL));
+    }
+
+    @Test
+    public void testGet() throws IOException {
+        // 反序列化
+        ObjectMapper mapper = new ObjectMapper();
+        UserAccountStatusEnum userAccountStatusEnum = mapper.readValue(mapper.writeValueAsString(UserAccountStatusEnum.NORMAL), UserAccountStatusEnum.class);
+        assertEquals(UserAccountStatusEnum.NORMAL.getCode(), userAccountStatusEnum.getCode());
+        assertEquals(UserAccountStatusEnum.NORMAL.getDesc(), userAccountStatusEnum.getDesc());
+    }
+}

src/test/java/cn/celess/blog/filter/AuthorizationFilter.java

@@ -1,12 +1,12 @@
 package cn.celess.blog.filter;
 
 import cn.celess.blog.BaseTest;
-import net.sf.json.JSONObject;
+import cn.celess.blog.entity.Response;
 import org.junit.Test;
 
-import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
-import static org.junit.Assert.*;
 import static cn.celess.blog.enmu.ResponseEnum.*;
+import static org.junit.Assert.*;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 
 /**
  * @Author: 小海
@@ -17,60 +17,37 @@ public class AuthorizationFilter extends BaseTest {
 
     @Test
     public void UserAccess() throws Exception {
-        String token = "";
         // 未登录
-        mockMvc.perform(get("/user/userInfo").header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code));
-        });
-        token = userLogin();
-        mockMvc.perform(get("/user/userInfo").header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-        });
+        getMockData(get("/user/userInfo")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
+        // user权限登录
+        getMockData(get("/user/userInfo"), userLogin()).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
     }
 
     @Test
     public void AdminAccess() throws Exception {
-        String token = "";
         // 未登录
-        mockMvc.perform(get("/admin/articles?page=1&count=1").header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code));
-        });
-        token = userLogin();
-        mockMvc.perform(get("/admin/articles?page=1&count=1").header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(PERMISSION_ERROR.getCode(), object.getInt(Code));
-        });
-        token = adminLogin();
-        mockMvc.perform(get("/admin/articles?page=1&count=1").header("Authorization", token)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
-        });
+        getMockData(get("/admin/articles?page=1&count=1")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
+        // user权限
+        getMockData(get("/admin/articles?page=1&count=1"), userLogin()).andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), getResponse(result).getCode()));
+        // admin 权限
+        getMockData(get("/admin/articles?page=1&count=1"), adminLogin()).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
     }
 
     @Test
     public void VisitorAccess() throws Exception {
-        mockMvc.perform(get("/user/userInfo")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code));
-        });
-        mockMvc.perform(get("/admin/articles?page=1&count=1")).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code));
-        });
+        getMockData(get("/user/userInfo")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
+        getMockData(get("/admin/articles?page=1&count=1")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
     }
 
     @Test
     public void authorizationTest() throws Exception {
         // 测试response中有无Authorization字段
-        String s = userLogin();
-        mockMvc.perform(get("/user/userInfo").header("Authorization", s)).andDo(result -> {
-            JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
-            assertEquals(SUCCESS.getCode(), object.getInt(Code));
+        String token = userLogin();
+        getMockData(get("/user/userInfo"), token).andDo(result -> {
+            Response<Object> response = getResponse(result);
+            assertEquals(SUCCESS.getCode(), response.getCode());
             assertNotNull(result.getResponse().getHeader("Authorization"));
-            assertNotEquals(s, result.getResponse().getHeader("Authorization"));
+            assertNotEquals(token, result.getResponse().getHeader("Authorization"));
         });
     }
 }