mirror of https://github.com/kubernetes/kops.git
114 lines
3.2 KiB
Go
114 lines
3.2 KiB
Go
package gcetasks
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/golang/glog"
|
|
"google.golang.org/api/compute/v1"
|
|
"k8s.io/kops/upup/pkg/fi"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/gce"
|
|
"k8s.io/kops/upup/pkg/fi/cloudup/terraform"
|
|
"time"
|
|
)
|
|
|
|
//go:generate fitask -type=ManagedInstanceGroup
|
|
type ManagedInstanceGroup struct {
|
|
Name *string
|
|
|
|
Zone *string
|
|
BaseInstanceName *string
|
|
InstanceTemplate *InstanceTemplate
|
|
TargetSize *int64
|
|
}
|
|
|
|
var _ fi.CompareWithID = &ManagedInstanceGroup{}
|
|
|
|
func (e *ManagedInstanceGroup) CompareWithID() *string {
|
|
return e.Name
|
|
}
|
|
|
|
func (e *ManagedInstanceGroup) Find(c *fi.Context) (*ManagedInstanceGroup, error) {
|
|
cloud := c.Cloud.(*gce.GCECloud)
|
|
|
|
r, err := cloud.Compute.InstanceGroupManagers.Get(cloud.Project, *e.Zone, *e.Name).Do()
|
|
if err != nil {
|
|
if gce.IsNotFound(err) {
|
|
return nil, nil
|
|
}
|
|
return nil, fmt.Errorf("error listing ManagedInstanceGroups: %v", err)
|
|
}
|
|
|
|
actual := &ManagedInstanceGroup{}
|
|
actual.Name = &r.Name
|
|
actual.Zone = fi.String(lastComponent(r.Zone))
|
|
actual.BaseInstanceName = &r.BaseInstanceName
|
|
actual.TargetSize = &r.TargetSize
|
|
actual.InstanceTemplate = &InstanceTemplate{Name: fi.String(lastComponent(r.InstanceTemplate))}
|
|
|
|
return actual, nil
|
|
}
|
|
|
|
func (e *ManagedInstanceGroup) Run(c *fi.Context) error {
|
|
return fi.DefaultDeltaRunMethod(e, c)
|
|
}
|
|
|
|
func (_ *ManagedInstanceGroup) CheckChanges(a, e, changes *ManagedInstanceGroup) error {
|
|
return nil
|
|
}
|
|
|
|
func BuildInstanceTemplateURL(project, name string) string {
|
|
return fmt.Sprintf("https://www.googleapis.com/compute/v1/projects/%s/global/instanceTemplates/%s", project, name)
|
|
}
|
|
|
|
func (_ *ManagedInstanceGroup) RenderGCE(t *gce.GCEAPITarget, a, e, changes *ManagedInstanceGroup) error {
|
|
project := t.Cloud.Project
|
|
|
|
i := &compute.InstanceGroupManager{
|
|
Name: *e.Name,
|
|
Zone: *e.Zone,
|
|
BaseInstanceName: *e.BaseInstanceName,
|
|
TargetSize: *e.TargetSize,
|
|
InstanceTemplate: BuildInstanceTemplateURL(project, *e.InstanceTemplate.Name),
|
|
}
|
|
|
|
if a == nil {
|
|
for {
|
|
_, err := t.Cloud.Compute.InstanceGroupManagers.Insert(t.Cloud.Project, *e.Zone, i).Do()
|
|
if err != nil {
|
|
if gce.IsNotReady(err) {
|
|
glog.Infof("Found resourceNotReady error - sleeping before retry: %v", err)
|
|
time.Sleep(5 * time.Second)
|
|
continue
|
|
}
|
|
return fmt.Errorf("error creating ManagedInstanceGroup: %v", err)
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
} else {
|
|
return fmt.Errorf("Cannot apply changes to ManagedInstanceGroup: %v", changes)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type terraformInstanceGroupManager struct {
|
|
Name *string `json:"name"`
|
|
Zone *string `json:"zone"`
|
|
BaseInstanceName *string `json:"base_instance_name"`
|
|
InstanceTemplate *terraform.Literal `json:"instance_template"`
|
|
TargetSize *int64 `json:"target_size"`
|
|
}
|
|
|
|
func (_ *ManagedInstanceGroup) RenderTerraform(t *terraform.TerraformTarget, a, e, changes *ManagedInstanceGroup) error {
|
|
tf := &terraformInstanceGroupManager{
|
|
Name: e.Name,
|
|
Zone: e.Zone,
|
|
BaseInstanceName: e.BaseInstanceName,
|
|
InstanceTemplate: e.InstanceTemplate.TerraformLink(),
|
|
TargetSize: e.TargetSize,
|
|
}
|
|
|
|
return t.RenderResource("google_compute_instance_group_manager", *e.Name, tf)
|
|
}
|