示例#1
0
文件: stack.go 项目: koding/koding
// getFromFamily calls GCE API and retrieves the size of provided image.
func (s *Stack) getDiskSize(currentProject string, computeService *compute.Service) func(string, string) int {
	if computeService == nil {
		return nil
	}

	imagesService := compute.NewImagesService(computeService)
	return func(project, image string) int {
		if project == "" {
			project = currentProject
		}

		computeImg, err := imagesService.Get(project, image).Do()
		if err != nil {
			s.Log.Warning("cannot find image %q: %v (project: %q)", image, err, project)
			if project == currentProject {
				return 0
			}

			project = currentProject

			// Fall-back to current project.
			if computeImg, err = imagesService.Get(project, image).Do(); err != nil {
				s.Log.Warning("cannot find image %q in stack's project: %v", image, err)
				return 0
			}
		}

		if computeImg == nil || computeImg.DiskSizeGb == 0 {
			s.Log.Warning("no size metadata for image: %q (project: %q)", image, project)
			return 0
		}

		return int(computeImg.DiskSizeGb)
	}
}
示例#2
0
文件: gce.go 项目: hanscj1/images
// New returns a new instance of GceImages
func New(conf *GCEConfig) (*GceImages, error) {
	var err error
	if conf.ProjectID == "" {
		return nil, errors.New("ProjectID is not set. Please check your configuration.")
	}

	// increase the timeout. Also we need to pass the client with the context itself
	timeout := time.Second * 30
	ctx := context.WithValue(oauth2.NoContext, oauth2.HTTPClient, &http.Client{
		Transport: &http.Transport{TLSHandshakeTimeout: timeout},
		Timeout:   timeout,
	})

	var client *http.Client

	// allowed scopes
	scopes := []string{compute.ComputeScope}

	// Recommended way is explicit passing of credentials json which can be
	// downloaded from console.developers.google under APIs & Auth/Credentials
	// section
	if conf.AccountFile != "" {
		// expand shell meta character
		path, err := homedir.Expand(conf.AccountFile)
		if err != nil {
			return nil, err
		}

		jsonContent, err := ioutil.ReadFile(path)
		if err != nil {
			return nil, err
		}

		jtwConfig, err := google.JWTConfigFromJSON(jsonContent, scopes...)
		if err != nil {
			return nil, err
		}

		client = jtwConfig.Client(ctx)
	} else {
		// Look for application default credentials, for more details, see:
		// https://developers.google.com/accounts/docs/application-default-credentials
		client, err = google.DefaultClient(ctx, scopes...)
		if err != nil {
			return nil, err
		}
	}

	svc, err := compute.New(client)
	if err != nil {
		return nil, err
	}

	return &GceImages{
		svc:    compute.NewImagesService(svc),
		config: conf,
	}, nil
}
示例#3
0
文件: stack.go 项目: koding/koding
// getFromFamily calls GCE API and retrieves the latest image that is a part of
// provided family. If an error occurs, this function is no-op.
func (s *Stack) getFromFamily(computeService *compute.Service) func(string, string) string {
	if computeService == nil {
		return nil
	}

	imagesService := compute.NewImagesService(computeService)
	return func(project, family string) string {
		image, err := imagesService.GetFromFamily(project, family).Do()
		if err != nil {
			s.Log.Warning("cannot create image service: %s", err)
			return ""
		}

		if image == nil || image.Name == "" {
			s.Log.Warning("no image for family: %s (project: %s)", family, project)
			return ""
		}

		return image.Name
	}
}