Skip to content

Commit

Permalink
add mutex to peering create
Browse files Browse the repository at this point in the history
  • Loading branch information
emilymye committed Jan 8, 2020
1 parent 1f9be95 commit 09ef189
Showing 1 changed file with 25 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -89,10 +89,22 @@ func resourceComputeNetworkPeeringCreate(d *schema.ResourceData, meta interface{
if err != nil {
return err
}
peerNetworkFieldValue, err := ParseNetworkFieldValue(d.Get("peer_network").(string), d, config)
if err != nil {
return err
}

request := &computeBeta.NetworksAddPeeringRequest{}
request.NetworkPeering = expandNetworkPeering(d)

// Only one peering operation at a time can be performed for a given network.
// Lock on both networks, sorted so we don't deadlock for A <--> B peering pairs.
peeringLockNames := sortedNetworkPeeringMutexKeys(networkFieldValue, peerNetworkFieldValue)
for _, kn := range peeringLockNames {
mutexKV.Lock(kn)
defer mutexKV.Unlock(kn)
}

addOp, err := config.clientComputeBeta.Networks.AddPeering(networkFieldValue.Project, networkFieldValue.Name, request).Do()
if err != nil {
return fmt.Errorf("Error adding network peering: %s", err)
Expand Down Expand Up @@ -159,10 +171,13 @@ func resourceComputeNetworkPeeringDelete(d *schema.ResourceData, meta interface{
Name: name,
}

// Only one delete peering operation at a time can be performed inside any peered VPCs.
peeringLockName := getNetworkPeeringLockName(networkFieldValue.Name, peerNetworkFieldValue.Name)
mutexKV.Lock(peeringLockName)
defer mutexKV.Unlock(peeringLockName)
// Only one peering operation at a time can be performed for a given network.
// Lock on both networks, sorted so we don't deadlock for A <--> B peering pairs.
peeringLockNames := sortedNetworkPeeringMutexKeys(networkFieldValue, peerNetworkFieldValue)
for _, kn := range peeringLockNames {
mutexKV.Lock(kn)
defer mutexKV.Unlock(kn)
}

removeOp, err := config.clientCompute.Networks.RemovePeering(networkFieldValue.Project, networkFieldValue.Name, request).Do()
if err != nil {
Expand Down Expand Up @@ -201,13 +216,15 @@ func expandNetworkPeering(d *schema.ResourceData) *computeBeta.NetworkPeering {
}
}

func getNetworkPeeringLockName(networkName, peerNetworkName string) string {
func sortedNetworkPeeringMutexKeys(networkName, peerNetworkName *GlobalFieldValue) []string {
// Whether you delete the peering from network A to B or the one from B to A, they
// cannot happen at the same time.
networks := []string{networkName, peerNetworkName}
networks := []string{
fmt.Sprintf("%s/peerings", networkName.RelativeLink()),
fmt.Sprintf("%s/peerings", peerNetworkName.RelativeLink()),
}
sort.Strings(networks)

return fmt.Sprintf("network_peering/%s/%s", networks[0], networks[1])
return networks
}

func resourceComputeNetworkPeeringImport(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
Expand Down

0 comments on commit 09ef189

Please sign in to comment.