persist run number of the active partition

Author: ironMann

src/StfBuilder/StfBuilderDevice.cxx

@@ -446,6 +446,7 @@ void StfBuilderDevice::PreRun()
   if (I().mDiscoveryConfig->enabled()) {
     auto& lStatus = I().mDiscoveryConfig->status();
     lStatus.mutable_info()->set_process_state(BasicInfo::RUNNING);
+    lStatus.mutable_partition()->set_run_number(DataDistLogger::sRunNumber);
     I().mDiscoveryConfig->write();
   }
 

src/StfSender/StfSenderDevice.cxx

@@ -161,6 +161,7 @@ void StfSenderDevice::PreRun()
   if (!standalone() && I().mDiscoveryConfig) {
     auto& lStatus = I().mDiscoveryConfig->status();
     lStatus.mutable_info()->set_process_state(BasicInfo::RUNNING);
+    lStatus.mutable_partition()->set_run_number(DataDistLogger::sRunNumber);
     I().mDiscoveryConfig->write();
   }
 

src/TfBuilder/TfBuilderDevice.cxx

@@ -171,6 +171,7 @@ void TfBuilderDevice::PreRun()
   // update running state
   auto& lStatus = mDiscoveryConfig->status();
   lStatus.mutable_info()->set_process_state(BasicInfo::RUNNING);
+  lStatus.mutable_partition()->set_run_number(DataDistLogger::sRunNumber);
   mDiscoveryConfig->write();
 
   // make directory for the file sink

src/common/discovery/discovery.proto

@@ -47,15 +47,14 @@ message BasicInfo {
 }
 
 message PartitionInfo {
-  string environment_id     = 1;
-  // AliECS environment ID (required)
-  string partition_id       = 2;
-  // Partition ID. (required)
+  string environment_id = 1;    // AliECS environment ID (required)
+  string partition_id   = 2;    // Partition ID. (required)
+  uint64 run_number     = 3;
 }
 
 message TfSchedulerServiceConfigStatus {
-  BasicInfo   info                  = 1;
-  uint32      number_instances      = 2;
+  BasicInfo   info             = 1;
+  uint32      number_instances = 2;
 }
 
 message TfSchedulerInstanceConfigStatus {