mirror of https://github.com/grafana/grafana.git
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/grafana/authlib/claims"
|
|
infraDB "github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/tracing"
|
|
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
|
"github.com/grafana/grafana/pkg/storage/unified/sql/db/dbimpl"
|
|
)
|
|
|
|
// Creates a new ResourceServer
|
|
func NewResourceServer(ctx context.Context, db infraDB.DB, cfg *setting.Cfg, features featuremgmt.FeatureToggles, tracer tracing.Tracer) (resource.ResourceServer, error) {
|
|
opts := resource.ResourceServerOptions{
|
|
Tracer: tracer,
|
|
}
|
|
|
|
eDB, err := dbimpl.ProvideResourceDB(db, cfg, tracer)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
store, err := NewBackend(BackendOptions{DBProvider: eDB, Tracer: tracer})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
opts.Backend = store
|
|
opts.Diagnostics = store
|
|
opts.Lifecycle = store
|
|
|
|
if features.IsEnabledGlobally(featuremgmt.FlagUnifiedStorageSearch) {
|
|
opts.Index = resource.NewResourceIndexServer()
|
|
server, err := resource.NewResourceServer(opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// initialze the search index
|
|
indexer, ok := server.(resource.ResourceIndexer)
|
|
if !ok {
|
|
return nil, errors.New("index server does not implement ResourceIndexer")
|
|
}
|
|
_, err = indexer.Index(ctx)
|
|
return server, err
|
|
}
|
|
|
|
if features.IsEnabledGlobally(featuremgmt.FlagKubernetesFolders) {
|
|
opts.WriteAccess = resource.WriteAccessHooks{
|
|
Folder: func(ctx context.Context, user claims.AuthInfo, uid string) bool {
|
|
// #TODO build on the logic here
|
|
// #TODO only enable write access when the resource being written in the folder
|
|
// is another folder
|
|
return true
|
|
},
|
|
}
|
|
}
|
|
|
|
return resource.NewResourceServer(opts)
|
|
}
|