mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
9f6144059a
* remove GRN and switch tenant to namespace * clean up remaining references * simplify and remove inconsistency in With* parameters * parse listing keys so we can use db index * bump the schema version --------- Co-authored-by: Ryan McKinley <ryantxu@gmail.com>
95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
package entity
|
|
|
|
import (
|
|
context "context"
|
|
"strconv"
|
|
|
|
grpc "google.golang.org/grpc"
|
|
codes "google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/metadata"
|
|
status "google.golang.org/grpc/status"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/appcontext"
|
|
)
|
|
|
|
var _ EntityStoreServer = (*entityStoreClientWrapper)(nil)
|
|
|
|
// wrapper for EntityStoreClient that implements EntityStore interface
|
|
type entityStoreClientWrapper struct {
|
|
EntityStoreClient
|
|
}
|
|
|
|
func (c *entityStoreClientWrapper) Read(ctx context.Context, in *ReadEntityRequest) (*Entity, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.Read(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) BatchRead(ctx context.Context, in *BatchReadEntityRequest) (*BatchReadEntityResponse, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.BatchRead(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) Create(ctx context.Context, in *CreateEntityRequest) (*CreateEntityResponse, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.Create(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) Update(ctx context.Context, in *UpdateEntityRequest) (*UpdateEntityResponse, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.Update(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) Delete(ctx context.Context, in *DeleteEntityRequest) (*DeleteEntityResponse, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.Delete(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) History(ctx context.Context, in *EntityHistoryRequest) (*EntityHistoryResponse, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.History(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) List(ctx context.Context, in *EntityListRequest) (*EntityListResponse, error) {
|
|
ctx, err := c.wrapContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.EntityStoreClient.List(ctx, in)
|
|
}
|
|
func (c *entityStoreClientWrapper) Watch(*EntityWatchRequest, EntityStore_WatchServer) error {
|
|
return status.Errorf(codes.Unimplemented, "method Watch not implemented")
|
|
}
|
|
|
|
func (c *entityStoreClientWrapper) wrapContext(ctx context.Context) (context.Context, error) {
|
|
user, err := appcontext.User(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// set grpc metadata into the context to pass to the grpc server
|
|
ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs(
|
|
"grafana-idtoken", user.IDToken,
|
|
"grafana-userid", strconv.FormatInt(user.UserID, 10),
|
|
"grafana-orgid", strconv.FormatInt(user.OrgID, 10),
|
|
"grafana-login", user.Login,
|
|
))
|
|
|
|
return ctx, nil
|
|
}
|
|
|
|
func NewEntityStoreClientWrapper(cc grpc.ClientConnInterface) EntityStoreServer {
|
|
return &entityStoreClientWrapper{&entityStoreClient{cc}}
|
|
}
|