implement beanstalk plugin

plugins/beanstalk/beanstalk.go

@@ -0,0 +1,59 @@
+package beanstalk
+
+import (
+	"context"
+	"encoding/json"
+	"os"
+	"runtime"
+	"time"
+
+	"github.com/shogo82148/aws-xray-yasdk-go/xray"
+	"github.com/shogo82148/aws-xray-yasdk-go/xray/schema"
+	"github.com/shogo82148/aws-xray-yasdk-go/xray/xraylog"
+)
+
+const configPath = "/var/elasticbeanstalk/xray/environment.conf"
+
+type plugin struct {
+	ElasticBeanstalk *schema.ElasticBeanstalk
+}
+
+// Init activates ECS Plugin at runtime.
+func Init() {
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	if runtime.GOOS != "linux" {
+		return
+	}
+
+	f, err := os.Open(configPath)
+	if err != nil {
+		// it seems not to be in Elastic Beanstalk environment.
+		// just ignore error.
+		xraylog.Debugf(ctx, "failed to read %q: %v", configPath, err)
+		return
+	}
+	defer f.Close()
+
+	var conf schema.ElasticBeanstalk
+	dec := json.NewDecoder(f)
+	if err := dec.Decode(&conf); err != nil {
+		xraylog.Debugf(ctx, "failed to decode: %v", err)
+		return
+	}
+	xray.AddPlugin(&plugin{
+		ElasticBeanstalk: &conf,
+	})
+}
+
+// HandleSegment implements Plugin.
+func (p *plugin) HandleSegment(seg *xray.Segment, doc *schema.Segment) {
+	if doc.AWS == nil {
+		doc.AWS = schema.AWS{}
+	}
+	doc.AWS.SetElasticBeanstalk(p.ElasticBeanstalk)
+}
+
+// Origin implements Plugin.
+func (*plugin) Origin() string { return schema.OriginElasticBeanstalk }

plugins/beanstalk/init/init.go

@@ -0,0 +1,7 @@
+package init
+
+import "github.com/shogo82148/aws-xray-yasdk-go/plugins/beanstalk"
+
+func init() {
+	beanstalk.Init()
+}