v9.0.0

pkg/cloudapi/disks/filter.go

@@ -0,0 +1,321 @@
+package disks
+
+import (
+	"context"
+	"strconv"
+
+	"repository.basistech.ru/BASIS/dynamix-golang-sdk/v9/interfaces"
+	"repository.basistech.ru/BASIS/dynamix-golang-sdk/v9/pkg/cloudapi/k8s"
+	"repository.basistech.ru/BASIS/dynamix-golang-sdk/v9/pkg/cloudapi/lb"
+)
+
+// FilterByID returns ListDisks with specified ID.
+func (ld ListDisks) FilterByID(id uint64) ListDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.ID == id
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByName returns ListDisks with specified Name.
+func (ld ListDisks) FilterByName(name string) ListDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.Name == name
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByStatus returns ListDisks with specified Status.
+func (ld ListDisks) FilterByStatus(status string) ListDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.Status == status
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByTechStatus returns ListDisks with specified TechStatus.
+func (ld ListDisks) FilterByTechStatus(techStatus string) ListDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.TechStatus == techStatus
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByComputeID is used to filter ListDisks attached to specified compute.
+func (ld ListDisks) FilterByComputeID(computeID uint64) ListDisks {
+	predicate := func(idisk ItemDisk) bool {
+		for k := range idisk.Computes {
+			if k == strconv.FormatUint(computeID, 10) {
+				return true
+			}
+		}
+
+		return false
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByK8SID is used to filter ListDisks by specified K8S cluster.
+func (ld ListDisks) FilterByK8SID(ctx context.Context, k8sID uint64, decortClient interfaces.Caller) (*ListDisks, error) {
+	caller := k8s.New(decortClient)
+
+	req := k8s.GetRequest{
+		K8SID: k8sID,
+	}
+
+	cluster, err := caller.Get(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	var result ListDisks
+
+	for _, masterCompute := range cluster.K8SGroups.Masters.DetailedInfo {
+		result.Data = append(result.Data, ld.FilterByComputeID(masterCompute.ID).Data...)
+	}
+
+	for _, workerGroup := range cluster.K8SGroups.Workers {
+		for _, workerCompute := range workerGroup.DetailedInfo {
+			result.Data = append(result.Data, ld.FilterByComputeID(workerCompute.ID).Data...)
+		}
+	}
+
+	result.EntryCount = uint64(len(result.Data))
+
+	return &result, nil
+}
+
+// FilterByLBID is used to filter ListDisks used by computes inside specified Load Balancer.
+func (ld ListDisks) FilterByLBID(ctx context.Context, lbID uint64, decortClient interfaces.Caller) (*ListDisks, error) {
+	caller := lb.New(decortClient)
+
+	req := lb.GetRequest{
+		LBID: lbID,
+	}
+
+	foundLB, err := caller.Get(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	var result ListDisks
+	result.Data = append(result.Data, ld.FilterByComputeID(foundLB.PrimaryNode.ComputeID).Data...)
+	result.Data = append(result.Data, ld.FilterByComputeID(foundLB.SecondaryNode.ComputeID).Data...)
+
+	result.EntryCount = uint64(len(result.Data))
+
+	return &result, nil
+}
+
+// FilterFunc allows filtering ListDisks based on a user-specified predicate.
+func (ld ListDisks) FilterFunc(predicate func(ItemDisk) bool) ListDisks {
+	var result ListDisks
+
+	for _, item := range ld.Data {
+		if predicate(item) {
+			result.Data = append(result.Data, item)
+		}
+	}
+
+	result.EntryCount = uint64(len(result.Data))
+
+	return result
+}
+
+// FindOne returns first found ItemDisk
+// If none was found, returns an empty struct.
+func (ld ListDisks) FindOne() ItemDisk {
+	if len(ld.Data) == 0 {
+		return ItemDisk{}
+	}
+
+	return ld.Data[0]
+}
+
+// FilterByID returns ListSearchDisks with specified ID.
+func (ld ListSearchDisks) FilterByID(id uint64) ListSearchDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.ID == id
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByName returns ListSearchDisks with specified Name.
+func (ld ListSearchDisks) FilterByName(name string) ListSearchDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.Name == name
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByStatus returns ListSearchDisks with specified Status.
+func (ld ListSearchDisks) FilterByStatus(status string) ListSearchDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.Status == status
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByTechStatus returns ListSearchDisks with specified TechStatus.
+func (ld ListSearchDisks) FilterByTechStatus(techStatus string) ListSearchDisks {
+	predicate := func(idisk ItemDisk) bool {
+		return idisk.TechStatus == techStatus
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByComputeID is used to filter ListSearchDisks attached to specified compute.
+func (ld ListSearchDisks) FilterByComputeID(computeID uint64) ListSearchDisks {
+	predicate := func(idisk ItemDisk) bool {
+		for k := range idisk.Computes {
+			if k == strconv.FormatUint(computeID, 10) {
+				return true
+			}
+		}
+
+		return false
+	}
+
+	return ld.FilterFunc(predicate)
+}
+
+// FilterByK8SID is used to filter ListSearchDisks by specified K8S cluster.
+func (ld ListSearchDisks) FilterByK8SID(ctx context.Context, k8sID uint64, decortClient interfaces.Caller) (ListSearchDisks, error) {
+	caller := k8s.New(decortClient)
+
+	req := k8s.GetRequest{
+		K8SID: k8sID,
+	}
+
+	cluster, err := caller.Get(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	var result ListSearchDisks
+
+	for _, masterCompute := range cluster.K8SGroups.Masters.DetailedInfo {
+		result = append(result, ld.FilterByComputeID(masterCompute.ID)...)
+	}
+
+	for _, workerGroup := range cluster.K8SGroups.Workers {
+		for _, workerCompute := range workerGroup.DetailedInfo {
+			result = append(result, ld.FilterByComputeID(workerCompute.ID)...)
+		}
+	}
+
+	return result, nil
+}
+
+// FilterByLBID is used to filter ListSearchDisks used by computes inside specified Load Balancer.
+func (ld ListSearchDisks) FilterByLBID(ctx context.Context, lbID uint64, decortClient interfaces.Caller) (ListSearchDisks, error) {
+	caller := lb.New(decortClient)
+
+	req := lb.GetRequest{
+		LBID: lbID,
+	}
+
+	foundLB, err := caller.Get(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+
+	var result ListSearchDisks
+	result = append(result, ld.FilterByComputeID(foundLB.PrimaryNode.ComputeID)...)
+	result = append(result, ld.FilterByComputeID(foundLB.SecondaryNode.ComputeID)...)
+
+	return result, nil
+}
+
+// FilterFunc allows filtering ListSearchDisks based on a user-specified predicate.
+func (ld ListSearchDisks) FilterFunc(predicate func(ItemDisk) bool) ListSearchDisks {
+	var result ListSearchDisks
+
+	for _, item := range ld {
+		if predicate(item) {
+			result = append(result, item)
+		}
+	}
+
+	return result
+}
+
+// FindOne returns first found ItemDisk
+// If none was found, returns an empty struct.
+func (ld ListSearchDisks) FindOne() ItemDisk {
+	if len(ld) == 0 {
+		return ItemDisk{}
+	}
+
+	return ld[0]
+}
+
+// FilterByID returns ListDisksUnattached with specified ID.
+func (lu ListDisksUnattached) FilterByID(id uint64) ListDisksUnattached {
+	predicate := func(idisk ItemDiskUnattached) bool {
+		return idisk.ID == id
+	}
+
+	return lu.FilterFunc(predicate)
+}
+
+// FilterByName returns ListDisksUnattached with specified Name.
+func (lu ListDisksUnattached) FilterByName(name string) ListDisksUnattached {
+	predicate := func(idisk ItemDiskUnattached) bool {
+		return idisk.Name == name
+	}
+
+	return lu.FilterFunc(predicate)
+}
+
+// FilterByStatus returns ListDisksUnattached with specified Status.
+func (lu ListDisksUnattached) FilterByStatus(status string) ListDisksUnattached {
+	predicate := func(idisk ItemDiskUnattached) bool {
+		return idisk.Status == status
+	}
+
+	return lu.FilterFunc(predicate)
+}
+
+// FilterByTechStatus returns ListDisksUnattached with specified TechStatus.
+func (lu ListDisksUnattached) FilterByTechStatus(techStatus string) ListDisksUnattached {
+	predicate := func(idisk ItemDiskUnattached) bool {
+		return idisk.TechStatus == techStatus
+	}
+
+	return lu.FilterFunc(predicate)
+}
+
+// FilterFunc allows filtering ListDisksUnattached based on a user-specified predicate.
+func (lu ListDisksUnattached) FilterFunc(predicate func(ItemDiskUnattached) bool) ListDisksUnattached {
+	var result ListDisksUnattached
+
+	for _, item := range lu.Data {
+		if predicate(item) {
+			result.Data = append(result.Data, item)
+		}
+	}
+
+	result.EntryCount = uint64(len(result.Data))
+
+	return result
+}
+
+// FindOne returns first found ItemDiskUnattached
+// If none was found, returns an empty struct.
+func (lu ListDisksUnattached) FindOne() ItemDiskUnattached {
+	if len(lu.Data) == 0 {
+		return ItemDiskUnattached{}
+	}
+
+	return lu.Data[0]
+}