Daniele Stefano Ferru
e4b79e2fc8
* WIP: mutator added, start working on validator * first validator iteration * second validator iteration * wip: working on integration tests * re-working mutation and validation, using Connection interface * fixing some rebase things * fixing integration tests * formatting * fixing unit tests * k8s codegen * linting * moving tests which are available only for enterprise * addressing comments: using repo config for connections, updating tests * addressing comments: adding some more info in the app and installation * fixing app data * addressing comments: updating connection implementation * addressing comments * formatting * fixing tests
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package connection
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
|
|
provisioning "github.com/grafana/grafana/apps/provisioning/pkg/apis/provisioning/v0alpha1"
|
|
)
|
|
|
|
//go:generate mockery --name=Extra --structname=MockExtra --inpackage --filename=extra_mock.go --with-expecter
|
|
type Extra interface {
|
|
Type() provisioning.ConnectionType
|
|
Build(ctx context.Context, r *provisioning.Connection) (Connection, error)
|
|
}
|
|
|
|
//go:generate mockery --name=Factory --structname=MockFactory --inpackage --filename=factory_mock.go --with-expecter
|
|
type Factory interface {
|
|
Types() []provisioning.ConnectionType
|
|
Build(ctx context.Context, r *provisioning.Connection) (Connection, error)
|
|
}
|
|
|
|
type factory struct {
|
|
extras map[provisioning.ConnectionType]Extra
|
|
enabled map[provisioning.ConnectionType]struct{}
|
|
}
|
|
|
|
func ProvideFactory(enabled map[provisioning.ConnectionType]struct{}, extras []Extra) (Factory, error) {
|
|
f := &factory{
|
|
enabled: enabled,
|
|
extras: make(map[provisioning.ConnectionType]Extra, len(extras)),
|
|
}
|
|
|
|
for _, e := range extras {
|
|
if _, exists := f.extras[e.Type()]; exists {
|
|
return nil, fmt.Errorf("connection type %q is already registered", e.Type())
|
|
}
|
|
f.extras[e.Type()] = e
|
|
}
|
|
|
|
return f, nil
|
|
}
|
|
|
|
func (f *factory) Types() []provisioning.ConnectionType {
|
|
var types []provisioning.ConnectionType
|
|
for t := range f.enabled {
|
|
if _, exists := f.extras[t]; exists {
|
|
types = append(types, t)
|
|
}
|
|
}
|
|
|
|
sort.Slice(types, func(i, j int) bool {
|
|
return string(types[i]) < string(types[j])
|
|
})
|
|
|
|
return types
|
|
}
|
|
|
|
func (f *factory) Build(ctx context.Context, c *provisioning.Connection) (Connection, error) {
|
|
for _, e := range f.extras {
|
|
if e.Type() == c.Spec.Type {
|
|
if _, enabled := f.enabled[e.Type()]; !enabled {
|
|
return nil, fmt.Errorf("connection type %q is not enabled", e.Type())
|
|
}
|
|
|
|
return e.Build(ctx, c)
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("connection type %q is not supported", c.Spec.Type)
|
|
}
|
|
|
|
var (
|
|
_ Factory = (*factory)(nil)
|
|
)
|