portainer/api/dataservices/base.go

package dataservices

import (
	portainer "github.com/portainer/portainer/api"

	"golang.org/x/exp/constraints"
)

type BaseCRUD[T any, I constraints.Integer] interface {
	Create(element *T) error
	Read(ID I) (*T, error)
	ReadAll() ([]T, error)
	Update(ID I, element *T) error
	Delete(ID I) error
}

type BaseDataService[T any, I constraints.Integer] struct {
	Bucket     string
	Connection portainer.Connection
}

func (s *BaseDataService[T, I]) BucketName() string {
	return s.Bucket
}

func (service *BaseDataService[T, I]) Tx(tx portainer.Transaction) BaseDataServiceTx[T, I] {
	return BaseDataServiceTx[T, I]{
		Bucket:     service.Bucket,
		Connection: service.Connection,
		Tx:         tx,
	}
}

func (service BaseDataService[T, I]) Read(ID I) (*T, error) {
	var element *T

	return element, service.Connection.ViewTx(func(tx portainer.Transaction) error {
		var err error
		element, err = service.Tx(tx).Read(ID)

		return err
	})
}

func (service BaseDataService[T, I]) ReadAll() ([]T, error) {
	var collection = make([]T, 0)

	return collection, service.Connection.ViewTx(func(tx portainer.Transaction) error {
		var err error
		collection, err = service.Tx(tx).ReadAll()

		return err
	})
}

func (service BaseDataService[T, I]) Update(ID I, element *T) error {
	return service.Connection.UpdateTx(func(tx portainer.Transaction) error {
		return service.Tx(tx).Update(ID, element)
	})
}

func (service BaseDataService[T, I]) Delete(ID I) error {
	return service.Connection.UpdateTx(func(tx portainer.Transaction) error {
		return service.Tx(tx).Delete(ID)
	})
}