set flag when storeresults is called

Signed-off-by: Ivan Milchev <[email protected]>
mondoohq · Mar 6, 2024 · 203dcc2 · 203dcc2
1 parent f8f595c
commit 203dcc2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/explorer/scan/local_scanner.go b/explorer/scan/local_scanner.go
@@ -616,8 +616,9 @@ func (s *localAssetScanner) runQueryPack() (*AssetReport, error) {
 			log.Debug().Msg("not storing resource data for this asset, nothing available")
 		} else {
 			_, err = conductor.StoreResults(context.Background(), &explorer.StoreResultsReq{
-				AssetMrn:  s.job.Asset.Mrn,
-				Resources: data,
+				AssetMrn:    s.job.Asset.Mrn,
+				Resources:   data,
+				IsLastBatch: true,
 			})
 			if err != nil {
 				return nil, err

diff --git a/mql/internal/collector.go b/mql/internal/collector.go
@@ -141,7 +141,7 @@ type FuncCollector struct {
 	SinkDataFunc func(results []*llx.RawResult)
 }
 
-func (c *FuncCollector) SinkData(results []*llx.RawResult, isDone bool) {
+func (c *FuncCollector) SinkData(results []*llx.RawResult) {
 	if len(results) == 0 || c.SinkDataFunc == nil {
 		return
 	}