Support cookie hash (easegress-io#1132)

* Support cookie hash * Use header Cookie instead of Set-Cookie
suchen-sci · Nov 9, 2023 · bb9e904 · bb9e904
1 parent 1c0e586
commit bb9e904
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/docs/07.Reference/7.02.Filters.md b/docs/07.Reference/7.02.Filters.md
@@ -1564,7 +1564,7 @@ Rules to revise request header.
 
 | Name | Type | Description | Required |
 | ------------- | ------ | ----------------------------------------------------------------------------------------------------------- | -------- |
-| policy | string | Load balance policy, valid values are `roundRobin`, `random`, `weightedRandom`, `ipHash`, `headerHash` and `forward`, the last one is only used in `GRPCProxy` | Yes |
+| policy | string | Load balance policy, valid values are `roundRobin`, `random`, `weightedRandom`, `ipHash`, `headerHash`, `cookieHash` and `forward`, the last one is only used in `GRPCProxy` | Yes |
 | headerHashKey | string | When `policy` is `headerHash`, this option is the name of a header whose value is used for hash calculation | No |
 | stickySession | [proxy.StickySession](#proxyStickySessionSpec) | Sticky session spec | No |
 | healthCheck | [proxy.HealthCheck](#proxyHealthCheckSpec) | Health check spec, note that healthCheck is not needed if you are using service registry | No |

diff --git a/pkg/filters/proxies/loadbalance.go b/pkg/filters/proxies/loadbalance.go
@@ -39,6 +39,9 @@ const (
  LoadBalancePolicyIPHash = "ipHash"
  // LoadBalancePolicyHeaderHash is the load balance policy of HTTP header hash.
  LoadBalancePolicyHeaderHash = "headerHash"
+ // LoadBalancePolicyCookieHash is the load balance policy of HTTP cookie hash,
+ // which is the shorthand of headerHash with hash key Set-Cookie.
+ LoadBalancePolicyCookieHash = "cookieHash"
 )
 
 // LoadBalancer is the interface of a load balancer.
@@ -108,6 +111,8 @@ func (glb *GeneralLoadBalancer) Init(
  lbp = &IPHashLoadBalancePolicy{}
  case LoadBalancePolicyHeaderHash:
  lbp = &HeaderHashLoadBalancePolicy{spec: glb.spec}
+ case LoadBalancePolicyCookieHash:
+ lbp = &HeaderHashLoadBalancePolicy{spec: &LoadBalanceSpec{HeaderHashKey: "Cookie"}}
  default:
  logger.Errorf("unsupported load balancing policy: %s", glb.spec.Policy)
  lbp = &RoundRobinLoadBalancePolicy{}
@@ -235,8 +240,7 @@ func (glb *GeneralLoadBalancer) Close() {
 }
 
 // RandomLoadBalancePolicy is a load balance policy that chooses a server randomly.
-type RandomLoadBalancePolicy struct {
-}
+type RandomLoadBalancePolicy struct{}
 
 // ChooseServer chooses a server randomly.
 func (lbp *RandomLoadBalancePolicy) ChooseServer(req protocols.Request, sg *ServerGroup) *Server {
@@ -255,8 +259,7 @@ func (lbp *RoundRobinLoadBalancePolicy) ChooseServer(req protocols.Request, sg *
 }
 
 // WeightedRandomLoadBalancePolicy is a load balance policy that chooses a server randomly by weight.
-type WeightedRandomLoadBalancePolicy struct {
-}
+type WeightedRandomLoadBalancePolicy struct{}
 
 // ChooseServer chooses a server randomly by weight.
 func (lbp *WeightedRandomLoadBalancePolicy) ChooseServer(req protocols.Request, sg *ServerGroup) *Server {
@@ -272,8 +275,7 @@ func (lbp *WeightedRandomLoadBalancePolicy) ChooseServer(req protocols.Request,
 }
 
 // IPHashLoadBalancePolicy is a load balance policy that chooses a server by ip hash.
-type IPHashLoadBalancePolicy struct {
-}
+type IPHashLoadBalancePolicy struct{}
 
 // ChooseServer chooses a server by ip hash.
 func (lbp *IPHashLoadBalancePolicy) ChooseServer(req protocols.Request, sg *ServerGroup) *Server {