diff options
Diffstat (limited to 'src/zennomad/include')
| -rw-r--r-- | src/zennomad/include/zennomad/nomadclient.h | 6 | ||||
| -rw-r--r-- | src/zennomad/include/zennomad/nomadprovisioner.h | 9 |
2 files changed, 13 insertions, 2 deletions
diff --git a/src/zennomad/include/zennomad/nomadclient.h b/src/zennomad/include/zennomad/nomadclient.h index 0a3411ace..cebf217e1 100644 --- a/src/zennomad/include/zennomad/nomadclient.h +++ b/src/zennomad/include/zennomad/nomadclient.h @@ -52,7 +52,11 @@ public: /** Build the Nomad job registration JSON for the given job ID and orchestrator endpoint. * The JSON structure varies based on the configured driver and distribution mode. */ - std::string BuildJobJson(const std::string& JobId, const std::string& OrchestratorEndpoint) const; + std::string BuildJobJson(const std::string& JobId, + const std::string& OrchestratorEndpoint, + const std::string& CoordinatorSession = {}, + bool CleanStart = false, + const std::string& TraceHost = {}) const; /** Submit a job via PUT /v1/jobs. On success, populates OutJob with the job info. */ bool SubmitJob(const std::string& JobJson, NomadJobInfo& OutJob); diff --git a/src/zennomad/include/zennomad/nomadprovisioner.h b/src/zennomad/include/zennomad/nomadprovisioner.h index 750693b3f..a8368e3dc 100644 --- a/src/zennomad/include/zennomad/nomadprovisioner.h +++ b/src/zennomad/include/zennomad/nomadprovisioner.h @@ -47,7 +47,11 @@ public: /** Construct a provisioner. * @param Config Nomad connection and job configuration. * @param OrchestratorEndpoint URL of the orchestrator that remote workers announce to. */ - NomadProvisioner(const NomadConfig& Config, std::string_view OrchestratorEndpoint); + NomadProvisioner(const NomadConfig& Config, + std::string_view OrchestratorEndpoint, + std::string_view CoordinatorSession = {}, + bool CleanStart = false, + std::string_view TraceHost = {}); /** Signals the management thread to exit and stops all tracked jobs. */ ~NomadProvisioner(); @@ -83,6 +87,9 @@ private: NomadConfig m_Config; std::string m_OrchestratorEndpoint; + std::string m_CoordinatorSession; + bool m_CleanStart = false; + std::string m_TraceHost; std::unique_ptr<NomadClient> m_Client; |