Add UpdateGuestNetworkInterface

While both Drive and NetworkInterface can have RateLimiters on creation,
updating RateLimiters is only supported by NetworkInterface
(via PartialNetworkInterface).

Fixes #66.

Author: kzys

firecracker.go

@@ -141,6 +141,27 @@ func (f *Client) PutGuestNetworkInterfaceByID(ctx context.Context, ifaceID strin
 	return f.client.Operations.PutGuestNetworkInterfaceByID(cfg)
 }
 
+// PatchGuestNetworkInterfaceByIDOpt is a functional option to be used for the
+// PatchGuestNetworkInterfaceByID API in setting any additional optional fields.
+type PatchGuestNetworkInterfaceByIDOpt func(*ops.PatchGuestNetworkInterfaceByIDParams)
+
+// PatchGuestNetworkInterfaceByID is a wrapper for the swagger generated client to make calling of the
+// API easier.
+func (f *Client) PatchGuestNetworkInterfaceByID(ctx context.Context, ifaceID string, ifaceCfg *models.PartialNetworkInterface, opts ...PatchGuestNetworkInterfaceByIDOpt) (*ops.PatchGuestNetworkInterfaceByIDNoContent, error) {
+	timeout, cancel := context.WithTimeout(ctx, firecrackerRequestTimeout)
+	defer cancel()
+
+	cfg := ops.NewPatchGuestNetworkInterfaceByIDParamsWithContext(timeout)
+	cfg.SetBody(ifaceCfg)
+	cfg.SetIfaceID(ifaceID)
+
+	for _, opt := range opts {
+		opt(cfg)
+	}
+
+	return f.client.Operations.PatchGuestNetworkInterfaceByID(cfg)
+}
+
 // PutGuestDriveByIDOpt is a functional option to be used for the
 // PutGuestDriveByID API in setting any additional optional fields.
 type PutGuestDriveByIDOpt func(*ops.PutGuestDriveByIDParams)

machine.go

@@ -564,6 +564,17 @@ func (m *Machine) createNetworkInterface(ctx context.Context, iface NetworkInter
 	return err
 }
 
+// UpdateGuestNetworkInterface modifies the current guest network interface of ID index
+func (m *Machine) UpdateGuestNetworkInterface(ctx context.Context, ifaceID string, iface models.PartialNetworkInterface, opts ...PatchGuestNetworkInterfaceByIDOpt) error {
+	if _, err := m.client.PatchGuestNetworkInterfaceByID(ctx, ifaceID, &iface, opts...); err != nil {
+		m.logger.Errorf("Update network interface failed: %s: %v", ifaceID, err)
+		return err
+	}
+
+	m.logger.Infof("Updated network interface: %s", ifaceID)
+	return nil
+}
+
 // attachDrive attaches a secondary block device
 func (m *Machine) attachDrive(ctx context.Context, dev models.Drive) error {
 	hostPath := StringValue(dev.PathOnHost)

machineiface.go

@@ -15,6 +15,8 @@ package firecracker
 
 import (
 	"context"
+
+	models "github.com/firecracker-microvm/firecracker-go-sdk/client/models"
 )
 
 // This ensures the interface method signatures match that of Machine
@@ -29,4 +31,5 @@ type MachineIface interface {
 	Wait(context.Context) error
 	SetMetadata(context.Context, interface{}) error
 	UpdateGuestDrive(context.Context, string, string, ...PatchGuestDriveByIDOpt) error
+	UpdateGuestNetworkInterface(context.Context, string, models.PartialNetworkInterface, ...PatchGuestNetworkInterfaceByIDOpt) error
 }