Halo开源项目学习（三）：注册与登录

基本介绍

首次启动 Halo 项目时需要安装博客并注册用户信息，当博客安装完成后用户就可以根据注册的信息登录到管理员界面，下面我们分析一下整个过程中代码是如何执行的。

博客安装

项目启动成功后，我们可以访问 http://127.0.0.1:8090 进入到博客首页，或者访问 http://127.0.0.1:8090/admin 进入到管理员页面。但如果博客未安装，那么页面会被重定向到安装页面：

这是因为 Halo 中定义了几个过滤器，分别为 ContentFilter、ApiAuthenticationFilter 和 AdminAuthenticationFilter。这三个过滤器均为 AbstractAuthenticationFilter 的子类，而 AbstractAuthenticationFilter 又继承自 OncePerRequestFilter，其重写的 doFilterInternal 方法如下：

protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response,    FilterChain filterChain) throws ServletException, IOException {    // Check whether the blog is installed or not    Boolean isInstalled =        optionService            .getByPropertyOrDefault(PrimaryProperties.IS_INSTALLED, Boolean.class, false);    // 如果博客未安装且当前并不是测试环境    if (!isInstalled && !Mode.TEST.equals(haloProperties.getMode())) {        // If not installed        getFailureHandler().onFailure(request, response, new NotInstallException("当前博客还没有初始化"));        return;    }    try {        // Check the one-time-token        // 进行一次性 token 检查        if (isSufficientOneTimeToken(request)) {            filterChain.doFilter(request, response);            return;        }        // 一次性 token 验证失败则需要做身份认证        // Do authenticate        doAuthenticate(request, response, filterChain);    } catch (AbstractHaloException e) {        getFailureHandler().onFailure(request, response, e);    } finally {        SecurityContextHolder.clearContext();    }}

doFilterInternal 方法的主要逻辑为：

1. 判断博客是否已安装，如果未安装且当前并非测试环境，那么由 failureHandler 处理 NotInstallException 异常并退出，否则继续向下执行。

2. 进行一次性 token 检查（本文并未使用到），如果一次性 token 验证成功则将该请求交付给下一个过滤器；如果失败则执行 doAuthenticate 方法对用户进行身份认证。若在发生异常，那么由 failureHandler 的 onFailure 方法处理该请求。

继承了 AbstractAuthenticationFilter 的子类都会根据上述逻辑处理用户的请求，只不过在不同的子类过滤器中，身份认证逻辑和 failureHandler 会有一定差异。下图展示了一个请求经过 Filter 的过程：

可见，不同的过滤器之间拦截的请求并没有交集，因此一个请求最多会被一个过滤器处理。当我们访问 http://127.0.0.1:8090 时，该请求会被 ContentFilter 拦截，然后执行 doFilterInternal 方法，由于博客未安装，所以由 failureHandler 处理 NotInstallException 异常。ContentFilter 中定义的 failureHandler 属于 ContentAuthenticationFailureHandler 类，该类中 onFailure 方法定义如下：

public void onFailure(HttpServletRequest request, HttpServletResponse response,    AbstractHaloException exception) throws IOException, ServletException {    if (exception instanceof NotInstallException) {        // 重定向到 /install        response.sendRedirect(request.getContextPath() + "/install");        return;    }    // Forward to error    request.getRequestDispatcher(request.getContextPath() + "/error")        .forward(request, response);}

上述代码表示，当异常为 NotInstallException，就将请求重定向到 /install：

/install 请求在 MainController 中定义，且该请求又会被重定向到 /admin/index.html#install：

@GetMapping("install")public void installation(HttpServletResponse response) throws IOException {    String installRedirectUri =        StringUtils.appendIfMissing(this.haloProperties.getAdminPath(), "/") + INSTALL_REDIRECT_URI;    // /admin/index.html#install    response.sendRedirect(installRedirectUri);}

index.html 文件位于 /resource/admin 目录下，#install 表示定位到 index.html 页面的 install 表单，也就是上文中展示的安装页面。

值得注意的是，当我们访问 http://127.0.0.1:8090/admin 时，请求并不会被过滤器处理（三个过滤器均放行了 /admin），但页面还是被重定向到了安装页面，这是因为 MainController 中也定义了 /admin 请求的重定向规则：

@GetMapping("${halo.admin-path:admin}")public void admin(HttpServletResponse response) throws IOException {    String adminIndexRedirectUri =        HaloUtils.ensureBoth(haloProperties.getAdminPath(), HaloUtils.URL_SEPARATOR)            + INDEX_REDIRECT_URI;    // /admin/index.html    response.sendRedirect(adminIndexRedirectUri);}

可见，访问 /admin 时，请求会被重定向到 /admin/index.html，但直接访问 index.html 还并不能显示安装页面，因为 URL 中并没有添加定位标识 #install。查看 index.html 中的代码后可以发现，当该页面打开时，浏览器会自动访问 /favicon.ico 和 /api/admin/is_installed，/api/admin/is_installed 会被过滤器放行，但 /favicon.ico 却会被 ContentFilter 拦截，之后又是两个重定向，最终让我们看到安装页面：

在安装页面填写完信息后，点击 "安装" 按钮，触发 /api/admin/installations 请求，请求中携带着我们填写的博客信息：

/api/admin/installations 在 InstallController 中定义，主要处理逻辑为：

public BaseResponse<String> installBlog(@RequestBody InstallParam installParam) {    // Validate manually    ValidationUtils.validate(installParam, CreateCheck.class);    // Check is installed    boolean isInstalled = optionService        .getByPropertyOrDefault(PrimaryProperties.IS_INSTALLED, Boolean.class, false);    if (isInstalled) {        throw new BadRequestException("该博客已初始化，不能再次安装！");    }    // Initialize settings    initSettings(installParam);    // Create default user    User user = createUser(installParam);    // Create default category    Category category = createDefaultCategoryIfAbsent();    // Create default post    PostDetailVO post = createDefaultPostIfAbsent(category);    // Create default sheet    createDefaultSheet();    // Create default postComment    createDefaultComment(post);    // Create default menu    createDefaultMenu();    eventPublisher.publishEvent(        new LogEvent(this, user.getId().toString(), LogType.BLOG_INITIALIZED, "博客已成功初始化")    );    return BaseResponse.ok("安装完成！");}

1. 初始化博客的系统设置：也可以称为初始化选项信息，例如将安装选项 is_installed 置为 true，将博客标题 blog_title 置为我们填写的标题等，这些信息会被保存到 options 表中。

2. 保存用户信息：也就是我们填写的姓名、email 等，在这些信息存储到 users 表之前，系统会将用户的密码进行加密处理，并为用户分配一个头像。

3. 创建默认的分类：分类名称为 "默认分类"。

4. 创建默认的文章：访问博客首页时看到的文章 "Hello Halo"。

5. 创建默认的页面：访问博客首页时看到的页面，标题为 "关于页面"。

6. 创建默认的评论：评论的 postId 为文章 "Hello Halo" 的 id，即表示该评论是属于 "Hello Halo" 的评论。

7. 创建默认的菜单：设置了 4 个一级菜单、菜单对应的 URL 以及菜单在首页排列的优先级，例如 "首页" 的优先级为 0（最高优先级），因此排列在第一位，访问的 URL 为 "/"，因此点击 "首页" 时会触发 "/" 请求。

8. 发布 LogEvent 事件：记录 "博客已成功初始化" 的系统日志。

用户登录

上文中提到，当用户访问 /admin 时，请求会被重定向到 /admin/index.html，而访问 index.html 时，默认显示的是登录表单，此时浏览器中的 URL 为 admin/index.html#/login?redirect=%2Fdashboard，这是由 index.html 引入的的 js 文件 https://cdn.jsdelivr.net/npm/halo-admin@1.4.13/dist/js/app.22ce7788.js（后文中将其简称为 js 文件）设置的，表示登录成功后重定向到 "Halo Dashboard" 界面（与定位 install 一样，这里是定位到 dashboard）。用户可填写 "用户名/邮箱" 和 "密码" 进行登录，登录按钮会触发 /api/admin/precheck 请求，该请求的处理逻辑为：

@PostMapping("login/precheck")@ApiOperation("Login")@CacheLock(autoDelete = false, prefix = "login_precheck")public LoginPreCheckDTO authPreCheck(@RequestBody @Valid LoginParam loginParam) {    final User user = adminService.authenticate(loginParam);    return new LoginPreCheckDTO(MFAType.useMFA(user.getMfaType()));}

上述方法首先调用 authenticate 方法验证用户的登录参数，然后告知前端登录参数是否正确以及是否需要输入两步验证码（默认关闭）。authenticate 方法会根据用户名/邮箱从 users 表中获取用户的信息，并判断当前用户账号是否有效，如果有效则继续判断登录的密码与设置的密码是否相同，如果密码正确则返回 User 对象：

public User authenticate(@NonNull LoginParam loginParam) {    Assert.notNull(loginParam, "Login param must not be null");    String username = loginParam.getUsername();    String mismatchTip = "用户名或者密码不正确";    final User user;    try {    // Get user by username or email    // userName 是用户名还是邮箱    user = ValidationUtils.isEmail(username)    ? userService.getByEmailOfNonNull(username) :    userService.getByUsernameOfNonNull(username);    } catch (NotFoundException e) {    log.error("Failed to find user by name: " + username);    // 记录登录失败的日志    eventPublisher.publishEvent(    new LogEvent(this, loginParam.getUsername(), LogType.LOGIN_FAILED,    loginParam.getUsername()));    throw new BadRequestException(mismatchTip);    }    // 用户账号的有效时间 expireTime 必须小于当前时间, 否则无法正常登录，这个东西就很奇怪    userService.mustNotExpire(user);    // 检查登录密码是否正确    if (!userService.passwordMatch(user, loginParam.getPassword())) {    // If the password is mismatch    eventPublisher.publishEvent(    new LogEvent(this, loginParam.getUsername(), LogType.LOGIN_FAILED,    loginParam.getUsername()));    throw new BadRequestException(mismatchTip);    }    return user;}

虽然 /api/login/precheck 返回的是一个 LoginPreCheckDTO 对象，但实际上前端收到的是一个 BaseResponse 对象，这是因为 Halo 中会使用 AOP 对 Controller 的响应进行封装：

默认情况下是不开启两步验证码的（MFAType 的默认值为 0），因此响应中的 needMFACode 为 false。如果需要，那么可在管理员页面的 "用户" -> "个人资料" -> "两步验证" 处开启。浏览器收到上图中的响应后，会自动发送 /api/admin/login 请求（由 js 文件设置），但如果开启了两步验证码，那么还需要输入验证码才能继续访问 /api/admin/login。

/api/admin/login 会向用户返回一个 AuthToken 对象：

@PostMapping("login")@ApiOperation("Login")@CacheLock(autoDelete = false, prefix = "login_auth")public AuthToken auth(@RequestBody @Valid LoginParam loginParam) {	return adminService.authCodeCheck(loginParam);}

authCodeCheck 方法的处理逻辑为：

public AuthToken authCodeCheck(@NonNull final LoginParam loginParam) {    // get user    final User user = this.authenticate(loginParam);    // check authCode    // 检查两步验证码    if (MFAType.useMFA(user.getMfaType())) {        if (StringUtils.isBlank(loginParam.getAuthcode())) {        throw new BadRequestException("请输入两步验证码");    }    TwoFactorAuthUtils.validateTFACode(user.getMfaKey(), loginParam.getAuthcode());    }    if (SecurityContextHolder.getContext().isAuthenticated()) {        // If the user has been logged in        throw new BadRequestException("您已登录，请不要重复登录");    }    // Log it then login successful    // 记录登录成功的日志    eventPublisher.publishEvent(    new LogEvent(this, user.getUsername(), LogType.LOGGED_IN, user.getNickname()));    // Generate new token    // 为用户生成 token    return buildAuthToken(user);}

上述方法首先调用 authenticate 方法获取用户，然后检查两步验证码（如果设置的话），接着记录登录成功的日志，最后为用户生成一个 token，token 可作为用户的身份标识，服务器可以根据 token 验证用户的身份，而无需用户名和密码。token 的生成逻辑如下：

private AuthToken buildAuthToken(@NonNull User user) {    Assert.notNull(user, "User must not be null");    // Generate new token    AuthToken token = new AuthToken();    token.setAccessToken(HaloUtils.randomUUIDWithoutDash());    token.setExpiredIn(ACCESS_TOKEN_EXPIRED_SECONDS);    token.setRefreshToken(HaloUtils.randomUUIDWithoutDash());    // Cache those tokens, just for clearing    cacheStore.putAny(SecurityUtils.buildAccessTokenKey(user), token.getAccessToken(),                      ACCESS_TOKEN_EXPIRED_SECONDS, TimeUnit.SECONDS);    cacheStore.putAny(SecurityUtils.buildRefreshTokenKey(user), token.getRefreshToken(),                      REFRESH_TOKEN_EXPIRED_DAYS, TimeUnit.DAYS);    // Cache those tokens with user id    cacheStore.putAny(SecurityUtils.buildTokenAccessKey(token.getAccessToken()), user.getId(),                      ACCESS_TOKEN_EXPIRED_SECONDS, TimeUnit.SECONDS);    cacheStore.putAny(SecurityUtils.buildTokenRefreshKey(token.getRefreshToken()), user.getId(),                      REFRESH_TOKEN_EXPIRED_DAYS, TimeUnit.DAYS);    return token;}

可以发现，token 中包含了 accessToken（随机生成的 UUID）、refreshToken（随机生成的 UUID）以及 accessToken 和 refreshToken 的过期时间。其中 accessToken 是用来做身份认证的，而 refreshToken 的作用是实现 token 的 "无痛刷新"。具体来讲，后端返回 token 信息后，浏览器会同时保存 accessToken 和 refreshToken，如果 accessToken 过期，那么当浏览器发送请求时，服务器会返回 "Token 已过期或不存在" 的失败响应，此时浏览器可以发送 /api/admin/refresh/{refreshToken} 请求，通过 refreshToken 向服务器申请一个新的 token（包括 accessToken 和 refreshToken），然后使用新的 accessToken 重新发送之前未处理成功的请求。因此，accessToken 和 refreshToken 是绑定在一起的，且 refreshToken 的过期时间（Halo 中设置的是 30 天）要大于 accessToken（1 天）。上述代码中，服务器使用 cacheStore 存储用户 id 和 token ，cacheStore 是项目中的内部缓存，它使用 ConcurrentHashMap 作为容器。

用户登录成功后浏览器获得的响应：

浏览器将 token 保存在了 Local Storate：

当浏览器下次请求资源时，会将 accessToken 存入到 Request Headers 中 Admin-Authorization 头域：

accessToken 过期后，浏览器使用 refreshToken 申请新的 token：

<img src

浏览器中 token 的保存、token 过期后的重新申请以及 Header 中 token 的添加都是由 js 文件设置的。另外，前文中提到，过滤器拦截请求后首先要进行一次性 token 检查，如果失败则需要验证用户的身份，而 Admin-Authorization 头域就是用于身份认证的，例如上图中的请求 api/admin/users/profiles 会被 AdminAuthenticationFilter 拦截，因为并未设置一次性 token，因此需要进行身份认证，而 AdminAuthenticationFilter 的身份认证逻辑为：

protected void doAuthenticate(HttpServletRequest request, HttpServletResponse response,FilterChain filterChain) throws ServletException, IOException {    // 如果未设置认证    if (!haloProperties.isAuthEnabled()) {        // Set security        userService.getCurrentUser().ifPresent(user ->        SecurityContextHolder.setContext(        new SecurityContextImpl(new AuthenticationImpl(new UserDetail(user)))));        // Do filter        filterChain.doFilter(request, response);    return;    }    // 获取 token, 从请求的 Query 参数中获取 admin_token 或者从 Header 中获取 Admin-Authorization    // Get token from request    String token = getTokenFromRequest(request);    if (StringUtils.isBlank(token)) {    throw new AuthenticationException("未登录，请登录后访问");    }    // 根据 token 从 cacheStore 缓存中获取用户 id    // Get user id from cache    Optional<Integer> optionalUserId =    cacheStore.getAny(SecurityUtils.buildTokenAccessKey(token), Integer.class);    if (!optionalUserId.isPresent()) {    	throw new AuthenticationException("Token 已过期或不存在").setErrorData(token);    }    // 获取用户    // Get the user    User user = userService.getById(optionalUserId.get());    // Build user detail    UserDetail userDetail = new UserDetail(user);    // 将用户信息存储到 ThreadLocal 中    // Set security    SecurityContextHolder    .setContext(new SecurityContextImpl(new AuthenticationImpl(userDetail)));    // Do filter    filterChain.doFilter(request, response);}

1. 如果博客未设置身份认证，那么将 users 表中的第一个用户作为当前用户，并存储到 ThreadLocal 容器中，ThreadLocal 可用于在同一个线程内的多个函数或者组件之间传递公共信息。如果开启了身份认证，则继续向下执行。
2. 获取 token，也就是从请求的 Query 参数中获取 admin_token 或者从 Header 中获取 Admin-Authorization。
3. 根据 token 从 cacheStore 缓存中获取用户 id，查询出用户后将用户存储到 ThreadLocal 中，身份认证通过。

以上便是用户输入账号密码来登录管理员页面的过程。

用户登出

用户退出登录时，触发 /api/admin/logout 请求，请求的处理逻辑是清除掉用户的 token：

public void logout() {	adminService.clearToken();}

clearToken 方法如下：

@PostMapping("logout")@ApiOperation("Logs out (Clear session)")@CacheLock(autoDelete = false)public void clearToken() {    // 检查 ThreadLocal 是否为空    // Check if the current is logging in    Authentication authentication = SecurityContextHolder.getContext().getAuthentication();    if (authentication == null) {    	throw new BadRequestException("您尚未登录，因此无法注销");    }    // 获取当前用户    // Get current user    User user = authentication.getDetail().getUser();    // 清除 accessToken    // Clear access token    cacheStore.getAny(SecurityUtils.buildAccessTokenKey(user), String.class)    .ifPresent(accessToken -> {    	// Delete token    	cacheStore.delete(SecurityUtils.buildTokenAccessKey(accessToken));    	cacheStore.delete(SecurityUtils.buildAccessTokenKey(user));    });    // 清除 refreshToken    // Clear refresh token    cacheStore.getAny(SecurityUtils.buildRefreshTokenKey(user), String.class)    .ifPresent(refreshToken -> {        cacheStore.delete(SecurityUtils.buildTokenRefreshKey(refreshToken));        cacheStore.delete(SecurityUtils.buildRefreshTokenKey(user));    });    eventPublisher.publishEvent(    new LogEvent(this, user.getUsername(), LogType.LOGGED_OUT, user.getNickname()));    log.info("You have been logged out, looking forward to your next visit!");}

1. 检查 ThreadLocal 是否为空，为空表示用户并未登陆。

2. 获取当前用户并清除 cacheStore 中与用户相关的 token。

3. 记录用户登出日志。

博客首页

上文介绍的登录和登出指的是在管理员界面上的操作，实际上 127.0.0.1:8090 才是博客的首页。当我们访问 / 时，ContentIndexController 中的 index 方法会处理请求：

@GetMappingpublic String index(Integer p, String token, Model model) {    PostPermalinkType permalinkType = optionService.getPostPermalinkType();    if (PostPermalinkType.ID.equals(permalinkType) && !Objects.isNull(p)) {        Post post = postService.getById(p);        return postModel.content(post, token, model);    }    return this.index(model, 1);}

index(model, 1) 指的是显示博客的第一页：

public String index(Model model,        @PathVariable(value = "page") Integer page) {    return postModel.list(page, model);}

postModel.list 方法的逻辑如下：

public String list(Integer page, Model model) {    // 获取每页显示的文章数量    int pageSize = optionService.getPostPageSize();    Pageable pageable = PageRequest        .of(page >= 1 ? page - 1 : page, pageSize, postService.getPostDefaultSort());    // 查询出所有已发布的文章, 默认按照发布时间降序排列    Page<Post> postPage = postService.pageBy(PostStatus.PUBLISHED, pageable);    Page<PostListVO> posts = postService.convertToListVo(postPage);    // 将文章以及相关属性存入到 model 中    model.addAttribute("is_index", true);    model.addAttribute("posts", posts);    model.addAttribute("meta_keywords", optionService.getSeoKeywords());    model.addAttribute("meta_description", optionService.getSeoDescription());    // 返回已激活主题文件中的 index.ftl    return themeService.render("index");}

1. 查看博客每页显示的文章数量，默认是 10。
2. 查询出所有已发布的文章并对其排序，默认按照发布时间降序排列。
3. 将文章以及相关属性存入到 model 中，Halo 中使用的是 FreeMaker 模板引擎，将信息存入到 model 后前端可通过 EL 表达式获取到这些内容。
4. 返回 "index" 路径，该路径指向已激活主题（默认主题为 caicai_anatole）的 index.ftl 文件，该文件可生成我们看到的博客主页。

博客首页：