From 2609bc801cdc386e029518db8c36ded7b16f1b31 Mon Sep 17 00:00:00 2001
From: Darin Kelkhoff <darin.kelkhoff@gmail.com>
Date: Mon, 22 Jul 2024 14:25:49 -0500
Subject: [PATCH] CE-1405 Add dataSource as argument to
 ReportCustomRecordSourceInterface.execute

---
 .../backend/core/actions/reporting/GenerateReportAction.java   | 2 +-
 .../customizers/ReportCustomRecordSourceInterface.java         | 3 ++-
 .../qqq/backend/core/instances/QInstanceValidatorTest.java     | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/GenerateReportAction.java b/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/GenerateReportAction.java
index 3d687344..eff97f4a 100644
--- a/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/GenerateReportAction.java
+++ b/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/GenerateReportAction.java
@@ -453,7 +453,7 @@ public class GenerateReportAction extends AbstractQActionFunction<ReportInput, R
          if(dataSource.getCustomRecordSource() != null)
          {
             ReportCustomRecordSourceInterface recordSource = QCodeLoader.getAdHoc(ReportCustomRecordSourceInterface.class, dataSource.getCustomRecordSource());
-            recordSource.execute(reportInput, recordPipe);
+            recordSource.execute(reportInput, dataSource, recordPipe);
             return (true);
          }
          else if(dataSource.getSourceTable() != null)
diff --git a/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/customizers/ReportCustomRecordSourceInterface.java b/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/customizers/ReportCustomRecordSourceInterface.java
index 42b1749b..522f3f88 100644
--- a/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/customizers/ReportCustomRecordSourceInterface.java
+++ b/qqq-backend-core/src/main/java/com/kingsrook/qqq/backend/core/actions/reporting/customizers/ReportCustomRecordSourceInterface.java
@@ -25,6 +25,7 @@ package com.kingsrook.qqq.backend.core.actions.reporting.customizers;
 import com.kingsrook.qqq.backend.core.actions.reporting.RecordPipe;
 import com.kingsrook.qqq.backend.core.exceptions.QException;
 import com.kingsrook.qqq.backend.core.model.actions.reporting.ReportInput;
+import com.kingsrook.qqq.backend.core.model.metadata.reporting.QReportDataSource;
 
 
 /*******************************************************************************
@@ -37,6 +38,6 @@ public interface ReportCustomRecordSourceInterface
    /***************************************************************************
     ** Given the report input, put records into the pipe, for the report.
     ***************************************************************************/
-   void execute(ReportInput reportInput, RecordPipe recordPipe) throws QException;
+   void execute(ReportInput reportInput, QReportDataSource reportDataSource, RecordPipe recordPipe) throws QException;
 
 }
diff --git a/qqq-backend-core/src/test/java/com/kingsrook/qqq/backend/core/instances/QInstanceValidatorTest.java b/qqq-backend-core/src/test/java/com/kingsrook/qqq/backend/core/instances/QInstanceValidatorTest.java
index 512cf7c5..e09cfe1f 100644
--- a/qqq-backend-core/src/test/java/com/kingsrook/qqq/backend/core/instances/QInstanceValidatorTest.java
+++ b/qqq-backend-core/src/test/java/com/kingsrook/qqq/backend/core/instances/QInstanceValidatorTest.java
@@ -2398,7 +2398,7 @@ public class QInstanceValidatorTest extends BaseTest
        **
        ***************************************************************************/
       @Override
-      public void execute(ReportInput reportInput, RecordPipe recordPipe) throws QException
+      public void execute(ReportInput reportInput, QReportDataSource reportDataSource, RecordPipe recordPipe) throws QException
       {
 
       }