Skip to content
Snippets Groups Projects
Commit dc6d2aed authored by Humam HusamEldeen's avatar Humam HusamEldeen
Browse files

TDP-104 Add Enable/Disable AutoPilot to AWS

parent dc1317a8
No related branches found
No related tags found
No related merge requests found
......@@ -125,6 +125,12 @@
}
}
},
"v1DisableAutoPilotResponse": {
"type": "object"
},
"v1EnableAutoPilotResponse": {
"type": "object"
},
"v1GenerateRecommendationsResponse": {
"type": "object",
"properties": {
......
This diff is collapsed.
......@@ -33,6 +33,8 @@ type AWSServiceClient interface {
GetUsage(ctx context.Context, in *GetUsageRequest, opts ...grpc.CallOption) (*GetUsageResponse, error)
SubscribeToProduct(ctx context.Context, in *SubscribeToProductRequest, opts ...grpc.CallOption) (*SubscribeToProductResponse, error)
GetProductSubscriptionStatus(ctx context.Context, in *GetProductSubscriptionStatusRequest, opts ...grpc.CallOption) (*GetProductSubscriptionStatusResponse, error)
EnableAutoPilot(ctx context.Context, in *EnableAutoPilotRequest, opts ...grpc.CallOption) (*EnableAutoPilotResponse, error)
DisableAutoPilot(ctx context.Context, in *DisableAutoPilotRequest, opts ...grpc.CallOption) (*DisableAutoPilotResponse, error)
}
type aWSServiceClient struct {
......@@ -142,6 +144,24 @@ func (c *aWSServiceClient) GetProductSubscriptionStatus(ctx context.Context, in
return out, nil
}
func (c *aWSServiceClient) EnableAutoPilot(ctx context.Context, in *EnableAutoPilotRequest, opts ...grpc.CallOption) (*EnableAutoPilotResponse, error) {
out := new(EnableAutoPilotResponse)
err := c.cc.Invoke(ctx, "/tadpole.aws_service.v1.AWSService/EnableAutoPilot", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *aWSServiceClient) DisableAutoPilot(ctx context.Context, in *DisableAutoPilotRequest, opts ...grpc.CallOption) (*DisableAutoPilotResponse, error) {
out := new(DisableAutoPilotResponse)
err := c.cc.Invoke(ctx, "/tadpole.aws_service.v1.AWSService/DisableAutoPilot", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// AWSServiceServer is the server API for AWSService service.
// All implementations must embed UnimplementedAWSServiceServer
// for forward compatibility
......@@ -157,6 +177,8 @@ type AWSServiceServer interface {
GetUsage(context.Context, *GetUsageRequest) (*GetUsageResponse, error)
SubscribeToProduct(context.Context, *SubscribeToProductRequest) (*SubscribeToProductResponse, error)
GetProductSubscriptionStatus(context.Context, *GetProductSubscriptionStatusRequest) (*GetProductSubscriptionStatusResponse, error)
EnableAutoPilot(context.Context, *EnableAutoPilotRequest) (*EnableAutoPilotResponse, error)
DisableAutoPilot(context.Context, *DisableAutoPilotRequest) (*DisableAutoPilotResponse, error)
mustEmbedUnimplementedAWSServiceServer()
}
......@@ -197,6 +219,12 @@ func (UnimplementedAWSServiceServer) SubscribeToProduct(context.Context, *Subscr
func (UnimplementedAWSServiceServer) GetProductSubscriptionStatus(context.Context, *GetProductSubscriptionStatusRequest) (*GetProductSubscriptionStatusResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetProductSubscriptionStatus not implemented")
}
func (UnimplementedAWSServiceServer) EnableAutoPilot(context.Context, *EnableAutoPilotRequest) (*EnableAutoPilotResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method EnableAutoPilot not implemented")
}
func (UnimplementedAWSServiceServer) DisableAutoPilot(context.Context, *DisableAutoPilotRequest) (*DisableAutoPilotResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method DisableAutoPilot not implemented")
}
func (UnimplementedAWSServiceServer) mustEmbedUnimplementedAWSServiceServer() {}
// UnsafeAWSServiceServer may be embedded to opt out of forward compatibility for this service.
......@@ -408,6 +436,42 @@ func _AWSService_GetProductSubscriptionStatus_Handler(srv interface{}, ctx conte
return interceptor(ctx, in, info, handler)
}
func _AWSService_EnableAutoPilot_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(EnableAutoPilotRequest)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(AWSServiceServer).EnableAutoPilot(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/tadpole.aws_service.v1.AWSService/EnableAutoPilot",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(AWSServiceServer).EnableAutoPilot(ctx, req.(*EnableAutoPilotRequest))
}
return interceptor(ctx, in, info, handler)
}
func _AWSService_DisableAutoPilot_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DisableAutoPilotRequest)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(AWSServiceServer).DisableAutoPilot(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/tadpole.aws_service.v1.AWSService/DisableAutoPilot",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(AWSServiceServer).DisableAutoPilot(ctx, req.(*DisableAutoPilotRequest))
}
return interceptor(ctx, in, info, handler)
}
// AWSService_ServiceDesc is the grpc.ServiceDesc for AWSService service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
......@@ -459,6 +523,14 @@ var AWSService_ServiceDesc = grpc.ServiceDesc{
MethodName: "GetProductSubscriptionStatus",
Handler: _AWSService_GetProductSubscriptionStatus_Handler,
},
{
MethodName: "EnableAutoPilot",
Handler: _AWSService_EnableAutoPilot_Handler,
},
{
MethodName: "DisableAutoPilot",
Handler: _AWSService_DisableAutoPilot_Handler,
},
},
Streams: []grpc.StreamDesc{},
Metadata: "tadpole/aws-svc/v1/aws.proto",
......
......@@ -81,6 +81,46 @@ func (mr *MockAWSServiceClientMockRecorder) CreateOrganization(ctx, in any, opts
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrganization", reflect.TypeOf((*MockAWSServiceClient)(nil).CreateOrganization), varargs...)
}
// DisableAutoPilot mocks base method.
func (m *MockAWSServiceClient) DisableAutoPilot(ctx context.Context, in *v1.DisableAutoPilotRequest, opts ...grpc.CallOption) (*v1.DisableAutoPilotResponse, error) {
m.ctrl.T.Helper()
varargs := []any{ctx, in}
for _, a := range opts {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "DisableAutoPilot", varargs...)
ret0, _ := ret[0].(*v1.DisableAutoPilotResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// DisableAutoPilot indicates an expected call of DisableAutoPilot.
func (mr *MockAWSServiceClientMockRecorder) DisableAutoPilot(ctx, in any, opts ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{ctx, in}, opts...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DisableAutoPilot", reflect.TypeOf((*MockAWSServiceClient)(nil).DisableAutoPilot), varargs...)
}
// EnableAutoPilot mocks base method.
func (m *MockAWSServiceClient) EnableAutoPilot(ctx context.Context, in *v1.EnableAutoPilotRequest, opts ...grpc.CallOption) (*v1.EnableAutoPilotResponse, error) {
m.ctrl.T.Helper()
varargs := []any{ctx, in}
for _, a := range opts {
varargs = append(varargs, a)
}
ret := m.ctrl.Call(m, "EnableAutoPilot", varargs...)
ret0, _ := ret[0].(*v1.EnableAutoPilotResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// EnableAutoPilot indicates an expected call of EnableAutoPilot.
func (mr *MockAWSServiceClientMockRecorder) EnableAutoPilot(ctx, in any, opts ...any) *gomock.Call {
mr.mock.ctrl.T.Helper()
varargs := append([]any{ctx, in}, opts...)
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnableAutoPilot", reflect.TypeOf((*MockAWSServiceClient)(nil).EnableAutoPilot), varargs...)
}
// GenerateRecommendations mocks base method.
func (m *MockAWSServiceClient) GenerateRecommendations(ctx context.Context, in *v1.GenerateRecommendationsRequest, opts ...grpc.CallOption) (*v1.GenerateRecommendationsResponse, error) {
m.ctrl.T.Helper()
......@@ -314,6 +354,36 @@ func (mr *MockAWSServiceServerMockRecorder) CreateOrganization(arg0, arg1 any) *
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrganization", reflect.TypeOf((*MockAWSServiceServer)(nil).CreateOrganization), arg0, arg1)
}
// DisableAutoPilot mocks base method.
func (m *MockAWSServiceServer) DisableAutoPilot(arg0 context.Context, arg1 *v1.DisableAutoPilotRequest) (*v1.DisableAutoPilotResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "DisableAutoPilot", arg0, arg1)
ret0, _ := ret[0].(*v1.DisableAutoPilotResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// DisableAutoPilot indicates an expected call of DisableAutoPilot.
func (mr *MockAWSServiceServerMockRecorder) DisableAutoPilot(arg0, arg1 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DisableAutoPilot", reflect.TypeOf((*MockAWSServiceServer)(nil).DisableAutoPilot), arg0, arg1)
}
// EnableAutoPilot mocks base method.
func (m *MockAWSServiceServer) EnableAutoPilot(arg0 context.Context, arg1 *v1.EnableAutoPilotRequest) (*v1.EnableAutoPilotResponse, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "EnableAutoPilot", arg0, arg1)
ret0, _ := ret[0].(*v1.EnableAutoPilotResponse)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// EnableAutoPilot indicates an expected call of EnableAutoPilot.
func (mr *MockAWSServiceServerMockRecorder) EnableAutoPilot(arg0, arg1 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnableAutoPilot", reflect.TypeOf((*MockAWSServiceServer)(nil).EnableAutoPilot), arg0, arg1)
}
// GenerateRecommendations mocks base method.
func (m *MockAWSServiceServer) GenerateRecommendations(arg0 context.Context, arg1 *v1.GenerateRecommendationsRequest) (*v1.GenerateRecommendationsResponse, error) {
m.ctrl.T.Helper()
......
......@@ -30,6 +30,9 @@ service AWSService {
rpc SubscribeToProduct(SubscribeToProductRequest) returns (SubscribeToProductResponse);
rpc GetProductSubscriptionStatus(GetProductSubscriptionStatusRequest) returns (GetProductSubscriptionStatusResponse);
rpc EnableAutoPilot(EnableAutoPilotRequest) returns (EnableAutoPilotResponse);
rpc DisableAutoPilot(DisableAutoPilotRequest) returns (DisableAutoPilotResponse);
}
message CreateOrganizationRequest {
......@@ -193,3 +196,15 @@ message GetProductSubscriptionStatusRequest {
message GetProductSubscriptionStatusResponse {
bool status = 1;
}
message EnableAutoPilotRequest {
string account_id = 1;
}
message EnableAutoPilotResponse {}
message DisableAutoPilotRequest {
string account_id = 1;
}
message DisableAutoPilotResponse {}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment