3db8adc5d7
Allow plugins to be mapped and returned by their ID. Add skip plugin to allow plugins to decide whether they should be loaded. Signed-off-by: Derek McGowan <derek@mcgstyle.net>
43 lines
915 B
Go
43 lines
915 B
Go
package plugin
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
"github.com/containerd/containerd/events"
|
|
"github.com/containerd/containerd/log"
|
|
)
|
|
|
|
func NewContext(ctx context.Context, plugins map[PluginType]map[string]interface{}, root, id string) *InitContext {
|
|
return &InitContext{
|
|
plugins: plugins,
|
|
Root: filepath.Join(root, id),
|
|
Context: log.WithModule(ctx, id),
|
|
}
|
|
}
|
|
|
|
type InitContext struct {
|
|
Root string
|
|
Context context.Context
|
|
Config interface{}
|
|
Emitter *events.Emitter
|
|
|
|
plugins map[PluginType]map[string]interface{}
|
|
}
|
|
|
|
func (i *InitContext) Get(t PluginType) (interface{}, error) {
|
|
for _, v := range i.plugins[t] {
|
|
return v, nil
|
|
}
|
|
return nil, fmt.Errorf("no plugins registered for %s", t)
|
|
}
|
|
|
|
func (i *InitContext) GetAll(t PluginType) (map[string]interface{}, error) {
|
|
p, ok := i.plugins[t]
|
|
if !ok {
|
|
return nil, fmt.Errorf("no plugins registered for %s", t)
|
|
}
|
|
return p, nil
|
|
}
|