From 57b2edb17c81f4092b107ed5371b1202df58fa78 Mon Sep 17 00:00:00 2001
From: Taras Madan <tarasmadan@google.com>
Date: Tue, 23 Jul 2024 16:16:03 +0200
Subject: all: move spanner writes to dashboard/app

dashboard/app knows about subsystems more
---
 pkg/spanner/coveragedb/coverage.go | 98 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)

(limited to 'pkg')

diff --git a/pkg/spanner/coveragedb/coverage.go b/pkg/spanner/coveragedb/coverage.go
index b0ccedf07..7b99c6720 100644
--- a/pkg/spanner/coveragedb/coverage.go
+++ b/pkg/spanner/coveragedb/coverage.go
@@ -5,10 +5,14 @@ package coveragedb
 
 import (
 	"context"
+	"fmt"
 	"time"
 
 	"cloud.google.com/go/civil"
 	"cloud.google.com/go/spanner"
+	"github.com/google/syzkaller/pkg/subsystem"
+	_ "github.com/google/syzkaller/pkg/subsystem/lists"
+	"github.com/google/uuid"
 )
 
 type FilesRecord struct {
@@ -39,3 +43,97 @@ func NewClient(ctx context.Context, projectID string) (*spanner.Client, error) {
 	database := "projects/" + projectID + "/instances/syzbot/databases/coverage"
 	return spanner.NewClient(ctx, database)
 }
+
+// TODO: move to dashAPI once tested? I'm not sure we'll benefit.
+
+type Coverage struct {
+	Instrumented int64
+	Covered      int64
+}
+
+func SaveMergeResult(ctx context.Context, projectID string, covMap map[string]*Coverage,
+	template *HistoryRecord, totalRows int64, sss []*subsystem.Subsystem) error {
+	client, err := NewClient(ctx, projectID)
+	if err != nil {
+		return fmt.Errorf("spanner.NewClient() failed: %s", err.Error())
+	}
+	defer client.Close()
+
+	ssMatcher := subsystem.MakePathMatcher(sss)
+	ssCache := make(map[string][]string)
+
+	session := uuid.New().String()
+	mutations := []*spanner.Mutation{}
+	for filePath, record := range covMap {
+		mutations = append(mutations, fileRecordMutation(session, filePath, record))
+		subsystems := fileSubsystems(filePath, ssMatcher, ssCache)
+		mutations = append(mutations, fileSubsystemsMutation(template.Namespace, filePath, subsystems))
+		// 80k mutations is a DB limit. 4 fields * 2k records is apx 8k mutations
+		// let keep this value 10x lower to have a room for indexes
+		// indexes update are also counted
+		if len(mutations) > 2000 {
+			if _, err = client.Apply(ctx, mutations); err != nil {
+				return fmt.Errorf("failed to spanner.Apply(inserts): %s", err.Error())
+			}
+			mutations = nil
+		}
+	}
+	mutations = append(mutations, historyMutation(session, template, totalRows))
+	if _, err = client.Apply(ctx, mutations); err != nil {
+		return fmt.Errorf("failed to spanner.Apply(inserts): %s", err.Error())
+	}
+	return nil
+}
+
+func historyMutation(session string, template *HistoryRecord, totalRows int64) *spanner.Mutation {
+	historyInsert, err := spanner.InsertOrUpdateStruct("merge_history", &HistoryRecord{
+		Session:   session,
+		Time:      time.Now(),
+		Namespace: template.Namespace,
+		Repo:      template.Repo,
+		Commit:    template.Commit,
+		Duration:  template.Duration,
+		DateTo:    template.DateTo,
+		TotalRows: totalRows,
+	})
+	if err != nil {
+		panic(fmt.Sprintf("failed to spanner.InsertStruct(): %s", err.Error()))
+	}
+	return historyInsert
+}
+
+func fileRecordMutation(session, filePath string, record *Coverage) *spanner.Mutation {
+	insert, err := spanner.InsertOrUpdateStruct("files", &FilesRecord{
+		Session:      session,
+		FilePath:     filePath,
+		Instrumented: record.Instrumented,
+		Covered:      record.Covered,
+	})
+	if err != nil {
+		panic(fmt.Sprintf("failed to fileRecordMutation(): %s", err.Error()))
+	}
+	return insert
+}
+
+func fileSubsystemsMutation(ns, filePath string, subsystems []string) *spanner.Mutation {
+	insert, err := spanner.InsertOrUpdateStruct("file_subsystems", &FileSubsystems{
+		Namespace:  ns,
+		FilePath:   filePath,
+		Subsystems: subsystems,
+	})
+	if err != nil {
+		panic(fmt.Sprintf("failed to fileSubsystemsMutation(): %s", err.Error()))
+	}
+	return insert
+}
+
+func fileSubsystems(filePath string, ssMatcher *subsystem.PathMatcher, ssCache map[string][]string) []string {
+	sss, cached := ssCache[filePath]
+	if !cached {
+		for _, match := range ssMatcher.Match(filePath) {
+			sss = append(sss, match.Name)
+		}
+		ssCache[filePath] = sss
+	}
+	return sss
+}
-- 
cgit mrf-deployment