github.com/chnsz/golangsdk@v0.0.0-20240506093406-85a3fbfa605b/openstack/cce/v3/nodes/requests.go

package nodes

import (
	"reflect"

	"github.com/chnsz/golangsdk"
	"github.com/chnsz/golangsdk/openstack/common/tags"
)

var RequestOpts golangsdk.RequestOpts = golangsdk.RequestOpts{
	MoreHeaders: map[string]string{"Content-Type": "application/json"},
}

// ListOpts allows the filtering of list data using given parameters.
type ListOpts struct {
	Name  string `json:"name"`
	Uid   string `json:"uid"`
	Phase string `json:"phase"`
}

// List returns collection of nodes.
func List(client *golangsdk.ServiceClient, clusterID string, opts ListOpts) ([]Nodes, error) {
	var r ListResult
	_, r.Err = client.Get(rootURL(client, clusterID), &r.Body, &golangsdk.RequestOpts{
		OkCodes:     []int{200},
		MoreHeaders: RequestOpts.MoreHeaders, JSONBody: nil,
	})

	allNodes, err := r.ExtractNode()

	if err != nil {
		return nil, err
	}

	return FilterNodes(allNodes, opts), nil
}

func FilterNodes(nodes []Nodes, opts ListOpts) []Nodes {

	var refinedNodes []Nodes
	var matched bool

	m := map[string]FilterStruct{}

	if opts.Name != "" {
		m["Name"] = FilterStruct{Value: opts.Name, Driller: []string{"Metadata"}}
	}
	if opts.Uid != "" {
		m["Id"] = FilterStruct{Value: opts.Uid, Driller: []string{"Metadata"}}
	}

	if opts.Phase != "" {
		m["Phase"] = FilterStruct{Value: opts.Phase, Driller: []string{"Status"}}
	}

	if len(m) > 0 && len(nodes) > 0 {
		for _, nodes := range nodes {
			matched = true

			for key, value := range m {
				if sVal := GetStructNestedField(&nodes, key, value.Driller); !(sVal == value.Value) {
					matched = false
				}
			}
			if matched {
				refinedNodes = append(refinedNodes, nodes)
			}
		}
	} else {
		refinedNodes = nodes
	}
	return refinedNodes
}

func GetStructNestedField(v *Nodes, field string, structDriller []string) string {
	r := reflect.ValueOf(v)
	for _, drillField := range structDriller {
		f := reflect.Indirect(r).FieldByName(drillField).Interface()
		r = reflect.ValueOf(f)
	}
	f1 := reflect.Indirect(r).FieldByName(field)
	return string(f1.String())
}

type FilterStruct struct {
	Value   string
	Driller []string
}

// CreateOptsBuilder allows extensions to add additional parameters to the
// Create request.
type CreateOpts struct {
	// API type, fixed value Node
	Kind string `json:"kind" required:"true"`
	// API version, fixed value v3
	ApiVersion string `json:"apiversion" required:"true"`
	// Metadata required to create a Node
	Metadata CreateMetaData `json:"metadata"`
	// specifications to create a Node
	Spec Spec `json:"spec" required:"true"`
}

// Metadata required to create a Node
type CreateMetaData struct {
	// Node name
	Name string `json:"name,omitempty"`
	// Node tag, key value pair format
	Labels map[string]string `json:"labels,omitempty"`
	// Node annotation, key value pair format
	Annotations map[string]string `json:"annotations,omitempty"`
}

// Create accepts a CreateOpts struct and uses the values to create a new
// logical Node. When it is created, the Node does not have an internal
// interface
type CreateOptsBuilder interface {
	ToNodeCreateMap() (map[string]interface{}, error)
}

// ToNodeCreateMap builds a create request body from CreateOpts.
func (opts CreateOpts) ToNodeCreateMap() (map[string]interface{}, error) {
	return golangsdk.BuildRequestBody(opts, "")
}

// Create accepts a CreateOpts struct and uses the values to create a new
// logical node.
func Create(c *golangsdk.ServiceClient, clusterid string, opts CreateOptsBuilder) (r CreateResult) {
	b, err := opts.ToNodeCreateMap()
	if err != nil {
		r.Err = err
		return
	}
	reqOpt := &golangsdk.RequestOpts{OkCodes: []int{201}}
	_, r.Err = c.Post(rootURL(c, clusterid), b, &r.Body, reqOpt)
	return
}

type AddOpts struct {
	// API type, fixed value List
	Kind string `json:"kind" required:"true"`
	// API version, fixed value v3
	ApiVersion string `json:"apiversion" required:"true"`
	// List of nodes to add
	NodeList []AddNode `json:"nodeList" required:"true"`
}

type AddNode struct {
	ServerID string      `json:"serverID" required:"true"`
	Spec     AddNodeSpec `json:"spec" required:"true"`
}

type AddNodeSpec struct {
	// The OS of the node
	Os string `json:"os" required:"true"`
	// Node login parameters
	Login LoginSpec `json:"login" required:"true"`
	//Node name
	Name string `json:"name,omitempty"`
	// ECS server config of the node
	ServerConfig *ServerConfig `json:"serverConfig,omitempty"`
	// Volume management config of the node
	VolumeConfig *VolumeConfig `json:"volumeConfig,omitempty"`
	// Runtime config of the node
	RuntimeConfig *RuntimeConfig `json:"runtimeConfig,omitempty"`
	// Kubernetes options of the node
	K8sOptions *K8sOptions `json:"k8sOptions,omitempty"`
	// Custom lifecycle config of the node
	Lifecycle *Lifecycle `json:"lifecycle,omitempty"`
	// Extended parameter
	ExtendParam map[string]interface{} `json:"extendParam,omitempty"`
	// The initialized conditions
	InitializedConditions []string `json:"initializedConditions,omitempty"`
}

type ServerConfig struct {
	// Tag of a VM, key value pair format
	UserTags []tags.ResourceTag `json:"userTags,omitempty"`
	// System disk parameter of the node
	RootVolume *RootVolume
}

type RootVolume struct {
	// Custom image ID
	ImageID string `json:"imageID,omitempty"`
	// User master key ID, default to empty, means the disk in not encrypted
	CmkID string `json:"cmkID,omitempty"`
}

type VolumeConfig struct {
	// Docker data disk configurations
	LvmConfig string `json:"lvmConfig,omitempty"`
	// Disk initialization configuration management parameters
	Storage *StorageSpec `json:"storage,omitempty"`
}

type RuntimeConfig struct {
	// The available disk space of a single Docker container on the node in device mapper mode
	DockerBaseSize int `json:"dockerBaseSize,omitempty"`
	// The runtime spec
	Runtime *RunTimeSpec `json:"runtime,omitempty"`
}

type K8sOptions struct {
	// Tag of a Kubernetes node, key value pair format
	Labels map[string]string `json:"labels,omitempty"`
	// taints to created nodes to configure anti-affinity
	Taints []TaintSpec `json:"taints,omitempty"`
	// The maximum number of pods allowed to be created on a node
	MaxPods int `json:"maxPods,omitempty"`
	// NIC queue number configuration
	NicMultiQueue string `json:"nicMultiqueue,omitempty"`
	// NIC pre-binding ratio configuration
	NicThreshold string `json:"nicThreshold,omitempty"`
}

type Lifecycle struct {
	// Preinstall script
	Preinstall string `json:"preInstall,omitempty"`
	// Postinstall script
	PostInstall string `json:"postInstall,omitempty"`
}

type AddOptsBuilder interface {
	ToNodeAddMap() (map[string]interface{}, error)
}

func (opts AddOpts) ToNodeAddMap() (map[string]interface{}, error) {
	return golangsdk.BuildRequestBody(opts, "")
}

func Add(c *golangsdk.ServiceClient, clusterid string, opts AddOptsBuilder) (r AddResult) {
	b, err := opts.ToNodeAddMap()
	if err != nil {
		r.Err = err
		return
	}
	reqOpt := &golangsdk.RequestOpts{OkCodes: []int{200}}
	_, r.Err = c.Post(addNodeURL(c, clusterid), b, &r.Body, reqOpt)
	return
}

type ResetOpts struct {
	// API type, fixed value List
	Kind string `json:"kind" required:"true"`
	// API version, fixed value v3
	ApiVersion string `json:"apiversion" required:"true"`
	// List of nodes to reset
	NodeList []ResetNode `json:"nodeList" required:"true"`
}

type ResetNode struct {
	NodeID string      `json:"nodeID" required:"true"`
	Spec   AddNodeSpec `json:"spec" required:"true"`
}

type ResetOptsBuilder interface {
	ToNodeResetMap() (map[string]interface{}, error)
}

func (opts ResetOpts) ToNodeResetMap() (map[string]interface{}, error) {
	return golangsdk.BuildRequestBody(opts, "")
}

func Reset(c *golangsdk.ServiceClient, clusterid string, opts ResetOptsBuilder) (r AddResult) {
	b, err := opts.ToNodeResetMap()
	if err != nil {
		r.Err = err
		return
	}
	_, r.Err = c.Post(resetNodeURL(c, clusterid), b, &r.Body, nil)
	return
}

// Get retrieves a particular nodes based on its unique ID and cluster ID.
func Get(c *golangsdk.ServiceClient, clusterid, nodeid string) (r GetResult) {
	_, r.Err = c.Get(resourceURL(c, clusterid, nodeid), &r.Body, &golangsdk.RequestOpts{
		OkCodes:     []int{200},
		MoreHeaders: RequestOpts.MoreHeaders, JSONBody: nil,
	})
	return
}

// UpdateOptsBuilder allows extensions to add additional parameters to the
// Update request.
type UpdateOptsBuilder interface {
	ToNodeUpdateMap() (map[string]interface{}, error)
}

// UpdateOpts contains all the values needed to update a new node
type UpdateOpts struct {
	Metadata UpdateMetadata `json:"metadata,omitempty"`
}

type UpdateMetadata struct {
	Name string `json:"name,omitempty"`
}

// ToNodeUpdateMap builds an update body based on UpdateOpts.
func (opts UpdateOpts) ToNodeUpdateMap() (map[string]interface{}, error) {
	return golangsdk.BuildRequestBody(opts, "")
}

// Update allows nodes to be updated.
func Update(c *golangsdk.ServiceClient, clusterid, nodeid string, opts UpdateOptsBuilder) (r UpdateResult) {
	b, err := opts.ToNodeUpdateMap()
	if err != nil {
		r.Err = err
		return
	}
	_, r.Err = c.Put(resourceURL(c, clusterid, nodeid), b, &r.Body, &golangsdk.RequestOpts{
		OkCodes: []int{200},
	})
	return
}

// Delete will permanently delete a particular node based on its unique ID and cluster ID.
func Delete(c *golangsdk.ServiceClient, clusterid, nodeid string) (r DeleteResult) {
	_, r.Err = c.Delete(resourceURL(c, clusterid, nodeid), &golangsdk.RequestOpts{
		OkCodes:     []int{200},
		MoreHeaders: RequestOpts.MoreHeaders, JSONBody: nil,
	})
	return
}

type RemoveOptsBuilder interface {
	ToNodeRemoveMap() (map[string]interface{}, error)
}

type RemoveOpts struct {
	//  API type, fixed value RemoveNodesTask
	Kind string `json:"kind,omitempty"`
	// API version, fixed value v3
	Apiversion string `json:"apiVersion,omitempty"`

	Spec RemoveNodeSpec `json:"spec" required:"true"`
}

type RemoveNodeSpec struct {
	Login LoginSpec  `json:"login" required:"true"`
	Nodes []NodeItem `json:"nodes,omitempty"`
}

type NodeItem struct {
	Uid string `json:"uid,omitempty"`
}

func (opts RemoveOpts) ToNodeRemoveMap() (map[string]interface{}, error) {
	return golangsdk.BuildRequestBody(opts, "")
}

func Remove(c *golangsdk.ServiceClient, clusterid string, opts RemoveOptsBuilder) (r DeleteResult) {
	b, err := opts.ToNodeRemoveMap()
	if err != nil {
		r.Err = err
		return
	}

	_, r.Err = c.Put(removeNodeURL(c, clusterid), b, nil, &golangsdk.RequestOpts{
		OkCodes:     []int{200},
		MoreHeaders: RequestOpts.MoreHeaders, JSONBody: nil,
	})

	return
}

// GetJobDetails retrieves a particular job based on its unique ID
func GetJobDetails(c *golangsdk.ServiceClient, jobid string) (r GetResult) {
	_, r.Err = c.Get(getJobURL(c, jobid), &r.Body, &golangsdk.RequestOpts{
		OkCodes:     []int{200},
		MoreHeaders: RequestOpts.MoreHeaders, JSONBody: nil,
	})
	return
}