03cf302e9a
* logbridge with hclog and identical output * Initial search & replace This compiles, but there is a fair amount of TODO and commented out code, especially around the plugin logclient/logserver code. * strip logbridge * fix majority of tests * update logxi aliases * WIP fixing tests * more test fixes * Update test to hclog * Fix format * Rename hclog -> log * WIP making hclog and logxi love each other * update logger_test.go * clean up merged comments * Replace RawLogger interface with a Logger * Add some logger names * Replace Trace with Debug * update builtin logical logging patterns * Fix build errors * More log updates * update log approach in command and builtin * More log updates * update helper, http, and logical directories * Update loggers * Log updates * Update logging * Update logging * Update logging * Update logging * update logging in physical * prefixing and lowercase * Update logging * Move phyisical logging name to server command * Fix som tests * address jims feedback so far * incorporate brians feedback so far * strip comments * move vault.go to logging package * update Debug to Trace * Update go-plugin deps * Update logging based on review comments * Updates from review * Unvendor logxi * Remove null_logger.go
92 lines
2.8 KiB
Go
92 lines
2.8 KiB
Go
package plugin
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/vault/logical"
|
|
)
|
|
|
|
// backendPluginClient implements logical.Backend and is the
|
|
// go-plugin client.
|
|
type backendTracingMiddleware struct {
|
|
logger log.Logger
|
|
|
|
next logical.Backend
|
|
}
|
|
|
|
// Validate the backendTracingMiddle object satisfies the backend interface
|
|
var _ logical.Backend = &backendTracingMiddleware{}
|
|
|
|
func (b *backendTracingMiddleware) HandleRequest(ctx context.Context, req *logical.Request) (resp *logical.Response, err error) {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("handle request", "path", req.Path, "status", "finished", "err", err, "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("handle request", "path", req.Path, "status", "started")
|
|
return b.next.HandleRequest(ctx, req)
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) SpecialPaths() *logical.Paths {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("special paths", "status", "finished", "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("special paths", "status", "started")
|
|
return b.next.SpecialPaths()
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) System() logical.SystemView {
|
|
return b.next.System()
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) Logger() log.Logger {
|
|
return b.next.Logger()
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) HandleExistenceCheck(ctx context.Context, req *logical.Request) (found bool, exists bool, err error) {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("handle existence check", "path", req.Path, "status", "finished", "err", err, "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("handle existence check", "path", req.Path, "status", "started")
|
|
return b.next.HandleExistenceCheck(ctx, req)
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) Cleanup(ctx context.Context) {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("cleanup", "status", "finished", "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("cleanup", "status", "started")
|
|
b.next.Cleanup(ctx)
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) InvalidateKey(ctx context.Context, key string) {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("invalidate key", "key", key, "status", "finished", "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("invalidate key", "key", key, "status", "started")
|
|
b.next.InvalidateKey(ctx, key)
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) Setup(ctx context.Context, config *logical.BackendConfig) (err error) {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("setup", "status", "finished", "err", err, "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("setup", "status", "started")
|
|
return b.next.Setup(ctx, config)
|
|
}
|
|
|
|
func (b *backendTracingMiddleware) Type() logical.BackendType {
|
|
defer func(then time.Time) {
|
|
b.logger.Trace("type", "status", "finished", "took", time.Since(then))
|
|
}(time.Now())
|
|
|
|
b.logger.Trace("type", "status", "started")
|
|
return b.next.Type()
|
|
}
|