127.0.
智能大石头 authored at 2024-11-20 00:19:29
4.20 KiB
Stardust
using System.Web;
using Microsoft.AspNetCore.Mvc;
using NewLife;
using NewLife.Cube;
using NewLife.Cube.ViewModels;
using NewLife.Data;
using NewLife.Log;
using NewLife.Web;
using Stardust.Data.Nodes;
using Stardust.Server.Services;
using XCode.Membership;

namespace Stardust.Web.Areas.Redis.Controllers;

[Menu(50)]
[RedisArea]
public class RedisNodeController : EntityController<RedisNode>
{
    private readonly IRedisService _redisService;

    static RedisNodeController()
    {
        LogOnChange = true;

        ListFields.RemoveField("WebHook", "AlarmMemoryRate", "AlarmConnections", "AlarmSpeed", "AlarmInputKbps", "AlarmOutputKbps");
        ListFields.RemoveCreateField();
        ListFields.RemoveField("UpdateUser", "UpdateUserID", "UpdateIP", "Remark");

        {
            var df = ListFields.GetField("ProjectName") as ListField;
            df.Url = "/Platform/GalaxyProject?projectId={ProjectId}";
            df.Target = "_frame";
        }
        {
            var df = ListFields.AddListField("Monitor", "UpdateTime");
            df.DisplayName = "监控";
            df.Header = "监控";
            df.Url = "/Redis/RedisData?redisId={Id}";
        }
        {
            var df = ListFields.AddListField("Queue", "UpdateTime");
            df.DisplayName = "队列";
            df.Header = "队列";
            df.Url = "/Redis/RedisMessageQueue?redisId={Id}";
        }
        {
            var df = ListFields.AddListField("Refresh", "UpdateTime");
            df.DisplayName = "刷新";
            df.Header = "刷新";
            df.Url = "/Redis/RedisNode/Refresh?Id={Id}";
            df.DataAction = "action";
        }
        {
            var df = ListFields.AddListField("Log", "UpdateTime");
            df.DisplayName = "审计日志";
            df.Header = "审计日志";
            df.Url = $"/Admin/Log?category={HttpUtility.UrlEncode("Redis节点")}&linkId={{Id}}";
            df.Target = "_frame";
        }
    }

    public RedisNodeController(IRedisService redisService) => _redisService = redisService;

    protected override IEnumerable<RedisNode> Search(Pager p)
    {
        var nodeId = p["Id"].ToInt(-1);
        if (nodeId > 0)
        {
            var node = RedisNode.FindById(nodeId);
            if (node != null) return new[] { node };
        }

        var category = p["category"];
        var server = p["server"];
        var enable = p["enable"]?.ToBoolean();

        var start = p["dtStart"].ToDateTime();
        var end = p["dtEnd"].ToDateTime();

        return RedisNode.Search(server, category, enable, start, end, p["Q"], p);
    }

    /// <summary>搜索</summary>
    /// <param name="provinceId"></param>
    /// <param name="cityId"></param>
    /// <param name="key"></param>
    /// <returns></returns>
    public ActionResult NodeSearch(String category, String key = null)
    {
        var page = new PageParameter { PageSize = 20 };

        // 默认排序
        if (page.Sort.IsNullOrEmpty()) page.Sort = RedisNode._.Name;

        var list = RedisNode.Search(null, category, true, DateTime.MinValue, DateTime.MinValue, key, page);

        return Json(0, null, list.Select(e => new
        {
            e.Id,
            e.Name,
            e.Server,
            e.Category,
        }).ToArray());
    }

    [EntityAuthorize(PermissionFlags.Update)]
    public ActionResult Refresh(Int32 id)
    {
        var node = RedisNode.FindById(id);
        if (node != null)
        {
            XTrace.WriteLine("刷新 {0}/{1} {2}", node.Name, node.Id, node.Server);

            try
            {
                _redisService.TraceNode(node);

                var queues = RedisMessageQueue.FindAllByRedisId(node.Id);
                foreach (var item in queues.Where(e => e.Enable))
                {
                    _redisService.TraceQueue(item);
                    item.Update();
                }

                LogProvider.Provider.WriteLog("RedisNode", "Refresh", true, $"刷新Redis节点[{node}]成功");
            }
            catch (Exception ex)
            {
                LogProvider.Provider.WriteLog("RedisNode", "Refresh", false, ex?.GetMessage());

                throw;
            }
        }

        return JsonRefresh($"刷新[{node}]成功!");
    }
}