編寫儲存過程將異常節點移動到新建的一個節點下
阿新 • • 發佈:2018-12-24
預期效果
實現程式碼
-- 在處理之前,新節點的相關資訊已經插入到表中了 -- v_normal_node_res_id: 唯一正常的節點的res_id -- v_new_node_res_id: 新建的節點的res_id -- 功能:將異常節點及其子節點移動到新建的一個節點,作為其子節點 -- 注意:需要獲取新節點的級別,比如說為3 --則需要做的操作如下: /* (1)找到需要更新的記錄 a.排除正常節點及其子節點 特點:res_tree_level3_id=v_normal_node_res_id b.排除新節點 特點:res_tree_level3_id=v_new_node_res_id c.排除一二級節點 特點:res_tree_level3_id=0 (2)更新記錄:需要更新哪些欄位(以下是針對需要更新的節點) a.對於與新節點同級別的節點,需要將其parent_res_id設定為新節點的res_id b.對於所有節點,需要更新 設定res_level:= res_level + 1; res_tree_level3_id := 新節點的res_id res_tree_level4_id :=res_tree_level3_id res_tree_level5_id :=res_tree_level4_id 。。。 res_tree_level12_id:=res_tree_level1_id 直到該節點的res_level,都要進行更新 實際的寫程式碼的時候,需要逆序複製,否則值會被覆蓋 */ create or replace function func_move_abnormal_node_to_new_node(v_normal_node_res_id imos_id, v_new_node_res_id imos_id) returns integer as $BODY$ declare v_rec_tbl_res RECORD; -- 儲存當前節點資訊 v_current_node_res_id imos_id; -- 當前節點res_id v_current_node_res_level int32; -- 當前節點res_level v_tmp_counter int:=0; -- 計數器 v_str_sql_cmd text; -- 暫存動態SQL v_str_cmd text; -- 暫存動態SQL v_int_level_of_new_node int:=0; -- 新節點的級別,也等於正常節點的級別 v_rec record; begin select res_level into v_int_level_of_new_node from tbl_res where res_id=v_new_node_res_id; -- 獲取新節點的級別res_level==正常節點的級別 v_str_cmd = 'select res_id, res_level from tbl_res '; v_str_cmd = v_str_cmd || ' where res_tree_level'||v_int_level_of_new_node||'_id <> ' || v_normal_node_res_id ; --去除正常節點及其子節點 v_str_cmd = v_str_cmd || ' and res_tree_level' ||v_int_level_of_new_node||'_id <> 0 '; --去除一級、二..v_int_level_of_new_node-1級節點 v_str_cmd = v_str_cmd || ' and res_tree_level' ||v_int_level_of_new_node||'_id <> ' || v_new_node_res_id ; --去除新上任的v_int_level_of_new_node級節點 --修改parent_res_id for v_rec in (select res_id from tbl_res where res_level=v_int_level_of_new_node and res_id<>v_new_node_res_id and res_id <> v_normal_node_res_id ) loop update tbl_res set parent_res_id=v_new_node_res_id where res_id=v_rec.res_id; end loop; -- 獲取哪些記錄需要進行處理 for v_rec_tbl_res in execute v_str_cmd loop -- 獲取當前節點res_id v_current_node_res_id=v_rec_tbl_res.res_id; -- 更新新節點同級 & 異常節點的parent_res_id為v_new_node_res_id -- 更新當前節點res_tree_level[v_int_level_of_new_node]~res_tree_level[v_rec_tbl_res.res_level+1]以及res_level v_tmp_counter=v_rec_tbl_res.res_level+1; v_str_sql_cmd='update tbl_res set '; while v_tmp_counter > v_int_level_of_new_node loop v_str_sql_cmd = v_str_sql_cmd ||'res_tree_level'|| v_tmp_counter || '_id = res_tree_level' || (v_tmp_counter-1) || '_id,' ; v_tmp_counter = v_tmp_counter -1; end loop; v_str_sql_cmd = v_str_sql_cmd || 'res_level=res_level+1, res_tree_level'||v_int_level_of_new_node||'_id='||v_new_node_res_id; v_str_sql_cmd = v_str_sql_cmd || ' where res_id = '|| v_current_node_res_id|| '; '; execute v_str_sql_cmd; end loop; return 0; end; $BODY$ language plpgsql ; -- select * from tbl_res; -- select * from func_move_abnormal_node_to_new_node(10023,10024);