chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,86 @@
|
||||
/*
|
||||
* 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.Runtime.CompilerServices;
|
||||
|
||||
namespace QuantConnect.Lean.Engine.DataFeeds.WorkScheduling
|
||||
{
|
||||
/// <summary>
|
||||
/// Class to represent a work item
|
||||
/// </summary>
|
||||
public class WorkItem
|
||||
{
|
||||
/// <summary>
|
||||
/// Function to determine weight of item
|
||||
/// </summary>
|
||||
private Func<int> _weightFunc;
|
||||
|
||||
/// <summary>
|
||||
/// The current weight
|
||||
/// </summary>
|
||||
public int Weight { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// The work function to execute
|
||||
/// </summary>
|
||||
public Func<int, bool> Work { get; }
|
||||
|
||||
/// <summary>
|
||||
/// Creates a new instance
|
||||
/// </summary>
|
||||
/// <param name="work">The work function, takes an int, the amount of work to do
|
||||
/// and returns a bool, false if this work item is finished</param>
|
||||
/// <param name="weightFunc">The function used to determine the current weight</param>
|
||||
public WorkItem(Func<int, bool> work, Func<int> weightFunc)
|
||||
{
|
||||
Work = work;
|
||||
_weightFunc = weightFunc;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Updates the weight of this work item
|
||||
/// </summary>
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public int UpdateWeight()
|
||||
{
|
||||
Weight = _weightFunc();
|
||||
return Weight;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Compares two work items based on their weights
|
||||
/// </summary>
|
||||
public static int Compare(WorkItem obj, WorkItem other)
|
||||
{
|
||||
if (ReferenceEquals(obj, other))
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
// By definition, any object compares greater than null
|
||||
if (ReferenceEquals(obj, null))
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
if (ReferenceEquals(null, other))
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
|
||||
return other.Weight.CompareTo(obj.Weight);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user