summaryrefslogtreecommitdiffstats
path: root/examples/rgw/golang/put-bucket-notification-creation/put-bucket-notification-creation.go
blob: 5e8b7f4bc9f944e98568588a47dad6ed929d5e41 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package main

import (
	"flag"
	"fmt"
	"os"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/endpoints"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
)

func main() {
	bucket := flag.String("b", "", "Name of the bucket to add notification to")
	topic := flag.String("t", "", "The topic onto which the notification is attached to")
	flag.Parse()

	if *bucket == "" {
		fmt.Println("You must supply the name of the bucket")
		fmt.Println("-b BUCKET")
		return
	}

	if *topic == "" {
		fmt.Println("You must supply the name of the topic ARN")
		fmt.Println("-t TOPIC ARN")
		return
	}

	//Ceph RGW Credentials
	access_key := "0555b35654ad1656d804"
	secret_key := "h7GhxuBLTrlhVUyxSPUKUV8r/2EI4ngqJxD7iBdBYLhwluN30JaT3Q=="
	token_id := ""
	url := "http://127.0.0.1:8000"

	defaultResolver := endpoints.DefaultResolver()
	CustResolverFn := func(service, region string, optFns ...func(*endpoints.Options)) (endpoints.ResolvedEndpoint, error) {
		if service == "s3" {
			return endpoints.ResolvedEndpoint{
				URL: url,
			}, nil
		}

		return defaultResolver.EndpointFor(service, region, optFns...)
	}

	sess := session.Must(session.NewSessionWithOptions(session.Options{
		Config: aws.Config{
			Region:           aws.String("default"),
			Credentials:      credentials.NewStaticCredentials(access_key, secret_key, token_id),
			S3ForcePathStyle: aws.Bool(true),
			EndpointResolver: endpoints.ResolverFunc(CustResolverFn),
		},
	}))

	svc := s3.New(sess)

	suffixRule := []*s3.FilterRule{
		{
			Name:  aws.String("suffix"),
			Value: aws.String("jpg"),
		},
	}

	input := &s3.PutBucketNotificationConfigurationInput{
		Bucket: bucket,
		NotificationConfiguration: &s3.NotificationConfiguration{
			TopicConfigurations: []*s3.TopicConfiguration{
				{
					Events: []*string{aws.String("s3:ObjectCreated:*")},
					Filter: &s3.NotificationConfigurationFilter{
						Key: &s3.KeyFilter{
							FilterRules: suffixRule,
						},
					},
					Id:       aws.String("notif1"), //Raises MalformedXML if absent
					TopicArn: topic,
				},
			},
		},
	}

	_, err := svc.PutBucketNotificationConfiguration(input)

	if err != nil {
		exitErrorf("Unable to create Put Bucket Notification because of %s", err)
	}
	fmt.Println("Put bucket notification added to  ", *topic)
}

func exitErrorf(msg string, args ...interface{}) {
	fmt.Fprintf(os.Stderr, msg+"\n", args...)
	os.Exit(1)
}