klauvsteinen/go-unifi
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add context support

Browse Source
This commit is contained in:
Paul Tyng
2020-03-26 16:12:52 -04:00
parent 19709dff50
commit d076e78005
65 changed files with 642 additions and 405 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
unifi/unifi.go
View File

@@ -2,6 +2,7 @@ package unifi //
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
@@ -44,7 +45,7 @@ func (c *Client) SetHTTPClient(hc *http.Client) error {
return nil
}
func (c *Client) Login(user, pass string) error {
func (c *Client) Login(ctx context.Context, user, pass string) error {
if c.c == nil {
c.c = &http.Client{}
@@ -52,7 +53,7 @@ func (c *Client) Login(user, pass string) error {
c.c.Jar = jar
}
err := c.do("POST", "login", &struct {
err := c.do(ctx, "POST", "login", &struct {
Username string `json:"username"`
Password string `json:"password"`
}{
@@ -66,7 +67,7 @@ func (c *Client) Login(user, pass string) error {
return nil
}
func (c *Client) do(method, relativeURL string, reqBody interface{}, respBody interface{}) error {
func (c *Client) do(ctx context.Context, method, relativeURL string, reqBody interface{}, respBody interface{}) error {
var (
reqReader io.Reader
err error
@@ -88,7 +89,7 @@ func (c *Client) do(method, relativeURL string, reqBody interface{}, respBody in
url := c.baseURL.ResolveReference(reqURL)
req, err := http.NewRequest(method, url.String(), reqReader)
req, err := http.NewRequestWithContext(ctx, method, url.String(), reqReader)
if err != nil {
return err
}