mirror of
https://github.com/golang/go
synced 2024-11-17 07:45:09 -07:00
c2de759581
The computation for determining the shapes to use at the top of getInstantation was not always creating shapes with the proper indexes. If an instantiation is being called from another instantiated function, we cannot just copy the shape types unchanged, because their indexes may have changed. So, for type args that already shapes, we still call Shapify() with the correct index. Fixes #48645 Change-Id: Ibb61c6f9a3c317220fb85135ca87eb5ad4dcff9e Reviewed-on: https://go-review.googlesource.com/c/go/+/353030 Trust: Dan Scales <danscales@google.com> Run-TryBot: Dan Scales <danscales@google.com> TryBot-Result: Go Bot <gobot@golang.org> Reviewed-by: Keith Randall <khr@golang.org>
82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
// run -gcflags=-G=3
|
|
|
|
// Copyright 2021 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 main
|
|
|
|
type Iterator[T any] interface {
|
|
Iterate(fn func(T) bool)
|
|
}
|
|
|
|
type IteratorFunc[T any] func(fn func(T) bool)
|
|
|
|
func (f IteratorFunc[T]) Iterate(fn func(T) bool) {
|
|
f(fn)
|
|
}
|
|
|
|
type Stream[T any] struct {
|
|
it Iterator[T]
|
|
}
|
|
|
|
func (s Stream[T]) Iterate(fn func(T) bool) {
|
|
if s.it == nil {
|
|
return
|
|
}
|
|
s.it.Iterate(fn)
|
|
}
|
|
|
|
func FromIterator[T any](it Iterator[T]) Stream[T] {
|
|
return Stream[T]{it: it}
|
|
}
|
|
|
|
func (s Stream[T]) DropWhile(fn func(T) bool) Stream[T] {
|
|
return Pipe[T, T](s, func(t T) (T, bool) {
|
|
return t, true
|
|
})
|
|
}
|
|
|
|
func Pipe[T, R any](s Stream[T], op func(d T) (R, bool)) Stream[R] {
|
|
it := func(fn func(R) bool) {
|
|
// XXX Not getting the closure right when converting to interface.
|
|
// s.it.Iterate(func(t T) bool {
|
|
// r, ok := op(t)
|
|
// if !ok {
|
|
// return true
|
|
// }
|
|
|
|
// return fn(r)
|
|
// })
|
|
}
|
|
|
|
return FromIterator[R](IteratorFunc[R](it))
|
|
}
|
|
|
|
func Reduce[T, U any](s Stream[T], identity U, acc func(U, T) U) (r U) {
|
|
r = identity
|
|
s.Iterate(func(t T) bool {
|
|
r = acc(r, t)
|
|
return true
|
|
})
|
|
|
|
return r
|
|
}
|
|
|
|
type myIterator struct {
|
|
}
|
|
|
|
func (myIterator) Iterate(fn func(int) bool) {
|
|
}
|
|
|
|
func main() {
|
|
s := Stream[int]{}
|
|
s.it = myIterator{}
|
|
s = s.DropWhile(func(i int) bool {
|
|
return false
|
|
})
|
|
Reduce(s, nil, func(acc []int, e int) []int {
|
|
return append(acc, e)
|
|
})
|
|
}
|