profiles/gce/init.go

// +build linux

// Package gce provides a Profile for Google Compute Engine and should be
// used by binaries that only ever expect to be run on GCE.
package gce

import (
	"fmt"
	"net"

	"veyron.io/veyron/veyron/profiles"

	"veyron.io/veyron/veyron2"
	"veyron.io/veyron/veyron2/config"
	"veyron.io/veyron/veyron2/ipc"
	"veyron.io/veyron/veyron2/rt"

	"veyron.io/veyron/veyron/profiles/internal/gce"
)

var ListenSpec = &ipc.ListenSpec{
	Protocol: "tcp",
	Address:  "127.0.0.1:0",
}

func init() {
	rt.RegisterProfile(&profile{})
}

type profile struct{}

func (p *profile) Name() string {
	return "GCE"
}

func (p *profile) Runtime() string {
	return ""
}

func (p *profile) Platform() *veyron2.Platform {
	platform, _ := profiles.Platform()
	return platform
}

func (p *profile) String() string {
	return "net " + p.Platform().String()
}

func (p *profile) Init(rt veyron2.Runtime, publisher *config.Publisher) error {
	if !gce.RunningOnGCE() {
		return fmt.Errorf("GCE profile used on a non-GCE system")
	}
	if ip, err := gce.ExternalIPAddress(); err != nil {
		return err
	} else {
		ListenSpec.AddressChooser = func(network string, addrs []net.Addr) (net.Addr, error) {
			return &net.IPAddr{IP: ip}, nil
		}
	}
	return nil
}