load_balancing: support header session affinity policy

cloudflare · Jul 5, 2023 · 46798ef · 46798ef
1 parent bddcc22
commit 46798ef
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/.changelog/1302.txt b/.changelog/1302.txt
@@ -0,0 +1,3 @@
+```release-note:enhancement
+load_balancing: support header session affinity policy
+```
diff --git a/load_balancing.go b/load_balancing.go
@@ -253,17 +253,21 @@ type LocationStrategy struct {
 // LoadBalancerRuleOverridesSessionAffinityAttrs mimics SessionAffinityAttributes without the
 // DrainDuration field as that field can not be overwritten via rules.
 type LoadBalancerRuleOverridesSessionAffinityAttrs struct {
-	SameSite             string `json:"samesite,omitempty"`
-	Secure               string `json:"secure,omitempty"`
-	ZeroDowntimeFailover string `json:"zero_downtime_failover,omitempty"`
+	SameSite             string   `json:"samesite,omitempty"`
+	Secure               string   `json:"secure,omitempty"`
+	ZeroDowntimeFailover string   `json:"zero_downtime_failover,omitempty"`
+	Headers              []string `json:"headers,omitempty"`
+	RequireAllHeaders    *bool    `json:"require_all_headers,omitempty"`
 }
 
 // SessionAffinityAttributes represents the fields used to set attributes in a load balancer session affinity cookie.
 type SessionAffinityAttributes struct {
-	SameSite             string `json:"samesite,omitempty"`
-	Secure               string `json:"secure,omitempty"`
-	DrainDuration        int    `json:"drain_duration,omitempty"`
-	ZeroDowntimeFailover string `json:"zero_downtime_failover,omitempty"`
+	SameSite             string   `json:"samesite,omitempty"`
+	Secure               string   `json:"secure,omitempty"`
+	DrainDuration        int      `json:"drain_duration,omitempty"`
+	ZeroDowntimeFailover string   `json:"zero_downtime_failover,omitempty"`
+	Headers              []string `json:"headers,omitempty"`
+	RequireAllHeaders    bool     `json:"require_all_headers,omitempty"`
 }
 
 // LoadBalancerOriginHealth represents the health of the origin.