Create sdk/ and api/ submodules (#6583)

sdk/plugin/middleware.go

@@ -0,0 +1,91 @@
+package plugin
+
+import (
+	"context"
+	"time"
+
+	log "github.com/hashicorp/go-hclog"
+	"github.com/hashicorp/vault/sdk/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()
+}