Vert.x Web用户认证与会话管理：实现持久化登录体验

import io.vertx.core.AbstractVerticle;
import io.vertx.core.Promise;
import io.vertx.core.http.HttpServerOptions;
import io.vertx.ext.auth.User;
import io.vertx.ext.auth.authentication.AuthenticationProvider;
import io.vertx.ext.auth.authentication.UsernamePasswordCredentials;
import io.vertx.ext.web.Router;
import io.vertx.ext.web.handler.AuthenticationHandler;
import io.vertx.ext.web.handler.CookieHandler;
import io.vertx.ext.web.handler.SessionHandler;
import io.vertx.ext.web.sstore.LocalSessionStore;
import io.vertx.core.json.JsonObject;

public class PersistentAuthExample extends AbstractVerticle {

    @Override
    public void start(Promise<Void> startPromise) {
        // 1. 创建一个Router
        Router router = Router.router(vertx);

        // 2. 配置CookieHandler - 必须在SessionHandler之前
        // CookieHandler负责解析和设置HTTP Cookie
        router.route().handler(CookieHandler.create());

        // 3. 配置SessionHandler
        // SessionHandler依赖于CookieHandler，用于管理用户会话数据。
        // LocalSessionStore适用于开发测试，生产环境建议使用分布式存储。
        router.route().handler(SessionHandler.create(LocalSessionStore.create(vertx)));

        // 4. 创建一个模拟的认证提供者
        // 在实际应用中，这会是OAuth2Auth、JDBCAuth、JWTAuth等具体的认证实现。
        AuthenticationProvider myAuthProvider = credentials -> {
            // 这是一个简化示例，实际认证逻辑会更复杂，例如验证用户名密码
            if (credentials instanceof UsernamePasswordCredentials) {
                UsernamePasswordCredentials upc = (UsernamePasswordCredentials) credentials;
                if ("user".equals(upc.getUsername()) && "password".equals(upc.getPassword())) {
                    System.out.println("模拟认证成功！");
                    return Promise.succeededFuture(User.create(new JsonObject().put("username", upc.getUsername())));
                }
            }
            System.out.println("模拟认证失败！");
            return Promise.fail("Invalid credentials");
        };

        // 5. 创建AuthenticationHandler
        // 这个处理器将负责触发认证流程，并在用户认证后将User对象设置到RoutingContext。
        // Vert.x Web的AuthenticationHandler会自动将User对象存储到Session中。
        AuthenticationHandler authHandler = routingContext -> {
            // 如果会话中已经有User，说明用户已通过认证，直接放行
            if (routingContext.user() != null) {
                System.out.println("用户已在会话中认证: " + routingContext.user().principal().getString("username"));
                routingContext.next();
                return;
            }

            // 否则，模拟认证流程。在真实应用中，这会是重定向到登录页、处理表单提交或OAuth2回调等。
            System.out.println("会话中无用户，尝试进行认证...");

            // 模拟一个认证成功。在实际的OAuth2AuthHandler等中，
            // 认证成功后会调用 routingContext.setUser(user)，
            // 而SessionHandler会自动将此User对象存储到当前会话中。
            // 这里我们直接调用myAuthProvider进行认证，假设 credentials 可以从请求中获取
            // 比如从表单参数中获取用户名和密码
            UsernamePasswordCredentials credentials = new UsernamePasswordCredentials("user", "password"); // 模拟从请求中获取凭证
            myAuthProvider.authenticate(credentials)
                .onSuccess(user -> {
                    System.out.println("认证成功，将用户设置到RoutingContext中。SessionHandler将负责持久化。");
                    routingContext.setUser(user); // Vert.x Web的认证处理器会自动将User存入Session
                    // 如果需要存储更多自定义会话数据，可以使用 routingContext.session().put("key", "value");
                    routingContext.session().put("customData", "some_value_for_" + user.principal().getString("username"));
                    routingContext.next();
                })
                .onFailure(err -> {
                    System.err.println("认证失败: " + err.getMessage());
                    routingContext.fail(401); // 未授权
                });
        };

        // 6. 将AuthenticationHandler应用于受保护的路径
        router.route("/protected/*").handler(authHandler);

        // 7. 定义受保护的资源处理器
        router.get("/protected/resource").handler(ctx -> {
            User user = ctx.user();
            String username = user != null ? user.principal().getString("username") : "Guest";
            ctx.response().end("欢迎, " + username + "! 这是一个受保护的资源。会话ID: " + ctx.session().id() +
                               "\n自定义会话数据: " + ctx.session().get("customData", String.class));
        });

        // 8. 定义一个非受保护的公共路径
        router.get("/public").handler(ctx -> {
            ctx.response().end("这是一个公共资源。");
        });

        // 9. 启动HTTP服务器
        vertx.createHttpServer(new HttpServerOptions().setPort(8080).setHost("localhost"))
            .requestHandler(router)
            .listen()
            .onSuccess(server -> {
                System.out.println("HTTP 服务器已在端口 8080 启动: http://localhost:8080");
                System.out.println("访问 http://localhost:8080/protected/resource 进行测试");
                startPromise.complete();
            })
            .onFailure(startPromise::fail);
    }
}