blob: f17dbc0dbf3810fbb123e80272a9debd78cb59fe [file] [log] [blame]
// Copyright 2016 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build go1.16 && (linux || darwin)
// +build go1.16
// +build linux darwin
package main
import (
"context"
"fmt"
"log"
"time"
"cloud.google.com/go/datastore"
"golang.org/x/build/internal/coordinator/pool"
"golang.org/x/build/types"
)
// Process is a datastore record about the lifetime of a coordinator process.
//
// Example GQL query:
// SELECT * From Process where LastHeartbeat > datetime("2016-01-01T00:00:00Z")
type ProcessRecord struct {
ID string
Start time.Time
LastHeartbeat time.Time
// TODO: version, who deployed, CoreOS version, Docker version,
// GCE instance type?
}
func updateInstanceRecord() {
dsClient := pool.NewGCEConfiguration().DSClient()
if dsClient == nil {
return
}
ctx := context.Background()
for {
key := datastore.NameKey("Process", processID, nil)
_, err := dsClient.Put(ctx, key, &ProcessRecord{
ID: processID,
Start: processStartTime,
LastHeartbeat: time.Now(),
})
if err != nil {
log.Printf("datastore Process Put: %v", err)
}
time.Sleep(30 * time.Second)
}
}
func putBuildRecord(br *types.BuildRecord) {
dsClient := pool.NewGCEConfiguration().DSClient()
if dsClient == nil {
return
}
ctx := context.Background()
key := datastore.NameKey("Build", br.ID, nil)
if _, err := dsClient.Put(ctx, key, br); err != nil {
log.Printf("datastore Build Put: %v", err)
}
}
func putSpanRecord(sr *types.SpanRecord) {
dsClient := pool.NewGCEConfiguration().DSClient()
if dsClient == nil {
return
}
ctx := context.Background()
key := datastore.NameKey("Span", fmt.Sprintf("%s-%v-%v", sr.BuildID, sr.StartTime.UnixNano(), sr.Event), nil)
if _, err := dsClient.Put(ctx, key, sr); err != nil {
log.Printf("datastore Span Put: %v", err)
}
}