节点命令使用状态字段,区分命令的不同状态
大石头 authored at 2021-04-21 00:25:23
4.39 KiB
Stardust
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Runtime.Serialization;
using System.Xml.Serialization;
using NewLife;
using NewLife.Data;
using Stardust.Models;
using XCode;
using XCode.Membership;

namespace Stardust.Data.Nodes
{
    /// <summary>命令状态</summary>
    public enum CommandStatus
    {
        /// <summary>就绪</summary>
        就绪 = 0,

        /// <summary>处理中</summary>
        处理中 = 1,

        /// <summary>已完成</summary>
        已完成 = 2,

        /// <summary>取消</summary>
        取消 = 3,
    }

    /// <summary>节点命令</summary>
    public partial class NodeCommand : Entity<NodeCommand>
    {
        #region 对象操作
        static NodeCommand()
        {
            // 累加字段
            //var df = Meta.Factory.AdditionalFields;
            //df.Add(__.NodeID);

            // 过滤器 UserModule、TimeModule、IPModule
            Meta.Modules.Add<UserModule>();
            Meta.Modules.Add<TimeModule>();
            Meta.Modules.Add<IPModule>();
        }

        /// <summary>验证数据,通过抛出异常的方式提示验证失败。</summary>
        /// <param name="isNew">是否插入</param>
        public override void Valid(Boolean isNew)
        {
            // 如果没有脏数据,则不需要进行任何处理
            if (!HasDirty) return;
        }
        #endregion

        #region 扩展属性
        /// <summary>节点</summary>
        [XmlIgnore, IgnoreDataMember]
        //[ScriptIgnore]
        public Node Node => Extends.Get(nameof(Node), k => Node.FindByID(NodeID));

        /// <summary>节点</summary>
        [XmlIgnore, IgnoreDataMember]
        //[ScriptIgnore]
        [DisplayName("节点")]
        [Map(__.NodeID, typeof(Node), "ID")]
        public String NodeName => Node?.Name;
        #endregion

        #region 扩展查询
        /// <summary>根据编号查找</summary>
        /// <param name="id">编号</param>
        /// <returns>实体对象</returns>
        public static NodeCommand FindByID(Int32 id)
        {
            if (id <= 0) return null;

            // 实体缓存
            if (Meta.Session.Count < 1000) return Meta.Cache.Find(e => e.ID == id);

            // 单对象缓存
            return Meta.SingleCache[id];

            //return Find(_.ID == id);
        }
        #endregion

        #region 高级查询
        /// <summary>高级搜索</summary>
        /// <param name="nodeId"></param>
        /// <param name="command"></param>
        /// <param name="start"></param>
        /// <param name="end"></param>
        /// <param name="key"></param>
        /// <param name="page"></param>
        /// <returns></returns>
        public static IList<NodeCommand> Search(Int32 nodeId, String command, DateTime start, DateTime end, String key, PageParameter page)
        {
            var exp = new WhereExpression();

            if (nodeId > 0) exp &= _.NodeID == nodeId;
            if (!command.IsNullOrEmpty()) exp &= _.Command == command;

            exp &= _.UpdateTime.Between(start, end);

            if (!key.IsNullOrEmpty()) exp &= _.Command == key;

            return FindAll(exp, page);
        }
        #endregion

        #region 业务操作
        /// <summary>获取有效命令</summary>
        /// <param name="nodeId"></param>
        /// <param name="count"></param>
        /// <returns></returns>
        public static IList<NodeCommand> AcquireCommands(Int32 nodeId, Int32 count = 100) => FindAll(_.NodeID == nodeId & _.Status <= CommandStatus.处理中, _.ID.Asc(), null, 0, count);

        /// <summary>添加节点命令</summary>
        /// <param name="node"></param>
        /// <param name="command"></param>
        /// <returns></returns>
        public static NodeCommand Add(Node node, String command)
        {
            var cmd = new NodeCommand
            {
                NodeID = node.ID,
                Command = command,
                CreateTime = DateTime.Now,
            };

            cmd.Insert();

            return cmd;
        }

        /// <summary>转为模型</summary>
        /// <returns></returns>
        public CommandModel ToModel()
        {
            return new CommandModel
            {
                Id = ID,
                Command = Command,
                Argument = Argument,
                Expire = Expire,
            };
        }
        #endregion
    }
}