build/report/report/build.go

// Copyright 2022 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package report

import (
	"context"
	"sync"

	"tools/treble/build/report/app"
)

// Channel data structures, include explicit error field to reply to each input
type buildTargetData struct {
	input      *app.BuildInput
	buildSteps int
	error      bool
}
type buildSourceData struct {
	source string
	query  *app.BuildQuery
	error  bool
}
type buildPathData struct {
	filename string
	path     *app.BuildPath
	error    bool
}

//
// create build target from  using repo data
//
func createBuildTarget(ctx context.Context, repo *repo, buildTarget *buildTargetData) *app.BuildTarget {

	out := &app.BuildTarget{Name: buildTarget.input.Target,
		Steps:     buildTarget.buildSteps,
		Projects:  make(map[string]*app.BuildProject),
		FileCount: len(buildTarget.input.Files),
	}

	for i, _ := range buildTarget.input.Files {
		proj, buildFile := lookupProjectFile(ctx, repo, &buildTarget.input.Files[i])
		if buildFile != nil {
			if buildProj, exists := out.Projects[proj.Name]; exists {
				buildProj.Files = append(buildProj.Files, *buildFile)
			} else {
				out.Projects[proj.Name] =
					&app.BuildProject{Path: proj.RepoPath,
						Name: proj.Name, Revision: proj.GitProj.Revision,
						Files: []app.BuildFile{*buildFile}}
			}
		}
	}
	return (out)
}

// Setup routines to resolve target names to app.BuildInput objects
func targetResolvers(ctx context.Context, rtx *Context) (chan string, chan *buildTargetData) {
	var wg sync.WaitGroup
	inChan := make(chan string)
	outChan := make(chan *buildTargetData)
	for i := 0; i < rtx.WorkerCount; i++ {
		wg.Add(1)
		go func() {
			for targetName := range inChan {
				var buildSteps int
				cmds, err := rtx.Build.Command(ctx, targetName)
				if err == nil {
					buildSteps = len(cmds.Cmds)
				}
				input, err := rtx.Build.Input(ctx, targetName)
				outChan <- &buildTargetData{input: input, buildSteps: buildSteps, error: err != nil}
			}
			wg.Done()
		}()
	}
	go func() {
		wg.Wait()
		close(outChan)
	}()

	return inChan, outChan
}

//
// Setup routines to resolve build input targets to BuildTarget
func resolveBuildInputs(ctx context.Context, rtx *Context, repo *repo, inChan chan *buildTargetData) chan *app.BuildTarget {
	var wg sync.WaitGroup
	outChan := make(chan *app.BuildTarget)
	for i := 0; i < rtx.WorkerCount; i++ {
		wg.Add(1)
		go func() {
			for buildTarget := range inChan {
				outChan <- createBuildTarget(ctx, repo, buildTarget)
			}
			wg.Done()
		}()
	}
	go func() {
		wg.Wait()
		close(outChan)
	}()
	return outChan
}

// Setup routines to resolve source file to query
func queryResolvers(ctx context.Context, rtx *Context) (chan string, chan *buildSourceData) {
	var wg sync.WaitGroup
	inChan := make(chan string)
	outChan := make(chan *buildSourceData)
	for i := 0; i < rtx.WorkerCount; i++ {
		wg.Add(1)
		go func() {
			for srcName := range inChan {
				query, err := rtx.Build.Query(ctx, srcName)
				outChan <- &buildSourceData{source: srcName, query: query, error: err != nil}
			}
			wg.Done()
		}()
	}
	go func() {
		wg.Wait()
		close(outChan)
	}()

	return inChan, outChan
}

// Setup routines to resolve path
func pathResolvers(ctx context.Context, rtx *Context, target string) (chan string, chan *buildPathData) {
	var wg sync.WaitGroup
	inChan := make(chan string)
	outChan := make(chan *buildPathData)
	for i := 0; i < rtx.WorkerCount; i++ {
		wg.Add(1)
		go func() {
			for dep := range inChan {
				path, err := rtx.Build.Path(ctx, target, dep)
				outChan <- &buildPathData{filename: dep, path: path, error: err != nil}
			}
			wg.Done()
		}()
	}
	go func() {
		wg.Wait()
		close(outChan)
	}()

	return inChan, outChan
}

// Setup routines to resolve paths
func pathsResolvers(ctx context.Context, rtx *Context, target string) (chan string, chan *buildPathData) {
	var wg sync.WaitGroup
	inChan := make(chan string)
	outChan := make(chan *buildPathData)
	for i := 0; i < rtx.WorkerCount; i++ {
		wg.Add(1)
		go func() {
			for dep := range inChan {
				paths, err := rtx.Build.Paths(ctx, target, dep)
				if err != nil {
					outChan <- &buildPathData{filename: dep, path: nil, error: true}
				} else {
					for _, path := range paths {

						outChan <- &buildPathData{filename: dep, path: path, error: false}
					}
				}
			}
			wg.Done()
		}()
	}
	go func() {
		wg.Wait()
		close(outChan)
	}()

	return inChan, outChan
}