fuzzer/vm.go

package main

import (
	"fmt"
	"path"
	"runtime"

	"google.golang.org/api/logging/v2"

	"go.skia.org/infra/go/auth"
	"go.skia.org/infra/go/gce"
	"go.skia.org/infra/go/gce/server"
)

func FuzzerBase(name string) *gce.Instance {
	vm := server.Server20170613(name)
	vm.BootDisk.Type = gce.DISK_TYPE_PERSISTENT_SSD
	vm.DataDisk.SizeGb = 375
	vm.DataDisk.Type = gce.DISK_TYPE_LOCAL_SSD
	vm.Metadata["owner_primary"] = "kjlubick"
	vm.Metadata["owner_secondary"] = "jcgregorio"
	vm.Scopes = append(vm.Scopes, auth.SCOPE_READ_WRITE, logging.LoggingWriteScope)

	_, filename, _, _ := runtime.Caller(0)
	dir := path.Dir(filename)
	vm.SetupScript = path.Join(dir, "setup-script.sh")
	vm.StartupScript = path.Join(dir, "startup-script.sh")
	return vm
}

func FrontEnd() *gce.Instance {
	vm := FuzzerBase("skia-fuzzer-fe")
	vm.ExternalIpAddress = "104.154.112.170"
	vm.MachineType = gce.MACHINE_TYPE_STANDARD_8
	return vm
}

func BackEnd(num int) *gce.Instance {
	vm := FuzzerBase(fmt.Sprintf("skia-fuzzer-be-%d", num))
	vm.MachineType = gce.MACHINE_TYPE_STANDARD_32
	return vm
}

func main() {
	server.Main(gce.ZONE_DEFAULT, map[string]*gce.Instance{
		"fe":   FrontEnd(),
		"be-1": BackEnd(1),
		"be-2": BackEnd(2),
		"be-3": BackEnd(3),
	})
}