tracedb/go/importtile/main.go

// importtile allows importing a .gob based Tile into tracedb.
//
// It also has hooks for profiling.
package main

import (
	"flag"
	"log"
	"os"
	"runtime/pprof"
	"time"

	"go.skia.org/infra/go/common"
	"go.skia.org/infra/go/grpclog"
	"go.skia.org/infra/go/sklog"
	"go.skia.org/infra/go/trace/db"
	"go.skia.org/infra/go/util"
	"go.skia.org/infra/golden/go/types"
	"google.golang.org/grpc"
)

// flags
var (
	address    = flag.String("address", "localhost:9090", "The address of the traceserver gRPC endpoint.")
	cpuprofile = flag.String("cpuprofile", "", "Write cpu profile to file.")
)

func _main(ts db.DB) {
	week := time.Hour * 24 * 7
	commits, err := ts.List(time.Now().Add(-week), time.Now())
	if err != nil {
		sklog.Errorf("Failed to load commits: %s", err)
		return
	}
	if len(commits) > 50 {
		commits = commits[:50]
	}

	begin := time.Now()
	_, _, err = ts.TileFromCommits(commits)
	if err != nil {
		sklog.Errorf("Failed to load Tile: %s", err)
		return
	}
	sklog.Infof("Time to load tile: %v", time.Now().Sub(begin))
	// Now load a second time.
	begin = time.Now()
	_, _, err = ts.TileFromCommits(commits)
	if err != nil {
		sklog.Errorf("Failed to load Tile: %s", err)
		return
	}
	sklog.Infof("Time to load tile the second time: %v", time.Now().Sub(begin))
}

func main() {
	common.Init()
	grpclog.Init()

	// Set up a connection to the server.
	conn, err := grpc.Dial(*address, grpc.WithInsecure())
	if err != nil {
		sklog.Fatalf("did not connect: %v", err)
	}
	defer util.Close(conn)

	// Build a TraceService client.
	builder := types.GoldenTraceBuilder
	ts, err := db.NewTraceServiceDB(conn, builder)
	if err != nil {
		log.Fatalf("Failed to create db.DB: %s", err)
	}
	sklog.Infof("Opened tracedb")
	if *cpuprofile != "" {
		f, err := os.Create(*cpuprofile)
		if err != nil {
			sklog.Fatalf("Failed to open profiling file: %s", err)
		}
		if err := pprof.StartCPUProfile(f); err != nil {
			sklog.Fatalf("Failed to start profiling: %s", err)
		}
		defer pprof.StopCPUProfile()
		_main(ts)
	} else {
		_main(ts)
	}
}