// MergeSwarmSpecToGRPC merges a Spec with an initial grpc ClusterSpec func MergeSwarmSpecToGRPC(s types.Spec, spec swarmapi.ClusterSpec) (swarmapi.ClusterSpec, error) { // We take the initSpec (either created from scratch, or returned by swarmkit), // and will only change the value if the one taken from types.Spec is not nil or 0. // In other words, if the value taken from types.Spec is nil or 0, we will maintain the status quo. if s.Annotations.Name != "" { spec.Annotations.Name = s.Annotations.Name } if len(s.Annotations.Labels) != 0 { spec.Annotations.Labels = s.Annotations.Labels } if s.Orchestration.TaskHistoryRetentionLimit != nil { spec.Orchestration.TaskHistoryRetentionLimit = *s.Orchestration.TaskHistoryRetentionLimit } if s.Raft.SnapshotInterval != 0 { spec.Raft.SnapshotInterval = s.Raft.SnapshotInterval } if s.Raft.KeepOldSnapshots != nil { spec.Raft.KeepOldSnapshots = *s.Raft.KeepOldSnapshots } if s.Raft.LogEntriesForSlowFollowers != 0 { spec.Raft.LogEntriesForSlowFollowers = s.Raft.LogEntriesForSlowFollowers } if s.Raft.HeartbeatTick != 0 { spec.Raft.HeartbeatTick = uint32(s.Raft.HeartbeatTick) } if s.Raft.ElectionTick != 0 { spec.Raft.ElectionTick = uint32(s.Raft.ElectionTick) } if s.Dispatcher.HeartbeatPeriod != 0 { spec.Dispatcher.HeartbeatPeriod = ptypes.DurationProto(time.Duration(s.Dispatcher.HeartbeatPeriod)) } if s.CAConfig.NodeCertExpiry != 0 { spec.CAConfig.NodeCertExpiry = ptypes.DurationProto(s.CAConfig.NodeCertExpiry) } for _, ca := range s.CAConfig.ExternalCAs { protocol, ok := swarmapi.ExternalCA_CAProtocol_value[strings.ToUpper(string(ca.Protocol))] if !ok { return swarmapi.ClusterSpec{}, fmt.Errorf("invalid protocol: %q", ca.Protocol) } spec.CAConfig.ExternalCAs = append(spec.CAConfig.ExternalCAs, &swarmapi.ExternalCA{ Protocol: swarmapi.ExternalCA_CAProtocol(protocol), URL: ca.URL, Options: ca.Options, }) } spec.EncryptionConfig.AutoLockManagers = s.EncryptionConfig.AutoLockManagers return spec, nil }
// SwarmSpecToGRPCandMerge converts a Spec to a grpc ClusterSpec and merge AcceptancePolicy from an existing grpc ClusterSpec if provided. func SwarmSpecToGRPCandMerge(s types.Spec, existingSpec *swarmapi.ClusterSpec) (swarmapi.ClusterSpec, error) { spec := swarmapi.ClusterSpec{ Annotations: swarmapi.Annotations{ Name: s.Name, Labels: s.Labels, }, Orchestration: swarmapi.OrchestrationConfig{ TaskHistoryRetentionLimit: s.Orchestration.TaskHistoryRetentionLimit, }, Raft: swarmapi.RaftConfig{ SnapshotInterval: s.Raft.SnapshotInterval, KeepOldSnapshots: s.Raft.KeepOldSnapshots, LogEntriesForSlowFollowers: s.Raft.LogEntriesForSlowFollowers, HeartbeatTick: s.Raft.HeartbeatTick, ElectionTick: s.Raft.ElectionTick, }, Dispatcher: swarmapi.DispatcherConfig{ HeartbeatPeriod: ptypes.DurationProto(time.Duration(s.Dispatcher.HeartbeatPeriod)), }, CAConfig: swarmapi.CAConfig{ NodeCertExpiry: ptypes.DurationProto(s.CAConfig.NodeCertExpiry), }, } for _, ca := range s.CAConfig.ExternalCAs { protocol, ok := swarmapi.ExternalCA_CAProtocol_value[strings.ToUpper(string(ca.Protocol))] if !ok { return swarmapi.ClusterSpec{}, fmt.Errorf("invalid protocol: %q", ca.Protocol) } spec.CAConfig.ExternalCAs = append(spec.CAConfig.ExternalCAs, &swarmapi.ExternalCA{ Protocol: swarmapi.ExternalCA_CAProtocol(protocol), URL: ca.URL, Options: ca.Options, }) } if err := SwarmSpecUpdateAcceptancePolicy(&spec, s.AcceptancePolicy, existingSpec); err != nil { return swarmapi.ClusterSpec{}, err } return spec, nil }