Will Browne
1a0bc39ec3
* tackle some /pkg/infra/* packages * run make update-workspace * add owner for slugify dep
63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
package termination
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/trace"
|
|
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/plugins/config"
|
|
"github.com/grafana/grafana/pkg/plugins/log"
|
|
"github.com/grafana/grafana/pkg/plugins/tracing"
|
|
"github.com/grafana/grafana/pkg/semconv"
|
|
)
|
|
|
|
// Terminator is responsible for the Termination stage of the plugin loader pipeline.
|
|
type Terminator interface {
|
|
Terminate(ctx context.Context, p *plugins.Plugin) (*plugins.Plugin, error)
|
|
}
|
|
|
|
// TerminateFunc is the function used for the Terminate step of the Termination stage.
|
|
type TerminateFunc func(ctx context.Context, p *plugins.Plugin) error
|
|
|
|
type Terminate struct {
|
|
cfg *config.PluginManagementCfg
|
|
terminateSteps []TerminateFunc
|
|
log log.Logger
|
|
tracer trace.Tracer
|
|
}
|
|
|
|
type Opts struct {
|
|
TerminateFuncs []TerminateFunc
|
|
}
|
|
|
|
// New returns a new Termination stage.
|
|
func New(cfg *config.PluginManagementCfg, opts Opts) (*Terminate, error) {
|
|
if opts.TerminateFuncs == nil {
|
|
opts.TerminateFuncs = []TerminateFunc{}
|
|
}
|
|
|
|
return &Terminate{
|
|
cfg: cfg,
|
|
terminateSteps: opts.TerminateFuncs,
|
|
log: log.New("plugins.termination"),
|
|
tracer: otel.Tracer("github.com/grafana/grafana/pkg/plugins/manager/pipeline/termination"),
|
|
}, nil
|
|
}
|
|
|
|
// Terminate will execute the Terminate steps of the Termination stage.
|
|
func (t *Terminate) Terminate(ctx context.Context, p *plugins.Plugin) (*plugins.Plugin, error) {
|
|
ctx, span := t.tracer.Start(ctx, "termination.Terminate", trace.WithAttributes(
|
|
semconv.GrafanaPluginId(p.ID),
|
|
))
|
|
defer span.End()
|
|
|
|
for _, terminate := range t.terminateSteps {
|
|
if err := terminate(ctx, p); err != nil {
|
|
return nil, tracing.Error(span, err)
|
|
}
|
|
}
|
|
return p, nil
|
|
}
|