Systemd: Add support for MAINPID= and EXTEND_TIMEOUT_USEC

src/Systemd.cpp

@@ -29,6 +29,12 @@ const std::string isWatchdog = std::string("WATCHDOG=1");
 // systemd string prefix to publish information to systemd
 const std::string isStatusPrefix = std::string("STATUS=");
 
+// systemd string prefix to publish the main PID
+const std::string mainPidPrefix = std::string("MAINPID=");
+
+// systemd string prefix to publish timeout extension
+const std::string extendTimeoutPrefix = std::string("EXTEND_TIMEOUT_USEC=");
+
 // Multiplier to convert between milliseconds and nanoseconds
 const uint64_t msToNano = 1000000;
 
@@ -75,6 +81,19 @@ void notifyStatus(std::string status) {
     sd_pid_notify(getpid(), 0, statusPrefix.append(status).c_str());
 }
 
+// Sends the specified PID of the service to systemd in case the service manager did not fork off the process itself
+void sendMainPid(int pid) {
+    std::string mainPidStr = std::string(mainPidPrefix);
+    sd_pid_notify(getpid(), 0, mainPidStr.append(std::to_string(pid)).c_str());
+}
+
+// Allows the kdb process to extend the amount of time taken to transition the process (e.g. during process start, before ready to check TP log file)
+//  @param extendTimeUs The time extension required in microseconds
+void extendTimeout(int extendTimeUs) {
+    std::string timeoutStr = std::string(extendTimeoutPrefix);
+    sd_pid_notify(getpid(), 0, timeoutStr.append(std::to_string(extendTimeUs)).c_str());
+}
+
 }   // namespace kdbsystemd
 
 
@@ -117,3 +136,31 @@ extern "C" K sendStatus(K status) {
     kdbsystemd::notifyStatus(statusStr);
     return kb(1);
 }
+
+extern "C" K sendMainPid(K intOrNullArg) {
+    int pid = getpid();
+
+    if(intOrNullArg->t != -KI)
+        pid = intOrNullArg->i;
+
+    kdbsystemd::sendMainPid(pid);
+    return kb(1);
+}
+
+extern "C" K extendTimeout(K timespan) {
+    int timeoutUs;
+
+    if(timespan->t != -KN)
+        return krr((char*) "[lib-kdbsystemd] Incorrect type for timeout extension. Must be timespan");
+
+    if(timespan->j == nj)
+        return krr((char*) "[lib-kdbsystemd] Cannot specify null timeout extension");
+
+    timeoutUs = timespan->j / 1000;
+
+    if(timeoutUs <= 0)
+        return krr((char*) "[lib-kdbsystemd] Incorrect value for start timeout extension. Must be greater than 0 us");
+
+    kdbsystemd::extendTimeout(timeoutUs);
+    return kb(1);
+}