api: add query options on agent force-leave endpoint

hashicorp · Jan 17, 2023 · d4b4323 · d4b4323
1 parent e4a268e
commit d4b4323
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.changelog/15987.txt b/.changelog/15987.txt
@@ -0,0 +1,3 @@
+```release-note:improvement
+api: Enable setting query options on agent force-leave endpoint.
+```
diff --git a/api/agent.go b/api/agent.go
@@ -1048,7 +1048,14 @@ func (a *Agent) ForceLeavePrune(node string) error {
 // ForceLeaveOpts is used to have the agent eject a failed node or remove it
 // completely from the list of members.
 func (a *Agent) ForceLeaveOpts(node string, opts ForceLeaveOpts) error {
+	return a.ForceLeaveOptions(node, opts, nil)
+}
+
+// ForceLeaveOptions is used to have the agent eject a failed node or remove it
+// completely from the list of members. Allows usage of QueryOptions on-top of ForceLeaveOpts
+func (a *Agent) ForceLeaveOptions(node string, opts ForceLeaveOpts, q *QueryOptions) error {
 	r := a.c.newRequest("PUT", "/v1/agent/force-leave/"+node)
+	r.setQueryOptions(q)
 	if opts.Prune {
 		r.params.Set("prune", "1")
 	}