rudder-go-sdk/storage.go

package rudder

import (
	"context"
	"net/http"
	"net/url"
	"strconv"

	"hub.rudder.build/rudder/rudder-go-sdk/models"
)

type StorageService struct {
	transport *transport
}

type ListStorageParameters struct {
	Limit  int
	Cursor string
	Search string
}

func (s *StorageService) List(ctx context.Context, parameters ListStorageParameters) (*models.GetProjectStorageResponse, error) {
	query := url.Values{}
	if parameters.Limit > 0 {
		query.Set("limit", strconv.Itoa(parameters.Limit))
	}
	if parameters.Cursor != "" {
		query.Set("cursor", parameters.Cursor)
	}
	if parameters.Search != "" {
		query.Set("search", parameters.Search)
	}

	var response models.GetProjectStorageResponse
	if err := s.transport.do(ctx, http.MethodGet, "/project-storage", query, nil, &response); err != nil {
		return nil, err
	}
	return &response, nil
}

func (s *StorageService) Get(ctx context.Context, itemType string) (*models.ProjectStorageItem, error) {
	var response models.AdminGetProjectStorageResponse
	if err := s.transport.do(ctx, http.MethodGet, "/project-storage/"+url.PathEscape(itemType), nil, nil, &response); err != nil {
		return nil, err
	}
	return &response.Item, nil
}

type UpsertStorageParameters struct {
	IdempotencyKey string
	Items          []models.AdminProjectStorageItem
}

func (s *StorageService) Upsert(ctx context.Context, parameters UpsertStorageParameters) error {
	body := models.BatchUpsertProjectStorageRequest{
		IdempotencyKey: parameters.IdempotencyKey,
		Items:          parameters.Items,
	}
	return s.transport.do(ctx, http.MethodPut, "/project-storage", nil, body, nil)
}

type DeleteStorageParameters struct {
	IdempotencyKey string
	Types          []string
}

func (s *StorageService) Delete(ctx context.Context, parameters DeleteStorageParameters) error {
	body := models.BatchDeleteProjectStorageRequest{
		IdempotencyKey: parameters.IdempotencyKey,
		Types:          parameters.Types,
	}
	return s.transport.do(ctx, http.MethodDelete, "/project-storage", nil, body, nil)
}