github.com/rumpl/bof@v23.0.0-rc.2+incompatible/daemon/cluster/convert/node.go (about)

     1  package convert // import "github.com/docker/docker/daemon/cluster/convert"
     2  
     3  import (
     4  	"fmt"
     5  	"strings"
     6  
     7  	types "github.com/docker/docker/api/types/swarm"
     8  	gogotypes "github.com/gogo/protobuf/types"
     9  	swarmapi "github.com/moby/swarmkit/v2/api"
    10  )
    11  
    12  // NodeFromGRPC converts a grpc Node to a Node.
    13  func NodeFromGRPC(n swarmapi.Node) types.Node {
    14  	node := types.Node{
    15  		ID: n.ID,
    16  		Spec: types.NodeSpec{
    17  			Role:         types.NodeRole(strings.ToLower(n.Spec.DesiredRole.String())),
    18  			Availability: types.NodeAvailability(strings.ToLower(n.Spec.Availability.String())),
    19  		},
    20  		Status: types.NodeStatus{
    21  			State:   types.NodeState(strings.ToLower(n.Status.State.String())),
    22  			Message: n.Status.Message,
    23  			Addr:    n.Status.Addr,
    24  		},
    25  	}
    26  
    27  	// Meta
    28  	node.Version.Index = n.Meta.Version.Index
    29  	node.CreatedAt, _ = gogotypes.TimestampFromProto(n.Meta.CreatedAt)
    30  	node.UpdatedAt, _ = gogotypes.TimestampFromProto(n.Meta.UpdatedAt)
    31  
    32  	// Annotations
    33  	node.Spec.Annotations = annotationsFromGRPC(n.Spec.Annotations)
    34  
    35  	// Description
    36  	if n.Description != nil {
    37  		node.Description.Hostname = n.Description.Hostname
    38  		if n.Description.Platform != nil {
    39  			node.Description.Platform.Architecture = n.Description.Platform.Architecture
    40  			node.Description.Platform.OS = n.Description.Platform.OS
    41  		}
    42  		if n.Description.Resources != nil {
    43  			node.Description.Resources.NanoCPUs = n.Description.Resources.NanoCPUs
    44  			node.Description.Resources.MemoryBytes = n.Description.Resources.MemoryBytes
    45  			node.Description.Resources.GenericResources = GenericResourcesFromGRPC(n.Description.Resources.Generic)
    46  		}
    47  		if n.Description.Engine != nil {
    48  			node.Description.Engine.EngineVersion = n.Description.Engine.EngineVersion
    49  			node.Description.Engine.Labels = n.Description.Engine.Labels
    50  			for _, plugin := range n.Description.Engine.Plugins {
    51  				node.Description.Engine.Plugins = append(node.Description.Engine.Plugins, types.PluginDescription{Type: plugin.Type, Name: plugin.Name})
    52  			}
    53  		}
    54  		if n.Description.TLSInfo != nil {
    55  			node.Description.TLSInfo.TrustRoot = string(n.Description.TLSInfo.TrustRoot)
    56  			node.Description.TLSInfo.CertIssuerPublicKey = n.Description.TLSInfo.CertIssuerPublicKey
    57  			node.Description.TLSInfo.CertIssuerSubject = n.Description.TLSInfo.CertIssuerSubject
    58  		}
    59  		for _, csi := range n.Description.CSIInfo {
    60  			if csi != nil {
    61  				node.Description.CSIInfo = append(
    62  					node.Description.CSIInfo,
    63  					types.NodeCSIInfo{
    64  						PluginName:        csi.PluginName,
    65  						NodeID:            csi.NodeID,
    66  						MaxVolumesPerNode: csi.MaxVolumesPerNode,
    67  					},
    68  				)
    69  			}
    70  		}
    71  	}
    72  
    73  	// Manager
    74  	if n.ManagerStatus != nil {
    75  		node.ManagerStatus = &types.ManagerStatus{
    76  			Leader:       n.ManagerStatus.Leader,
    77  			Reachability: types.Reachability(strings.ToLower(n.ManagerStatus.Reachability.String())),
    78  			Addr:         n.ManagerStatus.Addr,
    79  		}
    80  	}
    81  
    82  	return node
    83  }
    84  
    85  // NodeSpecToGRPC converts a NodeSpec to a grpc NodeSpec.
    86  func NodeSpecToGRPC(s types.NodeSpec) (swarmapi.NodeSpec, error) {
    87  	spec := swarmapi.NodeSpec{
    88  		Annotations: swarmapi.Annotations{
    89  			Name:   s.Name,
    90  			Labels: s.Labels,
    91  		},
    92  	}
    93  	if role, ok := swarmapi.NodeRole_value[strings.ToUpper(string(s.Role))]; ok {
    94  		spec.DesiredRole = swarmapi.NodeRole(role)
    95  	} else {
    96  		return swarmapi.NodeSpec{}, fmt.Errorf("invalid Role: %q", s.Role)
    97  	}
    98  
    99  	if availability, ok := swarmapi.NodeSpec_Availability_value[strings.ToUpper(string(s.Availability))]; ok {
   100  		spec.Availability = swarmapi.NodeSpec_Availability(availability)
   101  	} else {
   102  		return swarmapi.NodeSpec{}, fmt.Errorf("invalid Availability: %q", s.Availability)
   103  	}
   104  
   105  	return spec, nil
   106  }