Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
58 lines (43 sloc) 1.19 KB
package main
import (
http "net/http"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3"
)
// s3Transport implements RoundTripper for the 's3' protocol.
type s3Transport struct {
svc *s3.S3
}
func newS3Transport() http.RoundTripper {
s3Conf := aws.NewConfig()
if len(conf.S3Region) != 0 {
s3Conf.Region = aws.String(conf.S3Region)
}
if len(conf.S3Endpoint) != 0 {
s3Conf.Endpoint = aws.String(conf.S3Endpoint)
s3Conf.S3ForcePathStyle = aws.Bool(true)
}
sess, err := session.NewSession()
if err != nil {
logFatal("Can't create S3 session: %s", err)
}
if sess.Config.Region == nil || len(*sess.Config.Region) == 0 {
sess.Config.Region = aws.String("us-west-1")
}
return s3Transport{s3.New(sess, s3Conf)}
}
func (t s3Transport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
input := &s3.GetObjectInput{
Bucket: aws.String(req.URL.Host),
Key: aws.String(req.URL.Path),
}
if len(req.URL.RawQuery) > 0 {
input.VersionId = aws.String(req.URL.RawQuery)
}
s3req, _ := t.svc.GetObjectRequest(input)
if err := s3req.Send(); err != nil {
return nil, err
}
return s3req.HTTPResponse, nil
}
You can’t perform that action at this time.