task_scheduler/vm.go

package main

import (
	"path"
	"runtime"

	"cloud.google.com/go/bigtable"
	"cloud.google.com/go/datastore"
	"go.skia.org/infra/go/auth"
	"go.skia.org/infra/go/gce"
	"go.skia.org/infra/go/gce/server"
)

func TaskSchedulerBase(name, ipAddress string) *gce.Instance {
	vm := server.Server20170928(name)
	vm.DataDisks[0].SizeGb = 200
	vm.DataDisks[0].Type = gce.DISK_TYPE_PERSISTENT_SSD
	vm.ExternalIpAddress = ipAddress
	vm.Metadata["owner_primary"] = "borenet"
	vm.Metadata["owner_secondary"] = "benjaminwagner"
	vm.Scopes = append(vm.Scopes,
		auth.SCOPE_GERRIT,
		datastore.ScopeDatastore,
		bigtable.Scope,
	)
	return vm
}

func TaskSchedulerProd(dir string) *gce.Instance {
	vm := TaskSchedulerBase("skia-task-scheduler", "35.202.175.145" /* Whitelisted in swarming, isolate and buildbucket servers */)
	vm.DataDisks[0].SizeGb = 500
	vm.ServiceAccount = "task-scheduler@skia-buildbots.google.com.iam.gserviceaccount.com"
	vm.StartupScript = path.Join(dir, "startup-script-production.sh")
	return vm
}

func TaskSchedulerInternal(dir string) *gce.Instance {
	vm := TaskSchedulerBase("skia-task-scheduler-internal", "35.184.167.88" /* Whitelisted in swarming, isolate and buildbucket servers */)
	vm.ServiceAccount = "task-scheduler-internal@skia-buildbots.google.com.iam.gserviceaccount.com"
	vm.StartupScript = path.Join(dir, "startup-script-internal.sh")
	return vm
}

func TaskSchedulerStaging(dir string) *gce.Instance {
	vm := TaskSchedulerBase("skia-task-scheduler-staging", "35.239.145.127" /* Whitelisted in swarming, isolate and buildbucket servers */)
	vm.ServiceAccount = "task-scheduler@skia-buildbots.google.com.iam.gserviceaccount.com"
	vm.StartupScript = path.Join(dir, "startup-script-staging.sh")
	return vm
}

func main() {
	_, filename, _, _ := runtime.Caller(0)
	dir := path.Dir(filename)
	server.Main(gce.ZONE_DEFAULT, map[string]*gce.Instance{
		"prod":     TaskSchedulerProd(dir),
		"internal": TaskSchedulerInternal(dir),
		"staging":  TaskSchedulerStaging(dir),
	})
}