方正证券量化交易接口-留个档 Unity 一个简易的红点树功能实现

---root: node_r1 node_r2 --- ↗ ↖ ↗ ↖ ---level1: node_l1_1 node_l1_2 node_l1_3 --- ↗ ↖ ---level2: node_l2_1 node_l2_2 --- ↗ ↖ ---level3: node_l3_1 node_l3_2

例命名规则：node_l_序号

只有叶子节点可以被设置具体数值，向上的树全部进行计算

中node_l3_1,node_l3_2,node_l2_2,node_l1_1,node_l1_3可以被设置数值

某个节点被设置数值时序： node_l3_1通知回调 node_l2_1计算node_l3_1和node_l3_2累加 node_l2_1通知回调 node_l1_2计算node_l2_1和node_l2_2累加 node_l1_2通知回调 node_r1计算node_l1_1和node_l1_2累加 node_r1通知回调 node_r2计算node_l1_3和node_l1_2累加 node_r2通知回调

支持x型树。例如中心节点node_l1_ 因为最外层展示可能会有多个模块需求，比如道具修改对应红点是在活动，任务，背包多个Root节点

不支持菱形结构。例如在中node_l3_2同时指向node_l2_ 这个需求可能是合理的，但是实际应用场景非常少菱形结构在递归中，可能会由于人为原因导致死循环，排查bug困难或者导致逻辑复杂度高可以新增一个节点，同时增加计数来解决菱形结构的问题

构建关系树，应该是静态的，固定的。这是设计层面的事情，不应该出现动态修改关系树的行为。动态创建单独的无Link的节点，是被允许的。public接口都是CreateorLoad

监听变化：AddListener和RemoveListener需要按照生命周期成对出现，有Add必有Remove

    local pCallBack = Bind(self, self.FunctionCallBack)  // 注：Lua里需要Bind
    function xxx:FunctionCallBack()
        ...
    end
    self:AddListener(RedDotKey.Activity, pCallBack)
    self:RemoveListener(RedDotKey.Activity, pCallBack)   // 注：lua内部实现C# delegate使用的是table<function>遍历对比

具体代码：

---@class RedDotKey    每100个跨度 为一个功能模块
local RedDotKey =
{
    Activity = 100,
    Activity_SevenDay = 101,
}

return ConstClass("RedDotKey", RedDotKey)


---@class RedDotNode
---@field eKey          RedDotKey      唯一ID 键
---@field nRef          number      计数，每一层都需要记录，可能是相邻节点变化，本节点不需要重新计算自己的子节点
---@field callBack      function<RedDotKey, number>[]  计数变更的回调
---@field parent        RedDotNode[]   父节点 可能会有多个
---@field child         RedDotNode[]   子节点 可能会有多个

---@class RedDotManager : Singleton
---@field map   table<RedDotKey, RedDotNode> 键值对hash
local RedDotManager = BaseClass("RedDotManager", Singleton)

function RedDotManager:__init()
    self.map = {}

    self:Link(RedDotKey.Activity, RedDotKey.Activity_SevenDay)
end

---单个节点--------------------------------------------------------------------------------------------------------------

---添加加入一个注册
---@public
---@param eKey      RedDotKey   唯一ID
---@param callBack  function<RedDotKey, number>        回调
function RedDotManager:AddListener(eKey, callBack)
    ---@type RedDotNode
    local node = self:CreateOrGetNode(eKey)
    local result = table.indexof(node.callBack, callBack)
    if result == false then
        ---没有找到 相同的注册, 正常加入 callback list
        table.insert(node.callBack, callBack)
    else
        ---找到了相同的callback
        Debug.Log("!!! RedDotManager:AddNode, add same callback : "..tostring(eKey))
    end
end

---移除一个注册
---@public
---@param eKey      RedDotKey   唯一ID
---@param callBack  function        回调
function RedDotManager:RemoveListener(eKey, callBack)
    ---@type RedDotNode
    local node = self.map[eKey]
    if node == nil then
        return
    end
    local result = table.indexof(node.callBack, callBack)
    if result ~= false then
        table.remove(node.callBack, result)
    else
        ---没有找到这个callback
        Debug.Log("!!! RedDotManager:Remove, add same callback : "..tostring(eKey))
    end
end

---创建或者获取一个节点
---@private
---@param eKey RedDotKey
function RedDotManager:CreateOrGetNode(eKey)
    ---@type RedDotNode
    local node = self.map[eKey]
    if node == nil then
        self.map[eKey] = {}
        node = self.map[eKey]
        ---初始化
        node.eKey = eKey
        node.nRef = 0
        node.callBack = {}
        node.parent = {}
        node.child = {}
    end
    return node
end

---父子节点关系--------------------------------------------------------------------------------------------------------------

---建立父子关系
---@public
---@param eKeyChild     RedDotKey 唯一ID
---@param eKeyParent    RedDotKey 唯一ID
function RedDotManager:Link(eKeyChild, eKeyParent)
    ---@type RedDotNode
    local nodeChild = self:CreateOrGetNode(eKeyChild)
    ---@type RedDotNode
    local nodeParent = self:CreateOrGetNode(eKeyParent)

    if self:CheckRepeat(nodeChild, nodeParent.child, true) then
        Debug.Log("!!! RedDotManager:LinkParent already exit in c "..tostring(eKeyChild).."  "..tostring(eKeyChild))
        return
    end
    if self:CheckRepeat(nodeChild, nodeParent.parent, false) then
        Debug.Log("!!! RedDotManager:LinkParent already exit in p "..tostring(eKeyChild).."  "..tostring(eKeyChild))
        return
    end
    ---之前没有连接关系，可以顺利连接
    table.insert(nodeChild.parent, nodeParent)
    table.insert(nodeParent.child, nodeChild)
end

---检查自己是不是在树里，会造成无限循环的情况
---@private
---@param node      RedDotNode 被检查的对象
---@param nodeLink  RedDotNode 当前树的某一个节点
---@param isCorP    boolean 递归向Child or parent， true=child
function RedDotManager:CheckRepeat(node, nodeLink, isCorP)
    local nCount = #nodeLink
    if nCount == 0 then
        return false
    end
    if table.indexof(nodeLink, node) ~= false then
        return true
    end
    for i = 1, nCount do
        ---@type RedDotNode
        local one = nodeLink[i]
        if self:CheckRepeat(node, isCorP and one.child or one.parent) then
            return true
        end
    end
    return false
end

---计数和通知--------------------------------------------------------------------------------------------------------------

---修改计数
---@public
---@param eKey      RedDotKey   唯一ID
---@param isReset   boolean     是否是重置
---@param nValue    number      增量 or 数值
function RedDotManager:ChangeNodeRef(eKey, isReset, nValue)
    ---@type RedDotNode
    local node = self:CreateOrGetNode(eKey)
    if not self:CheckIsLeafNode(node) then
        Debug.Log("!!! RedDotManager:ChangeNodeRef not Leaf node : "..tostring(eKey))
        return
    end
    ---缓存上一次的计数
    local nTemp = node.nRef
    if isReset then
        node.nRef = nValue
    else
        node.nRef = node.nRef + nValue
    end
    ---如果修改后计数不同，
    if nTemp ~= node.nRef then
        ---调用回调 通知修改
        self:DoCallBack(node)
        ---刷新上层
        self:RefreshParentRef(node)
    end
end

---检查自己是不是叶子节点
---@private
---@param node RedDotNode
---@return boolean 是否
function RedDotManager:CheckIsLeafNode(node)
    ---单个 无父 无子 节点，允许主动修改
    ---无子节点 被认为是 叶子节点， 可以主动修改
    return #node.child == 0
end

---刷新父节点的计数
---@private
---@param node RedDotNode
function RedDotManager:RefreshParentRef(node)
    local nCount = #node.parent
    if nCount == 0 then
        return
    end
    for i = 1,nCount do
        ---@type RedDotNode
        local one = node.parent[i]
        one.nRef = self:CollectChildRef(one)
        self:DoCallBack(one)
        ---递归向上
        self:RefreshParentRef(one)
    end
end

---累加子节点计数
---@private
---@param node RedDotNode
---@return number 子节点的计数all
function RedDotManager:CollectChildRef(node)
    local nResult = 0
    local nCount = #node.child
    for i = 1, nCount do
        ---@type RedDotNode
        local one = node.child[i]
        nResult = nResult + one.nRef
    end
    return nResult
end

---回调通知
---@private
---@param node RedDotNode
function RedDotManager:DoCallBack(node)
    ---调用回调 通知修改
    local nCount = #node.callBack
    for i = 1, nCount do
        node.callBack[i](node.eKey, node.nRef)
    end
end

程序学无止尽。欢迎大家沟通，有啥不明确的，或者不对的，也可以和我私聊我的QQ334524067神一般的狄狄

文章为作者独立观点，不代表股票程序化软件自动交易接口观点