awsmocker/middleware_initialize.go at master · webdestroya/awsmocker · GitHub

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
package awsmocker

import (
	"context"

	"github.com/aws/smithy-go/middleware"
)

var _ middleware.InitializeMiddleware = (*mockerMiddleware)(nil)

func (m *mockerMiddleware) HandleInitialize(ctx context.Context, in middleware.InitializeInput, next middleware.InitializeHandler) (middleware.InitializeOutput, middleware.Metadata, error) {

	// if _, ok := in.Parameters.(*ec2.DescribeSubnetsInput); ok {
	// 	out := middleware.InitializeOutput{
	// 		Result: &ec2.DescribeSubnetsOutput{
	// 			Subnets: []ec2Types.Subnet{
	// 				{
	// 					SubnetId: aws.String("subnet-aaaaaaaa"),
	// 					VpcId:    aws.String("vpc-11111111"),
	// 				},
	// 			},
	// 		},
	// 	}
	// 	return out, middleware.Metadata{}, nil
	// }

	reqId := m.mocker.mwReqCounter.Add(1)

	ctx = middleware.WithStackValue(ctx, mwCtxKeyReqId{}, reqId)
	ctx = middleware.WithStackValue(ctx, mwCtxKeyParams{}, in.Parameters)

	m.mocker.requestLog.Store(reqId, mwDBEntry{
		Parameters: in.Parameters,
	})

	defer m.mocker.requestLog.Delete(reqId)

	return next.HandleInitialize(ctx, in)
}