c1f1aacffe
There's no reason it should be at the top-level. Change-Id: I35b06e7baa0e425c6ff9a82964d0a1570d4eb6d0
83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package storelogger
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"sync/atomic"
|
|
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/storj/private/kvstore"
|
|
)
|
|
|
|
var mon = monkit.Package()
|
|
|
|
var id int64
|
|
|
|
// Logger implements a zap.Logger for kvstore.Store.
|
|
type Logger struct {
|
|
log *zap.Logger
|
|
store kvstore.Store
|
|
}
|
|
|
|
// New creates a new Logger with log and store.
|
|
func New(log *zap.Logger, store kvstore.Store) *Logger {
|
|
loggerid := atomic.AddInt64(&id, 1)
|
|
name := strconv.Itoa(int(loggerid))
|
|
return &Logger{log.Named(name), store}
|
|
}
|
|
|
|
// Put adds a value to store.
|
|
func (store *Logger) Put(ctx context.Context, key kvstore.Key, value kvstore.Value) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
store.log.Debug("Put", zap.ByteString("key", key), zap.Int("value length", len(value)), zap.Binary("truncated value", truncate(value)))
|
|
return store.store.Put(ctx, key, value)
|
|
}
|
|
|
|
// Get gets a value to store.
|
|
func (store *Logger) Get(ctx context.Context, key kvstore.Key) (_ kvstore.Value, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
store.log.Debug("Get", zap.ByteString("key", key))
|
|
return store.store.Get(ctx, key)
|
|
}
|
|
|
|
// Delete deletes key and the value.
|
|
func (store *Logger) Delete(ctx context.Context, key kvstore.Key) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
store.log.Debug("Delete", zap.ByteString("key", key))
|
|
return store.store.Delete(ctx, key)
|
|
}
|
|
|
|
// Range iterates over all items in unspecified order.
|
|
func (store *Logger) Range(ctx context.Context, fn func(context.Context, kvstore.Key, kvstore.Value) error) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
store.log.Debug("Range")
|
|
return store.store.Range(ctx, func(ctx context.Context, key kvstore.Key, value kvstore.Value) error {
|
|
store.log.Debug(" ",
|
|
zap.ByteString("key", key),
|
|
zap.Int("value length", len(value)),
|
|
zap.Binary("truncated value", truncate(value)),
|
|
)
|
|
return fn(ctx, key, value)
|
|
})
|
|
}
|
|
|
|
// Close closes the store.
|
|
func (store *Logger) Close() error {
|
|
store.log.Debug("Close")
|
|
return store.store.Close()
|
|
}
|
|
|
|
func truncate(v kvstore.Value) (t []byte) {
|
|
if len(v)-1 < 10 {
|
|
t = []byte(v)
|
|
} else {
|
|
t = v[:10]
|
|
}
|
|
return t
|
|
}
|