allow resource query to get multiple types

Swind · Apr 27, 2022 · 62d7aa0 · 62d7aa0
1 parent fd45952
commit 62d7aa0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/main.go b/main.go
@@ -320,7 +320,7 @@ func main() {
 		fmt.Println(string(nodeList))
 
 	case "getResourceList":
-		resource, err := c.GetResourceList()
+		resource, err := c.GetResourceList("")
 		if err != nil {
 			log.Printf("Error listing resources %+v\n", err)
 			os.Exit(1)

diff --git a/proxmox/client.go b/proxmox/client.go
@@ -142,13 +142,17 @@ func (c *Client) GetNodeList() (list map[string]interface{}, err error) {
 	return
 }
 
-func (c *Client) GetResourceList() (list map[string]interface{}, err error) {
-	err = c.GetJsonRetryable("/cluster/resources", &list, 3)
+func (c *Client) GetResourceList(resourceType string) (list map[string]interface{}, err error) {
+	var endpoint = "/cluster/resources"
+	if resourceType != "" {
+		endpoint = fmt.Sprintf("%s?type=%s", endpoint, resourceType)
+	}
+	err = c.GetJsonRetryable(endpoint, &list, 3)
 	return
 }
 
 func (c *Client) GetVmList() (list map[string]interface{}, err error) {
-	err = c.GetJsonRetryable("/cluster/resources?type=vm", &list, 3)
+	list, err = c.GetResourceList("vm")
 	return
 }