K8s: Remove restore functionality; can be done with list (#102560)
This commit is contained in:
@@ -164,7 +164,6 @@ Experimental features might be changed or removed without prior notice.
|
||||
| `disableClassicHTTPHistogram` | Disables classic HTTP Histogram (use with enableNativeHTTPHistogram) |
|
||||
| `kubernetesSnapshots` | Routes snapshot requests from /api to the /apis endpoint |
|
||||
| `kubernetesDashboards` | Use the kubernetes API in the frontend for dashboards |
|
||||
| `kubernetesRestore` | Allow restoring objects in k8s |
|
||||
| `kubernetesClientDashboardsFolders` | Route the folder and dashboard service requests to k8s |
|
||||
| `datasourceQueryTypes` | Show query type endpoints in datasource API servers (currently hardcoded for testdata, expressions, and prometheus) |
|
||||
| `queryService` | Register /apis/query.grafana.app/ -- will eventually replace /api/ds/query |
|
||||
|
||||
@@ -390,10 +390,6 @@ export interface FeatureToggles {
|
||||
*/
|
||||
kubernetesDashboards?: boolean;
|
||||
/**
|
||||
* Allow restoring objects in k8s
|
||||
*/
|
||||
kubernetesRestore?: boolean;
|
||||
/**
|
||||
* Route the folder and dashboard service requests to k8s
|
||||
*/
|
||||
kubernetesClientDashboardsFolders?: boolean;
|
||||
|
||||
@@ -41,11 +41,9 @@ func GetAuthorizer(dashboardService dashboards.DashboardService, l log.Logger) a
|
||||
}
|
||||
|
||||
// expensive path to lookup permissions for a single dashboard
|
||||
// must include deleted to allow for restores
|
||||
dto, err := dashboardService.GetDashboard(ctx, &dashboards.GetDashboardQuery{
|
||||
UID: attr.GetName(),
|
||||
OrgID: info.OrgID,
|
||||
IncludeDeleted: true,
|
||||
UID: attr.GetName(),
|
||||
OrgID: info.OrgID,
|
||||
})
|
||||
if err != nil {
|
||||
return authorizer.DecisionDeny, "error loading dashboard", err
|
||||
|
||||
@@ -1,103 +0,0 @@
|
||||
package dashboard
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
||||
"k8s.io/apimachinery/pkg/runtime"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/apiserver/pkg/registry/rest"
|
||||
"k8s.io/apiserver/pkg/storage"
|
||||
|
||||
"github.com/grafana/grafana/pkg/services/apiserver/endpoints/request"
|
||||
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
||||
)
|
||||
|
||||
// LatestConnector will return the latest version of the resource - even if it is deleted
|
||||
type LatestConnector interface {
|
||||
rest.Storage
|
||||
rest.Connecter
|
||||
rest.StorageMetadata
|
||||
}
|
||||
|
||||
func NewLatestConnector(unified resource.ResourceClient, gr schema.GroupResource) LatestConnector {
|
||||
return &latestREST{
|
||||
unified: unified,
|
||||
gr: gr,
|
||||
}
|
||||
}
|
||||
|
||||
type latestREST struct {
|
||||
unified resource.ResourceClient
|
||||
gr schema.GroupResource
|
||||
}
|
||||
|
||||
func (l *latestREST) New() runtime.Object {
|
||||
return &metav1.PartialObjectMetadataList{}
|
||||
}
|
||||
|
||||
func (l *latestREST) Destroy() {
|
||||
}
|
||||
|
||||
func (l *latestREST) ConnectMethods() []string {
|
||||
return []string{"GET"}
|
||||
}
|
||||
|
||||
func (l *latestREST) ProducesMIMETypes(verb string) []string {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (l *latestREST) ProducesObject(verb string) interface{} {
|
||||
return &metav1.PartialObjectMetadataList{}
|
||||
}
|
||||
|
||||
func (l *latestREST) NewConnectOptions() (runtime.Object, bool, string) {
|
||||
return nil, false, ""
|
||||
}
|
||||
|
||||
func (l *latestREST) Connect(ctx context.Context, uid string, opts runtime.Object, responder rest.Responder) (http.Handler, error) {
|
||||
info, err := request.NamespaceInfoFrom(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
key := &resource.ResourceKey{
|
||||
Namespace: info.Value,
|
||||
Group: l.gr.Group,
|
||||
Resource: l.gr.Resource,
|
||||
Name: uid,
|
||||
}
|
||||
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
||||
rsp, err := l.unified.Read(ctx, &resource.ReadRequest{
|
||||
Key: key,
|
||||
ResourceVersion: 0, // 0 will return the latest version that was not a delete event
|
||||
IncludeDeleted: true,
|
||||
})
|
||||
if err != nil {
|
||||
responder.Error(err)
|
||||
return
|
||||
} else if rsp == nil || (rsp.Error != nil && rsp.Error.Code == http.StatusNotFound) {
|
||||
responder.Error(storage.NewKeyNotFoundError(uid, 0))
|
||||
return
|
||||
} else if rsp.Error != nil {
|
||||
responder.Error(fmt.Errorf("could not retrieve object: %s", rsp.Error.Message))
|
||||
return
|
||||
}
|
||||
|
||||
uncastObj, err := runtime.Decode(unstructured.UnstructuredJSONScheme, rsp.Value)
|
||||
if err != nil {
|
||||
responder.Error(fmt.Errorf("could not convert object: %s", err.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
finalObj := uncastObj.(*unstructured.Unstructured)
|
||||
finalObj.SetResourceVersion(strconv.FormatInt(rsp.ResourceVersion, 10))
|
||||
|
||||
responder.Object(http.StatusOK, finalObj)
|
||||
}), nil
|
||||
}
|
||||
@@ -1,99 +0,0 @@
|
||||
package dashboard
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/mock"
|
||||
"github.com/stretchr/testify/require"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/apiserver/pkg/endpoints/request"
|
||||
)
|
||||
|
||||
func TestLatest(t *testing.T) {
|
||||
gr := schema.GroupResource{
|
||||
Group: "group",
|
||||
Resource: "resource",
|
||||
}
|
||||
ctx := context.Background()
|
||||
mockResponder := &mockResponder{}
|
||||
mockClient := &mockResourceClient{}
|
||||
r := &latestREST{
|
||||
unified: mockClient,
|
||||
gr: gr,
|
||||
}
|
||||
|
||||
t.Run("no namespace in context", func(t *testing.T) {
|
||||
_, err := r.Connect(ctx, "test-uid", nil, mockResponder)
|
||||
require.Error(t, err)
|
||||
})
|
||||
|
||||
ctx = request.WithNamespace(context.Background(), "default")
|
||||
|
||||
t.Run("happy path", func(t *testing.T) {
|
||||
req := httptest.NewRequest("GET", "/latest", nil)
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
readReq := &resource.ReadRequest{
|
||||
Key: &resource.ResourceKey{
|
||||
Namespace: "default",
|
||||
Group: "group",
|
||||
Resource: "resource",
|
||||
Name: "uid",
|
||||
},
|
||||
ResourceVersion: 0,
|
||||
IncludeDeleted: true,
|
||||
}
|
||||
|
||||
expectedObject := &metav1.PartialObjectMetadata{
|
||||
TypeMeta: metav1.TypeMeta{
|
||||
Kind: "resource",
|
||||
APIVersion: "v0alpha1",
|
||||
},
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: "uid",
|
||||
Namespace: "default",
|
||||
ResourceVersion: strconv.FormatInt(123, 10),
|
||||
},
|
||||
}
|
||||
|
||||
val, err := json.Marshal(expectedObject)
|
||||
require.NoError(t, err)
|
||||
mockClient.On("Read", ctx, readReq).Return(&resource.ReadResponse{
|
||||
ResourceVersion: 123,
|
||||
Value: val,
|
||||
}, nil).Once()
|
||||
|
||||
mockResponder.On("Object", http.StatusOK, mock.MatchedBy(func(obj interface{}) bool {
|
||||
unstructuredObj, ok := obj.(*unstructured.Unstructured)
|
||||
expectedMap := map[string]interface{}{
|
||||
"apiVersion": expectedObject.APIVersion,
|
||||
"kind": expectedObject.Kind,
|
||||
"metadata": map[string]interface{}{
|
||||
"name": expectedObject.Name,
|
||||
"namespace": expectedObject.Namespace,
|
||||
"resourceVersion": expectedObject.ResourceVersion,
|
||||
"creationTimestamp": nil,
|
||||
},
|
||||
}
|
||||
return ok && reflect.DeepEqual(unstructuredObj.Object, expectedMap)
|
||||
}))
|
||||
|
||||
handler, err := r.Connect(ctx, "uid", nil, mockResponder)
|
||||
require.NoError(t, err)
|
||||
handler.ServeHTTP(w, req)
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
|
||||
mockClient.AssertExpectations(t)
|
||||
mockResponder.AssertExpectations(t)
|
||||
})
|
||||
}
|
||||
@@ -70,11 +70,6 @@ func (d *directResourceClient) Read(ctx context.Context, in *resource.ReadReques
|
||||
return d.server.Read(ctx, in)
|
||||
}
|
||||
|
||||
// Restore implements ResourceClient.
|
||||
func (d *directResourceClient) Restore(ctx context.Context, in *resource.RestoreRequest, opts ...grpc.CallOption) (*resource.RestoreResponse, error) {
|
||||
return d.server.Restore(ctx, in)
|
||||
}
|
||||
|
||||
// Search implements ResourceClient.
|
||||
func (d *directResourceClient) Search(ctx context.Context, in *resource.ResourceSearchRequest, opts ...grpc.CallOption) (*resource.ResourceSearchResponse, error) {
|
||||
return d.server.Search(ctx, in)
|
||||
|
||||
@@ -280,11 +280,6 @@ func (b *DashboardsAPIBuilder) storageForVersion(
|
||||
return err
|
||||
}
|
||||
|
||||
if b.features.IsEnabledGlobally(featuremgmt.FlagKubernetesRestore) {
|
||||
storage[dashboards.StoragePath("restore")] = NewRestoreConnector(b.unified, gr)
|
||||
storage[dashboards.StoragePath("latest")] = NewLatestConnector(b.unified, gr)
|
||||
}
|
||||
|
||||
// Register the DTO endpoint that will consolidate all dashboard bits
|
||||
storage[dashboards.StoragePath("dto")], err = NewDTOConnector(
|
||||
storage[dashboards.StoragePath()].(rest.Getter),
|
||||
|
||||
@@ -1,122 +0,0 @@
|
||||
package dashboard
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/runtime"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/apiserver/pkg/registry/rest"
|
||||
"k8s.io/apiserver/pkg/storage"
|
||||
|
||||
"github.com/grafana/grafana/pkg/services/apiserver/endpoints/request"
|
||||
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
||||
)
|
||||
|
||||
type RestoreConnector interface {
|
||||
rest.Storage
|
||||
rest.Connecter
|
||||
rest.StorageMetadata
|
||||
}
|
||||
|
||||
func NewRestoreConnector(unified resource.ResourceClient, gr schema.GroupResource) RestoreConnector {
|
||||
return &restoreREST{
|
||||
unified: unified,
|
||||
gr: gr,
|
||||
}
|
||||
}
|
||||
|
||||
type restoreREST struct {
|
||||
unified resource.ResourceClient
|
||||
gr schema.GroupResource
|
||||
}
|
||||
|
||||
func (r *restoreREST) New() runtime.Object {
|
||||
return &metav1.PartialObjectMetadataList{}
|
||||
}
|
||||
|
||||
func (r *restoreREST) Destroy() {
|
||||
}
|
||||
|
||||
func (r *restoreREST) ConnectMethods() []string {
|
||||
return []string{"POST"}
|
||||
}
|
||||
|
||||
func (r *restoreREST) ProducesMIMETypes(verb string) []string {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *restoreREST) ProducesObject(verb string) interface{} {
|
||||
return &metav1.PartialObjectMetadataList{}
|
||||
}
|
||||
|
||||
func (r *restoreREST) NewConnectOptions() (runtime.Object, bool, string) {
|
||||
return nil, false, ""
|
||||
}
|
||||
|
||||
type RestoreOptions struct {
|
||||
ResourceVersion int64 `json:"resourceVersion"`
|
||||
}
|
||||
|
||||
func (r *restoreREST) Connect(ctx context.Context, uid string, opts runtime.Object, responder rest.Responder) (http.Handler, error) {
|
||||
info, err := request.NamespaceInfoFrom(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
key := &resource.ResourceKey{
|
||||
Namespace: info.Value,
|
||||
Group: r.gr.Group,
|
||||
Resource: r.gr.Resource,
|
||||
Name: uid,
|
||||
}
|
||||
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
||||
body, err := io.ReadAll(req.Body)
|
||||
if err != nil {
|
||||
responder.Error(fmt.Errorf("unable to read request body: %s", err.Error()))
|
||||
return
|
||||
}
|
||||
reqBody := &RestoreOptions{}
|
||||
err = json.Unmarshal(body, &reqBody)
|
||||
if err != nil {
|
||||
responder.Error(fmt.Errorf("unable to unmarshal request body: %s", err.Error()))
|
||||
return
|
||||
}
|
||||
|
||||
if reqBody.ResourceVersion == 0 {
|
||||
responder.Error(fmt.Errorf("resource version required"))
|
||||
return
|
||||
}
|
||||
|
||||
rsp, err := r.unified.Restore(ctx, &resource.RestoreRequest{
|
||||
ResourceVersion: reqBody.ResourceVersion,
|
||||
Key: key,
|
||||
})
|
||||
if err != nil {
|
||||
responder.Error(err)
|
||||
return
|
||||
} else if rsp == nil || (rsp.Error != nil && rsp.Error.Code == http.StatusNotFound) {
|
||||
responder.Error(storage.NewKeyNotFoundError(uid, reqBody.ResourceVersion))
|
||||
return
|
||||
} else if rsp.Error != nil {
|
||||
responder.Error(fmt.Errorf("could not re-create object: %s", rsp.Error.Message))
|
||||
return
|
||||
}
|
||||
|
||||
obj := metav1.PartialObjectMetadata{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: key.Name,
|
||||
Namespace: key.Namespace,
|
||||
ResourceVersion: strconv.FormatInt(rsp.ResourceVersion, 10),
|
||||
},
|
||||
}
|
||||
|
||||
responder.Object(http.StatusOK, &obj)
|
||||
}), nil
|
||||
}
|
||||
@@ -1,126 +0,0 @@
|
||||
package dashboard
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/grafana/grafana/pkg/storage/unified/resource"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/mock"
|
||||
"google.golang.org/grpc"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/runtime"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/apiserver/pkg/endpoints/request"
|
||||
)
|
||||
|
||||
type mockResourceClient struct {
|
||||
mock.Mock
|
||||
resource.ResourceClient
|
||||
}
|
||||
|
||||
func (m *mockResourceClient) Restore(ctx context.Context, req *resource.RestoreRequest, opts ...grpc.CallOption) (*resource.RestoreResponse, error) {
|
||||
args := m.Called(ctx, req)
|
||||
return args.Get(0).(*resource.RestoreResponse), args.Error(1)
|
||||
}
|
||||
|
||||
func (m *mockResourceClient) Read(ctx context.Context, req *resource.ReadRequest, opts ...grpc.CallOption) (*resource.ReadResponse, error) {
|
||||
args := m.Called(ctx, req)
|
||||
return args.Get(0).(*resource.ReadResponse), args.Error(1)
|
||||
}
|
||||
|
||||
type mockResponder struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
func (m *mockResponder) Object(statusCode int, obj runtime.Object) {
|
||||
m.Called(statusCode, obj)
|
||||
}
|
||||
|
||||
func (m *mockResponder) Error(err error) {
|
||||
m.Called(err)
|
||||
}
|
||||
|
||||
func TestRestore(t *testing.T) {
|
||||
gr := schema.GroupResource{
|
||||
Group: "group",
|
||||
Resource: "resource",
|
||||
}
|
||||
ctx := context.Background()
|
||||
mockResponder := &mockResponder{}
|
||||
mockClient := &mockResourceClient{}
|
||||
r := &restoreREST{
|
||||
unified: mockClient,
|
||||
gr: gr,
|
||||
}
|
||||
|
||||
t.Run("no namespace in context", func(t *testing.T) {
|
||||
_, err := r.Connect(ctx, "test-uid", nil, mockResponder)
|
||||
assert.Error(t, err)
|
||||
})
|
||||
|
||||
ctx = request.WithNamespace(context.Background(), "default")
|
||||
|
||||
t.Run("invalid resourceVersion", func(t *testing.T) {
|
||||
req := httptest.NewRequest("POST", "/restore", bytes.NewReader([]byte(`{"resourceVersion":0}`)))
|
||||
w := httptest.NewRecorder()
|
||||
|
||||
expectedError := fmt.Errorf("resource version required")
|
||||
mockResponder.On("Error", mock.MatchedBy(func(err error) bool {
|
||||
return err.Error() == expectedError.Error()
|
||||
}))
|
||||
|
||||
handler, err := r.Connect(ctx, "test-uid", nil, mockResponder)
|
||||
assert.NoError(t, err)
|
||||
|
||||
handler.ServeHTTP(w, req)
|
||||
mockResponder.AssertExpectations(t)
|
||||
})
|
||||
|
||||
t.Run("happy path", func(t *testing.T) {
|
||||
req := httptest.NewRequest("POST", "/restore", bytes.NewReader([]byte(`{"resourceVersion":123}`)))
|
||||
w := httptest.NewRecorder()
|
||||
restoreReq := &resource.RestoreRequest{
|
||||
ResourceVersion: 123,
|
||||
Key: &resource.ResourceKey{
|
||||
Namespace: "default",
|
||||
Group: "group",
|
||||
Resource: "resource",
|
||||
Name: "uid",
|
||||
},
|
||||
}
|
||||
|
||||
expectedObject := &metav1.PartialObjectMetadata{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: "uid",
|
||||
Namespace: "default",
|
||||
ResourceVersion: strconv.FormatInt(123, 10),
|
||||
},
|
||||
}
|
||||
|
||||
mockClient.On("Restore", ctx, restoreReq).Return(&resource.RestoreResponse{
|
||||
ResourceVersion: 123,
|
||||
}, nil).Once()
|
||||
|
||||
mockResponder.On("Object", http.StatusOK, mock.MatchedBy(func(obj interface{}) bool {
|
||||
metadata, ok := obj.(*metav1.PartialObjectMetadata)
|
||||
return ok &&
|
||||
metadata.ObjectMeta.Name == "uid" &&
|
||||
metadata.ObjectMeta.Namespace == "default" &&
|
||||
metadata.ObjectMeta.ResourceVersion == "123"
|
||||
})).Return(expectedObject)
|
||||
|
||||
handler, err := r.Connect(ctx, "uid", nil, mockResponder)
|
||||
assert.NoError(t, err)
|
||||
handler.ServeHTTP(w, req)
|
||||
assert.Equal(t, http.StatusOK, w.Code)
|
||||
|
||||
mockClient.AssertExpectations(t)
|
||||
mockResponder.AssertExpectations(t)
|
||||
})
|
||||
}
|
||||
@@ -459,7 +459,7 @@ func (s *SearchHandler) getDashboardsUIDsSharedWithUser(ctx context.Context, use
|
||||
}
|
||||
|
||||
if folderUidIdx == -1 {
|
||||
return sharedDashboards, fmt.Errorf("Error retrieving folder information")
|
||||
return sharedDashboards, fmt.Errorf("error retrieving folder information")
|
||||
}
|
||||
|
||||
// populate list of unique folder UIDs in the list of dashboards user has read permissions
|
||||
|
||||
@@ -692,9 +692,6 @@ func (m *MockClient) Update(ctx context.Context, in *resource.UpdateRequest, opt
|
||||
func (m *MockClient) Read(ctx context.Context, in *resource.ReadRequest, opts ...grpc.CallOption) (*resource.ReadResponse, error) {
|
||||
return nil, nil
|
||||
}
|
||||
func (m *MockClient) Restore(ctx context.Context, in *resource.RestoreRequest, opts ...grpc.CallOption) (*resource.RestoreResponse, error) {
|
||||
return nil, nil
|
||||
}
|
||||
func (m *MockClient) GetBlob(ctx context.Context, in *resource.GetBlobRequest, opts ...grpc.CallOption) (*resource.GetBlobResponse, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
@@ -265,8 +265,6 @@ type GetDashboardQuery struct {
|
||||
FolderID *int64
|
||||
FolderUID *string
|
||||
OrgID int64
|
||||
|
||||
IncludeDeleted bool // only supported when using unified storage
|
||||
}
|
||||
|
||||
type DashboardTagCloudItem struct {
|
||||
|
||||
@@ -757,7 +757,7 @@ func (dr *DashboardServiceImpl) saveDashboard(ctx context.Context, cmd *dashboar
|
||||
|
||||
func (dr *DashboardServiceImpl) GetSoftDeletedDashboard(ctx context.Context, orgID int64, uid string) (*dashboards.Dashboard, error) {
|
||||
if dr.features.IsEnabledGlobally(featuremgmt.FlagKubernetesClientDashboardsFolders) {
|
||||
return dr.getDashboardThroughK8s(ctx, &dashboards.GetDashboardQuery{OrgID: orgID, UID: uid, IncludeDeleted: true})
|
||||
return dr.getDashboardThroughK8s(ctx, &dashboards.GetDashboardQuery{OrgID: orgID, UID: uid})
|
||||
}
|
||||
|
||||
return dr.dashboardStore.GetSoftDeletedDashboard(ctx, orgID, uid)
|
||||
@@ -1520,12 +1520,6 @@ func (dr *DashboardServiceImpl) CleanUpDeletedDashboards(ctx context.Context) (i
|
||||
// -----------------------------------------------------------------------------------------
|
||||
|
||||
func (dr *DashboardServiceImpl) getDashboardThroughK8s(ctx context.Context, query *dashboards.GetDashboardQuery) (*dashboards.Dashboard, error) {
|
||||
// if including deleted dashboards for restore, use the /latest subresource
|
||||
subresource := ""
|
||||
if query.IncludeDeleted && dr.features.IsEnabledGlobally(featuremgmt.FlagKubernetesRestore) {
|
||||
subresource = "latest"
|
||||
}
|
||||
|
||||
// get uid if not passed in
|
||||
if query.UID == "" {
|
||||
result, err := dr.GetDashboardUIDByID(ctx, &dashboards.GetDashboardRefByIDQuery{
|
||||
@@ -1538,7 +1532,7 @@ func (dr *DashboardServiceImpl) getDashboardThroughK8s(ctx context.Context, quer
|
||||
query.UID = result.UID
|
||||
}
|
||||
|
||||
out, err := dr.k8sclient.Get(ctx, query.UID, query.OrgID, v1.GetOptions{}, subresource)
|
||||
out, err := dr.k8sclient.Get(ctx, query.UID, query.OrgID, v1.GetOptions{}, "")
|
||||
if err != nil && !apierrors.IsNotFound(err) {
|
||||
return nil, err
|
||||
} else if err != nil || out == nil {
|
||||
|
||||
@@ -656,12 +656,6 @@ var (
|
||||
Owner: grafanaAppPlatformSquad,
|
||||
FrontendOnly: true,
|
||||
},
|
||||
{
|
||||
Name: "kubernetesRestore",
|
||||
Description: "Allow restoring objects in k8s",
|
||||
Stage: FeatureStageExperimental,
|
||||
Owner: grafanaAppPlatformSquad,
|
||||
},
|
||||
{
|
||||
Name: "kubernetesClientDashboardsFolders",
|
||||
Description: "Route the folder and dashboard service requests to k8s",
|
||||
|
||||
@@ -413,7 +413,6 @@ unifiedStorageSearchUI,2024-12-19T18:21:48Z,,a8f347144ddc16f2033fdeb4f3474e49239
|
||||
playlistsReconciler,2024-12-20T03:09:31Z,,24bf337c562dc9b9d8684cc9acb7ea171ea83414,Charandas
|
||||
k8SFolderCounts,2024-12-27T17:10:44Z,,df36e77cd31d2ad77e3d708748d040367a0c8c9c,Leonor Oliveira
|
||||
k8SFolderMove,2024-12-27T17:10:44Z,,df36e77cd31d2ad77e3d708748d040367a0c8c9c,Leonor Oliveira
|
||||
kubernetesRestore,2025-01-03T14:48:47Z,,5429512779bd5f25b88ff728ea91efdef7dfafa0,Stephanie Hingtgen
|
||||
improvedExternalSessionHandlingSAML,2025-01-09T17:02:49Z,,c52ec21c75ab72c2f7d28259bac0364edae560d0,Misi
|
||||
teamHttpHeadersMimir,2025-01-13T10:42:47Z,,04acbcdef23f673bd6bbfdbbece29c9769ce155a,Eric Leijonmarck
|
||||
ABTestFeatureToggleA,2025-01-13T21:13:13Z,,009d7f42b3d09b3a6be1f00f07314e2b25af7ebc,Nathan Marrs
|
||||
|
||||
|
@@ -86,7 +86,6 @@ formatString,GA,@grafana/dataviz-squad,false,false,true
|
||||
kubernetesPlaylists,GA,@grafana/grafana-app-platform-squad,false,true,false
|
||||
kubernetesSnapshots,experimental,@grafana/grafana-app-platform-squad,false,true,false
|
||||
kubernetesDashboards,experimental,@grafana/grafana-app-platform-squad,false,false,true
|
||||
kubernetesRestore,experimental,@grafana/grafana-app-platform-squad,false,false,false
|
||||
kubernetesClientDashboardsFolders,experimental,@grafana/grafana-app-platform-squad,false,false,false
|
||||
datasourceQueryTypes,experimental,@grafana/grafana-app-platform-squad,false,true,false
|
||||
queryService,experimental,@grafana/grafana-app-platform-squad,false,true,false
|
||||
|
||||
|
@@ -355,10 +355,6 @@ const (
|
||||
// Use the kubernetes API in the frontend for dashboards
|
||||
FlagKubernetesDashboards = "kubernetesDashboards"
|
||||
|
||||
// FlagKubernetesRestore
|
||||
// Allow restoring objects in k8s
|
||||
FlagKubernetesRestore = "kubernetesRestore"
|
||||
|
||||
// FlagKubernetesClientDashboardsFolders
|
||||
// Route the folder and dashboard service requests to k8s
|
||||
FlagKubernetesClientDashboardsFolders = "kubernetesClientDashboardsFolders"
|
||||
|
||||
@@ -2404,18 +2404,6 @@
|
||||
"expression": "true"
|
||||
}
|
||||
},
|
||||
{
|
||||
"metadata": {
|
||||
"name": "kubernetesRestore",
|
||||
"resourceVersion": "1735880498698",
|
||||
"creationTimestamp": "2025-01-03T14:48:47Z"
|
||||
},
|
||||
"spec": {
|
||||
"description": "Allow restoring objects in k8s",
|
||||
"stage": "experimental",
|
||||
"codeowner": "@grafana/grafana-app-platform-squad"
|
||||
}
|
||||
},
|
||||
{
|
||||
"metadata": {
|
||||
"name": "kubernetesSnapshots",
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -177,8 +177,6 @@ message ReadRequest {
|
||||
|
||||
// Optionally pick an explicit resource version
|
||||
int64 resource_version = 2;
|
||||
// Optionally decide to return the latest RV if deleted
|
||||
bool include_deleted = 3;
|
||||
}
|
||||
|
||||
message ReadResponse {
|
||||
@@ -702,26 +700,6 @@ message ResourceTableRow {
|
||||
bytes object = 4;
|
||||
}
|
||||
|
||||
//----------------------------
|
||||
// Restore Support
|
||||
//----------------------------
|
||||
|
||||
message RestoreRequest {
|
||||
// Full key must be set
|
||||
ResourceKey key = 1;
|
||||
|
||||
// The resource version to restore
|
||||
int64 resource_version = 2;
|
||||
}
|
||||
|
||||
message RestoreResponse {
|
||||
// Error details
|
||||
ErrorResult error = 1;
|
||||
|
||||
// The updated resource version
|
||||
int64 resource_version = 2;
|
||||
}
|
||||
|
||||
//----------------------------
|
||||
// Blob Support
|
||||
//----------------------------
|
||||
@@ -811,7 +789,6 @@ service ResourceStore {
|
||||
rpc Create(CreateRequest) returns (CreateResponse);
|
||||
rpc Update(UpdateRequest) returns (UpdateResponse);
|
||||
rpc Delete(DeleteRequest) returns (DeleteResponse);
|
||||
rpc Restore(RestoreRequest) returns (RestoreResponse);
|
||||
|
||||
// The results *may* include values that should not be returned to the user
|
||||
// This will perform best-effort filtering to increase performace.
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
||||
// versions:
|
||||
// - protoc-gen-go-grpc v1.4.0
|
||||
// - protoc-gen-go-grpc v1.5.1
|
||||
// - protoc (unknown)
|
||||
// source: resource.proto
|
||||
|
||||
@@ -15,17 +15,16 @@ import (
|
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the grpc package it is being compiled against.
|
||||
// Requires gRPC-Go v1.62.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion8
|
||||
// Requires gRPC-Go v1.64.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion9
|
||||
|
||||
const (
|
||||
ResourceStore_Read_FullMethodName = "/resource.ResourceStore/Read"
|
||||
ResourceStore_Create_FullMethodName = "/resource.ResourceStore/Create"
|
||||
ResourceStore_Update_FullMethodName = "/resource.ResourceStore/Update"
|
||||
ResourceStore_Delete_FullMethodName = "/resource.ResourceStore/Delete"
|
||||
ResourceStore_Restore_FullMethodName = "/resource.ResourceStore/Restore"
|
||||
ResourceStore_List_FullMethodName = "/resource.ResourceStore/List"
|
||||
ResourceStore_Watch_FullMethodName = "/resource.ResourceStore/Watch"
|
||||
ResourceStore_Read_FullMethodName = "/resource.ResourceStore/Read"
|
||||
ResourceStore_Create_FullMethodName = "/resource.ResourceStore/Create"
|
||||
ResourceStore_Update_FullMethodName = "/resource.ResourceStore/Update"
|
||||
ResourceStore_Delete_FullMethodName = "/resource.ResourceStore/Delete"
|
||||
ResourceStore_List_FullMethodName = "/resource.ResourceStore/List"
|
||||
ResourceStore_Watch_FullMethodName = "/resource.ResourceStore/Watch"
|
||||
)
|
||||
|
||||
// ResourceStoreClient is the client API for ResourceStore service.
|
||||
@@ -41,7 +40,6 @@ type ResourceStoreClient interface {
|
||||
Create(ctx context.Context, in *CreateRequest, opts ...grpc.CallOption) (*CreateResponse, error)
|
||||
Update(ctx context.Context, in *UpdateRequest, opts ...grpc.CallOption) (*UpdateResponse, error)
|
||||
Delete(ctx context.Context, in *DeleteRequest, opts ...grpc.CallOption) (*DeleteResponse, error)
|
||||
Restore(ctx context.Context, in *RestoreRequest, opts ...grpc.CallOption) (*RestoreResponse, error)
|
||||
// The results *may* include values that should not be returned to the user
|
||||
// This will perform best-effort filtering to increase performace.
|
||||
// NOTE: storage.Interface is ultimatly responsible for the final filtering
|
||||
@@ -49,7 +47,7 @@ type ResourceStoreClient interface {
|
||||
// The results *may* include values that should not be returned to the user
|
||||
// This will perform best-effort filtering to increase performace.
|
||||
// NOTE: storage.Interface is ultimatly responsible for the final filtering
|
||||
Watch(ctx context.Context, in *WatchRequest, opts ...grpc.CallOption) (ResourceStore_WatchClient, error)
|
||||
Watch(ctx context.Context, in *WatchRequest, opts ...grpc.CallOption) (grpc.ServerStreamingClient[WatchEvent], error)
|
||||
}
|
||||
|
||||
type resourceStoreClient struct {
|
||||
@@ -100,16 +98,6 @@ func (c *resourceStoreClient) Delete(ctx context.Context, in *DeleteRequest, opt
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *resourceStoreClient) Restore(ctx context.Context, in *RestoreRequest, opts ...grpc.CallOption) (*RestoreResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(RestoreResponse)
|
||||
err := c.cc.Invoke(ctx, ResourceStore_Restore_FullMethodName, in, out, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *resourceStoreClient) List(ctx context.Context, in *ListRequest, opts ...grpc.CallOption) (*ListResponse, error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
out := new(ListResponse)
|
||||
@@ -120,13 +108,13 @@ func (c *resourceStoreClient) List(ctx context.Context, in *ListRequest, opts ..
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *resourceStoreClient) Watch(ctx context.Context, in *WatchRequest, opts ...grpc.CallOption) (ResourceStore_WatchClient, error) {
|
||||
func (c *resourceStoreClient) Watch(ctx context.Context, in *WatchRequest, opts ...grpc.CallOption) (grpc.ServerStreamingClient[WatchEvent], error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
stream, err := c.cc.NewStream(ctx, &ResourceStore_ServiceDesc.Streams[0], ResourceStore_Watch_FullMethodName, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
x := &resourceStoreWatchClient{ClientStream: stream}
|
||||
x := &grpc.GenericClientStream[WatchRequest, WatchEvent]{ClientStream: stream}
|
||||
if err := x.ClientStream.SendMsg(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
@@ -136,26 +124,12 @@ func (c *resourceStoreClient) Watch(ctx context.Context, in *WatchRequest, opts
|
||||
return x, nil
|
||||
}
|
||||
|
||||
type ResourceStore_WatchClient interface {
|
||||
Recv() (*WatchEvent, error)
|
||||
grpc.ClientStream
|
||||
}
|
||||
|
||||
type resourceStoreWatchClient struct {
|
||||
grpc.ClientStream
|
||||
}
|
||||
|
||||
func (x *resourceStoreWatchClient) Recv() (*WatchEvent, error) {
|
||||
m := new(WatchEvent)
|
||||
if err := x.ClientStream.RecvMsg(m); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
// This type alias is provided for backwards compatibility with existing code that references the prior non-generic stream type by name.
|
||||
type ResourceStore_WatchClient = grpc.ServerStreamingClient[WatchEvent]
|
||||
|
||||
// ResourceStoreServer is the server API for ResourceStore service.
|
||||
// All implementations should embed UnimplementedResourceStoreServer
|
||||
// for forward compatibility
|
||||
// for forward compatibility.
|
||||
//
|
||||
// This provides the CRUD+List+Watch support needed for a k8s apiserver
|
||||
// The semantics and behaviors of this service are constrained by kubernetes
|
||||
@@ -166,7 +140,6 @@ type ResourceStoreServer interface {
|
||||
Create(context.Context, *CreateRequest) (*CreateResponse, error)
|
||||
Update(context.Context, *UpdateRequest) (*UpdateResponse, error)
|
||||
Delete(context.Context, *DeleteRequest) (*DeleteResponse, error)
|
||||
Restore(context.Context, *RestoreRequest) (*RestoreResponse, error)
|
||||
// The results *may* include values that should not be returned to the user
|
||||
// This will perform best-effort filtering to increase performace.
|
||||
// NOTE: storage.Interface is ultimatly responsible for the final filtering
|
||||
@@ -174,12 +147,15 @@ type ResourceStoreServer interface {
|
||||
// The results *may* include values that should not be returned to the user
|
||||
// This will perform best-effort filtering to increase performace.
|
||||
// NOTE: storage.Interface is ultimatly responsible for the final filtering
|
||||
Watch(*WatchRequest, ResourceStore_WatchServer) error
|
||||
Watch(*WatchRequest, grpc.ServerStreamingServer[WatchEvent]) error
|
||||
}
|
||||
|
||||
// UnimplementedResourceStoreServer should be embedded to have forward compatible implementations.
|
||||
type UnimplementedResourceStoreServer struct {
|
||||
}
|
||||
// UnimplementedResourceStoreServer should be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedResourceStoreServer struct{}
|
||||
|
||||
func (UnimplementedResourceStoreServer) Read(context.Context, *ReadRequest) (*ReadResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method Read not implemented")
|
||||
@@ -193,15 +169,13 @@ func (UnimplementedResourceStoreServer) Update(context.Context, *UpdateRequest)
|
||||
func (UnimplementedResourceStoreServer) Delete(context.Context, *DeleteRequest) (*DeleteResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method Delete not implemented")
|
||||
}
|
||||
func (UnimplementedResourceStoreServer) Restore(context.Context, *RestoreRequest) (*RestoreResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method Restore not implemented")
|
||||
}
|
||||
func (UnimplementedResourceStoreServer) List(context.Context, *ListRequest) (*ListResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method List not implemented")
|
||||
}
|
||||
func (UnimplementedResourceStoreServer) Watch(*WatchRequest, ResourceStore_WatchServer) error {
|
||||
func (UnimplementedResourceStoreServer) Watch(*WatchRequest, grpc.ServerStreamingServer[WatchEvent]) error {
|
||||
return status.Errorf(codes.Unimplemented, "method Watch not implemented")
|
||||
}
|
||||
func (UnimplementedResourceStoreServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeResourceStoreServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to ResourceStoreServer will
|
||||
@@ -211,6 +185,13 @@ type UnsafeResourceStoreServer interface {
|
||||
}
|
||||
|
||||
func RegisterResourceStoreServer(s grpc.ServiceRegistrar, srv ResourceStoreServer) {
|
||||
// If the following call pancis, it indicates UnimplementedResourceStoreServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&ResourceStore_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
@@ -286,24 +267,6 @@ func _ResourceStore_Delete_Handler(srv interface{}, ctx context.Context, dec fun
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _ResourceStore_Restore_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(RestoreRequest)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(ResourceStoreServer).Restore(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: ResourceStore_Restore_FullMethodName,
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(ResourceStoreServer).Restore(ctx, req.(*RestoreRequest))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
func _ResourceStore_List_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(ListRequest)
|
||||
if err := dec(in); err != nil {
|
||||
@@ -327,21 +290,11 @@ func _ResourceStore_Watch_Handler(srv interface{}, stream grpc.ServerStream) err
|
||||
if err := stream.RecvMsg(m); err != nil {
|
||||
return err
|
||||
}
|
||||
return srv.(ResourceStoreServer).Watch(m, &resourceStoreWatchServer{ServerStream: stream})
|
||||
return srv.(ResourceStoreServer).Watch(m, &grpc.GenericServerStream[WatchRequest, WatchEvent]{ServerStream: stream})
|
||||
}
|
||||
|
||||
type ResourceStore_WatchServer interface {
|
||||
Send(*WatchEvent) error
|
||||
grpc.ServerStream
|
||||
}
|
||||
|
||||
type resourceStoreWatchServer struct {
|
||||
grpc.ServerStream
|
||||
}
|
||||
|
||||
func (x *resourceStoreWatchServer) Send(m *WatchEvent) error {
|
||||
return x.ServerStream.SendMsg(m)
|
||||
}
|
||||
// This type alias is provided for backwards compatibility with existing code that references the prior non-generic stream type by name.
|
||||
type ResourceStore_WatchServer = grpc.ServerStreamingServer[WatchEvent]
|
||||
|
||||
// ResourceStore_ServiceDesc is the grpc.ServiceDesc for ResourceStore service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
@@ -366,10 +319,6 @@ var ResourceStore_ServiceDesc = grpc.ServiceDesc{
|
||||
MethodName: "Delete",
|
||||
Handler: _ResourceStore_Delete_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "Restore",
|
||||
Handler: _ResourceStore_Restore_Handler,
|
||||
},
|
||||
{
|
||||
MethodName: "List",
|
||||
Handler: _ResourceStore_List_Handler,
|
||||
@@ -396,7 +345,7 @@ type BulkStoreClient interface {
|
||||
// Write multiple resources to the same Namespace/Group/Resource
|
||||
// Events will not be sent until the stream is complete
|
||||
// Only the *create* permissions is checked
|
||||
BulkProcess(ctx context.Context, opts ...grpc.CallOption) (BulkStore_BulkProcessClient, error)
|
||||
BulkProcess(ctx context.Context, opts ...grpc.CallOption) (grpc.ClientStreamingClient[BulkRequest, BulkResponse], error)
|
||||
}
|
||||
|
||||
type bulkStoreClient struct {
|
||||
@@ -407,58 +356,40 @@ func NewBulkStoreClient(cc grpc.ClientConnInterface) BulkStoreClient {
|
||||
return &bulkStoreClient{cc}
|
||||
}
|
||||
|
||||
func (c *bulkStoreClient) BulkProcess(ctx context.Context, opts ...grpc.CallOption) (BulkStore_BulkProcessClient, error) {
|
||||
func (c *bulkStoreClient) BulkProcess(ctx context.Context, opts ...grpc.CallOption) (grpc.ClientStreamingClient[BulkRequest, BulkResponse], error) {
|
||||
cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
|
||||
stream, err := c.cc.NewStream(ctx, &BulkStore_ServiceDesc.Streams[0], BulkStore_BulkProcess_FullMethodName, cOpts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
x := &bulkStoreBulkProcessClient{ClientStream: stream}
|
||||
x := &grpc.GenericClientStream[BulkRequest, BulkResponse]{ClientStream: stream}
|
||||
return x, nil
|
||||
}
|
||||
|
||||
type BulkStore_BulkProcessClient interface {
|
||||
Send(*BulkRequest) error
|
||||
CloseAndRecv() (*BulkResponse, error)
|
||||
grpc.ClientStream
|
||||
}
|
||||
|
||||
type bulkStoreBulkProcessClient struct {
|
||||
grpc.ClientStream
|
||||
}
|
||||
|
||||
func (x *bulkStoreBulkProcessClient) Send(m *BulkRequest) error {
|
||||
return x.ClientStream.SendMsg(m)
|
||||
}
|
||||
|
||||
func (x *bulkStoreBulkProcessClient) CloseAndRecv() (*BulkResponse, error) {
|
||||
if err := x.ClientStream.CloseSend(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
m := new(BulkResponse)
|
||||
if err := x.ClientStream.RecvMsg(m); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
// This type alias is provided for backwards compatibility with existing code that references the prior non-generic stream type by name.
|
||||
type BulkStore_BulkProcessClient = grpc.ClientStreamingClient[BulkRequest, BulkResponse]
|
||||
|
||||
// BulkStoreServer is the server API for BulkStore service.
|
||||
// All implementations should embed UnimplementedBulkStoreServer
|
||||
// for forward compatibility
|
||||
// for forward compatibility.
|
||||
type BulkStoreServer interface {
|
||||
// Write multiple resources to the same Namespace/Group/Resource
|
||||
// Events will not be sent until the stream is complete
|
||||
// Only the *create* permissions is checked
|
||||
BulkProcess(BulkStore_BulkProcessServer) error
|
||||
BulkProcess(grpc.ClientStreamingServer[BulkRequest, BulkResponse]) error
|
||||
}
|
||||
|
||||
// UnimplementedBulkStoreServer should be embedded to have forward compatible implementations.
|
||||
type UnimplementedBulkStoreServer struct {
|
||||
}
|
||||
// UnimplementedBulkStoreServer should be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedBulkStoreServer struct{}
|
||||
|
||||
func (UnimplementedBulkStoreServer) BulkProcess(BulkStore_BulkProcessServer) error {
|
||||
func (UnimplementedBulkStoreServer) BulkProcess(grpc.ClientStreamingServer[BulkRequest, BulkResponse]) error {
|
||||
return status.Errorf(codes.Unimplemented, "method BulkProcess not implemented")
|
||||
}
|
||||
func (UnimplementedBulkStoreServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeBulkStoreServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to BulkStoreServer will
|
||||
@@ -468,34 +399,22 @@ type UnsafeBulkStoreServer interface {
|
||||
}
|
||||
|
||||
func RegisterBulkStoreServer(s grpc.ServiceRegistrar, srv BulkStoreServer) {
|
||||
// If the following call pancis, it indicates UnimplementedBulkStoreServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&BulkStore_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
func _BulkStore_BulkProcess_Handler(srv interface{}, stream grpc.ServerStream) error {
|
||||
return srv.(BulkStoreServer).BulkProcess(&bulkStoreBulkProcessServer{ServerStream: stream})
|
||||
return srv.(BulkStoreServer).BulkProcess(&grpc.GenericServerStream[BulkRequest, BulkResponse]{ServerStream: stream})
|
||||
}
|
||||
|
||||
type BulkStore_BulkProcessServer interface {
|
||||
SendAndClose(*BulkResponse) error
|
||||
Recv() (*BulkRequest, error)
|
||||
grpc.ServerStream
|
||||
}
|
||||
|
||||
type bulkStoreBulkProcessServer struct {
|
||||
grpc.ServerStream
|
||||
}
|
||||
|
||||
func (x *bulkStoreBulkProcessServer) SendAndClose(m *BulkResponse) error {
|
||||
return x.ServerStream.SendMsg(m)
|
||||
}
|
||||
|
||||
func (x *bulkStoreBulkProcessServer) Recv() (*BulkRequest, error) {
|
||||
m := new(BulkRequest)
|
||||
if err := x.ServerStream.RecvMsg(m); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
// This type alias is provided for backwards compatibility with existing code that references the prior non-generic stream type by name.
|
||||
type BulkStore_BulkProcessServer = grpc.ClientStreamingServer[BulkRequest, BulkResponse]
|
||||
|
||||
// BulkStore_ServiceDesc is the grpc.ServiceDesc for BulkStore service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
@@ -561,7 +480,7 @@ func (c *resourceIndexClient) GetStats(ctx context.Context, in *ResourceStatsReq
|
||||
|
||||
// ResourceIndexServer is the server API for ResourceIndex service.
|
||||
// All implementations should embed UnimplementedResourceIndexServer
|
||||
// for forward compatibility
|
||||
// for forward compatibility.
|
||||
//
|
||||
// Unlike the ResourceStore, this service can be exposed to clients directly
|
||||
// It should be implemented with efficient indexes and does not need read-after-write semantics
|
||||
@@ -571,9 +490,12 @@ type ResourceIndexServer interface {
|
||||
GetStats(context.Context, *ResourceStatsRequest) (*ResourceStatsResponse, error)
|
||||
}
|
||||
|
||||
// UnimplementedResourceIndexServer should be embedded to have forward compatible implementations.
|
||||
type UnimplementedResourceIndexServer struct {
|
||||
}
|
||||
// UnimplementedResourceIndexServer should be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedResourceIndexServer struct{}
|
||||
|
||||
func (UnimplementedResourceIndexServer) Search(context.Context, *ResourceSearchRequest) (*ResourceSearchResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method Search not implemented")
|
||||
@@ -581,6 +503,7 @@ func (UnimplementedResourceIndexServer) Search(context.Context, *ResourceSearchR
|
||||
func (UnimplementedResourceIndexServer) GetStats(context.Context, *ResourceStatsRequest) (*ResourceStatsResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method GetStats not implemented")
|
||||
}
|
||||
func (UnimplementedResourceIndexServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeResourceIndexServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to ResourceIndexServer will
|
||||
@@ -590,6 +513,13 @@ type UnsafeResourceIndexServer interface {
|
||||
}
|
||||
|
||||
func RegisterResourceIndexServer(s grpc.ServiceRegistrar, srv ResourceIndexServer) {
|
||||
// If the following call pancis, it indicates UnimplementedResourceIndexServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&ResourceIndex_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
@@ -697,7 +627,7 @@ func (c *managedObjectIndexClient) ListManagedObjects(ctx context.Context, in *L
|
||||
|
||||
// ManagedObjectIndexServer is the server API for ManagedObjectIndex service.
|
||||
// All implementations should embed UnimplementedManagedObjectIndexServer
|
||||
// for forward compatibility
|
||||
// for forward compatibility.
|
||||
//
|
||||
// Query managed objects
|
||||
// Results access control is based on access to the repository *not* the items
|
||||
@@ -708,9 +638,12 @@ type ManagedObjectIndexServer interface {
|
||||
ListManagedObjects(context.Context, *ListManagedObjectsRequest) (*ListManagedObjectsResponse, error)
|
||||
}
|
||||
|
||||
// UnimplementedManagedObjectIndexServer should be embedded to have forward compatible implementations.
|
||||
type UnimplementedManagedObjectIndexServer struct {
|
||||
}
|
||||
// UnimplementedManagedObjectIndexServer should be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedManagedObjectIndexServer struct{}
|
||||
|
||||
func (UnimplementedManagedObjectIndexServer) CountManagedObjects(context.Context, *CountManagedObjectsRequest) (*CountManagedObjectsResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method CountManagedObjects not implemented")
|
||||
@@ -718,6 +651,7 @@ func (UnimplementedManagedObjectIndexServer) CountManagedObjects(context.Context
|
||||
func (UnimplementedManagedObjectIndexServer) ListManagedObjects(context.Context, *ListManagedObjectsRequest) (*ListManagedObjectsResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method ListManagedObjects not implemented")
|
||||
}
|
||||
func (UnimplementedManagedObjectIndexServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeManagedObjectIndexServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to ManagedObjectIndexServer will
|
||||
@@ -727,6 +661,13 @@ type UnsafeManagedObjectIndexServer interface {
|
||||
}
|
||||
|
||||
func RegisterManagedObjectIndexServer(s grpc.ServiceRegistrar, srv ManagedObjectIndexServer) {
|
||||
// If the following call pancis, it indicates UnimplementedManagedObjectIndexServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&ManagedObjectIndex_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
@@ -832,7 +773,7 @@ func (c *blobStoreClient) GetBlob(ctx context.Context, in *GetBlobRequest, opts
|
||||
|
||||
// BlobStoreServer is the server API for BlobStore service.
|
||||
// All implementations should embed UnimplementedBlobStoreServer
|
||||
// for forward compatibility
|
||||
// for forward compatibility.
|
||||
type BlobStoreServer interface {
|
||||
// Upload a blob that will be saved in a resource
|
||||
PutBlob(context.Context, *PutBlobRequest) (*PutBlobResponse, error)
|
||||
@@ -841,9 +782,12 @@ type BlobStoreServer interface {
|
||||
GetBlob(context.Context, *GetBlobRequest) (*GetBlobResponse, error)
|
||||
}
|
||||
|
||||
// UnimplementedBlobStoreServer should be embedded to have forward compatible implementations.
|
||||
type UnimplementedBlobStoreServer struct {
|
||||
}
|
||||
// UnimplementedBlobStoreServer should be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedBlobStoreServer struct{}
|
||||
|
||||
func (UnimplementedBlobStoreServer) PutBlob(context.Context, *PutBlobRequest) (*PutBlobResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method PutBlob not implemented")
|
||||
@@ -851,6 +795,7 @@ func (UnimplementedBlobStoreServer) PutBlob(context.Context, *PutBlobRequest) (*
|
||||
func (UnimplementedBlobStoreServer) GetBlob(context.Context, *GetBlobRequest) (*GetBlobResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method GetBlob not implemented")
|
||||
}
|
||||
func (UnimplementedBlobStoreServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeBlobStoreServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to BlobStoreServer will
|
||||
@@ -860,6 +805,13 @@ type UnsafeBlobStoreServer interface {
|
||||
}
|
||||
|
||||
func RegisterBlobStoreServer(s grpc.ServiceRegistrar, srv BlobStoreServer) {
|
||||
// If the following call pancis, it indicates UnimplementedBlobStoreServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&BlobStore_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
@@ -954,7 +906,7 @@ func (c *diagnosticsClient) IsHealthy(ctx context.Context, in *HealthCheckReques
|
||||
|
||||
// DiagnosticsServer is the server API for Diagnostics service.
|
||||
// All implementations should embed UnimplementedDiagnosticsServer
|
||||
// for forward compatibility
|
||||
// for forward compatibility.
|
||||
//
|
||||
// Clients can use this service directly
|
||||
// NOTE: This is read only, and no read afer write guarantees
|
||||
@@ -963,13 +915,17 @@ type DiagnosticsServer interface {
|
||||
IsHealthy(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error)
|
||||
}
|
||||
|
||||
// UnimplementedDiagnosticsServer should be embedded to have forward compatible implementations.
|
||||
type UnimplementedDiagnosticsServer struct {
|
||||
}
|
||||
// UnimplementedDiagnosticsServer should be embedded to have
|
||||
// forward compatible implementations.
|
||||
//
|
||||
// NOTE: this should be embedded by value instead of pointer to avoid a nil
|
||||
// pointer dereference when methods are called.
|
||||
type UnimplementedDiagnosticsServer struct{}
|
||||
|
||||
func (UnimplementedDiagnosticsServer) IsHealthy(context.Context, *HealthCheckRequest) (*HealthCheckResponse, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method IsHealthy not implemented")
|
||||
}
|
||||
func (UnimplementedDiagnosticsServer) testEmbeddedByValue() {}
|
||||
|
||||
// UnsafeDiagnosticsServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to DiagnosticsServer will
|
||||
@@ -979,6 +935,13 @@ type UnsafeDiagnosticsServer interface {
|
||||
}
|
||||
|
||||
func RegisterDiagnosticsServer(s grpc.ServiceRegistrar, srv DiagnosticsServer) {
|
||||
// If the following call pancis, it indicates UnimplementedDiagnosticsServer was
|
||||
// embedded by pointer and is nil. This will cause panics if an
|
||||
// unimplemented method is ever invoked, so we test this at initialization
|
||||
// time to prevent it from happening at runtime later due to I/O.
|
||||
if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
|
||||
t.testEmbeddedByValue()
|
||||
}
|
||||
s.RegisterService(&Diagnostics_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
|
||||
@@ -10,14 +10,12 @@ import (
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"go.opentelemetry.io/otel/trace"
|
||||
"go.opentelemetry.io/otel/trace/noop"
|
||||
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
||||
"k8s.io/apimachinery/pkg/types"
|
||||
|
||||
claims "github.com/grafana/authlib/types"
|
||||
"github.com/grafana/grafana/pkg/apimachinery/utils"
|
||||
@@ -393,6 +391,7 @@ func (s *server) newEvent(ctx context.Context, user claims.AuthInfo, key *Resour
|
||||
if oldValue == nil {
|
||||
event.Type = WatchEvent_ADDED
|
||||
} else {
|
||||
event.Type = WatchEvent_MODIFIED
|
||||
check.Verb = utils.VerbUpdate
|
||||
|
||||
temp := &unstructured.Unstructured{}
|
||||
@@ -404,13 +403,6 @@ func (s *server) newEvent(ctx context.Context, user claims.AuthInfo, key *Resour
|
||||
if err != nil {
|
||||
return nil, AsErrorResult(err)
|
||||
}
|
||||
|
||||
// restores will restore with a different k8s uid
|
||||
if event.ObjectOld.GetUID() != obj.GetUID() {
|
||||
event.Type = WatchEvent_ADDED
|
||||
} else {
|
||||
event.Type = WatchEvent_MODIFIED
|
||||
}
|
||||
}
|
||||
|
||||
if key.Namespace != obj.GetNamespace() {
|
||||
@@ -792,126 +784,6 @@ func (s *server) List(ctx context.Context, req *ListRequest) (*ListResponse, err
|
||||
return rsp, err
|
||||
}
|
||||
|
||||
func (s *server) Restore(ctx context.Context, req *RestoreRequest) (*RestoreResponse, error) {
|
||||
ctx, span := s.tracer.Start(ctx, "storage_server.List")
|
||||
defer span.End()
|
||||
|
||||
// check that the user has access
|
||||
user, ok := claims.AuthInfoFrom(ctx)
|
||||
if !ok || user == nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Message: "no user found in context",
|
||||
Code: http.StatusUnauthorized,
|
||||
}}, nil
|
||||
}
|
||||
|
||||
if err := s.Init(ctx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
checker, err := s.access.Compile(ctx, user, claims.ListRequest{
|
||||
Group: req.Key.Group,
|
||||
Resource: req.Key.Resource,
|
||||
Namespace: req.Key.Namespace,
|
||||
Verb: utils.VerbGet,
|
||||
})
|
||||
if err != nil {
|
||||
return &RestoreResponse{Error: AsErrorResult(err)}, nil
|
||||
}
|
||||
if checker == nil {
|
||||
return &RestoreResponse{Error: &ErrorResult{
|
||||
Code: http.StatusForbidden,
|
||||
}}, nil
|
||||
}
|
||||
|
||||
// get the asked for resource version to restore
|
||||
readRsp, err := s.Read(ctx, &ReadRequest{
|
||||
Key: req.Key,
|
||||
ResourceVersion: req.ResourceVersion,
|
||||
IncludeDeleted: true,
|
||||
})
|
||||
if err != nil || readRsp == nil || readRsp.Error != nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusNotFound,
|
||||
Message: fmt.Sprintf("could not find old resource: %s", readRsp.Error.Message),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// generate a new k8s UID when restoring. The name will remain the same
|
||||
// (for dashboards, this will be the dashboard uid), but since controllers
|
||||
// will see this as a create event, we do not want the same k8s UID, or
|
||||
// there may be unintended behavior
|
||||
newUid := types.UID(uuid.NewString())
|
||||
tmp := &unstructured.Unstructured{}
|
||||
err = tmp.UnmarshalJSON(readRsp.Value)
|
||||
if err != nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusNotFound,
|
||||
Message: fmt.Sprintf("could not unmarhsal: %s", err.Error()),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
obj, err := utils.MetaAccessor(tmp)
|
||||
if err != nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusNotFound,
|
||||
Message: fmt.Sprintf("could not get object: %s", err.Error()),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
obj.SetUID(newUid)
|
||||
|
||||
rtObj, ok := obj.GetRuntimeObject()
|
||||
if !ok {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusNotFound,
|
||||
Message: "could not get runtime object",
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
newObj, err := json.Marshal(rtObj)
|
||||
if err != nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusNotFound,
|
||||
Message: fmt.Sprintf("could not marshal object: %s", err.Error()),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// finally, send to the backend to create & update the history of the restored object
|
||||
event, errRes := s.newEvent(ctx, user, req.Key, newObj, readRsp.Value)
|
||||
if errRes != nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusInternalServerError,
|
||||
Message: fmt.Sprintf("could not create restore resource event: %s", errRes.Message),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
rv, err := s.backend.WriteEvent(ctx, *event)
|
||||
if err != nil {
|
||||
return &RestoreResponse{
|
||||
Error: &ErrorResult{
|
||||
Code: http.StatusInternalServerError,
|
||||
Message: fmt.Sprintf("could not restore resource: %s", err.Error()),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
return &RestoreResponse{
|
||||
Error: nil,
|
||||
ResourceVersion: rv,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *server) initWatcher() error {
|
||||
var err error
|
||||
s.broadcaster, err = NewBroadcaster(s.ctx, func(out chan<- *WrittenEvent) error {
|
||||
|
||||
@@ -233,79 +233,4 @@ func TestSimpleServer(t *testing.T) {
|
||||
ResourceVersion: created.ResourceVersion})
|
||||
require.ErrorIs(t, err, ErrOptimisticLockingFailed)
|
||||
})
|
||||
|
||||
t.Run("playlist restore", func(t *testing.T) {
|
||||
uid := "zzz"
|
||||
raw := []byte(`{
|
||||
"apiVersion": "playlist.grafana.app/v0alpha1",
|
||||
"kind": "Playlist",
|
||||
"metadata": {
|
||||
"name": "fdgsv37qslr0ga",
|
||||
"namespace": "default",
|
||||
"uid": "` + uid + `",
|
||||
"annotations": {
|
||||
"grafana.app/repoName": "elsewhere",
|
||||
"grafana.app/repoPath": "path/to/item",
|
||||
"grafana.app/repoTimestamp": "2024-02-02T00:00:00Z"
|
||||
}
|
||||
},
|
||||
"spec": {
|
||||
"title": "hello",
|
||||
"interval": "5m",
|
||||
"items": [
|
||||
{
|
||||
"type": "dashboard_by_uid",
|
||||
"value": "vmie2cmWz"
|
||||
}
|
||||
]
|
||||
}
|
||||
}`)
|
||||
|
||||
key := &ResourceKey{
|
||||
Group: "playlist.grafana.app",
|
||||
Resource: "rrrr",
|
||||
Namespace: "default",
|
||||
Name: "fdgsv37qslr0ga",
|
||||
}
|
||||
|
||||
// create
|
||||
created, err := server.Create(ctx, &CreateRequest{
|
||||
Value: raw,
|
||||
Key: key,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
// make sure it exists
|
||||
found, err := server.Read(ctx, &ReadRequest{Key: key})
|
||||
require.NoError(t, err)
|
||||
require.Nil(t, found.Error)
|
||||
fmt.Println(found.ResourceVersion)
|
||||
|
||||
// delete it
|
||||
deleted, err := server.Delete(ctx, &DeleteRequest{Key: key, ResourceVersion: created.ResourceVersion})
|
||||
require.NoError(t, err)
|
||||
require.True(t, deleted.ResourceVersion > created.ResourceVersion)
|
||||
|
||||
// restore it
|
||||
restored, err := server.Restore(ctx, &RestoreRequest{
|
||||
Key: key,
|
||||
ResourceVersion: found.ResourceVersion,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
require.Nil(t, restored.Error)
|
||||
require.True(t, restored.ResourceVersion > deleted.ResourceVersion)
|
||||
|
||||
// ensure it exists now
|
||||
found, err = server.Read(ctx, &ReadRequest{Key: key})
|
||||
require.NoError(t, err)
|
||||
require.Nil(t, found.Error)
|
||||
require.Equal(t, restored.ResourceVersion, found.ResourceVersion)
|
||||
foundUnstructured := &unstructured.Unstructured{}
|
||||
err = foundUnstructured.UnmarshalJSON(found.Value)
|
||||
require.NoError(t, err)
|
||||
foundObj, err := utils.MetaAccessor(foundUnstructured)
|
||||
require.NoError(t, err)
|
||||
// the UID should be different now
|
||||
require.NotEqual(t, uid, string(foundObj.GetUID()))
|
||||
})
|
||||
}
|
||||
|
||||
@@ -308,9 +308,6 @@ func (b *backend) WriteEvent(ctx context.Context, event resource.WriteEvent) (in
|
||||
// TODO: validate key ?
|
||||
switch event.Type {
|
||||
case resource.WatchEvent_ADDED:
|
||||
if event.ObjectOld != nil {
|
||||
return b.restore(ctx, event)
|
||||
}
|
||||
return b.create(ctx, event)
|
||||
case resource.WatchEvent_MODIFIED:
|
||||
return b.update(ctx, event)
|
||||
@@ -488,73 +485,6 @@ func (b *backend) delete(ctx context.Context, event resource.WriteEvent) (int64,
|
||||
return rv, nil
|
||||
}
|
||||
|
||||
func (b *backend) restore(ctx context.Context, event resource.WriteEvent) (int64, error) {
|
||||
ctx, span := b.tracer.Start(ctx, tracePrefix+"Restore")
|
||||
defer span.End()
|
||||
guid := uuid.New().String()
|
||||
folder := ""
|
||||
if event.Object != nil {
|
||||
folder = event.Object.GetFolder()
|
||||
}
|
||||
rv, err := b.rvManager.ExecWithRV(ctx, event.Key, func(tx db.Tx) (string, error) {
|
||||
// 1. Re-create resource
|
||||
// Note: we may want to replace the write event with a create event, tbd.
|
||||
if _, err := dbutil.Exec(ctx, tx, sqlResourceInsert, sqlResourceRequest{
|
||||
SQLTemplate: sqltemplate.New(b.dialect),
|
||||
WriteEvent: event,
|
||||
Folder: folder,
|
||||
GUID: guid,
|
||||
}); err != nil {
|
||||
return guid, fmt.Errorf("insert into resource: %w", err)
|
||||
}
|
||||
|
||||
// 2. Insert into resource history
|
||||
if _, err := dbutil.Exec(ctx, tx, sqlResourceHistoryInsert, sqlResourceRequest{
|
||||
SQLTemplate: sqltemplate.New(b.dialect),
|
||||
WriteEvent: event,
|
||||
Folder: folder,
|
||||
GUID: guid,
|
||||
}); err != nil {
|
||||
return guid, fmt.Errorf("insert into resource history: %w", err)
|
||||
}
|
||||
_ = b.historyPruner.Add(pruningKey{
|
||||
namespace: event.Key.Namespace,
|
||||
group: event.Key.Group,
|
||||
resource: event.Key.Resource,
|
||||
name: event.Key.Name,
|
||||
})
|
||||
|
||||
// 3. Update all resource history entries with the new UID
|
||||
// Note: we do not update any history entries that have a deletion timestamp included. This will become
|
||||
// important once we start using finalizers, as the initial delete will show up as an update with a deletion timestamp included.
|
||||
if _, err := dbutil.Exec(ctx, tx, sqlResoureceHistoryUpdateUid, sqlResourceHistoryUpdateRequest{
|
||||
SQLTemplate: sqltemplate.New(b.dialect),
|
||||
WriteEvent: event,
|
||||
OldUID: string(event.ObjectOld.GetUID()),
|
||||
NewUID: string(event.Object.GetUID()),
|
||||
}); err != nil {
|
||||
return guid, fmt.Errorf("update history uid: %w", err)
|
||||
}
|
||||
|
||||
return guid, nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
b.notifier.send(ctx, &resource.WrittenEvent{
|
||||
Type: event.Type,
|
||||
Key: event.Key,
|
||||
PreviousRV: event.PreviousRV,
|
||||
Value: event.Value,
|
||||
ResourceVersion: rv,
|
||||
Folder: folder,
|
||||
})
|
||||
|
||||
return rv, nil
|
||||
}
|
||||
|
||||
func (b *backend) ReadResource(ctx context.Context, req *resource.ReadRequest) *resource.BackendReadResponse {
|
||||
_, span := b.tracer.Start(ctx, tracePrefix+".Read")
|
||||
defer span.End()
|
||||
@@ -577,17 +507,6 @@ func (b *backend) ReadResource(ctx context.Context, req *resource.ReadRequest) *
|
||||
err := b.db.WithTx(ctx, ReadCommittedRO, func(ctx context.Context, tx db.Tx) error {
|
||||
var err error
|
||||
res, err = dbutil.QueryRow(ctx, tx, sr, readReq)
|
||||
// if not found, look for latest deleted version (if requested)
|
||||
if errors.Is(err, sql.ErrNoRows) && req.IncludeDeleted {
|
||||
sr = sqlResourceHistoryRead
|
||||
readReq2 := &sqlResourceReadRequest{
|
||||
SQLTemplate: sqltemplate.New(b.dialect),
|
||||
Request: req,
|
||||
Response: NewReadResponse(),
|
||||
}
|
||||
res, err = dbutil.QueryRow(ctx, tx, sr, readReq2)
|
||||
return err
|
||||
}
|
||||
return err
|
||||
})
|
||||
|
||||
|
||||
@@ -376,88 +376,6 @@ func TestBackend_delete(t *testing.T) {
|
||||
})
|
||||
}
|
||||
|
||||
func TestBackend_restore(t *testing.T) {
|
||||
t.Parallel()
|
||||
meta, err := utils.MetaAccessor(&unstructured.Unstructured{
|
||||
Object: map[string]any{},
|
||||
})
|
||||
require.NoError(t, err)
|
||||
meta.SetUID("new-uid")
|
||||
oldMeta, err := utils.MetaAccessor(&unstructured.Unstructured{
|
||||
Object: map[string]any{},
|
||||
})
|
||||
require.NoError(t, err)
|
||||
oldMeta.SetUID("old-uid")
|
||||
event := resource.WriteEvent{
|
||||
Type: resource.WatchEvent_ADDED,
|
||||
Key: resKey,
|
||||
Object: meta,
|
||||
ObjectOld: oldMeta,
|
||||
}
|
||||
|
||||
t.Run("happy path", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
b, ctx := setupBackendTest(t)
|
||||
|
||||
b.SQLMock.ExpectBegin()
|
||||
expectSuccessfulResourceVersionExec(t, b.TestDBProvider,
|
||||
func() { b.ExecWithResult("insert resource", 0, 1) },
|
||||
func() { b.ExecWithResult("insert resource_history", 0, 1) },
|
||||
func() { b.ExecWithResult("update resource_history", 0, 1) },
|
||||
)
|
||||
b.SQLMock.ExpectCommit()
|
||||
|
||||
v, err := b.restore(ctx, event)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, int64(200), v)
|
||||
})
|
||||
|
||||
t.Run("error restoring resource", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
b, ctx := setupBackendTest(t)
|
||||
|
||||
b.SQLMock.ExpectBegin()
|
||||
b.ExecWithErr("insert resource", errTest)
|
||||
b.SQLMock.ExpectRollback()
|
||||
|
||||
v, err := b.restore(ctx, event)
|
||||
require.Zero(t, v)
|
||||
require.Error(t, err)
|
||||
require.ErrorContains(t, err, "insert into resource")
|
||||
})
|
||||
|
||||
t.Run("error inserting into resource history", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
b, ctx := setupBackendTest(t)
|
||||
|
||||
b.SQLMock.ExpectBegin()
|
||||
b.ExecWithResult("insert resource", 0, 1)
|
||||
b.ExecWithErr("insert resource_history", errTest)
|
||||
b.SQLMock.ExpectRollback()
|
||||
|
||||
v, err := b.restore(ctx, event)
|
||||
require.Zero(t, v)
|
||||
require.Error(t, err)
|
||||
require.ErrorContains(t, err, "insert into resource history")
|
||||
})
|
||||
|
||||
t.Run("error updating resource history uid", func(t *testing.T) {
|
||||
t.Parallel()
|
||||
b, ctx := setupBackendTest(t)
|
||||
|
||||
b.SQLMock.ExpectBegin()
|
||||
b.ExecWithResult("insert resource", 0, 1)
|
||||
b.ExecWithResult("insert resource_history", 0, 1)
|
||||
b.ExecWithErr("update resource_history", errTest)
|
||||
b.SQLMock.ExpectRollback()
|
||||
|
||||
v, err := b.restore(ctx, event)
|
||||
require.Zero(t, v)
|
||||
require.Error(t, err)
|
||||
require.ErrorContains(t, err, "update history uid")
|
||||
})
|
||||
}
|
||||
|
||||
func TestBackend_getHistory(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
|
||||
@@ -14,12 +14,8 @@ SELECT
|
||||
AND {{ .Ident "group" }} = {{ .Arg .Request.Key.Group }}
|
||||
AND {{ .Ident "resource" }} = {{ .Arg .Request.Key.Resource }}
|
||||
AND {{ .Ident "name" }} = {{ .Arg .Request.Key.Name }}
|
||||
{{ if .Request.IncludeDeleted }}
|
||||
AND {{ .Ident "action" }} != 3
|
||||
AND {{ .Ident "value" }} NOT LIKE '%deletionTimestamp%'
|
||||
{{ end }}
|
||||
{{ if gt .Request.ResourceVersion 0 }}
|
||||
AND {{ .Ident "resource_version" }} {{ if .Request.IncludeDeleted }}={{ else }}<={{ end }} {{ .Arg .Request.ResourceVersion }}
|
||||
AND {{ .Ident "resource_version" }} <= {{ .Arg .Request.ResourceVersion }}
|
||||
{{ end }}
|
||||
ORDER BY {{ .Ident "resource_version" }} DESC
|
||||
LIMIT 1
|
||||
|
||||
@@ -1,8 +0,0 @@
|
||||
UPDATE {{ .Ident "resource_history" }}
|
||||
SET {{ .Ident "value" }} = REPLACE({{ .Ident "value" }}, CONCAT('"uid":"', {{ .Arg .OldUID }}, '"'), CONCAT('"uid":"', {{ .Arg .NewUID }}, '"'))
|
||||
WHERE {{ .Ident "name" }} = {{ .Arg .WriteEvent.Key.Name }}
|
||||
AND {{ .Ident "namespace" }} = {{ .Arg .WriteEvent.Key.Namespace }}
|
||||
AND {{ .Ident "group" }} = {{ .Arg .WriteEvent.Key.Group }}
|
||||
AND {{ .Ident "resource" }} = {{ .Arg .WriteEvent.Key.Resource }}
|
||||
AND {{ .Ident "action" }} != 3
|
||||
AND {{ .Ident "value" }} NOT LIKE '%deletionTimestamp%';
|
||||
@@ -39,7 +39,6 @@ var (
|
||||
sqlResourceUpdateRV = mustTemplate("resource_update_rv.sql")
|
||||
sqlResourceHistoryRead = mustTemplate("resource_history_read.sql")
|
||||
sqlResourceHistoryUpdateRV = mustTemplate("resource_history_update_rv.sql")
|
||||
sqlResoureceHistoryUpdateUid = mustTemplate("resource_history_update_uid.sql")
|
||||
sqlResourceHistoryInsert = mustTemplate("resource_history_insert.sql")
|
||||
sqlResourceHistoryPoll = mustTemplate("resource_history_poll.sql")
|
||||
sqlResourceHistoryGet = mustTemplate("resource_history_get.sql")
|
||||
@@ -281,19 +280,6 @@ func (r *sqlPruneHistoryRequest) Validate() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// update resource history
|
||||
|
||||
type sqlResourceHistoryUpdateRequest struct {
|
||||
sqltemplate.SQLTemplate
|
||||
WriteEvent resource.WriteEvent
|
||||
OldUID string
|
||||
NewUID string
|
||||
}
|
||||
|
||||
func (r sqlResourceHistoryUpdateRequest) Validate() error {
|
||||
return nil // TODO
|
||||
}
|
||||
|
||||
type sqlResourceBlobInsertRequest struct {
|
||||
sqltemplate.SQLTemplate
|
||||
Now time.Time
|
||||
|
||||
@@ -176,26 +176,6 @@ func TestUnifiedStorageQueries(t *testing.T) {
|
||||
},
|
||||
},
|
||||
|
||||
sqlResoureceHistoryUpdateUid: {
|
||||
{
|
||||
Name: "modify uids in history",
|
||||
Data: &sqlResourceHistoryUpdateRequest{
|
||||
SQLTemplate: mocks.NewTestingSQLTemplate(),
|
||||
WriteEvent: resource.WriteEvent{
|
||||
Key: &resource.ResourceKey{
|
||||
Namespace: "nn",
|
||||
Group: "gg",
|
||||
Resource: "rr",
|
||||
Name: "name",
|
||||
},
|
||||
PreviousRV: 1234,
|
||||
},
|
||||
OldUID: "old-uid",
|
||||
NewUID: "new-uid",
|
||||
},
|
||||
},
|
||||
},
|
||||
|
||||
sqlResourceHistoryInsert: {
|
||||
{
|
||||
Name: "insert into resource_history",
|
||||
|
||||
Vendored
-8
@@ -1,8 +0,0 @@
|
||||
UPDATE `resource_history`
|
||||
SET `value` = REPLACE(`value`, CONCAT('"uid":"', 'old-uid', '"'), CONCAT('"uid":"', 'new-uid', '"'))
|
||||
WHERE `name` = 'name'
|
||||
AND `namespace` = 'nn'
|
||||
AND `group` = 'gg'
|
||||
AND `resource` = 'rr'
|
||||
AND `action` != 3
|
||||
AND `value` NOT LIKE '%deletionTimestamp%';
|
||||
Vendored
-8
@@ -1,8 +0,0 @@
|
||||
UPDATE "resource_history"
|
||||
SET "value" = REPLACE("value", CONCAT('"uid":"', 'old-uid', '"'), CONCAT('"uid":"', 'new-uid', '"'))
|
||||
WHERE "name" = 'name'
|
||||
AND "namespace" = 'nn'
|
||||
AND "group" = 'gg'
|
||||
AND "resource" = 'rr'
|
||||
AND "action" != 3
|
||||
AND "value" NOT LIKE '%deletionTimestamp%';
|
||||
Vendored
-8
@@ -1,8 +0,0 @@
|
||||
UPDATE "resource_history"
|
||||
SET "value" = REPLACE("value", CONCAT('"uid":"', 'old-uid', '"'), CONCAT('"uid":"', 'new-uid', '"'))
|
||||
WHERE "name" = 'name'
|
||||
AND "namespace" = 'nn'
|
||||
AND "group" = 'gg'
|
||||
AND "resource" = 'rr'
|
||||
AND "action" != 3
|
||||
AND "value" NOT LIKE '%deletionTimestamp%';
|
||||
Reference in New Issue
Block a user