Add organization id to packet (QuantConnect#5733)

Coulbe · Jul 2, 2021 · 12b62f6 · 12b62f6
1 parent 8d2455b
commit 12b62f6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/Common/Packets/AlgorithmNodePacket.cs b/Common/Packets/AlgorithmNodePacket.cs
@@ -1,4 +1,4 @@
-/*
+/*
  * QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
  * Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
  *
@@ -49,6 +49,10 @@ public AlgorithmNodePacket(PacketType type)
         [JsonProperty(PropertyName = "sUserToken")]
         public string UserToken = "";
 
+        /// User Organization Id
+        [JsonProperty(PropertyName = "sOrganizationID")]
+        public string OrganizationId = "";
+
         /// <summary>
         /// Project Id of the request
         /// </summary>

diff --git a/Queues/JobQueue.cs b/Queues/JobQueue.cs
@@ -39,6 +39,7 @@ public class JobQueue : IJobQueueHandler
         private const string DefaultDataChannelProvider = "DataChannelProvider";
         private bool _liveMode = Config.GetBool("live-mode");
         private static readonly string AccessToken = Config.Get("api-access-token");
+        private static readonly string OrganizationId = Config.Get("job-organization-id");
         private static readonly int UserId = Config.GetInt("job-user-id", 0);
         private static readonly int ProjectId = Config.GetInt("job-project-id", 0);
         private readonly string AlgorithmTypeName = Config.Get("algorithm-type-name");
@@ -109,6 +110,7 @@ public AlgorithmNodePacket NextJob(out string location)
                     UserToken = AccessToken,
                     UserId = UserId,
                     ProjectId = ProjectId,
+                    OrganizationId = OrganizationId,
                     Version = Globals.Version,
                     DeployId = algorithmId,
                     Parameters = parameters,
@@ -140,6 +142,7 @@ public AlgorithmNodePacket NextJob(out string location)
                 UserToken = AccessToken,
                 UserId = UserId,
                 ProjectId = ProjectId,
+                OrganizationId = OrganizationId,
                 Version = Globals.Version,
                 BacktestId = algorithmId,
                 Language = Language,