Upgrade versions of vendored libs
This commit is contained in:
139
vendor/github.com/cockroachdb/pebble/objstorage/shared/logging.go
generated
vendored
Normal file
139
vendor/github.com/cockroachdb/pebble/objstorage/shared/logging.go
generated
vendored
Normal file
@@ -0,0 +1,139 @@
|
||||
// Copyright 2023 The LevelDB-Go and Pebble 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 shared
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"sort"
|
||||
)
|
||||
|
||||
// WithLogging wraps the given Storage implementation and emits logs for various
|
||||
// operations.
|
||||
func WithLogging(wrapped Storage, logf func(fmt string, args ...interface{})) Storage {
|
||||
return &loggingStore{
|
||||
logf: logf,
|
||||
wrapped: wrapped,
|
||||
}
|
||||
}
|
||||
|
||||
// loggingStore wraps a shared.Storage implementation and emits logs of the
|
||||
// operations.
|
||||
type loggingStore struct {
|
||||
logf func(fmt string, args ...interface{})
|
||||
wrapped Storage
|
||||
}
|
||||
|
||||
var _ Storage = (*loggingStore)(nil)
|
||||
|
||||
func (l *loggingStore) Close() error {
|
||||
l.logf("close")
|
||||
return l.wrapped.Close()
|
||||
}
|
||||
|
||||
func (l *loggingStore) ReadObject(
|
||||
ctx context.Context, objName string,
|
||||
) (_ ObjectReader, objSize int64, _ error) {
|
||||
r, size, err := l.wrapped.ReadObject(ctx, objName)
|
||||
l.logf("create reader for object %q: %s", objName, errOrPrintf(err, "%d bytes", size))
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
return &loggingReader{
|
||||
l: l,
|
||||
name: objName,
|
||||
wrapped: r,
|
||||
}, size, nil
|
||||
}
|
||||
|
||||
type loggingReader struct {
|
||||
l *loggingStore
|
||||
name string
|
||||
wrapped ObjectReader
|
||||
}
|
||||
|
||||
var _ ObjectReader = (*loggingReader)(nil)
|
||||
|
||||
func (l *loggingReader) ReadAt(ctx context.Context, p []byte, offset int64) error {
|
||||
if err := l.wrapped.ReadAt(ctx, p, offset); err != nil {
|
||||
l.l.logf("read object %q at %d (length %d): error %v", l.name, offset, len(p), err)
|
||||
return err
|
||||
}
|
||||
l.l.logf("read object %q at %d (length %d)", l.name, offset, len(p))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (l *loggingReader) Close() error {
|
||||
l.l.logf("close reader for %q", l.name)
|
||||
return l.wrapped.Close()
|
||||
}
|
||||
|
||||
func (l *loggingStore) CreateObject(objName string) (io.WriteCloser, error) {
|
||||
l.logf("create object %q", objName)
|
||||
writer, err := l.wrapped.CreateObject(objName)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &loggingWriter{
|
||||
l: l,
|
||||
name: objName,
|
||||
WriteCloser: writer,
|
||||
}, nil
|
||||
}
|
||||
|
||||
type loggingWriter struct {
|
||||
l *loggingStore
|
||||
name string
|
||||
bytesWritten int64
|
||||
io.WriteCloser
|
||||
}
|
||||
|
||||
func (l *loggingWriter) Write(p []byte) (n int, err error) {
|
||||
n, err = l.WriteCloser.Write(p)
|
||||
l.bytesWritten += int64(n)
|
||||
return n, err
|
||||
}
|
||||
|
||||
func (l *loggingWriter) Close() error {
|
||||
l.l.logf("close writer for %q after %d bytes", l.name, l.bytesWritten)
|
||||
return l.WriteCloser.Close()
|
||||
}
|
||||
|
||||
func (l *loggingStore) List(prefix, delimiter string) ([]string, error) {
|
||||
l.logf("list (prefix=%q, delimiter=%q)", prefix, delimiter)
|
||||
res, err := l.wrapped.List(prefix, delimiter)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sorted := append([]string(nil), res...)
|
||||
sort.Strings(sorted)
|
||||
for _, s := range sorted {
|
||||
l.logf(" - %s", s)
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
func (l *loggingStore) Delete(objName string) error {
|
||||
l.logf("delete object %q", objName)
|
||||
return l.wrapped.Delete(objName)
|
||||
}
|
||||
|
||||
func (l *loggingStore) Size(objName string) (int64, error) {
|
||||
size, err := l.wrapped.Size(objName)
|
||||
l.logf("size of object %q: %s", objName, errOrPrintf(err, "%d", size))
|
||||
return size, err
|
||||
}
|
||||
|
||||
func errOrPrintf(err error, format string, args ...interface{}) string {
|
||||
if err != nil {
|
||||
return fmt.Sprintf("error: %v", err)
|
||||
}
|
||||
return fmt.Sprintf(format, args...)
|
||||
}
|
||||
|
||||
func (l *loggingStore) IsNotExistError(err error) bool {
|
||||
return l.wrapped.IsNotExistError(err)
|
||||
}
|
||||
161
vendor/github.com/cockroachdb/pebble/objstorage/shared/mem.go
generated
vendored
Normal file
161
vendor/github.com/cockroachdb/pebble/objstorage/shared/mem.go
generated
vendored
Normal file
@@ -0,0 +1,161 @@
|
||||
// Copyright 2023 The LevelDB-Go and Pebble 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 shared
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"io"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// NewInMem returns an in-memory implementation of the shared.Storage
|
||||
// interface (for testing).
|
||||
func NewInMem() Storage {
|
||||
store := &inMemStore{}
|
||||
store.mu.objects = make(map[string]*inMemObj)
|
||||
return store
|
||||
}
|
||||
|
||||
// inMemStore is an in-memory implementation of the shared.Storage interface
|
||||
// (for testing).
|
||||
type inMemStore struct {
|
||||
mu struct {
|
||||
sync.Mutex
|
||||
objects map[string]*inMemObj
|
||||
}
|
||||
}
|
||||
|
||||
var _ Storage = (*inMemStore)(nil)
|
||||
|
||||
type inMemObj struct {
|
||||
name string
|
||||
data []byte
|
||||
}
|
||||
|
||||
func (s *inMemStore) Close() error {
|
||||
*s = inMemStore{}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *inMemStore) ReadObject(
|
||||
ctx context.Context, objName string,
|
||||
) (_ ObjectReader, objSize int64, _ error) {
|
||||
obj, err := s.getObj(objName)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
return &inMemReader{data: obj.data}, int64(len(obj.data)), nil
|
||||
}
|
||||
|
||||
type inMemReader struct {
|
||||
data []byte
|
||||
}
|
||||
|
||||
var _ ObjectReader = (*inMemReader)(nil)
|
||||
|
||||
func (r *inMemReader) ReadAt(ctx context.Context, p []byte, offset int64) error {
|
||||
if offset+int64(len(p)) > int64(len(r.data)) {
|
||||
return io.EOF
|
||||
}
|
||||
copy(p, r.data[offset:])
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *inMemReader) Close() error {
|
||||
r.data = nil
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *inMemStore) CreateObject(objName string) (io.WriteCloser, error) {
|
||||
return &inMemWriter{
|
||||
store: s,
|
||||
name: objName,
|
||||
}, nil
|
||||
}
|
||||
|
||||
type inMemWriter struct {
|
||||
store *inMemStore
|
||||
name string
|
||||
buf bytes.Buffer
|
||||
}
|
||||
|
||||
var _ io.WriteCloser = (*inMemWriter)(nil)
|
||||
|
||||
func (o *inMemWriter) Write(p []byte) (n int, err error) {
|
||||
if o.store == nil {
|
||||
panic("Write after Close")
|
||||
}
|
||||
return o.buf.Write(p)
|
||||
}
|
||||
|
||||
func (o *inMemWriter) Close() error {
|
||||
if o.store != nil {
|
||||
o.store.addObj(&inMemObj{
|
||||
name: o.name,
|
||||
data: o.buf.Bytes(),
|
||||
})
|
||||
o.store = nil
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *inMemStore) List(prefix, delimiter string) ([]string, error) {
|
||||
if delimiter != "" {
|
||||
panic("delimiter unimplemented")
|
||||
}
|
||||
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
res := make([]string, 0, len(s.mu.objects))
|
||||
for name := range s.mu.objects {
|
||||
if strings.HasPrefix(name, prefix) {
|
||||
res = append(res, name)
|
||||
}
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
func (s *inMemStore) Delete(objName string) error {
|
||||
s.rmObj(objName)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Size returns the length of the named object in bytesWritten.
|
||||
func (s *inMemStore) Size(objName string) (int64, error) {
|
||||
obj, err := s.getObj(objName)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return int64(len(obj.data)), nil
|
||||
}
|
||||
|
||||
func (s *inMemStore) IsNotExistError(err error) bool {
|
||||
return err == os.ErrNotExist
|
||||
}
|
||||
|
||||
func (s *inMemStore) getObj(name string) (*inMemObj, error) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
obj, ok := s.mu.objects[name]
|
||||
if !ok {
|
||||
return nil, os.ErrNotExist
|
||||
}
|
||||
return obj, nil
|
||||
}
|
||||
|
||||
func (s *inMemStore) addObj(o *inMemObj) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.mu.objects[o.name] = o
|
||||
}
|
||||
|
||||
func (s *inMemStore) rmObj(name string) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
delete(s.mu.objects, name)
|
||||
}
|
||||
74
vendor/github.com/cockroachdb/pebble/objstorage/shared/storage.go
generated
vendored
Normal file
74
vendor/github.com/cockroachdb/pebble/objstorage/shared/storage.go
generated
vendored
Normal file
@@ -0,0 +1,74 @@
|
||||
// Copyright 2023 The LevelDB-Go and Pebble 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 shared
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
)
|
||||
|
||||
// Storage is an interface for a blob storage driver. This is lower-level
|
||||
// than an FS-like interface, however FS/File-like abstractions can be built on
|
||||
// top of these methods.
|
||||
//
|
||||
// TODO(bilal): Consider pushing shared file obsoletion as well as path
|
||||
// generation behind this interface.
|
||||
type Storage interface {
|
||||
io.Closer
|
||||
|
||||
// ReadObject returns an ObjectReader that can be used to perform reads on an
|
||||
// object, along with the total size of the object.
|
||||
ReadObject(ctx context.Context, objName string) (_ ObjectReader, objSize int64, _ error)
|
||||
|
||||
// CreateObject returns a writer for the object at the request name. A new
|
||||
// empty object is created if CreateObject is called on an existing object.
|
||||
//
|
||||
// A Writer *must* be closed via either Close, and if closing returns a
|
||||
// non-nil error, that error should be handled or reported to the user -- an
|
||||
// implementation may buffer written data until Close and only then return
|
||||
// an error, or Write may return an opaque io.EOF with the underlying cause
|
||||
// returned by the subsequent Close().
|
||||
//
|
||||
// TODO(radu): if we encounter some unrelated error while writing to the
|
||||
// WriteCloser, we'd want to abort the whole thing rather than letting Close
|
||||
// finalize the upload.
|
||||
CreateObject(objName string) (io.WriteCloser, error)
|
||||
|
||||
// List enumerates files within the supplied prefix, returning a list of
|
||||
// objects within that prefix. If delimiter is non-empty, names which have the
|
||||
// same prefix, prior to the delimiter but after the prefix, are grouped into a
|
||||
// single result which is that prefix. The order that results are returned is
|
||||
// undefined. If a prefix is specified, the prefix is trimmed from the result
|
||||
// list.
|
||||
//
|
||||
// An example would be, if the storage contains objects a, b/4, b/5 and b/6,
|
||||
// these would be the return values:
|
||||
// List("", "") -> ["a", "b/4", "b/5", "b/6"]
|
||||
// List("", "/") -> ["a", "b"]
|
||||
// List("b", "/") -> ["4", "5", "6"]
|
||||
// List("b", "") -> ["/4", "/5", "/6"]
|
||||
List(prefix, delimiter string) ([]string, error)
|
||||
|
||||
// Delete removes the named object from the store.
|
||||
Delete(objName string) error
|
||||
|
||||
// Size returns the length of the named object in bytesWritten.
|
||||
Size(objName string) (int64, error)
|
||||
|
||||
// IsNotExistError returns true if the given error (returned by a method in
|
||||
// this interface) indicates that the object does not exist.
|
||||
IsNotExistError(err error) bool
|
||||
}
|
||||
|
||||
// ObjectReader is used to perform reads on an object.
|
||||
type ObjectReader interface {
|
||||
// ReadAt reads len(p) bytes into p starting at offset off.
|
||||
//
|
||||
// Clients of ReadAt can execute parallel ReadAt calls on the
|
||||
// same ObjectReader.
|
||||
ReadAt(ctx context.Context, p []byte, offset int64) error
|
||||
|
||||
Close() error
|
||||
}
|
||||
Reference in New Issue
Block a user