Add option to quit on successful sending

wiechula · shahor02 · commit db2b72e6a786 · 2022-12-02T09:14:36.000+01:00
diff --git a/Detectors/DCS/testWorkflow/README.md b/Detectors/DCS/testWorkflow/README.md
@@ -104,6 +104,7 @@ You can use it as e.g.:
 echo "blabla" > TOFfile.txt  # this is the file you want to send from the DCS to the config processor
 xterm -e "alienv load cppzmq/latest O2/latest; ./dcssend -f TOFfile.txt -o 5556 -a 5557"& # run the server emulator in separate terminal
 ```
+In case the process should end upon successful sending of the object, add `-q true`. This is helpful e.g. in case several objects should be sent from a script. Otherwise the object is sent over and over again.
 
 Then, in other terminal you can run your DCS config processor, as described above (make sure the ports of sender and receiver are consistent.
 In case of problems you can validate the receiving process using `dcsclient` test utility (emulates `o2-dcs-config-proxy ...` workflow by receiving the file from the `DCS server` and sending it an acknowledgment):
diff --git a/Detectors/DCS/testWorkflow/src/dcssend.cpp b/Detectors/DCS/testWorkflow/src/dcssend.cpp
@@ -35,6 +35,7 @@ int main(int argc, char** argv)
     add_option("file-port,o", bpo::value<int>()->default_value(5556), "port to send the file");
     add_option("ack-port,a", bpo::value<int>()->default_value(5557), "port to receive the acknowledgment");
     add_option("timeout,t", bpo::value<int>()->default_value(5), "timeout for acknowledgment");
+    add_option("quit-on-ack,q", bpo::value<bool>()->default_value(false), "quit if acknowledgment is ok");
     opt_all.add(opt_general).add(opt_hidden);
     bpo::store(bpo::command_line_parser(argc, argv).options(opt_all).positional(opt_pos).run(), vm);
 
@@ -60,6 +61,8 @@ int main(int argc, char** argv)
 
   int recv_timeout = vm["timeout"].as<int>();
 
+  const bool quitOnAck = vm["quit-on-ack"].as<bool>();
+
   collector.set(zmq::sockopt::rcvtimeo, recv_timeout * 1000);
 
   int pub_port = vm["file-port"].as<int>(), col_port = vm["ack-port"].as<int>();
@@ -112,6 +115,9 @@ int main(int argc, char** argv)
     ans.assign(ack.to_string());
     std::cout << ans << std::endl;
     // end of ack -------------------
+    if (quitOnAck && (ans.find("ok") == (ans.size() - 2))) {
+      break;
+    }
 
     trial++;
   }
