feat/implementation of parallel resource limit (#103)

* implementation of parallel resource limit
cloudquery · Oct 30, 2021 · d48afc8 · d48afc8
1 parent a5b33ff
commit d48afc8
Show file tree

Hide file tree

Showing 6 changed files with 292 additions and 150 deletions.
diff --git a/cqproto/grpc.go b/cqproto/grpc.go
@@ -67,6 +67,7 @@ func (g GRPCClient) FetchResources(ctx context.Context, request *FetchResourcesR
 	res, err := g.client.FetchResources(ctx, &internal.FetchResources_Request{
 		Resources:              request.Resources,
 		PartialFetchingEnabled: request.PartialFetchingEnabled,
+		ParallelFetchingLimit:  request.ParallelFetchingLimit,
 	})
 	if err != nil {
 		return nil, err
@@ -156,7 +157,7 @@ func (g *GRPCServer) ConfigureProvider(ctx context.Context, request *internal.Co
 func (g *GRPCServer) FetchResources(request *internal.FetchResources_Request, server internal.Provider_FetchResourcesServer) error {
 	return g.Impl.FetchResources(
 		server.Context(),
-		&FetchResourcesRequest{Resources: request.GetResources(), PartialFetchingEnabled: request.PartialFetchingEnabled},
+		&FetchResourcesRequest{Resources: request.GetResources(), PartialFetchingEnabled: request.PartialFetchingEnabled, ParallelFetchingLimit: request.ParallelFetchingLimit},
 		&GRPCFetchResourcesServer{server: server},
 	)
 }