Apache APISIX 实现基于 Cookie 的流量分流与会话保持方案

在实际业务场景中，我们经常需要根据特定条件对流量进行精细化控制。本文将详细介绍如何使用 Apache APISIX 实现基于 Cookie 的智能流量分流方案，该方案能够实现初始请求按比例分流，并通过 Cookie 机制保持后续请求的会话一致性。

业务需求分析

假设我们有一个国际化网站，URL 结构为 https://xyz.com/ca/fr/index.html，其中 ca 表示国家（加拿大），fr 表示法语版本。业务需求如下：

1. 初始请求需要进行流量分流：10% 的流量导向英语版本（/ca/en/index.html），90% 保持法语版本
2. 对于被分流的 10% 流量，APISIX 需要设置一个 Cookie
3. 后续请求如果携带该 Cookie，则直接路由到英语版本，不再进行流量分流
4. 没有 Cookie 的请求继续按 10:90 的比例分流

这实际上是一种会话保持（Session Affinity）的实现方案，确保用户一旦被分流到特定版本后，后续访问保持一致性。

技术实现方案

1. 流量分流基础实现

Apache APISIX 的 traffic-split 插件可以实现基本的流量分流功能。配置示例如下：

routes:
  - uri: /ca/fr/index.html
    plugins:
      traffic-split:
        rules:
          - weighted_upstreams:
              - weight: 90
                upstream_id: 1  # 法语版本上游
              - weight: 10
                upstream_id: 2  # 英语版本上游

2. Cookie 设置与会话保持

要实现完整的业务需求，我们需要结合多个插件：

1. 流量分流：使用 traffic-split 插件
2. Cookie 设置：使用 proxy-rewrite 插件添加 Set-Cookie 头
3. 会话保持：使用 traffic-split 的条件分流功能

完整配置方案如下：

routes:
  - uri: /ca/fr/index.html
    plugins:
      traffic-split:
        rules:
          - match:
              - vars: [["cookie_lang", "==", "en"]]
            weighted_upstreams:
              - weight: 100
                upstream_id: 2  # 直接路由到英语版本
          - weighted_upstreams:
              - weight: 90
                upstream_id: 1  # 法语版本
              - weight: 10
                upstream:
                  name: english_upstream
                  nodes:
                    "english.service:80": 1
                  type: roundrobin
                  plugins:
                    proxy-rewrite:
                      headers:
                        Set-Cookie: "lang=en; Path=/; Max-Age=86400"

3. 方案解析

1. 流量分流逻辑：

   • 首先检查请求是否携带 lang=en 的 Cookie
   • 如果存在，100% 路由到英语版本
   • 如果不存在，按 10:90 比例分流

2. Cookie 设置机制：

   • 对于被分流的 10% 流量，APISIX 会在响应中添加 Set-Cookie 头
   • Cookie 设置为 lang=en，有效期 1 天（86400 秒）
   • 路径设置为 / 确保全站有效

3. 会话保持实现：

   • 浏览器后续请求会自动携带该 Cookie
   • APISIX 检测到 Cookie 后直接路由到英语版本
   • 实现用户会话的一致性

高级配置建议

1. Cookie 安全性：

   • 生产环境建议添加 Secure 和 HttpOnly 属性
   • 可以考虑使用 SameSite 属性防止 CSRF 攻击

2. 分流算法优化：

   • 默认使用随机分流，可能导致短时间内比例不准确
   • 可考虑基于客户端 IP 的哈希算法保证更稳定的分流比例

3. 监控与告警：

   • 监控各版本的实际流量比例
   • 设置异常告警，如某版本流量突降等

常见问题处理

1. Cookie 不生效：

   • 检查域名和路径设置是否正确
   • 确保客户端没有禁用 Cookie
   • 验证 Set-Cookie 头是否被正确返回

2. 分流比例不准确：

   • 检查流量是否足够大（小流量下比例可能波动较大）
   • 确认是否有缓存或 CDN 影响了分流

3. 性能考虑：

   • 大量 Cookie 检查可能增加性能开销
   • 对于高并发场景，建议进行压力测试

总结

通过 Apache APISIX 的灵活配置，我们可以实现复杂的流量控制策略。本文介绍的基于 Cookie 的流量分流方案不仅满足了初始按比例分流的需求，还通过 Cookie 机制实现了会话保持，确保用户体验的一致性。这种方案特别适用于 A/B 测试、多语言版本切换、灰度发布等场景。

实际部署时，建议先在测试环境充分验证，逐步调整参数，确保系统稳定性和业务需求的完美平衡。