github.com/jiasir/docker@v1.3.3-0.20170609024000-252e610103e7/daemon/cluster/convert/node.go (about)

     1  package convert
     2  
     3  import (
     4  	"fmt"
     5  	"strings"
     6  
     7  	types "github.com/docker/docker/api/types/swarm"
     8  	swarmapi "github.com/docker/swarmkit/api"
     9  	gogotypes "github.com/gogo/protobuf/types"
    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  		}
    46  		if n.Description.Engine != nil {
    47  			node.Description.Engine.EngineVersion = n.Description.Engine.EngineVersion
    48  			node.Description.Engine.Labels = n.Description.Engine.Labels
    49  			for _, plugin := range n.Description.Engine.Plugins {
    50  				node.Description.Engine.Plugins = append(node.Description.Engine.Plugins, types.PluginDescription{Type: plugin.Type, Name: plugin.Name})
    51  			}
    52  		}
    53  		if n.Description.TLSInfo != nil {
    54  			node.Description.TLSInfo.TrustRoot = string(n.Description.TLSInfo.TrustRoot)
    55  			node.Description.TLSInfo.CertIssuerPublicKey = n.Description.TLSInfo.CertIssuerPublicKey
    56  			node.Description.TLSInfo.CertIssuerSubject = n.Description.TLSInfo.CertIssuerSubject
    57  		}
    58  	}
    59  
    60  	//Manager
    61  	if n.ManagerStatus != nil {
    62  		node.ManagerStatus = &types.ManagerStatus{
    63  			Leader:       n.ManagerStatus.Leader,
    64  			Reachability: types.Reachability(strings.ToLower(n.ManagerStatus.Reachability.String())),
    65  			Addr:         n.ManagerStatus.Addr,
    66  		}
    67  	}
    68  
    69  	return node
    70  }
    71  
    72  // NodeSpecToGRPC converts a NodeSpec to a grpc NodeSpec.
    73  func NodeSpecToGRPC(s types.NodeSpec) (swarmapi.NodeSpec, error) {
    74  	spec := swarmapi.NodeSpec{
    75  		Annotations: swarmapi.Annotations{
    76  			Name:   s.Name,
    77  			Labels: s.Labels,
    78  		},
    79  	}
    80  	if role, ok := swarmapi.NodeRole_value[strings.ToUpper(string(s.Role))]; ok {
    81  		spec.DesiredRole = swarmapi.NodeRole(role)
    82  	} else {
    83  		return swarmapi.NodeSpec{}, fmt.Errorf("invalid Role: %q", s.Role)
    84  	}
    85  
    86  	if availability, ok := swarmapi.NodeSpec_Availability_value[strings.ToUpper(string(s.Availability))]; ok {
    87  		spec.Availability = swarmapi.NodeSpec_Availability(availability)
    88  	} else {
    89  		return swarmapi.NodeSpec{}, fmt.Errorf("invalid Availability: %q", s.Availability)
    90  	}
    91  
    92  	return spec, nil
    93  }