ds8-solution-pro/ds-wms-service/DS.Module.Core/Data/TaskFlowDataContext.cs

namespace DS.Module.Core.Data
{
    /// <summary>
    /// 任务模块之间用于传入、获取数据的容器
    /// </summary>
    public class TaskFlowDataContext
    {
        private readonly Dictionary<string, object?> dataContext = new();

        /// <summary>
        /// 
        /// </summary>
        public TaskFlowDataContext(string key, object? value)
        {
            dataContext.Add(key, value);
        }

        /// <summary>
        /// 
        /// </summary>
        public TaskFlowDataContext(params (string key, object? value)[] keyValues)
        {
            foreach (var item in keyValues)
            {
                dataContext.Add(item.key, item.value);
            }
        }

        /// <summary>
        /// 
        /// </summary>
        public void Set(string key, object? value)
        {
            if (ContainsKey(key))
            {
                dataContext[key] = value;
            }
            else
            {
                dataContext.Add(key, value);
            }
        }
        /// <summary>
        /// 
        /// </summary>
        public T? Get<T>(string key)
        {
            if (dataContext.ContainsKey(key))
            {
                var value = dataContext[key];
                if (value is T t)
                {
                    return t;
                }
            }
            return default;
        }

        /// <summary>
        /// 
        /// </summary>
        public bool ContainsKey(string key) => dataContext.ContainsKey(key);
    }
}