MySQL如何實現(xiàn)查詢樹結(jié)構(gòu)

這篇文章給大家分享的是有關(guān)MySQL如何實現(xiàn)查詢樹結(jié)構(gòu)的內(nèi)容。小編覺得挺實用的，因此分享給大家做個參考，一起跟隨小編過來看看吧。

MySQL 查詢樹結(jié)構(gòu)

1. 關(guān)于樹結(jié)構(gòu)

此類結(jié)構(gòu)的數(shù)據(jù)，通常需要表結(jié)構(gòu)中含有id 、parentId等自關(guān)聯(lián)字段，有時為了提高查詢效率還可增加更多冗余字段，如index，index的值為所有父級目錄的id字符串集合。

關(guān)于樹結(jié)構(gòu)數(shù)據(jù)的組裝，常見的寫法是在程序中通過遞歸的方式去構(gòu)建出一顆完整的樹，單純通過sql的方式其實并不常用，下面分別給出兩種方式的例子。

2. MySQL自定義函數(shù)的方式

什么是MySQL自定義函數(shù)：聚合函數(shù)，日期函數(shù)之類的都是MySQL的函數(shù)，此處我們定義的函數(shù)可同他們一樣使用，不過只能在定義的數(shù)據(jù)庫中使用，自定義函數(shù)和存儲過程類似，不同的是，函數(shù)只會返回一個值，不允許返回一個結(jié)果集。

2.1 創(chuàng)建測試數(shù)據(jù)

CREATE TABLE `tree`  (
  `id` bigint(11) NOT NULL,
  `pid` bigint(11) NULL DEFAULT NULL,
  `name` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,
  PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;
INSERT INTO `tree` VALUES (1, 0, '中國');
INSERT INTO `tree` VALUES (2, 1, '四川省');
INSERT INTO `tree` VALUES (3, 2, '成都市');
INSERT INTO `tree` VALUES (4, 3, '武侯區(qū)');
INSERT INTO `tree` VALUES (5, 4, '紅牌樓');
INSERT INTO `tree` VALUES (6, 1, '廣東省');
INSERT INTO `tree` VALUES (7, 1, '浙江省');
INSERT INTO `tree` VALUES (8, 6, '廣州市');

2.2 獲取 某節(jié)點下所有子節(jié)點

CREATE FUNCTION `GET_CHILD_NODE`(rootId varchar(100))   
RETURNS varchar(2000)  
BEGIN   
DECLARE str varchar(2000);  
DECLARE cid varchar(100);   
SET str = '$';   
SET cid = rootId;   
WHILE cid is not null DO   
    SET str = concat(str, ',', cid);   
    SELECT group_concat(id) INTO cid FROM tree where FIND_IN_SET(pid, cid);   
END WHILE;   
RETURN str;   
END

調(diào)用自定義函數(shù)

select * from tree where FIND_IN_SET(id, GET_CHILD_NODE(2));

2.3 獲取 某節(jié)點的所有父節(jié)點

CREATE FUNCTION `GET_PARENT_NODE`(rootId varchar(100))   
RETURNS varchar(1000)   
BEGIN   
DECLARE fid varchar(100) default '';   
DECLARE str varchar(1000) default rootId;   
  
WHILE rootId is not null do   
    SET fid =(SELECT pid FROM tree WHERE id = rootId);   
    IF fid is not null THEN   
        SET str = concat(str, ',', fid);   
        SET rootId = fid;   
    ELSE   
        SET rootId = fid;   
    END IF;   
END WHILE;   
return str;  
END

調(diào)用自定義函數(shù)

select * from tree where FIND_IN_SET(id, GET_PARENT_NODE(5));

3. Oracle數(shù)據(jù)庫的方式

只需要使用start with connect by prior語句即可完成遞歸的樹查詢，詳情請自己查閱相關(guān)資料。

4. 程序代碼遞歸的方式構(gòu)建樹

這里我就不給出完整代碼了，遞歸的方式很簡單，就是先查出所有樹節(jié)點，然后通過一個TreeNode類中的add方法遞歸把所有子節(jié)點給加進來。核心代碼如下：

public class TreeNodeDTO {
    
    private String id;
    private String parentId;
    private String name;
    private List<TreeNodeDTO> children = new ArrayList<>();
    public void add(TreeNodeDTO node) {
        if ("0".equals(node.parentId)) {
            this.children.add(node);
        } else if (node.parentId.equals(this.id)) {
            this.children.add(node);
        } else {
         	//遞歸調(diào)用add()添加子節(jié)點
            for (TreeNodeDTO tmp_node : children) {
                tmp_node.add(node);
            }
        }
    }
 }

5. 通過hashMap，只需要遍歷一次

就可以完成樹的生成：五星推薦

List<TreeNodeDTO> list = dbMapper.getNodeList();
ArrayList<TreeNodeDTO> rootNodes = new ArrayList<>();
Map<Integer, TreeNodeDTO> map = new HashMap<>();
for (TreeNodeDTO node :list) {
    map.put(node.getId(), node);
    Integer parentId = node.getParentId();
    // 判斷是否有父節(jié)點 (沒有父節(jié)點本身就是個父菜單)
    if (parentId.equals('0')){
        rootNodes.add(node);
        // 找出不是父級菜單的且集合中包括其父菜單ID
    } else if (map.containsKey(parentId)){
        map.get(parentId).getChildren().add(node);
    }
}

MySQL 查詢帶樹狀結(jié)構(gòu)的信息

在Oracle中有函數(shù)應(yīng)用直接能夠查詢出樹狀的樹狀結(jié)構(gòu)信息，例如有下面樹狀結(jié)構(gòu)的組織成員架構(gòu)，那么如果我們想查其中一個節(jié)點下的所有節(jié)點信息

在Oracle中可以直接用下面的語法可以進行直接查詢

START WITH CONNECT BY PRIOR

但是在Mysql中是沒有這個語法的

而如果你也是想要查詢這樣的數(shù)據(jù)結(jié)構(gòu)信息該怎么做呢？我們可以自定義函數(shù)。我們將上面的信息初始化信息進數(shù)據(jù)庫中。首先先創(chuàng)建一張表用于存儲這些信息，ID為存儲自身的ID信息，PARENT_ID存儲父ID信息

CREATE TABLE `company_inf` (
  `ID` varchar(32) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
  `NAME` varchar(255) COLLATE utf8mb4_unicode_ci DEFAULT NULL,
  `PARENT_ID` varchar(32) COLLATE utf8mb4_unicode_ci DEFAULT NULL
)

然后將圖中的信息初始化表中

INSERT INTO company_inf VALUES ('1','總經(jīng)理王大麻子','1');
INSERT INTO company_inf VALUES ('2','研發(fā)部經(jīng)理劉大瘸子','1');
INSERT INTO company_inf VALUES ('3','銷售部經(jīng)理馬二愣子','1');
INSERT INTO company_inf VALUES ('4','財務(wù)部經(jīng)理趙三駝子','1');
INSERT INTO company_inf VALUES ('5','秘書員工J','1');
INSERT INTO company_inf VALUES ('6','研發(fā)一組組長吳大棒槌','2');
INSERT INTO company_inf VALUES ('7','研發(fā)二組組長鄭老六','2');
INSERT INTO company_inf VALUES ('8','銷售人員G','3');
INSERT INTO company_inf VALUES ('9','銷售人員H','3');
INSERT INTO company_inf VALUES ('10','財務(wù)人員I','4');
INSERT INTO company_inf VALUES ('11','開發(fā)人員A','6');
INSERT INTO company_inf VALUES ('12','開發(fā)人員B','6');
INSERT INTO company_inf VALUES ('13','開發(fā)人員C','6');
INSERT INTO company_inf VALUES ('14','開發(fā)人員D','7');
INSERT INTO company_inf VALUES ('15','開發(fā)人員E','7');
INSERT INTO company_inf VALUES ('16','開發(fā)人員F','7');

例如我們想要查詢研發(fā)部門經(jīng)理劉大瘸子下的所有員工，在Oracle中我們可以這樣寫

  SELECT *
  FROM T_PORTAL_AUTHORITY
  START WITH ID='1'
  CONNECT BY PRIOR ID = PARENT_ID

而在Mysql中我們需要下面這樣自定義函數(shù)

CREATE FUNCTION getChild(parentId VARCHAR(1000))
RETURNS VARCHAR(1000)
BEGIN
    DECLARE oTemp VARCHAR(1000);
    DECLARE oTempChild VARCHAR(1000);
    SET oTemp = '';
    SET oTempChild =parentId;
    WHILE oTempChild is not null DO
        IF oTemp != '' THEN
            SET oTemp = concat(oTemp,',',oTempChild);
        ELSE
            SET oTemp = oTempChild;
        END IF;
        SELECT group_concat(ID) INTO oTempChild FROM company_inf where parentId<>ID and FIND_IN_SET(parent_id,oTempChild)>0;
    END WHILE;
RETURN oTemp;
END

然后這樣查詢即可

SELECT * FROM company_inf WHERE FIND_IN_SET(ID,getChild('2'));

此時查看查詢出來的信息就是劉大瘸子下所有的員工信息了

感謝各位的閱讀！關(guān)于“MySQL如何實現(xiàn)查詢樹結(jié)構(gòu)”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，讓大家可以學(xué)到更多知識，如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！