google / go-cloud

The Go Cloud Development Kit (Go CDK): A library and tools for open cloud development in Go.
https://gocloud.dev/
Apache License 2.0
9.57k stars 812 forks source link

blob: `Bucket.List()` with Go 1.23 iterators #3488

Open myaaaaaaaaa opened 1 month ago

myaaaaaaaaa commented 1 month ago

Is your feature request related to a problem? Please describe.

Bucket.List() is somewhat cumbersome to use since it was designed long before Go 1.23. Now that Go 1.23 standardized what iterators should look like, there's an opportunity to design simpler iteration APIs.

For details, see https://go.dev/blog/range-functions

Describe the solution you'd like

Option 1: ListIterator.All()

func (i *ListIterator) All(ctx context.Context, err *error) iter.Seq2[*ListObject, func(io.Writer)]
// Note: The iter.Seq2 type cannot be used until Go CDK phases out support for Go versions older than 1.23.
// For the time being, this function will need to return func(func(*ListObject, func(io.Writer)) bool)

Consumes the remaining objects in the ListIterator. The second value is a convenient way to call Bucket.Download() on the current object without the typical overhead of error handling, as this is already handled by All()

    var err error
    iter := bucket.List(options)
    for obj, download := range iter.All(ctx, &err) {
        fmt.Println(obj.Key)
        if obj.Size < 1024 {
            var buf bytes.Buffer
            download(&buf)
            fmt.Println(buf.String())
        }
    }
    // loop terminates early on error
    if err != nil {
        // also happens if download() fails or ctx is cancelled
        log.Fatal(err)
    }

Option 2: Bucket.All()

Avoids the need for the ListIterator object entirely.

    var err error
    for obj, download := range bucket.All(ctx, options, &err) {
        fmt.Println(obj.Key)
        if obj.Size < 1024 {
            var buf bytes.Buffer
            download(&buf)
            fmt.Println(buf.String())
        }
    }
    if err != nil {
        log.Fatal(err)
    }

One potential downside to this is that if a standard io.FS.All() method is ever introduced, it will almost certainly have an incompatible signature with whatever is decided here. It seems safer to avoid adding iterator methods directly to Bucket at this time.