github.com/portworx/docker@v1.12.1/daemon/cluster/convert/swarm.go (about)

     1  package convert
     2  
     3  import (
     4  	"fmt"
     5  	"strings"
     6  	"time"
     7  
     8  	types "github.com/docker/engine-api/types/swarm"
     9  	swarmapi "github.com/docker/swarmkit/api"
    10  	"github.com/docker/swarmkit/protobuf/ptypes"
    11  )
    12  
    13  // SwarmFromGRPC converts a grpc Cluster to a Swarm.
    14  func SwarmFromGRPC(c swarmapi.Cluster) types.Swarm {
    15  	swarm := types.Swarm{
    16  		ClusterInfo: types.ClusterInfo{
    17  			ID: c.ID,
    18  			Spec: types.Spec{
    19  				Orchestration: types.OrchestrationConfig{
    20  					TaskHistoryRetentionLimit: c.Spec.Orchestration.TaskHistoryRetentionLimit,
    21  				},
    22  				Raft: types.RaftConfig{
    23  					SnapshotInterval:           c.Spec.Raft.SnapshotInterval,
    24  					KeepOldSnapshots:           c.Spec.Raft.KeepOldSnapshots,
    25  					LogEntriesForSlowFollowers: c.Spec.Raft.LogEntriesForSlowFollowers,
    26  					HeartbeatTick:              c.Spec.Raft.HeartbeatTick,
    27  					ElectionTick:               c.Spec.Raft.ElectionTick,
    28  				},
    29  			},
    30  		},
    31  		JoinTokens: types.JoinTokens{
    32  			Worker:  c.RootCA.JoinTokens.Worker,
    33  			Manager: c.RootCA.JoinTokens.Manager,
    34  		},
    35  	}
    36  
    37  	heartbeatPeriod, _ := ptypes.Duration(c.Spec.Dispatcher.HeartbeatPeriod)
    38  	swarm.Spec.Dispatcher.HeartbeatPeriod = uint64(heartbeatPeriod)
    39  
    40  	swarm.Spec.CAConfig.NodeCertExpiry, _ = ptypes.Duration(c.Spec.CAConfig.NodeCertExpiry)
    41  
    42  	for _, ca := range c.Spec.CAConfig.ExternalCAs {
    43  		swarm.Spec.CAConfig.ExternalCAs = append(swarm.Spec.CAConfig.ExternalCAs, &types.ExternalCA{
    44  			Protocol: types.ExternalCAProtocol(strings.ToLower(ca.Protocol.String())),
    45  			URL:      ca.URL,
    46  			Options:  ca.Options,
    47  		})
    48  	}
    49  
    50  	// Meta
    51  	swarm.Version.Index = c.Meta.Version.Index
    52  	swarm.CreatedAt, _ = ptypes.Timestamp(c.Meta.CreatedAt)
    53  	swarm.UpdatedAt, _ = ptypes.Timestamp(c.Meta.UpdatedAt)
    54  
    55  	// Annotations
    56  	swarm.Spec.Name = c.Spec.Annotations.Name
    57  	swarm.Spec.Labels = c.Spec.Annotations.Labels
    58  
    59  	return swarm
    60  }
    61  
    62  // SwarmSpecToGRPC converts a Spec to a grpc ClusterSpec.
    63  func SwarmSpecToGRPC(s types.Spec) (swarmapi.ClusterSpec, error) {
    64  	spec := swarmapi.ClusterSpec{
    65  		Annotations: swarmapi.Annotations{
    66  			Name:   s.Name,
    67  			Labels: s.Labels,
    68  		},
    69  		Orchestration: swarmapi.OrchestrationConfig{
    70  			TaskHistoryRetentionLimit: s.Orchestration.TaskHistoryRetentionLimit,
    71  		},
    72  		Raft: swarmapi.RaftConfig{
    73  			SnapshotInterval:           s.Raft.SnapshotInterval,
    74  			KeepOldSnapshots:           s.Raft.KeepOldSnapshots,
    75  			LogEntriesForSlowFollowers: s.Raft.LogEntriesForSlowFollowers,
    76  			HeartbeatTick:              s.Raft.HeartbeatTick,
    77  			ElectionTick:               s.Raft.ElectionTick,
    78  		},
    79  		Dispatcher: swarmapi.DispatcherConfig{
    80  			HeartbeatPeriod: ptypes.DurationProto(time.Duration(s.Dispatcher.HeartbeatPeriod)),
    81  		},
    82  		CAConfig: swarmapi.CAConfig{
    83  			NodeCertExpiry: ptypes.DurationProto(s.CAConfig.NodeCertExpiry),
    84  		},
    85  	}
    86  
    87  	for _, ca := range s.CAConfig.ExternalCAs {
    88  		protocol, ok := swarmapi.ExternalCA_CAProtocol_value[strings.ToUpper(string(ca.Protocol))]
    89  		if !ok {
    90  			return swarmapi.ClusterSpec{}, fmt.Errorf("invalid protocol: %q", ca.Protocol)
    91  		}
    92  		spec.CAConfig.ExternalCAs = append(spec.CAConfig.ExternalCAs, &swarmapi.ExternalCA{
    93  			Protocol: swarmapi.ExternalCA_CAProtocol(protocol),
    94  			URL:      ca.URL,
    95  			Options:  ca.Options,
    96  		})
    97  	}
    98  
    99  	return spec, nil
   100  }