TreeViewVisitor: 一个快捷访问 TreeView 控件节点的帮助类

TreeViewVisitor: 一个快捷访问 TreeView 控件节点的帮助类

Web 开发中经常用到 TreeView 控件，一个很常见的操作就是遍历所有的节点。但是该控件本身并未提供此功能，每次都要手工写递归访问的函数，非常烦琐。所以我写了一个简单的帮助类来解决这个问题，这样就不需要每次用递归来访问了：

using System;
using System.Data;
using System.Configuration;
using System.Web;
using System.Web.Security;
using System.Web.UI;
using System.Web.UI.WebControls;
using System.Web.UI.WebControls.WebParts;
using System.Web.UI.HtmlControls;
using System.Collections.Generic;

namespace CmsNet2.Components {
    /// <summary>
    /// TreeViewVisitor by Neil Chen, 2006-09-28
    /// </summary>
    public class TreeViewVisitor {

        public TreeViewVisitor(TreeView treeView) {
            this.treeView = treeView;
        }

        private TreeView treeView;

        public TreeView TreeView {
            get { return treeView; }
            set { treeView = value; }
        }

        private void Walk(TreeNode node) {
            if (node == null) {
                nodes.Add(treeView.Nodes[0]);

                foreach (TreeNode childNode in treeView.Nodes[0].ChildNodes) {
                    Walk(childNode);
                }
            } else {
                nodes.Add(node);

                foreach (TreeNode childNode in node.ChildNodes) {
                    Walk(childNode);
                }
            }                        
        }

        private List<TreeNode> nodes;

        public IEnumerable<TreeNode> GetAllTreeNodes() {
            nodes = new List<TreeNode>();

            Walk(null);

            foreach (TreeNode node in nodes) {
                yield return node;
            }
        }
    }
}

一个简单的使用范例：遍历所有节点，并将符合条件的节点选中。

public void SetCheckedCategories(List<Category> categories) {
    // 创建访问器对象
    TreeViewVisitor visitor = new TreeViewVisitor(tvCategories);
    
    // 得到“铺平”了的所有节点集合
    IEnumerable<TreeNode> nodes = visitor.GetAllTreeNodes();

    // 进行你需要的访问操作
    foreach (TreeNode node in nodes) {
        foreach (Category category in categories) {
            if (category.CategoryId == int.Parse(node.Value)) {
                node.Checked = true;                        
            }
        }
    }           
}

当然，目前这个类的应用范围很局限，仅仅限于 TreeView 控件。但如果我们进一步思考，则可以很容易的推广到对普通的树状结构，或者多级菜单的访问逻辑中。从而实现与具体控件无关的访问逻辑。

因为业余时间有限，先到此打住。