add context to windows functions

superfly · Aug 13, 2021 · 6b18e6d · 6b18e6d
1 parent 6c53a71
commit 6b18e6d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pkg/agent/client_noagent.go b/pkg/agent/client_noagent.go
@@ -53,7 +53,7 @@ func (c *Client) Ping() (int, error) {
 	return 0, nil
 }
 
-func (c *Client) Establish(slug string) error {
+func (c *Client) Establish(ctx context.Context, slug string) error {
 	if c.Client == nil {
 		return fmt.Errorf("no client set for stub agent")
 	}
@@ -79,7 +79,7 @@ func (c *Client) Establish(slug string) error {
 	return nil
 }
 
-func (c *Client) Probe(o *api.Organization) error {
+func (c *Client) Probe(ctx context.Context, o *api.Organization) error {
 	tunnel, err := c.tunnelFor(o.Slug)
 	if err != nil {
 		return fmt.Errorf("probe: can't build tunnel: %s", err)
@@ -92,7 +92,7 @@ func (c *Client) Probe(o *api.Organization) error {
 	return nil
 }
 
-func (c *Client) Instances(o *api.Organization, app string) (*Instances, error) {
+func (c *Client) Instances(ctx context.Context, o *api.Organization, app string) (*Instances, error) {
 	tunnel, err := c.tunnelFor(o.Slug)
 	if err != nil {
 		return nil, fmt.Errorf("can't build tunnel: %s", err)
@@ -115,7 +115,7 @@ type Dialer struct {
 	tunnel *wg.Tunnel
 }
 
-func (c *Client) Dialer(o *api.Organization) (*Dialer, error) {
+func (c *Client) Dialer(ctx context.Context, o *api.Organization) (*Dialer, error) {
 	if err := c.Establish(o.Slug); err != nil {
 		return nil, fmt.Errorf("dial: can't establish tunel: %s", err)
 	}