Skip to content

Commit b02ed0f

Browse files
BrianHicksBrianHicks
committed
triton: add nics
1 parent c6b768e commit b02ed0f

2 files changed

Lines changed: 297 additions & 13 deletions

File tree

‎builtin/providers/triton/resource_machine.go‎

Lines changed: 127 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"regexp"
77
"time"
88

9+
"github.com/hashicorp/terraform/helper/hashcode"
910
"github.com/hashicorp/terraform/helper/schema"
1011
"github.com/joyent/gosdc/cloudapi"
1112
)
@@ -108,17 +109,53 @@ func resourceMachine() *schema.Resource {
108109
Type: schema.TypeString,
109110
Computed: true,
110111
},
111-
"networks": {
112-
Description: "desired network IDs",
113-
Type: schema.TypeList,
114-
Optional: true,
112+
"nic": {
113+
Description: "network interface",
114+
Type: schema.TypeSet,
115115
Computed: true,
116-
// TODO: this really should ForceNew but the Network IDs don't seem to
117-
// be returned by the API, meaning if we track them here TF will replace
118-
// the resource on every run.
119-
// ForceNew: true,
120-
Elem: &schema.Schema{
121-
Type: schema.TypeString,
116+
Optional: true,
117+
Set: func(v interface{}) int {
118+
m := v.(map[string]interface{})
119+
return hashcode.String(m["network"].(string))
120+
},
121+
Elem: &schema.Resource{
122+
Schema: map[string]*schema.Schema{
123+
"ip": {
124+
Description: "NIC's IPv4 address",
125+
Computed: true,
126+
Type: schema.TypeString,
127+
},
128+
"mac": {
129+
Description: "NIC's MAC address",
130+
Computed: true,
131+
Type: schema.TypeString,
132+
},
133+
"primary": {
134+
Description: "Whether this is the machine's primary NIC",
135+
Computed: true,
136+
Type: schema.TypeBool,
137+
},
138+
"netmask": {
139+
Description: "IPv4 netmask",
140+
Computed: true,
141+
Type: schema.TypeString,
142+
},
143+
"gateway": {
144+
Description: "IPv4 gateway",
145+
Computed: true,
146+
Type: schema.TypeString,
147+
},
148+
"state": {
149+
Description: "describes the state of the NIC (e.g. provisioning, running, or stopped)",
150+
Computed: true,
151+
Type: schema.TypeString,
152+
},
153+
"network": {
154+
Description: "Network ID this NIC is attached to",
155+
Required: true,
156+
Type: schema.TypeString,
157+
},
158+
},
122159
},
123160
},
124161
"firewall_enabled": {
@@ -153,6 +190,18 @@ func resourceMachine() *schema.Resource {
153190
Optional: true,
154191
Computed: true,
155192
},
193+
194+
// deprecated fields
195+
"networks": {
196+
Description: "desired network IDs",
197+
Type: schema.TypeList,
198+
Optional: true,
199+
Computed: true,
200+
Deprecated: "Networks is deprecated, please use `nic`",
201+
Elem: &schema.Schema{
202+
Type: schema.TypeString,
203+
},
204+
},
156205
},
157206
}
158207
}
@@ -164,6 +213,11 @@ func resourceMachineCreate(d *schema.ResourceData, meta interface{}) error {
164213
for _, network := range d.Get("networks").([]interface{}) {
165214
networks = append(networks, network.(string))
166215
}
216+
nics := d.Get("nic").(*schema.Set)
217+
for _, nicI := range nics.List() {
218+
nic := nicI.(map[string]interface{})
219+
networks = append(networks, nic["network"].(string))
220+
}
167221

168222
metadata := map[string]string{}
169223
for schemaName, metadataKey := range resourceMachineMetadataKeys {
@@ -221,6 +275,11 @@ func resourceMachineRead(d *schema.ResourceData, meta interface{}) error {
221275
return err
222276
}
223277

278+
nics, err := client.ListNICs(d.Id())
279+
if err != nil {
280+
return err
281+
}
282+
224283
d.SetId(machine.Id)
225284
d.Set("name", machine.Name)
226285
d.Set("type", machine.Type)
@@ -235,9 +294,31 @@ func resourceMachineRead(d *schema.ResourceData, meta interface{}) error {
235294
d.Set("package", machine.Package)
236295
d.Set("image", machine.Image)
237296
d.Set("primaryip", machine.PrimaryIP)
238-
d.Set("networks", machine.Networks)
239297
d.Set("firewall_enabled", machine.FirewallEnabled)
240298

299+
// create and update NICs
300+
var (
301+
machineNICs []map[string]interface{}
302+
networks []string
303+
)
304+
for _, nic := range nics {
305+
machineNICs = append(
306+
machineNICs,
307+
map[string]interface{}{
308+
"ip": nic.IP,
309+
"mac": nic.MAC,
310+
"primary": nic.Primary,
311+
"netmask": nic.Netmask,
312+
"gateway": nic.Gateway,
313+
"state": nic.State,
314+
"network": nic.Network,
315+
},
316+
)
317+
networks = append(networks, nic.Network)
318+
}
319+
d.Set("nic", machineNICs)
320+
d.Set("networks", networks)
321+
241322
// computed attributes from metadata
242323
for schemaName, metadataKey := range resourceMachineMetadataKeys {
243324
d.Set(schemaName, machine.Metadata[metadataKey])
@@ -336,6 +417,41 @@ func resourceMachineUpdate(d *schema.ResourceData, meta interface{}) error {
336417
d.SetPartial("firewall_enabled")
337418
}
338419

420+
if d.HasChange("nic") {
421+
o, n := d.GetChange("nic")
422+
if o == nil {
423+
o = new(schema.Set)
424+
}
425+
if n == nil {
426+
n = new(schema.Set)
427+
}
428+
429+
oldNICs := o.(*schema.Set)
430+
newNICs := o.(*schema.Set)
431+
432+
// add new NICs that are not in old NICs
433+
for _, nicI := range newNICs.Difference(oldNICs).List() {
434+
nic := nicI.(map[string]interface{})
435+
fmt.Printf("adding %+v\n", nic)
436+
_, err := client.AddNIC(d.Id(), nic["network"].(string))
437+
if err != nil {
438+
return err
439+
}
440+
}
441+
442+
// remove old NICs that are not in new NICs
443+
for _, nicI := range oldNICs.Difference(newNICs).List() {
444+
nic := nicI.(map[string]interface{})
445+
fmt.Printf("removing %+v\n", nic)
446+
err := client.RemoveNIC(d.Id(), nic["mac"].(string))
447+
if err != nil {
448+
return err
449+
}
450+
}
451+
452+
d.SetPartial("nic")
453+
}
454+
339455
// metadata stuff
340456
metadata := map[string]string{}
341457
for schemaName, metadataKey := range resourceMachineMetadataKeys {

‎builtin/providers/triton/resource_machine_test.go‎

Lines changed: 170 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,62 @@ func TestAccTritonMachine_basic(t *testing.T) {
3434
})
3535
}
3636

37+
func TestAccTritonMachine_nic(t *testing.T) {
38+
machineName := fmt.Sprintf("acctest-%d", acctest.RandInt())
39+
config := fmt.Sprintf(testAccTritonMachine_withnic, machineName, machineName)
40+
41+
resource.Test(t, resource.TestCase{
42+
PreCheck: func() { testAccPreCheck(t) },
43+
Providers: testAccProviders,
44+
CheckDestroy: testCheckTritonMachineDestroy,
45+
Steps: []resource.TestStep{
46+
resource.TestStep{
47+
Config: config,
48+
Check: resource.ComposeTestCheckFunc(
49+
testCheckTritonMachineExists("triton_machine.test"),
50+
func(*terraform.State) error {
51+
time.Sleep(10 * time.Second)
52+
return nil
53+
},
54+
testCheckTritonMachineHasFabric("triton_machine.test", "triton_fabric.test"),
55+
),
56+
},
57+
},
58+
})
59+
}
60+
61+
func TestAccTritonMachine_addnic(t *testing.T) {
62+
machineName := fmt.Sprintf("acctest-%d", acctest.RandInt())
63+
without := fmt.Sprintf(testAccTritonMachine_withoutnic, machineName, machineName)
64+
with := fmt.Sprintf(testAccTritonMachine_withnic, machineName, machineName)
65+
66+
resource.Test(t, resource.TestCase{
67+
PreCheck: func() { testAccPreCheck(t) },
68+
Providers: testAccProviders,
69+
CheckDestroy: testCheckTritonMachineDestroy,
70+
Steps: []resource.TestStep{
71+
resource.TestStep{
72+
Config: without,
73+
Check: resource.ComposeTestCheckFunc(
74+
testCheckTritonMachineExists("triton_machine.test"),
75+
func(*terraform.State) error {
76+
time.Sleep(10 * time.Second)
77+
return nil
78+
},
79+
testCheckTritonMachineHasNoFabric("triton_machine.test", "triton_fabric.test"),
80+
),
81+
},
82+
resource.TestStep{
83+
Config: with,
84+
Check: resource.ComposeTestCheckFunc(
85+
testCheckTritonMachineExists("triton_machine.test"),
86+
testCheckTritonMachineHasFabric("triton_machine.test", "triton_fabric.test"),
87+
),
88+
},
89+
},
90+
})
91+
}
92+
3793
func testCheckTritonMachineExists(name string) resource.TestCheckFunc {
3894
return func(s *terraform.State) error {
3995
// Ensure we have enough information in state to look up in API
@@ -56,6 +112,66 @@ func testCheckTritonMachineExists(name string) resource.TestCheckFunc {
56112
}
57113
}
58114

115+
func testCheckTritonMachineHasFabric(name, fabricName string) resource.TestCheckFunc {
116+
return func(s *terraform.State) error {
117+
// Ensure we have enough information in state to look up in API
118+
machine, ok := s.RootModule().Resources[name]
119+
if !ok {
120+
return fmt.Errorf("Not found: %s", name)
121+
}
122+
123+
network, ok := s.RootModule().Resources[fabricName]
124+
if !ok {
125+
return fmt.Errorf("Not found: %s", fabricName)
126+
}
127+
conn := testAccProvider.Meta().(*cloudapi.Client)
128+
129+
nics, err := conn.ListNICs(machine.Primary.ID)
130+
if err != nil {
131+
return fmt.Errorf("Bad: Check NICs Exist: %s", err)
132+
}
133+
134+
fmt.Printf("%+v\n", machine.Primary)
135+
for _, nic := range nics {
136+
fmt.Printf("---\n%+v\n%+v\n", nic, network.Primary)
137+
if nic.Network == network.Primary.ID {
138+
return nil
139+
}
140+
}
141+
142+
return fmt.Errorf("Bad: Machine %q does not have Fabric %q", machine.Primary.ID, network.Primary.ID)
143+
}
144+
}
145+
146+
func testCheckTritonMachineHasNoFabric(name, fabricName string) resource.TestCheckFunc {
147+
return func(s *terraform.State) error {
148+
// Ensure we have enough information in state to look up in API
149+
machine, ok := s.RootModule().Resources[name]
150+
if !ok {
151+
return fmt.Errorf("Not found: %s", name)
152+
}
153+
154+
network, ok := s.RootModule().Resources[fabricName]
155+
if !ok {
156+
return fmt.Errorf("Not found: %s", fabricName)
157+
}
158+
conn := testAccProvider.Meta().(*cloudapi.Client)
159+
160+
nics, err := conn.ListNICs(machine.Primary.ID)
161+
if err != nil {
162+
return fmt.Errorf("Bad: Check NICs Exist: %s", err)
163+
}
164+
165+
for _, nic := range nics {
166+
if nic.Network == network.Primary.ID {
167+
return fmt.Errorf("Bad: Machine %q has Fabric %q", machine.Primary.ID, network.Primary.ID)
168+
}
169+
}
170+
171+
return nil
172+
}
173+
}
174+
59175
func testCheckTritonMachineDestroy(s *terraform.State) error {
60176
conn := testAccProvider.Meta().(*cloudapi.Client)
61177

@@ -84,7 +200,59 @@ resource "triton_machine" "test" {
84200
image = "842e6fa6-6e9b-11e5-8402-1b490459e334"
85201
86202
tags = {
87-
test = "hello!"
88-
}
203+
test = "hello!"
204+
}
205+
}
206+
`
207+
208+
var testAccTritonMachine_withnic = `
209+
resource "triton_fabric" "test" {
210+
name = "%s-network"
211+
description = "test network"
212+
vlan_id = 2 # every DC seems to have a vlan 2 available
213+
214+
subnet = "10.0.0.0/22"
215+
gateway = "10.0.0.1"
216+
provision_start_ip = "10.0.0.5"
217+
provision_end_ip = "10.0.3.250"
218+
219+
resolvers = ["8.8.8.8", "8.8.4.4"]
220+
}
221+
222+
resource "triton_machine" "test" {
223+
name = "%s"
224+
package = "g3-standard-0.25-smartos"
225+
image = "842e6fa6-6e9b-11e5-8402-1b490459e334"
226+
227+
tags = {
228+
test = "hello!"
229+
}
230+
231+
nic { network = "${triton_fabric.test.id}" }
232+
}
233+
`
234+
235+
var testAccTritonMachine_withoutnic = `
236+
resource "triton_fabric" "test" {
237+
name = "%s-network"
238+
description = "test network"
239+
vlan_id = 2 # every DC seems to have a vlan 2 available
240+
241+
subnet = "10.0.0.0/22"
242+
gateway = "10.0.0.1"
243+
provision_start_ip = "10.0.0.5"
244+
provision_end_ip = "10.0.3.250"
245+
246+
resolvers = ["8.8.8.8", "8.8.4.4"]
247+
}
248+
249+
resource "triton_machine" "test" {
250+
name = "%s"
251+
package = "g3-standard-0.25-smartos"
252+
image = "842e6fa6-6e9b-11e5-8402-1b490459e334"
253+
254+
tags = {
255+
test = "hello!"
256+
}
89257
}
90258
`

0 commit comments

Comments
 (0)