Skip to content

Commit

Permalink
cluster_id is vulnerable to race condition when called concurrently (#94
Browse files Browse the repository at this point in the history
)

* fix(cluster_id): Get() is "insert if not exist", needs read-write lock protection

* fix(jobs): remove static clusterID getter in sendVersionsImpl

1) add generic data.ClusterID type in sendVersions type struct
2) enable common clusterID reference to be included in multiple, concurrent jobs, so that appropriate locking can be enforced
  • Loading branch information
jackfrancis authored Aug 11, 2016
1 parent 2582640 commit 31fe3bc
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 2 deletions.
1 change: 1 addition & 0 deletions boot.go
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@ func main() {

svPeriodic := jobs.NewSendVersionsPeriodic(
apiClient,
clusterID,
deisK8sResources,
availableVersion,
pollDur,
Expand Down
2 changes: 2 additions & 0 deletions data/cluster_id.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@ func NewClusterIDFromPersistentStorage(sgc k8s.KubeSecretGetterCreator) ClusterI

// Get is the ClusterID interface implementation
func (c clusterIDFromPersistentStorage) Get() (string, error) {
c.rwm.Lock()
defer c.rwm.Unlock()
secret, err := c.secretGetterCreator.Get(wfmSecretName)
//If we don't have the secret we shouldn't be returning error and instead a create a new one
if err != nil && !apierrors.IsNotFound(err) {
Expand Down
8 changes: 6 additions & 2 deletions jobs/jobs.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ type Periodic interface {
// SendVersions fulfills the Periodic interface
type sendVersions struct {
k8sResources *k8s.ResourceInterfaceNamespaced
clusterID data.ClusterID
apiClient *apiclient.WorkflowManager
availableVersions data.AvailableVersions
frequency time.Duration
Expand All @@ -30,12 +31,14 @@ type sendVersions struct {
// NewSendVersionsPeriodic creates a new SendVersions using sgc and rcl as the the secret getter / creator and replication controller lister implementations (respectively)
func NewSendVersionsPeriodic(
apiClient *apiclient.WorkflowManager,
clusterID data.ClusterID,
ri *k8s.ResourceInterfaceNamespaced,
availableVersions data.AvailableVersions,
frequency time.Duration,
) Periodic {
return &sendVersions{
k8sResources: ri,
clusterID: clusterID,
apiClient: apiClient,
availableVersions: availableVersions,
frequency: frequency,
Expand All @@ -45,7 +48,7 @@ func NewSendVersionsPeriodic(
// Do is the Periodic interface implementation
func (s sendVersions) Do() error {
if config.Spec.CheckVersions {
err := sendVersionsImpl(s.apiClient, s.k8sResources, s.availableVersions)
err := sendVersionsImpl(s.apiClient, s.clusterID, s.k8sResources, s.availableVersions)
if err != nil {
return err
}
Expand Down Expand Up @@ -135,12 +138,13 @@ func DoPeriodic(pSlice []Periodic) chan<- struct{} {
// sendVersions sends cluster version data
func sendVersionsImpl(
apiClient *apiclient.WorkflowManager,
clusterID data.ClusterID,
k8sResources *k8s.ResourceInterfaceNamespaced,
availableVersions data.AvailableVersions,
) error {
cluster, err := data.GetCluster(
data.NewInstalledDeisData(k8sResources),
data.NewClusterIDFromPersistentStorage(k8sResources.Secrets()),
clusterID,
data.NewLatestReleasedComponent(k8sResources, availableVersions),
)
if err != nil {
Expand Down

0 comments on commit 31fe3bc

Please sign in to comment.