set limit parameter for get methods

skilld-labs · Dec 21, 2022 · 950968e · 950968e
1 parent b194282
commit 950968e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 11 deletions.
diff --git a/rest/sites.go b/rest/sites.go
@@ -49,8 +49,8 @@ type Membership struct {
 	Id   string `json:"id"`
 }
 
-func (c *Client) GetSites() (*SitesRes, error) {
-	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/sites", nil)
+func (c *Client) GetSites(limit int) (*SitesRes, error) {
+	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/sites?maxItems="+strconv.Itoa(limit), nil)
 	if err != nil {
 		return &SitesRes{}, err
 	}
@@ -136,8 +136,8 @@ func (c *Client) RemoveMemberFromSite(user string, id string) error {
 	return nil
 }
 
-func (c *Client) GetUserSites(user string) (*SitesRes, error) {
-	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/people/"+user+"/sites", nil)
+func (c *Client) GetUserSites(user string, limit int) (*SitesRes, error) {
+	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/people/"+user+"/sites?maxItems="+strconv.Itoa(limit), nil)
 	if err != nil {
 		return &SitesRes{}, err
 	}
@@ -148,8 +148,8 @@ func (c *Client) GetUserSites(user string) (*SitesRes, error) {
 	return response, nil
 }
 
-func (c *Client) GetArchivedSites() (*SitesRes, error) {
-	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/deleted-nodes", nil)
+func (c *Client) GetArchivedSites(limit int) (*SitesRes, error) {
+	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/deleted-nodes?maxItems="+strconv.Itoa(limit), nil)
 	if err != nil {
 		return &SitesRes{}, err
 	}
@@ -173,7 +173,7 @@ func (c *Client) RestoreArchivedNode(nodeID string) (*SiteRes, error) {
 }
 
 func (c *Client) IsArchived(shortName string) (bool, error) {
-	sitesRes, err := c.GetArchivedSites()
+	sitesRes, err := c.GetArchivedSites(100000)
 	if err != nil {
 		return false, err
 	}

diff --git a/rest/users.go b/rest/users.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
+	"strconv"
 )
 
 type Users []User
@@ -56,8 +57,8 @@ func (c *Client) Login(username string, password string) error {
 	return nil
 }
 
-func (c *Client) GetUsers() (*UsersRes, error) {
-	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/people", nil)
+func (c *Client) GetUsers(limit int) (*UsersRes, error) {
+	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/people?maxItems="+strconv.Itoa(limit), nil)
 	if err != nil {
 		return &UsersRes{}, err
 	}
@@ -80,8 +81,8 @@ func (c *Client) GetUser(userName string) (User, error) {
 	return response.Entry, nil
 }
 
-func (c *Client) GetUsersFromSiteName(siteName string) (*SiteUsers, error) {
-	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/sites/"+siteName+"/members", nil)
+func (c *Client) GetUsersFromSiteName(siteName string, limit int) (*SiteUsers, error) {
+	req, err := http.NewRequest("GET", c.getUrl()+"/alfresco/api/-default-/public/alfresco/versions/1/sites/"+siteName+"/members?maxItems="+strconv.Itoa(limit), nil)
 	if err != nil {
 		return &SiteUsers{}, err
 	}