Merge pull request #3622 from ddelange/patch-1

Support all S3 instant retrieval storage classes

registry/storage/driver/s3-aws/s3.go

@@ -75,6 +75,18 @@ const listMax = 1000
 // noStorageClass defines the value to be used if storage class is not supported by the S3 endpoint
 const noStorageClass = "NONE"
 
+// s3StorageClasses lists all compatible (instant retrieval) S3 storage classes
+var s3StorageClasses = []string{
+	noStorageClass,
+	s3.StorageClassStandard,
+	s3.StorageClassReducedRedundancy,
+	s3.StorageClassStandardIa,
+	s3.StorageClassOnezoneIa,
+	s3.StorageClassIntelligentTiering,
+	s3.StorageClassOutposts,
+	s3.StorageClassGlacierIr,
+}
+
 // validRegions maps known s3 region identifiers to region descriptors
 var validRegions = map[string]struct{}{}
 
@@ -327,16 +339,27 @@ func FromParameters(parameters map[string]interface{}) (*Driver, error) {
 	if storageClassParam != nil {
 		storageClassString, ok := storageClassParam.(string)
 		if !ok {
-			return nil, fmt.Errorf("the storageclass parameter must be one of %v, %v invalid",
+			return nil, fmt.Errorf(
-				[]string{s3.StorageClassStandard, s3.StorageClassReducedRedundancy}, storageClassParam)
+				"the storageclass parameter must be one of %v, %v invalid",
+				s3StorageClasses,
+				storageClassParam,
+			)
 		}
 		// All valid storage class parameters are UPPERCASE, so be a bit more flexible here
 		storageClassString = strings.ToUpper(storageClassString)
 		if storageClassString != noStorageClass &&
 			storageClassString != s3.StorageClassStandard &&
-			storageClassString != s3.StorageClassReducedRedundancy {
+			storageClassString != s3.StorageClassReducedRedundancy &&
-			return nil, fmt.Errorf("the storageclass parameter must be one of %v, %v invalid",
+			storageClassString != s3.StorageClassStandardIa &&
-				[]string{noStorageClass, s3.StorageClassStandard, s3.StorageClassReducedRedundancy}, storageClassParam)
+			storageClassString != s3.StorageClassOnezoneIa &&
+			storageClassString != s3.StorageClassIntelligentTiering &&
+			storageClassString != s3.StorageClassOutposts &&
+			storageClassString != s3.StorageClassGlacierIr {
+			return nil, fmt.Errorf(
+				"the storageclass parameter must be one of %v, %v invalid",
+				s3StorageClasses,
+				storageClassParam,
+			)
 		}
 		storageClass = storageClassString
 	}

registry/storage/driver/s3-aws/s3_test.go

@@ -250,66 +250,51 @@ func TestStorageClass(t *testing.T) {
 	}
 	defer os.Remove(rootDir)
 
-	standardDriver, err := s3DriverConstructor(rootDir, s3.StorageClassStandard)
-	if err != nil {
-		t.Fatalf("unexpected error creating driver with standard storage: %v", err)
-	}
-
-	rrDriver, err := s3DriverConstructor(rootDir, s3.StorageClassReducedRedundancy)
-	if err != nil {
-		t.Fatalf("unexpected error creating driver with reduced redundancy storage: %v", err)
-	}
-
-	if _, err = s3DriverConstructor(rootDir, noStorageClass); err != nil {
-		t.Fatalf("unexpected error creating driver without storage class: %v", err)
-	}
-
-	standardFilename := "/test-standard"
-	rrFilename := "/test-rr"
 	contents := []byte("contents")
 	ctx := context.Background()
+	for _, storageClass := range s3StorageClasses {
+		filename := "/test-" + storageClass
+		s3Driver, err := s3DriverConstructor(rootDir, storageClass)
+		if err != nil {
+			t.Fatalf("unexpected error creating driver with storage class %v: %v", storageClass, err)
+		}
 
-	err = standardDriver.PutContent(ctx, standardFilename, contents)
+		err = s3Driver.PutContent(ctx, filename, contents)
-	if err != nil {
+		if err != nil {
-		t.Fatalf("unexpected error creating content: %v", err)
+			t.Fatalf("unexpected error creating content with storage class %v: %v", storageClass, err)
-	}
+		}
-	defer standardDriver.Delete(ctx, standardFilename)
+		defer s3Driver.Delete(ctx, filename)
 
-	err = rrDriver.PutContent(ctx, rrFilename, contents)
+		driverUnwrapped := s3Driver.Base.StorageDriver.(*driver)
-	if err != nil {
+		resp, err := driverUnwrapped.S3.GetObject(&s3.GetObjectInput{
-		t.Fatalf("unexpected error creating content: %v", err)
+			Bucket: aws.String(driverUnwrapped.Bucket),
+			Key:    aws.String(driverUnwrapped.s3Path(filename)),
+		})
+		if err != nil {
+			t.Fatalf("unexpected error retrieving file with storage class %v: %v", storageClass, err)
+		}
+		defer resp.Body.Close()
+		// Amazon only populates this header value for non-standard storage classes
+		if storageClass == s3.StorageClassStandard && resp.StorageClass != nil {
+			t.Fatalf(
+				"unexpected response storage class for file with storage class %v: %v",
+				storageClass,
+				*resp.StorageClass,
+			)
+		} else if storageClass != s3.StorageClassStandard && resp.StorageClass == nil {
+			t.Fatalf(
+				"unexpected response storage class for file with storage class %v: %v",
+				storageClass,
+				s3.StorageClassStandard,
+			)
+		} else if storageClass != s3.StorageClassStandard && storageClass != *resp.StorageClass {
+			t.Fatalf(
+				"unexpected response storage class for file with storage class %v: %v",
+				storageClass,
+				*resp.StorageClass,
+			)
+		}
 	}
-	defer rrDriver.Delete(ctx, rrFilename)
-
-	standardDriverUnwrapped := standardDriver.Base.StorageDriver.(*driver)
-	resp, err := standardDriverUnwrapped.S3.GetObject(&s3.GetObjectInput{
-		Bucket: aws.String(standardDriverUnwrapped.Bucket),
-		Key:    aws.String(standardDriverUnwrapped.s3Path(standardFilename)),
-	})
-	if err != nil {
-		t.Fatalf("unexpected error retrieving standard storage file: %v", err)
-	}
-	defer resp.Body.Close()
-	// Amazon only populates this header value for non-standard storage classes
-	if resp.StorageClass != nil {
-		t.Fatalf("unexpected storage class for standard file: %v", resp.StorageClass)
-	}
-
-	rrDriverUnwrapped := rrDriver.Base.StorageDriver.(*driver)
-	resp, err = rrDriverUnwrapped.S3.GetObject(&s3.GetObjectInput{
-		Bucket: aws.String(rrDriverUnwrapped.Bucket),
-		Key:    aws.String(rrDriverUnwrapped.s3Path(rrFilename)),
-	})
-	if err != nil {
-		t.Fatalf("unexpected error retrieving reduced-redundancy storage file: %v", err)
-	}
-	defer resp.Body.Close()
-	if resp.StorageClass == nil {
-		t.Fatalf("unexpected storage class for reduced-redundancy file: %v", s3.StorageClassStandard)
-	} else if *resp.StorageClass != s3.StorageClassReducedRedundancy {
-		t.Fatalf("unexpected storage class for reduced-redundancy file: %v", *resp.StorageClass)
-	}
-
 }
 
 func TestDelete(t *testing.T) {