Handle workspace in Kong's client

kong/client.go

@@ -9,6 +9,7 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"os"
+	"sync"
 	"time"
 
 	"github.com/pkg/errors"
@@ -37,7 +38,9 @@ var (
 // Kong cluster
 type Client struct {
 	client                  *http.Client
-	baseURL                 string
+	defaultRootURL          string
+	workspace               string       // Do not access directly. Use Get/SetWorkspace().
+	workspaceLock           sync.RWMutex // Synchronizes access to workspace.
 	common                  service
 	Consumers               AbstractConsumerService
 	Developers              AbstractDeveloperService
@@ -117,7 +120,7 @@ func NewClient(baseURL *string, client *http.Client) (*Client, error) {
 	if err != nil {
 		return nil, errors.Wrap(err, "parsing URL")
 	}
-	kong.baseURL = url.String()
+	kong.defaultRootURL = url.String()
 
 	kong.common.client = kong
 	kong.Consumers = (*ConsumerService)(&kong.common)
@@ -164,6 +167,29 @@ func NewClient(baseURL *string, client *http.Client) (*Client, error) {
 	return kong, nil
 }
 
+// SetWorkspace sets the Kong Enteprise workspace in the client.
+// Calling this function with an empty string resets the workspace to default workspace.
+func (c *Client) SetWorkspace(workspace string) {
+	c.workspaceLock.Lock()
+	defer c.workspaceLock.Unlock()
+	c.workspace = workspace
+}
+
+//Workspace return the workspace
+func (c *Client) Workspace() string {
+	c.workspaceLock.RLock()
+	defer c.workspaceLock.RUnlock()
+	return c.workspace
+}
+
+// baseURL build the base URL from the rootURL and the workspace
+func (c *Client) baseURL(workspace string) string {
+	if len(workspace) > 0 {
+		return c.defaultRootURL + "/" + workspace
+	}
+	return c.defaultRootURL
+}
+
 // Do executes a HTTP request and returns a response
 func (c *Client) Do(ctx context.Context, req *http.Request,
 	v interface{}) (*Response, error) {
@@ -280,18 +306,20 @@ func (c *Client) Status(ctx context.Context) (*Status, error) {
 	return &s, nil
 }
 
-// Root returns the response of GET request on root of
-// Admin API (GET /).
+// Root returns the response of GET request on root of Admin API (GET / or /kong with a workspace).
 func (c *Client) Root(ctx context.Context) (map[string]interface{}, error) {
-	req, err := c.NewRequest("GET", "/", nil, nil)
+	endpoint := "/"
+	if len(c.Workspace()) > 0 {
+		endpoint = "/kong"
+	}
+	req, err := c.NewRequest("GET", endpoint, nil, nil)
 	if err != nil {
 		return nil, err
 	}
-
-	var root map[string]interface{}
-	_, err = c.Do(ctx, req, &root)
+	var info map[string]interface{}
+	_, err = c.Do(ctx, req, &info)
 	if err != nil {
 		return nil, err
 	}
-	return root, nil
+	return info, nil
 }

kong/request.go

@@ -9,13 +9,9 @@ import (
 	"github.com/google/go-querystring/query"
 )
 
-// NewRequest creates a request based on the inputs.
-// endpoint should be relative to the baseURL specified during
-// client creation.
-// body is always marshaled into JSON.
-func (c *Client) NewRequest(method, endpoint string, qs interface{},
+//NewRequestRaw creates a request based on the inputs.
+func (c *Client) NewRequestRaw(method, baseURL string, endpoint string, qs interface{},
 	body interface{}) (*http.Request, error) {
-
 	if endpoint == "" {
 		return nil, errors.New("endpoint can't be nil")
 	}
@@ -30,8 +26,7 @@ func (c *Client) NewRequest(method, endpoint string, qs interface{},
 	}
 
 	//Create a new request
-	req, err := http.NewRequest(method, c.baseURL+endpoint,
-		bytes.NewBuffer(buf))
+	req, err := http.NewRequest(method, baseURL+endpoint, bytes.NewBuffer(buf))
 	if err != nil {
 		return nil, err
 	}
@@ -51,3 +46,12 @@ func (c *Client) NewRequest(method, endpoint string, qs interface{},
 	}
 	return req, nil
 }
+
+// NewRequest creates a request based on the inputs.
+// endpoint should be relative to the baseURL specified during
+// client creation.
+// body is always marshaled into JSON.
+func (c *Client) NewRequest(method, endpoint string, qs interface{},
+	body interface{}) (*http.Request, error) {
+	return c.NewRequestRaw(method, c.baseURL(c.GetWorkspace()), endpoint, qs, body)
+}