编写UI

289次阅读

共计 3366 个字符，预计需要花费 9 分钟才能阅读完成。

我们已经实现了 API 系统、交易系统、定序系统、行情系统和推送系统，最后就差一个 UI 系统，让用户可以登录并通过浏览器下订单。UI 系统就是一个标准的 Web 系统，相对比较简单。

UI 系统本质上是一个 MVC 模型的 Web 系统，我们先引入一个视图的第三方依赖：

<dependency>
    <groupId>io.pebbletemplates</groupId>
    <artifactId>pebble-spring-boot-starter</artifactId>
    <version>${pebble.version}</version>
</dependency>

在 ui.yml 加入最基本的配置：

pebble:
  prefix: /templates/
  suffix: .html

注意到视图页面都放在 src/main/resources/templates/ 目录下。编写MvcController，实现登录功能：

@Controller
public class MvcController extends LoggerSupport {// 显示登录页
    @GetMapping("/signin")
    public ModelAndView signin(HttpServletRequest request) {if (UserContext.getUserId() != null) {return redirect("/");
        }
        return prepareModelAndView("signin");
    }

    // 登录
    @PostMapping("/signin")
    public ModelAndView signIn(@RequestParam("email") String email, @RequestParam("password") String password, HttpServletRequest request, HttpServletResponse response) {try {UserProfileEntity userProfile = userService.signin(email, password);
            // 登录成功后设置 Cookie:
            AuthToken token = new AuthToken(userProfile.userId, System.currentTimeMillis() + 1000 * cookieService.getExpiresInSeconds());
            cookieService.setSessionCookie(request, response, token);
        } catch (ApiException e) {// 登录失败:
            return prepareModelAndView("signin", Map.of("email", email, "error", "Invalid email or password."));
        } catch (Exception e) {// 登录失败:
            return prepareModelAndView("signin", Map.of("email", email, "error", "Internal server error."));
        }
        // 登录成功跳转:
        return redirect("/");
    }
}

登录成功后，设置一个 Cookie 代表用户身份，以 userId:expiresAt:hash 表示。由于计算哈希引入了HmacKey，因此，客户端无法伪造 Cookie。

继续编写 UIFilter，用于验证 Cookie 并把特定用户的身份绑定到UserContext 中：

public class UIFilter {@Override
    public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain)
            throws IOException, ServletException {// 查找 Cookie:
        AuthToken auth = cookieService.findSessionCookie(req);
        Long userId = auth == null ? null : auth.userId();
        try (UserContext ctx = new UserContext(userId)) {chain.doFilter(request, response);
        }
    }
}

我们再编写一个ProxyFilter，它的目的是将页面 JavaScript 对 API 的调用转发给 API 系统：

public class ProxyFilter {@Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {Long userId = UserContext.getUserId();
        // 构造一次性 Token:
        String authToken = null;
        if (userId != null) {AuthToken token = new AuthToken(userId, System.currentTimeMillis() + 60_000);
            authToken = "Bearer" + token.toSecureString(hmacKey);
        }
        // 转发到 API 并读取响应：
        String responseJson = null;
        try {if ("GET".equals(request.getMethod())) {Map<String, String[]> params = request.getParameterMap();
                Map<String, String> query = params.isEmpty() ? null : convertParams(params);
                responseJson = tradingApiClient.get(String.class, request.getRequestURI(), authToken, query);
            } else if ("POST".equals(request.getMethod())) {responseJson = tradingApiClient.post(String.class, request.getRequestURI(), authToken,
                        readBody(request));
            }
            // 写入响应:
            response.setContentType("application/json;charset=utf-8");
            PrintWriter pw = response.getWriter();
            pw.write(responseJson);
            pw.flush();} catch (ApiException e) {// 写入错误响应:
            writeApiException(request, response, e);
        } catch (Exception e) {// 写入错误响应:
            writeApiException(request, response,
                    new ApiException(ApiError.INTERNAL_SERVER_ERROR, null, e.getMessage()));
        }
    }
}

把 ProxyFilter 挂载到 /api/*，通过 UI 转发请求的目的是简化页面 JavaScript 调用 API，一是不再需要跨域，二是 UI 已经经过了登录认证，转发过程中自动生成一次性 Token 来调用 API，这样 JavaScript 不再关心如何生成Authorization 头。

下面我们就可以开始编写页面了：