其中用到了遊標(cursor),循環(loop),動態SQL預處理(prepare)等技術,特此記錄一下。
CREATE PROCEDURE statis_rows(in v_schema varchar(50))
BEGIN
DECLARE sql_str VARCHAR (200);
declare no_more_departments integer DEFAULT 0;
DECLARE sql_cur CURSOR FOR (
SELECT
CONCAT(
'select ',"'",table_schema,"'",",'",table_name,"',",'count(1)',' into @v_tab_schema,@v_tab_name,@v_count from ',
TABLE_SCHEMA,
'.',
TABLE_NAME,
''
)
FROM
information_schema.TABLES
WHERE table_schema = v_schema
and table_name <> 'table_rows'
);
DECLARE CONTINUE HANDLER FOR NOT FOUND SET no_more_departments=1;
/* 打開遊標,進入循環統計各表的記錄 */
OPEN sql_cur;
myLoop:LOOP
fetch sql_cur into sql_str;
if no_more_departments = 1 THEN
leave myLoop;
end if;
set @v_sql=sql_str;
prepare stmt from @v_sql;
EXECUTE stmt;
/* 將統計結果保存到表中,以便隨時都可以查看 */
INSERT INTO table_rows
VALUES
(
@v_tab_schema,
@v_tab_name,
@v_count,
sysdate()
);
DEALLOCATE prepare stmt;
end loop myLoop;
close sql_cur;
END
最後測試一下
call statis_rows('mysql');
call statis_rows('scott');
select * from table_rows;