Add handling for custom host name (QuantConnect#5043)

* Add handling for custom host name * Credit cost as decimal
ebengtso · Dec 17, 2020 · 6d824b4 · 6d824b4
1 parent 8e410fc
commit 6d824b4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/Common/Packets/AlgorithmNodePacket.cs b/Common/Packets/AlgorithmNodePacket.cs
@@ -33,6 +33,12 @@ public AlgorithmNodePacket(PacketType type)
             : base(type)
         { }
 
+        /// <summary>
+        /// The host name to use if any
+        /// </summary>
+        [JsonProperty(PropertyName = "sHostName")]
+        public string HostName;
+
         /// <summary>
         /// User Id placing request
         /// </summary>

diff --git a/Common/Packets/Controls.cs b/Common/Packets/Controls.cs
@@ -146,8 +146,8 @@ public class Controls
         /// <summary>
         /// The cost associated with running this job
         /// </summary>
-        [JsonProperty(PropertyName = "iCreditCost")]
-        public uint CreditCost;
+        [JsonProperty(PropertyName = "dCreditCost")]
+        public decimal CreditCost;
 
         /// <summary>
         /// Initializes a new default instance of the <see cref="Controls"/> class

diff --git a/Engine/Results/BaseResultsHandler.cs b/Engine/Results/BaseResultsHandler.cs
@@ -42,6 +42,8 @@ public abstract class BaseResultsHandler
         private static readonly TextWriter StandardOut = Console.Out;
         private static readonly TextWriter StandardError = Console.Error;
 
+        private string _hostName;
+
         /// <summary>
         /// The main loop update interval
         /// </summary>
@@ -232,6 +234,7 @@ public virtual void Exit()
         protected virtual Dictionary<string, string> GetServerStatistics(DateTime utcNow)
         {
             var serverStatistics = OS.GetServerStatistics();
+            serverStatistics["Hostname"] = _hostName;
             var upTime = utcNow - StartTime;
             serverStatistics["Up Time"] = $"{upTime.Days}d {upTime:hh\\:mm\\:ss}";
             serverStatistics["Total RAM (MB)"] = RamAllocation;
@@ -305,6 +308,7 @@ protected virtual Dictionary<int, Order> GetDeltaOrders(int orderEventsStartPosi
         /// <param name="transactionHandler">The transaction handler used to get the algorithms <see cref="Order"/> information</param>
         public virtual void Initialize(AlgorithmNodePacket job, IMessagingHandler messagingHandler, IApi api, ITransactionHandler transactionHandler)
         {
+            _hostName = job.HostName ?? Environment.MachineName;
             MessagingHandler = messagingHandler;
             TransactionHandler = transactionHandler;
             CompileId = job.CompileId;