Extended protobuf definition to pass query result limit to workers

lsst · Jul 20, 2023 · 66b9ab8 · 66b9ab8
1 parent 1d3ed34
commit 66b9ab8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/proto/worker.proto b/src/proto/worker.proto
@@ -74,6 +74,7 @@ message TaskMsg {
     optional bool scaninteractive = 12;
     optional int32 attemptcount = 13;
     optional uint32 czarid = 14;
+    optional int32 maxtablesize_mb = 15 [default = 0];
 }
 
 // Result message received from worker

diff --git a/src/qproc/TaskMsgFactory.cc b/src/qproc/TaskMsgFactory.cc
@@ -41,6 +41,7 @@
 #include "lsst/log/Log.h"
 
 // Qserv headers
+#include "czar/CzarConfig.h"
 #include "global/intTypes.h"
 #include "qmeta/types.h"
 #include "qproc/ChunkQuerySpec.h"
@@ -84,6 +85,7 @@ std::shared_ptr<proto::TaskMsg> TaskMsgFactory::_makeMsg(ChunkQuerySpec const& c
 
     taskMsg->set_scanpriority(chunkQuerySpec.scanInfo.scanRating);
     taskMsg->set_scaninteractive(chunkQuerySpec.scanInteractive);
+    taskMsg->set_maxtablesize_mb(czar::CzarConfig::instance()->getMaxTableSizeMB());
 
     // per-chunk
     taskMsg->set_chunkid(chunkQuerySpec.chunkId);