-
Notifications
You must be signed in to change notification settings - Fork 128
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Rough draft for combining the coverage files
- Loading branch information
yihuaf
committed
Feb 16, 2024
1 parent
bc98120
commit 55e31ce
Showing
4 changed files
with
219 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| package coverprofile | ||
|
|
||
| // gocovmerge takes the results from multiple `go test -coverprofile` runs and | ||
| // merges them into one profile | ||
|
|
||
| // Taken from: https://github.com/wadey/gocovmerge @ b5bfa59ec0adc420475f97f89b58045c721d761c | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "io" | ||
| "log" | ||
| "sort" | ||
|
|
||
| "golang.org/x/tools/cover" | ||
| ) | ||
|
|
||
| func mergeProfiles(p *cover.Profile, merge *cover.Profile) { | ||
| if p.Mode != merge.Mode { | ||
| log.Fatalf("cannot merge profiles with different modes") | ||
| } | ||
| // Since the blocks are sorted, we can keep track of where the last block | ||
| // was inserted and only look at the blocks after that as targets for merge | ||
| startIndex := 0 | ||
| for _, b := range merge.Blocks { | ||
| startIndex = mergeProfileBlock(p, b, startIndex) | ||
| } | ||
| } | ||
|
|
||
| func mergeProfileBlock(p *cover.Profile, pb cover.ProfileBlock, startIndex int) int { | ||
| sortFunc := func(i int) bool { | ||
| pi := p.Blocks[i+startIndex] | ||
| return pi.StartLine >= pb.StartLine && (pi.StartLine != pb.StartLine || pi.StartCol >= pb.StartCol) | ||
| } | ||
|
|
||
| i := 0 | ||
| if !sortFunc(i) { | ||
| i = sort.Search(len(p.Blocks)-startIndex, sortFunc) | ||
| } | ||
| i += startIndex | ||
| if i < len(p.Blocks) && p.Blocks[i].StartLine == pb.StartLine && p.Blocks[i].StartCol == pb.StartCol { | ||
| if p.Blocks[i].EndLine != pb.EndLine || p.Blocks[i].EndCol != pb.EndCol { | ||
| log.Fatalf("OVERLAP MERGE: %v %v %v", p.FileName, p.Blocks[i], pb) | ||
| } | ||
| switch p.Mode { | ||
| case "set": | ||
| p.Blocks[i].Count |= pb.Count | ||
| case "count", "atomic": | ||
| p.Blocks[i].Count += pb.Count | ||
| default: | ||
| log.Fatalf("unsupported covermode: '%s'", p.Mode) | ||
| } | ||
| } else { | ||
| if i > 0 { | ||
| pa := p.Blocks[i-1] | ||
| if pa.EndLine >= pb.EndLine && (pa.EndLine != pb.EndLine || pa.EndCol > pb.EndCol) { | ||
| log.Fatalf("OVERLAP BEFORE: %v %v %v", p.FileName, pa, pb) | ||
| } | ||
| } | ||
| if i < len(p.Blocks)-1 { | ||
| pa := p.Blocks[i+1] | ||
| if pa.StartLine <= pb.StartLine && (pa.StartLine != pb.StartLine || pa.StartCol < pb.StartCol) { | ||
| log.Fatalf("OVERLAP AFTER: %v %v %v", p.FileName, pa, pb) | ||
| } | ||
| } | ||
| p.Blocks = append(p.Blocks, cover.ProfileBlock{}) | ||
| copy(p.Blocks[i+1:], p.Blocks[i:]) | ||
| p.Blocks[i] = pb | ||
| } | ||
| return i + 1 | ||
| } | ||
|
|
||
| func addProfile(profiles []*cover.Profile, p *cover.Profile) []*cover.Profile { | ||
| i := sort.Search(len(profiles), func(i int) bool { return profiles[i].FileName >= p.FileName }) | ||
| if i < len(profiles) && profiles[i].FileName == p.FileName { | ||
| mergeProfiles(profiles[i], p) | ||
| } else { | ||
| profiles = append(profiles, nil) | ||
| copy(profiles[i+1:], profiles[i:]) | ||
| profiles[i] = p | ||
| } | ||
| return profiles | ||
| } | ||
|
|
||
| func dumpProfiles(profiles []*cover.Profile, out io.Writer) { | ||
| if len(profiles) == 0 { | ||
| return | ||
| } | ||
| fmt.Fprintf(out, "mode: %s\n", profiles[0].Mode) | ||
| for _, p := range profiles { | ||
| for _, b := range p.Blocks { | ||
| fmt.Fprintf(out, "%s:%d.%d,%d.%d %d %d\n", p.FileName, b.StartLine, | ||
| b.StartCol, b.EndLine, b.EndCol, b.NumStmt, b.Count) | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,64 @@ | ||
| package coverprofile | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "os" | ||
| "strings" | ||
|
|
||
| "golang.org/x/tools/cover" | ||
| ) | ||
|
|
||
| // ParseCoverProfile parse the coverprofile file from the flag | ||
| func ParseCoverProfile(args []string) (bool, string) { | ||
| for _, arg := range args { | ||
| if strings.HasPrefix(arg, "-coverprofile=") { | ||
| return true, strings.TrimPrefix(arg, "-coverprofile=") | ||
| } | ||
| } | ||
|
|
||
| return false, "" | ||
| } | ||
|
|
||
| func Combine(main string, other string) error { | ||
| var merged []*cover.Profile | ||
|
|
||
| mainProfiles, err := cover.ParseProfiles(main) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to parse coverprofile %s: %v", main, err) | ||
| } | ||
|
|
||
| for _, p := range mainProfiles { | ||
| merged = addProfile(merged, p) | ||
| } | ||
|
|
||
| otherProfiles, err := cover.ParseProfiles(other) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to parse coverprofile %s: %v", other, err) | ||
| } | ||
|
|
||
| for _, p := range otherProfiles { | ||
| merged = addProfile(merged, p) | ||
| } | ||
|
|
||
| // Create a tmp file to write the merged profiles to. Then use os.Rename to | ||
| // atomically move the file to the main profile to mimic the effect of | ||
| // atomic replacement of the file. Note, we can't put the file on tempfs | ||
| // using the all the nice utilities around tempfiles. In places like docker | ||
| // containers, calling os.Rename on a file that is on tempfs to a file on | ||
| // normal filesystem partition will fail with errno 18 invalid cross device | ||
| // link. | ||
| tempFile := fmt.Sprintf("%s.tmp", other) | ||
| f, err := os.Create(tempFile) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create temp file %s: %v", tempFile, err) | ||
| } | ||
| dumpProfiles(merged, f) | ||
| if err := f.Close(); err != nil { | ||
| return fmt.Errorf("failed to close temp file %s: %v", tempFile, err) | ||
| } | ||
| if err := os.Rename(tempFile, main); err != nil { | ||
| return fmt.Errorf("failed to rename temp file %s to %s: %v", tempFile, main, err) | ||
| } | ||
|
|
||
| return nil | ||
| } |