Commitef4a9a0

authored

internal: capture user-agent in metrics (#2644)

1 parentc964dbd commitef4a9a0Copy full SHA for ef4a9a0

File tree

+41

-0

lines changed

+41

-0

lines changed

Lines changed: 8 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,8 @@`
	`1`	`+{`
	`2`	`+"id":"33f76639-3bb3-44b9-b26b-55846650e8a0",`
	`3`	`+"type":"feature",`
	`4`	`+"description":"Capture user agent in internal metrics.",`
	`5`	`+"modules": [`
	`6`	`+"."`
	`7`	`+ ]`
	`8`	`+}`

Lines changed: 1 addition & 0 deletions

Lines changed: 3 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -54,6 +54,9 @@ func WithMetricMiddlewares(`
`54`	`54`	`iferr:=timeSigning(stack);err!=nil {`
`55`	`55`	`returnerr`
`56`	`56`	`}`
	`57`	`+iferr:=stack.Build.Add(&captureUserAgent{},middleware.After);err!=nil {`
	`58`	`+returnerr`
	`59`	`+}`
`57`	`60`	`returnnil`
`58`	`61`	`}`
`59`	`62`	`}`

Lines changed: 29 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,29 @@`
	`1`	`+package middleware`
	`2`	`+`
	`3`	`+import (`
	`4`	`+"context"`
	`5`	`+"fmt"`
	`6`	`+`
	`7`	`+"github.com/aws/aws-sdk-go-v2/aws/middleware/private/metrics"`
	`8`	`+"github.com/aws/smithy-go/middleware"`
	`9`	`+smithyhttp"github.com/aws/smithy-go/transport/http"`
	`10`	`+)`
	`11`	`+`
	`12`	`+typecaptureUserAgentstruct{}`
	`13`	`+`
	`14`	`+func (*captureUserAgent)ID()string {return"captureUserAgent" }`
	`15`	`+`
	`16`	`+func (*captureUserAgent)HandleBuild(`
	`17`	`+ctx context.Context,in middleware.BuildInput,next middleware.BuildHandler,`
	`18`	`+) (`
	`19`	`+out middleware.BuildOutput,md middleware.Metadata,errerror,`
	`20`	`+) {`
	`21`	`+r,ok:=in.Request.(*smithyhttp.Request)`
	`22`	`+if!ok {`
	`23`	`+returnout,md,fmt.Errorf("unexpected transport type %T",in.Request)`
	`24`	`+}`
	`25`	`+`
	`26`	`+mctx:=metrics.Context(ctx)`
	`27`	`+mctx.Data().UserAgent=r.Header.Get("User-Agent")`
	`28`	`+returnnext.HandleBuild(ctx,in)`
	`29`	`+}`

Comments

(0)