Skip to content

Commit

Permalink
registry: add loglevel support for aws s3 storage driver
Browse files Browse the repository at this point in the history
based on the work from
distribution#3057.

Co-authored-by: Simon Compston <compston@gmail.com>
Signed-off-by: Flavian Missi <fmissi@redhat.com>
  • Loading branch information
flavianmissi and vleurgat committed Sep 27, 2023
1 parent 58a7634 commit e0aed35
Show file tree
Hide file tree
Showing 4 changed files with 41 additions and 3 deletions.
2 changes: 2 additions & 0 deletions docs/configuration.md
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,7 @@ storage:
multipartcopythresholdsize: 33554432
rootdirectory: /s3/object/name/prefix
usedualstack: false
loglevel: logdebug
inmemory: # This driver takes no parameters
delete:
enabled: false
Expand Down Expand Up @@ -410,6 +411,7 @@ storage:
multipartcopymaxconcurrency: 100
multipartcopythresholdsize: 33554432
rootdirectory: /s3/object/name/prefix
loglevel: logdebug
inmemory:
delete:
enabled: false
Expand Down
2 changes: 2 additions & 0 deletions docs/storage-drivers/s3.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ Amazon S3 or S3 compatible services for object storage.
| `rootdirectory` | no | This is a prefix that is applied to all S3 keys to allow you to segment data in your bucket if necessary. |
| `storageclass` | no | The S3 storage class applied to each registry file. The default is `STANDARD`. |
| `objectacl` | no | The S3 Canned ACL for objects. The default value is "private". |
| `loglevel` | no | The log level for the S3 client. The default value is `logoff`. |

> **Note** You can provide empty strings for your access and secret keys to run the driver
> on an ec2 instance and handles authentication with the instance's credentials. If you
Expand Down Expand Up @@ -56,6 +57,7 @@ Amazon S3 or S3 compatible services for object storage.

`objectacl`: (optional) The canned object ACL to be applied to each registry object. Defaults to `private`. If you are using a bucket owned by another AWS account, it is recommended that you set this to `bucket-owner-full-control` so that the bucket owner can access your objects. Other valid options are available in the [AWS S3 documentation](http://docs.aws.amazon.com/AmazonS3/latest/dev/acl-overview.html#canned-acl).

`loglevel`: (optional) Valid values are: `logoff` (default), `logdebug`, `logdebugwithsigning`, `logdebugwithhttpbody`, `logdebugwithrequestretries`, `logdebugwithrequesterrors` and `logdebugwitheventstreambody`. See the [AWS SDK for Go API reference](https://docs.aws.amazon.com/sdk-for-go/api/aws/#LogLevelType) for details.

## S3 permission scopes

Expand Down
32 changes: 31 additions & 1 deletion registry/storage/driver/s3-aws/s3.go
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,7 @@ type DriverParameters struct {
SessionToken string
UseDualStack bool
Accelerate bool
LogLevel aws.LogLevelType
}

func init() {
Expand Down Expand Up @@ -461,11 +462,40 @@ func FromParameters(parameters map[string]interface{}) (*Driver, error) {
fmt.Sprint(sessionToken),
useDualStackBool,
accelerateBool,
getS3LogLevelFromParam(parameters["loglevel"]),
}

return New(params)
}

func getS3LogLevelFromParam(param interface{}) aws.LogLevelType {
if param == nil {
return aws.LogOff
}
logLevelParam := param.(string)
var logLevel aws.LogLevelType
switch strings.ToLower(logLevelParam) {
case "logoff":
logLevel = aws.LogOff
case "logdebug":
logLevel = aws.LogDebug
case "logdebugwithsigning":
logLevel = aws.LogDebugWithSigning
case "logdebugwithhttpbody":
logLevel = aws.LogDebugWithHTTPBody
case "logdebugwithrequestretries":
logLevel = aws.LogDebugWithRequestRetries
case "logdebugwithrequesterrors":
logLevel = aws.LogDebugWithRequestErrors
case "logdebugwitheventstreambody":
logLevel = aws.LogDebugWithEventStreamBody
default:
logLevel = aws.LogOff
}
fmt.Println("=====================", logLevel, "=====================")
return logLevel
}

// getParameterAsInt64 converts parameters[name] to an int64 value (using
// defaultt if nil), verifies it is no smaller than min, and returns it.
func getParameterAsInt64(parameters map[string]interface{}, name string, defaultt int64, min int64, max int64) (int64, error) {
Expand Down Expand Up @@ -504,7 +534,7 @@ func New(params DriverParameters) (*Driver, error) {
return nil, fmt.Errorf("on Amazon S3 this storage driver can only be used with v4 authentication")
}

awsConfig := aws.NewConfig()
awsConfig := aws.NewConfig().WithLogLevel(params.LogLevel)

if params.AccessKey != "" && params.SecretKey != "" {
creds := credentials.NewStaticCredentials(
Expand Down
8 changes: 6 additions & 2 deletions registry/storage/driver/s3-aws/s3_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,10 @@ import (
"github.com/distribution/distribution/v3/registry/storage/driver/testsuites"
)

var s3DriverConstructor func(rootDirectory, storageClass string) (*Driver, error)
var skipS3 func() string
var (
s3DriverConstructor func(rootDirectory, storageClass string) (*Driver, error)
skipS3 func() string
)

func init() {
var (
Expand All @@ -42,6 +44,7 @@ func init() {
useDualStack = os.Getenv("S3_USE_DUALSTACK")
combineSmallPart = os.Getenv("MULTIPART_COMBINE_SMALL_PART")
accelerate = os.Getenv("S3_ACCELERATE")
logLevel = os.Getenv("S3_LOGLEVEL")
)

root, err := os.MkdirTemp("", "driver-")
Expand Down Expand Up @@ -135,6 +138,7 @@ func init() {
sessionToken,
useDualStackBool,
accelerateBool,
getS3LogLevelFromParam(logLevel),
}

return New(parameters)
Expand Down

0 comments on commit e0aed35

Please sign in to comment.