chore: import upstream snapshot with attribution

2026-07-13 13:02:50 +08:00
commit 0fc60fdcb1
5008 changed files with 910633 additions and 0 deletions
@@ -0,0 +1,118 @@
+/*
+ * QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
+ * Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NodaTime;
+using QuantConnect.Interfaces;
+
+namespace QuantConnect
+{
+    /// <summary>
+    /// Provides a means of centralizing time for various time zones.
+    /// </summary>
+    public class TimeKeeper : ITimeKeeper
+    {
+        private DateTime _utcDateTime;
+
+        private readonly Dictionary<string, LocalTimeKeeper> _localTimeKeepers;
+
+        /// <summary>
+        /// Gets the current time in UTC
+        /// </summary>
+        public DateTime UtcTime
+        {
+            get { return _utcDateTime; }
+        }
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="TimeKeeper"/> class at the specified
+        /// UTC time and for the specified time zones. Each time zone specified will cause the
+        /// creation of a <see cref="LocalTimeKeeper"/> to handle conversions for that time zone.
+        /// </summary>
+        /// <param name="utcDateTime">The initial time</param>
+        /// <param name="timeZones">The time zones used to instantiate <see cref="LocalTimeKeeper"/> instances.</param>
+        public TimeKeeper(DateTime utcDateTime, params DateTimeZone[] timeZones)
+            : this(utcDateTime, timeZones ?? Enumerable.Empty<DateTimeZone>())
+        {
+        }
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="TimeKeeper"/> class at the specified
+        /// UTC time and for the specified time zones. Each time zone specified will cause the
+        /// creation of a <see cref="LocalTimeKeeper"/> to handle conversions for that time zone.
+        /// </summary>
+        /// <param name="utcDateTime">The initial time</param>
+        /// <param name="timeZones">The time zones used to instantiate <see cref="LocalTimeKeeper"/> instances.</param>
+        public TimeKeeper(DateTime utcDateTime, IEnumerable<DateTimeZone> timeZones)
+        {
+            _utcDateTime = utcDateTime;
+            _localTimeKeepers = timeZones.Distinct().Select(x => new LocalTimeKeeper(utcDateTime, x)).ToDictionary(x => x.TimeZone.Id);
+        }
+
+        /// <summary>
+        /// Sets the current UTC time for this time keeper and the attached child <see cref="LocalTimeKeeper"/> instances.
+        /// </summary>
+        /// <param name="utcDateTime">The current time in UTC</param>
+        public virtual void SetUtcDateTime(DateTime utcDateTime)
+        {
+            _utcDateTime = utcDateTime;
+            foreach (var timeZone in _localTimeKeepers)
+            {
+                timeZone.Value.UpdateTime(utcDateTime);
+            }
+        }
+
+        /// <summary>
+        /// Gets the local time in the specified time zone. If the specified <see cref="DateTimeZone"/>
+        /// has not already been added, this will throw a <see cref="KeyNotFoundException"/>.
+        /// </summary>
+        /// <param name="timeZone">The time zone to get local time for</param>
+        /// <returns>The local time in the specifed time zone</returns>
+        public DateTime GetTimeIn(DateTimeZone timeZone)
+        {
+            return GetLocalTimeKeeper(timeZone).LocalTime;
+        }
+
+        /// <summary>
+        /// Gets the <see cref="LocalTimeKeeper"/> instance for the specified time zone
+        /// </summary>
+        /// <param name="timeZone">The time zone whose <see cref="LocalTimeKeeper"/> we seek</param>
+        /// <returns>The <see cref="LocalTimeKeeper"/> instance for the specified time zone</returns>
+        public LocalTimeKeeper GetLocalTimeKeeper(DateTimeZone timeZone)
+        {
+            LocalTimeKeeper localTimeKeeper;
+            if (!_localTimeKeepers.TryGetValue(timeZone.Id, out localTimeKeeper))
+            {
+                localTimeKeeper = new LocalTimeKeeper(UtcTime, timeZone);
+                _localTimeKeepers[timeZone.Id] = localTimeKeeper;
+            }
+            return localTimeKeeper;
+        }
+
+        /// <summary>
+        /// Adds the specified time zone to this time keeper
+        /// </summary>
+        /// <param name="timeZone"></param>
+        public void AddTimeZone(DateTimeZone timeZone)
+        {
+            if (!_localTimeKeepers.ContainsKey(timeZone.Id))
+            {
+                _localTimeKeepers[timeZone.Id] = new LocalTimeKeeper(_utcDateTime, timeZone);
+            }
+        }
+    }
+}