Add IsProductionSafe to Blockstore metadata

cmd/lakefs/cmd/run.go

@@ -139,12 +139,6 @@ var runCmd = &cobra.Command{
 			logger.WithError(err).Fatal("failed to create login token provider")
 		}
 
-		blockstoreType := baseCfg.Blockstore.Type
-		if blockstoreType == "mem" {
-			printLocalWarning(os.Stderr, fmt.Sprintf("blockstore type %s", blockstoreType))
-			logger.WithField("adapter_type", blockstoreType).Warn("Block adapter NOT SUPPORTED for production use")
-		}
-
 		metadata := initStatsMetadata(ctx, logger, authMetadataManager, cfg)
 		bufferedCollector := stats.NewBufferedCollector(metadata.InstallationID, stats.Config(baseCfg.Stats),
 			stats.WithLogger(logger.WithField("service", "stats_collector")))
@@ -155,6 +149,16 @@ var runCmd = &cobra.Command{
 			logger.WithError(err).Fatal("Failed to create block adapter")
 		}
 
+		blockstoreMetadata, err := blockStore.BlockstoreMetadata(ctx)
+		if err != nil {
+			logger.WithError(err).Fatal("Failed to get block adapter metadata")
+		}
+		if !blockstoreMetadata.IsProductionSafe {
+			blockstoreType := blockStore.BlockstoreType()
+			printLocalWarning(os.Stderr, fmt.Sprintf("blockstore type %s", blockstoreType))
+			logger.WithField("adapter_type", blockstoreType).Warn("Block adapter NOT SUPPORTED for production use")
+		}
+
 		bufferedCollector.SetRuntimeCollector(blockStore.RuntimeStats)
 		// send metadata
 		bufferedCollector.CollectMetadata(metadata)

pkg/authentication/service.go

@@ -166,12 +166,9 @@ func (s *APIService) OauthCallback(w http.ResponseWriter, r *http.Request, _ ses
 // context on every client request.
 func addRequestID(headerName string) apiclient.RequestEditorFn {
 	return func(ctx context.Context, req *http.Request) error {
-		reqIDField := ctx.Value(httputil.RequestIDContextKey)
-		if reqIDField == nil {
-			return nil
+		if reqID, ok := ctx.Value(httputil.RequestIDContextKey).(string); ok {
+			req.Header.Add(headerName, reqID)
 		}
-		reqID := reqIDField.(string)
-		req.Header.Add(headerName, reqID)
 		return nil
 	}
 }

pkg/block/adapter.go

@@ -166,7 +166,8 @@ type Properties struct {
 }
 
 type BlockstoreMetadata struct {
-	Region *string
+	IsProductionSafe bool
+	Region           *string
 }
 
 type PutResponse struct {

pkg/block/azure/adapter.go

@@ -616,7 +616,9 @@ func (a *Adapter) BlockstoreType() string {
 }
 
 func (a *Adapter) BlockstoreMetadata(_ context.Context) (*block.BlockstoreMetadata, error) {
-	return nil, block.ErrOperationNotSupported
+	return &block.BlockstoreMetadata{
+		IsProductionSafe: true,
+	}, nil
 }
 
 func (a *Adapter) CompleteMultiPartUpload(ctx context.Context, obj block.ObjectPointer, _ string, multipartList *block.MultipartUploadCompletion) (*block.CompleteMultiPartUploadResponse, error) {

pkg/block/gs/adapter.go

@@ -681,7 +681,9 @@ func (a *Adapter) BlockstoreType() string {
 }
 
 func (a *Adapter) BlockstoreMetadata(_ context.Context) (*block.BlockstoreMetadata, error) {
-	return nil, block.ErrOperationNotSupported
+	return &block.BlockstoreMetadata{
+		IsProductionSafe: true,
+	}, nil
 }
 
 func (a *Adapter) GetStorageNamespaceInfo(string) *block.StorageNamespaceInfo {

pkg/block/local/adapter.go

@@ -533,7 +533,9 @@ func (l *Adapter) BlockstoreType() string {
 }
 
 func (l *Adapter) BlockstoreMetadata(_ context.Context) (*block.BlockstoreMetadata, error) {
-	return nil, block.ErrOperationNotSupported
+	return &block.BlockstoreMetadata{
+		IsProductionSafe: true,
+	}, nil
 }
 
 func (l *Adapter) GetStorageNamespaceInfo(string) *block.StorageNamespaceInfo {

pkg/block/mem/adapter.go

@@ -434,7 +434,9 @@ func (a *Adapter) BlockstoreType() string {
 }
 
 func (a *Adapter) BlockstoreMetadata(_ context.Context) (*block.BlockstoreMetadata, error) {
-	return nil, fmt.Errorf("blockstore metadata: %w", block.ErrOperationNotSupported)
+	return &block.BlockstoreMetadata{
+		IsProductionSafe: false,
+	}, nil
 }
 
 func (a *Adapter) GetStorageNamespaceInfo(string) *block.StorageNamespaceInfo {

pkg/block/s3/adapter.go

@@ -895,7 +895,10 @@ func (a *Adapter) BlockstoreMetadata(ctx context.Context) (*block.BlockstoreMeta
 	if err != nil {
 		return nil, err
 	}
-	return &block.BlockstoreMetadata{Region: &region}, nil
+	return &block.BlockstoreMetadata{
+		IsProductionSafe: true,
+		Region:           &region,
+	}, nil
 }
 
 func (a *Adapter) GetStorageNamespaceInfo(string) *block.StorageNamespaceInfo {

pkg/block/transient/adapter.go

@@ -143,7 +143,9 @@ func (a *Adapter) BlockstoreType() string {
 }
 
 func (a *Adapter) BlockstoreMetadata(_ context.Context) (*block.BlockstoreMetadata, error) {
-	return nil, block.ErrOperationNotSupported
+	return &block.BlockstoreMetadata{
+		IsProductionSafe: false,
+	}, nil
 }
 
 func (a *Adapter) GetStorageNamespaceInfo(string) *block.StorageNamespaceInfo {

pkg/block/validations.go

@@ -2,17 +2,17 @@ package block
 
 import (
 	"context"
-	"errors"
 	"fmt"
 )
 
 func ValidateInterRegionStorage(ctx context.Context, adapter Adapter, storageID, storageNamespace string) error {
 	blockstoreMetadata, err := adapter.BlockstoreMetadata(ctx)
-	if errors.Is(err, ErrOperationNotSupported) {
+	if err != nil {
+		return err
+	}
+	if blockstoreMetadata.Region == nil {
 		// region detection not supported for the server's blockstore, skip validation
 		return nil
-	} else if err != nil {
-		return fmt.Errorf("failed to get blockstore region: %w", err)
 	}
 
 	bucketRegion, err := adapter.GetRegion(ctx, storageID, storageNamespace)

pkg/block/validations_test.go

@@ -15,7 +15,7 @@ func TestController_ValidateInterRegionStorage(t *testing.T) {
 
 	t.Run("namespace with the same region as the storage", func(t *testing.T) {
 		opts := []testutil.MockAdapterOption{
-			testutil.WithBlockstoreMetadata(&block.BlockstoreMetadata{Region: swag.String("us-west-2")}),
+			testutil.WithBlockstoreMetadata(block.BlockstoreMetadata{Region: swag.String("us-west-2")}),
 			testutil.WithNamespaceRegion("us-west-2"),
 		}
 		adapter := testutil.NewMockAdapter(opts...)
@@ -26,7 +26,7 @@ func TestController_ValidateInterRegionStorage(t *testing.T) {
 
 	t.Run("namespace region different from storage region", func(t *testing.T) {
 		opts := []testutil.MockAdapterOption{
-			testutil.WithBlockstoreMetadata(&block.BlockstoreMetadata{Region: swag.String("us-east-1")}),
+			testutil.WithBlockstoreMetadata(block.BlockstoreMetadata{Region: swag.String("us-east-1")}),
 			testutil.WithNamespaceRegion("us-west-2"),
 		}
 		adapter := testutil.NewMockAdapter(opts...)

pkg/testutil/adapter.go

@@ -16,7 +16,7 @@ type MockAdapter struct {
 	LastStorageNamespace string
 	LastStorageClass     *string
 
-	blockstoreMetadata *block.BlockstoreMetadata
+	blockstoreMetadata block.BlockstoreMetadata
 	namespaceRegion    *string
 }
 
@@ -39,7 +39,7 @@ func NewMockAdapter(opts ...MockAdapterOption) *MockAdapter {
 	return adapter
 }
 
-func WithBlockstoreMetadata(bm *block.BlockstoreMetadata) func(a *MockAdapter) {
+func WithBlockstoreMetadata(bm block.BlockstoreMetadata) func(a *MockAdapter) {
 	return func(a *MockAdapter) {
 		a.blockstoreMetadata = bm
 	}
@@ -132,11 +132,7 @@ func (a *MockAdapter) BlockstoreType() string {
 }
 
 func (a *MockAdapter) BlockstoreMetadata(_ context.Context) (*block.BlockstoreMetadata, error) {
-	if a.blockstoreMetadata != nil {
-		return a.blockstoreMetadata, nil
-	} else {
-		return nil, block.ErrOperationNotSupported
-	}
+	return &a.blockstoreMetadata, nil
 }
 
 func (a *MockAdapter) GetStorageNamespaceInfo(string) *block.StorageNamespaceInfo {

webui/test/e2e/poms/repositoryPage.ts

@@ -57,7 +57,7 @@
     // Wait a bit for the button to be stable after hover
     await this.page.waitForTimeout(500);
 
     await firstRow.locator('button').last().click();
     await this.page.getByRole('button', { name: 'Delete' }).click();
     await this.page.getByRole("button", { name: "Yes" }).click();
   }
@@ -134,7 +134,7 @@
   }
 
   async uploadObject(filePath: string): Promise<void> {
-    await this.page.getByRole("button", { name: "Upload" }).click();
+    await this.page.getByRole("button", { name: "Upload", exact: true }).click();
     await this.page.getByText("Drag & drop files or folders here").click();
     const fileInput = await this.page.locator('input[type="file"]');
     await fileInput.setInputFiles(filePath);