diff options
author | Russ Cox <rsc@golang.org> | 2018-06-08 13:59:17 -0400 |
---|---|---|
committer | Russ Cox <rsc@golang.org> | 2018-06-15 18:57:58 +0000 |
commit | 5756895877492e3427c92e9ec8784eb1f4b01474 (patch) | |
tree | 5ea749c030e1c6a353873477603969c745e6f676 /src/cmd/go/internal/mvs/mvs.go | |
parent | 1a92cdbfc10e0c66f2e015264a39159c055a5c15 (diff) | |
download | go-5756895877492e3427c92e9ec8784eb1f4b01474.tar.gz go-5756895877492e3427c92e9ec8784eb1f4b01474.zip |
cmd/go: add dark copy of golang.org/x/vgo
This CL corresponds to golang.org/cl/118096 (7fbc8df48a7)
in the vgo repo.
It copies the bulk of the code from vgo back into the main repo,
but completely disabled - vgo.Init is a no-op and vgo.Enabled
returns false unconditionally.
The point of this CL is to make the two trees easier to diff and
to make future syncs smaller.
Change-Id: Ic34fd5ddd8272a70c5a3b3437b5169e967d0ed03
Reviewed-on: https://go-review.googlesource.com/118095
Reviewed-by: Bryan C. Mills <bcmills@google.com>
Diffstat (limited to 'src/cmd/go/internal/mvs/mvs.go')
-rw-r--r-- | src/cmd/go/internal/mvs/mvs.go | 308 |
1 files changed, 308 insertions, 0 deletions
diff --git a/src/cmd/go/internal/mvs/mvs.go b/src/cmd/go/internal/mvs/mvs.go new file mode 100644 index 0000000000..47670ff5a6 --- /dev/null +++ b/src/cmd/go/internal/mvs/mvs.go @@ -0,0 +1,308 @@ +// Copyright 2018 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. + +// Package mvs implements Minimal Version Selection. +// See https://research.swtch.com/vgo-mvs. +package mvs + +import ( + "fmt" + "sort" + + "cmd/go/internal/module" +) + +type Reqs interface { + Required(m module.Version) ([]module.Version, error) + Max(v1, v2 string) string + Latest(path string) (module.Version, error) + Previous(m module.Version) (module.Version, error) +} + +type MissingModuleError struct { + Module module.Version +} + +func (e *MissingModuleError) Error() string { + return fmt.Sprintf("missing module: %v", e.Module) +} + +// BuildList returns the build list for the target module. +func BuildList(target module.Version, reqs Reqs) ([]module.Version, error) { + return buildList(target, reqs, nil, nil) +} + +func buildList(target module.Version, reqs Reqs, uses map[module.Version][]module.Version, vers map[string]string) ([]module.Version, error) { + var ( + min = map[string]string{target.Path: target.Version} + todo = []module.Version{target} + seen = map[module.Version]bool{target: true} + ) + for len(todo) > 0 { + m := todo[len(todo)-1] + todo = todo[:len(todo)-1] + required, _ := reqs.Required(m) + for _, r := range required { + if uses != nil { + uses[r] = append(uses[r], m) + } + if !seen[r] { + if v, ok := min[r.Path]; !ok { + min[r.Path] = r.Version + } else if max := reqs.Max(v, r.Version); max != v { + min[r.Path] = max + } + todo = append(todo, r) + seen[r] = true + } + } + } + + if min[target.Path] != target.Version { + panic("unbuildable") // TODO + } + + if vers == nil { + vers = make(map[string]string) + } + list := []module.Version{target} + for i := 0; i < len(list); i++ { + m := list[i] + required, err := reqs.Required(m) + if err != nil { + // TODO: Check error is decent. + return nil, err + } + for _, r := range required { + v := min[r.Path] + if reqs.Max(v, r.Version) != v { + panic("mistake") // TODO + } + if _, ok := vers[r.Path]; !ok { + vers[r.Path] = v + list = append(list, module.Version{Path: r.Path, Version: v}) + } + } + } + tail := list[1:] + sort.Slice(tail, func(i, j int) bool { + return tail[i].Path < tail[j].Path + }) + return list, nil +} + +// Req returns the minimal requirement list for the target module +// that result in the given build list. +func Req(target module.Version, list []module.Version, reqs Reqs) ([]module.Version, error) { + // Compute postorder, cache requirements. + var postorder []module.Version + reqCache := map[module.Version][]module.Version{} + reqCache[target] = nil + var walk func(module.Version) error + walk = func(m module.Version) error { + _, ok := reqCache[m] + if ok { + return nil + } + required, err := reqs.Required(m) + if err != nil { + return err + } + reqCache[m] = required + for _, m1 := range required { + if err := walk(m1); err != nil { + return err + } + } + postorder = append(postorder, m) + return nil + } + for _, m := range list { + if err := walk(m); err != nil { + return nil, err + } + } + + // Walk modules in reverse post-order, only adding those not implied already. + have := map[string]string{} + walk = func(m module.Version) error { + if v, ok := have[m.Path]; ok && reqs.Max(m.Version, v) == v { + return nil + } + have[m.Path] = m.Version + for _, m1 := range reqCache[m] { + walk(m1) + } + return nil + } + max := map[string]string{} + for _, m := range list { + if max[m.Path] == "" { + max[m.Path] = m.Version + } else { + max[m.Path] = reqs.Max(m.Version, max[m.Path]) + } + } + var min []module.Version + for i := len(postorder) - 1; i >= 0; i-- { + m := postorder[i] + if max[m.Path] != m.Version { + // Older version. + continue + } + if have[m.Path] != m.Version { + min = append(min, m) + walk(m) + } + } + sort.Slice(min, func(i, j int) bool { + return min[i].Path < min[j].Path + }) + return min, nil +} + +// UpgradeAll returns a build list for the target module +// in which every module is upgraded to its latest version. +func UpgradeAll(target module.Version, reqs Reqs) ([]module.Version, error) { + have := map[string]bool{target.Path: true} + list := []module.Version{target} + for i := 0; i < len(list); i++ { + m := list[i] + required, err := reqs.Required(m) + if err != nil { + panic(err) // TODO + } + for _, r := range required { + latest, err := reqs.Latest(r.Path) + if err != nil { + panic(err) // TODO + } + if reqs.Max(latest.Version, r.Version) != latest.Version { + panic("mistake") // TODO + } + if !have[r.Path] { + have[r.Path] = true + list = append(list, module.Version{Path: r.Path, Version: latest.Version}) + } + } + } + tail := list[1:] + sort.Slice(tail, func(i, j int) bool { + return tail[i].Path < tail[j].Path + }) + return list, nil +} + +// Upgrade returns a build list for the target module +// in which the given additional modules are upgraded. +func Upgrade(target module.Version, reqs Reqs, upgrade ...module.Version) ([]module.Version, error) { + list, err := reqs.Required(target) + if err != nil { + panic(err) // TODO + } + // TODO: Maybe if an error is given, + // rerun with BuildList(upgrade[0], reqs) etc + // to find which ones are the buggy ones. + list = append([]module.Version(nil), list...) + list = append(list, upgrade...) + return BuildList(target, &override{target, list, reqs}) +} + +// Downgrade returns a build list for the target module +// in which the given additional modules are downgraded. +func Downgrade(target module.Version, reqs Reqs, downgrade ...module.Version) ([]module.Version, error) { + list, err := reqs.Required(target) + if err != nil { + panic(err) // TODO + } + max := make(map[string]string) + for _, r := range list { + max[r.Path] = r.Version + } + for _, d := range downgrade { + if v, ok := max[d.Path]; !ok || reqs.Max(v, d.Version) != d.Version { + max[d.Path] = d.Version + } + } + + var ( + added = make(map[module.Version]bool) + rdeps = make(map[module.Version][]module.Version) + excluded = make(map[module.Version]bool) + ) + var exclude func(module.Version) + exclude = func(m module.Version) { + if excluded[m] { + return + } + excluded[m] = true + for _, p := range rdeps[m] { + exclude(p) + } + } + var add func(module.Version) + add = func(m module.Version) { + if added[m] { + return + } + added[m] = true + if v, ok := max[m.Path]; ok && reqs.Max(m.Version, v) != v { + exclude(m) + return + } + list, err := reqs.Required(m) + if err != nil { + panic(err) // TODO + } + for _, r := range list { + add(r) + if excluded[r] { + exclude(m) + return + } + rdeps[r] = append(rdeps[r], m) + } + } + + var out []module.Version + out = append(out, target) +List: + for _, r := range list { + add(r) + for excluded[r] { + p, err := reqs.Previous(r) + if err != nil { + return nil, err // TODO + } + // If the target version is a pseudo-version, it may not be + // included when iterating over prior versions using reqs.Previous. + // Insert it into the right place in the iteration. + // If v is excluded, p should be returned again by reqs.Previous on the next iteration. + if v := max[r.Path]; reqs.Max(v, r.Version) != v && reqs.Max(p.Version, v) != p.Version { + p.Version = v + } + if p.Version == "none" { + continue List + } + add(p) + r = p + } + out = append(out, r) + } + + return out, nil +} + +type override struct { + target module.Version + list []module.Version + Reqs +} + +func (r *override) Required(m module.Version) ([]module.Version, error) { + if m == r.target { + return r.list, nil + } + return r.Reqs.Required(m) +} |