storj/satellite/satellitedb/util.go

63 lines
1.5 KiB
Go
Raw Normal View History

// Copyright (C) 2023 Storj Labs, Inc.
// See LICENSE for copying information.
package satellitedb
import (
"github.com/zeebo/errs"
"storj.io/private/tagsql"
)
// withRows ensures that rows get properly closed after the callback finishes.
func withRows(rows tagsql.Rows, err error) func(func(tagsql.Rows) error) error {
return func(callback func(tagsql.Rows) error) error {
if err != nil {
return err
}
err := callback(rows)
return errs.Combine(rows.Err(), rows.Close(), err)
}
}
// convertSlice converts xs by applying fn to each element.
// If there's an error during conversion, the function
// returns an empty slice and the error.
func convertSlice[In, Out any](xs []In, fn func(In) (Out, error)) ([]Out, error) {
rs := make([]Out, len(xs))
for i := range xs {
var err error
rs[i], err = fn(xs[i])
if err != nil {
return nil, err
}
}
return rs, nil
}
// convertSliceNoError converts xs by applying fn to each element.
func convertSliceNoError[In, Out any](xs []In, fn func(In) Out) []Out {
rs := make([]Out, len(xs))
for i := range xs {
rs[i] = fn(xs[i])
}
return rs
}
// convertSliceWithErrors converts xs by applying fn to each element.
// It returns all the successfully converted values and returns the list of
// errors separately.
func convertSliceWithErrors[In, Out any](xs []In, fn func(In) (Out, error)) ([]Out, []error) {
var errs []error
rs := make([]Out, 0, len(xs))
for i := range xs {
r, err := fn(xs[i])
if err != nil {
errs = append(errs, err)
continue
}
rs = append(rs, r)
}
return rs, errs
}