custom-kube-scheduler/kubernetes-v1.15.4/vendor/github.com/codedellemc/goscaleio/sds.go

package goscaleio

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"reflect"

	types "github.com/codedellemc/goscaleio/types/v1"
)

type Sds struct {
	Sds    *types.Sds
	client *Client
}

func NewSds(client *Client) *Sds {
	return &Sds{
		Sds:    new(types.Sds),
		client: client,
	}
}

func NewSdsEx(client *Client, sds *types.Sds) *Sds {
	return &Sds{
		Sds:    sds,
		client: client,
	}
}

func (protectionDomain *ProtectionDomain) CreateSds(name string, ipList []string) (string, error) {
	endpoint := protectionDomain.client.SIOEndpoint

	sdsParam := &types.SdsParam{}
	sdsParam.Name = name
	sdsParam.ProtectionDomainID = protectionDomain.ProtectionDomain.ID

	if len(ipList) == 0 {
		return "", fmt.Errorf("Must provide at least 1 SDS IP")
	} else if len(ipList) == 1 {
		sdsIP := types.SdsIp{IP: ipList[0], Role: "all"}
		sdsIPList := &types.SdsIpList{sdsIP}
		sdsParam.IPList = append(sdsParam.IPList, sdsIPList)
	} else if len(ipList) >= 2 {
		sdsIP1 := types.SdsIp{IP: ipList[0], Role: "sdcOnly"}
		sdsIP2 := types.SdsIp{IP: ipList[1], Role: "sdsOnly"}
		sdsIPList1 := &types.SdsIpList{sdsIP1}
		sdsIPList2 := &types.SdsIpList{sdsIP2}
		sdsParam.IPList = append(sdsParam.IPList, sdsIPList1)
		sdsParam.IPList = append(sdsParam.IPList, sdsIPList2)
	}

	jsonOutput, err := json.Marshal(&sdsParam)
	if err != nil {
		return "", fmt.Errorf("error marshaling: %s", err)
	}
	endpoint.Path = fmt.Sprintf("/api/types/Sds/instances")

	req := protectionDomain.client.NewRequest(map[string]string{}, "POST", endpoint, bytes.NewBufferString(string(jsonOutput)))
	req.SetBasicAuth("", protectionDomain.client.Token)
	req.Header.Add("Accept", "application/json;version="+protectionDomain.client.configConnect.Version)
	req.Header.Add("Content-Type", "application/json;version="+protectionDomain.client.configConnect.Version)

	resp, err := protectionDomain.client.retryCheckResp(&protectionDomain.client.Http, req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	bs, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return "", errors.New("error reading body")
	}

	var sds types.SdsResp
	err = json.Unmarshal(bs, &sds)
	if err != nil {
		return "", err
	}

	return sds.ID, nil
}

func (protectionDomain *ProtectionDomain) GetSds() (sdss []types.Sds, err error) {
	endpoint := protectionDomain.client.SIOEndpoint
	endpoint.Path = fmt.Sprintf("/api/instances/ProtectionDomain::%v/relationships/Sds", protectionDomain.ProtectionDomain.ID)

	req := protectionDomain.client.NewRequest(map[string]string{}, "GET", endpoint, nil)
	req.SetBasicAuth("", protectionDomain.client.Token)
	req.Header.Add("Accept", "application/json;version="+protectionDomain.client.configConnect.Version)

	resp, err := protectionDomain.client.retryCheckResp(&protectionDomain.client.Http, req)
	if err != nil {
		return []types.Sds{}, fmt.Errorf("problem getting response: %v", err)
	}
	defer resp.Body.Close()

	if err = protectionDomain.client.decodeBody(resp, &sdss); err != nil {
		return []types.Sds{}, fmt.Errorf("error decoding instances response: %s", err)
	}

	return sdss, nil
}

func (protectionDomain *ProtectionDomain) FindSds(field, value string) (sds *types.Sds, err error) {
	sdss, err := protectionDomain.GetSds()
	if err != nil {
		return &types.Sds{}, nil
	}

	for _, sds := range sdss {
		valueOf := reflect.ValueOf(sds)
		switch {
		case reflect.Indirect(valueOf).FieldByName(field).String() == value:
			return &sds, nil
		}
	}

	return &types.Sds{}, errors.New("Couldn't find SDS")
}