Skip to content

Commit

Permalink
Add mutex to peering create (#5338)
Browse files Browse the repository at this point in the history
Signed-off-by: Modular Magician <magic-modules@google.com>

Co-authored-by: emily <emilyye@google.com>
  • Loading branch information
modular-magician and emilymye committed Jan 8, 2020
1 parent 4018ee1 commit f32dd69
Showing 1 changed file with 25 additions and 8 deletions.
33 changes: 25 additions & 8 deletions google/resource_compute_network_peering.go
Original file line number Diff line number Diff line change
Expand Up @@ -73,10 +73,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 @@ -139,10 +151,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 @@ -177,13 +192,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 f32dd69

Please sign in to comment.