Skip to content

Commit

Permalink
Merge pull request #494 from hashicorp/b-dynamic-port-conflict
Browse files Browse the repository at this point in the history
Remove shared reference to DynamicPorts across allocs
  • Loading branch information
dadgar committed Nov 24, 2015
2 parents 2fccadc + 7ca2bac commit 2d69a40
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 7 deletions.
20 changes: 13 additions & 7 deletions nomad/structs/network.go
Original file line number Diff line number Diff line change
Expand Up @@ -95,11 +95,14 @@ func (idx *NetworkIndex) AddReserved(n *NetworkResource) (collide bool) {
used = make(map[int]struct{})
idx.UsedPorts[n.IP] = used
}
for _, port := range n.ReservedPorts {
if _, ok := used[port.Value]; ok {
collide = true
} else {
used[port.Value] = struct{}{}

for _, ports := range [][]Port{n.ReservedPorts, n.DynamicPorts} {
for _, port := range ports {
if _, ok := used[port.Value]; ok {
collide = true
} else {
used[port.Value] = struct{}{}
}
}
}

Expand Down Expand Up @@ -179,8 +182,11 @@ func (idx *NetworkIndex) AssignNetwork(ask *NetworkResource) (out *NetworkResour
if _, ok := idx.UsedPorts[ipStr][randPort]; ok {
goto PICK
}
if isPortReserved(offer.ReservedPorts, randPort) {
goto PICK

for _, ports := range [][]Port{offer.ReservedPorts, offer.DynamicPorts} {
if isPortReserved(ports, randPort) {
goto PICK
}
}
offer.DynamicPorts[i].Value = randPort
}
Expand Down
4 changes: 4 additions & 0 deletions nomad/structs/structs.go
Original file line number Diff line number Diff line change
Expand Up @@ -645,6 +645,10 @@ func (n *NetworkResource) Copy() *NetworkResource {
newR.ReservedPorts = make([]Port, len(n.ReservedPorts))
copy(newR.ReservedPorts, n.ReservedPorts)
}
if n.DynamicPorts != nil {
newR.DynamicPorts = make([]Port, len(n.DynamicPorts))
copy(newR.DynamicPorts, n.DynamicPorts)
}
return newR
}

Expand Down
13 changes: 13 additions & 0 deletions scheduler/generic_sched_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,19 @@ func TestServiceSched_JobRegister(t *testing.T) {
t.Fatalf("bad: %#v", out)
}

// Ensure different ports were used.
used := make(map[int]struct{})
for _, alloc := range out {
for _, resource := range alloc.TaskResources {
for _, port := range resource.Networks[0].DynamicPorts {
if _, ok := used[port.Value]; ok {
t.Fatalf("Port collision %v", port.Value)
}
used[port.Value] = struct{}{}
}
}
}

h.AssertEvalStatus(t, structs.EvalStatusComplete)
}

Expand Down

0 comments on commit 2d69a40

Please sign in to comment.