resource pools

Author: sp-yduck

api/pool_type.go

@@ -0,0 +1,24 @@
+package api
+
+type ResourcePool struct {
+	PoolID  string `json:"poolid"`
+	Comment string `json:"comment"`
+}
+
+type ResourcePoolMember struct {
+	Members []map[string]interface{} `json:"members"`
+}
+
+type UpdateResourcePoolOption struct {
+	PoolID  string `json:"poolid"`
+	Comment string `json:"comment,omitempty"`
+
+	// set true for removing object from resource pool
+	Delete bool `json:"delete,omitempty"`
+
+	// array of storage name
+	Storage []string `json:"storage,omitempty"`
+
+	// array of vmid
+	VMs []string `json:"vms,omitempty"`
+}

proxmox/pool.go

@@ -0,0 +1,78 @@
+package proxmox
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/sp-yduck/proxmox-go/api"
+)
+
+type Pool struct {
+	service *Service
+	Pool    *api.ResourcePool
+}
+
+func (s *Service) Pools(ctx context.Context) ([]*api.ResourcePool, error) {
+	return s.restclient.GetResourcePools(ctx)
+}
+
+func (s *Service) Pool(ctx context.Context, id string) (*Pool, error) {
+	pool, err := s.restclient.GetResourcPool(ctx, id)
+	if err != nil {
+		return nil, err
+	}
+	return &Pool{service: s, Pool: pool}, nil
+}
+
+func (s *Service) CreatePool(ctx context.Context, pool api.ResourcePool) (*Pool, error) {
+	if err := s.restclient.CreateResourcePool(ctx, pool); err != nil {
+		return nil, err
+	}
+	return s.Pool(ctx, pool.PoolID)
+}
+
+func (s *Service) DeletePool(ctx context.Context, id string) error {
+	return s.restclient.DeleteResourcePool(ctx, id)
+}
+
+func (p *Pool) AddVMs(ctx context.Context, vmids []int) error {
+	opts := api.UpdateResourcePoolOption{
+		PoolID: p.Pool.PoolID,
+		VMs:    itoaSlice(vmids),
+	}
+	return p.service.restclient.UpdateResourcePool(ctx, opts)
+}
+
+func (p *Pool) RemoveVMs(ctx context.Context, vmids []int) error {
+	opts := api.UpdateResourcePoolOption{
+		PoolID: p.Pool.PoolID,
+		VMs:    itoaSlice(vmids),
+		Delete: true,
+	}
+	return p.service.restclient.UpdateResourcePool(ctx, opts)
+}
+
+func (p *Pool) AddStorages(ctx context.Context, storageNames []string) error {
+	opts := api.UpdateResourcePoolOption{
+		PoolID:  p.Pool.PoolID,
+		Storage: storageNames,
+	}
+	return p.service.restclient.UpdateResourcePool(ctx, opts)
+}
+
+func (p *Pool) RemoveStorages(ctx context.Context, storageNames []string) error {
+	opts := api.UpdateResourcePoolOption{
+		PoolID:  p.Pool.PoolID,
+		Storage: storageNames,
+		Delete:  true,
+	}
+	return p.service.restclient.UpdateResourcePool(ctx, opts)
+}
+
+func itoaSlice(i []int) []string {
+	a := []string{}
+	for _, x := range i {
+		a = append(a, fmt.Sprintf("%d", x))
+	}
+	return a
+}

rest/pool.go

@@ -0,0 +1,44 @@
+package rest
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/sp-yduck/proxmox-go/api"
+)
+
+func (c *RESTClient) GetResourcePools(ctx context.Context) ([]*api.ResourcePool, error) {
+	var pools []*api.ResourcePool
+	if err := c.Get(ctx, "/pools", &pools); err != nil {
+		return nil, err
+	}
+	return pools, nil
+}
+
+func (c *RESTClient) GetResourcPool(ctx context.Context, id string) (*api.ResourcePool, error) {
+	pools, err := c.GetResourcePools(ctx)
+	if err != nil {
+		return nil, err
+	}
+	for _, p := range pools {
+		if p.PoolID == id {
+			return p, nil
+		}
+	}
+	return nil, NotFoundErr
+}
+
+func (c *RESTClient) CreateResourcePool(ctx context.Context, pool api.ResourcePool) error {
+	return c.Post(ctx, "/pools", pool, nil)
+}
+
+func (c *RESTClient) DeleteResourcePool(ctx context.Context, id string) error {
+	var res map[string]interface{}
+	path := fmt.Sprintf("/pools/%s", id)
+	return c.Delete(ctx, path, nil, &res)
+}
+
+func (c *RESTClient) UpdateResourcePool(ctx context.Context, option api.UpdateResourcePoolOption) error {
+	path := fmt.Sprintf("/pools/%s", option.PoolID)
+	return c.Put(ctx, path, option, nil)
+}

rest/pool_test.go

@@ -0,0 +1,42 @@
+package rest
+
+import (
+	"context"
+
+	"github.com/sp-yduck/proxmox-go/api"
+)
+
+func (s *TestSuite) TestGetResourcePools() {
+	pools, err := s.restclient.GetResourcePools(context.Background())
+	if err != nil {
+		s.T().Fatalf("failed to get pools: %v", err)
+	}
+	s.T().Logf("get pools: %v", pools)
+}
+
+func (s *TestSuite) TestCreateResourcePool() {
+	pool := api.ResourcePool{
+		PoolID:  "proxmox-go-test",
+		Comment: "proxmox-go-test comment",
+	}
+	if err := s.restclient.CreateResourcePool(context.Background(), pool); err != nil {
+		s.T().Fatalf("failed to create pool: %v", err)
+	}
+}
+
+func (s *TestSuite) TestDeleteResourcePool() {
+	if err := s.restclient.DeleteResourcePool(context.Background(), "proxmox-go-test"); err != nil {
+		s.T().Fatalf("failed to delete pool: %v", err)
+	}
+}
+
+func (s *TestSuite) TestUpdateResourcePool() {
+	opts := api.UpdateResourcePoolOption{
+		PoolID: "proxmox-go-test",
+		VMs:    []string{"102", "103"},
+		Delete: true,
+	}
+	if err := s.restclient.UpdateResourcePool(context.Background(), opts); err != nil {
+		s.T().Fatalf("failed to update pool: %v", err)
+	}
+}