0

我有一张包含培训记录的表格。每条记录都有一个带有代理值的字段。我有另一张只是代理价值的表格。我想将记录导出到每个机构的 CSV 文件中。有超过 200 万条记录，所以我不想导出整个表并手动执行。

我创建了一个存储过程，它使用游标从 Agency_codes 表中提取一个值，并将该值用于WHERE子句的 select 语句和INTO OUTFILE名称的一部分。

该过程有效，但仅适用于 Agency_codes 表中的前两个值。在第三个（ACB）值上，它在“where 子句”中给出错误未知列“ACB”，我很困惑为什么它与前两个值一起工作，然后在第三个值上停止。

这是程序：

DELIMITER $$
DROP PROCEDURE IF EXISTS  export_csv $$
CREATE PROCEDURE export_csv()
BEGIN
DECLARE agency_name VARCHAR(255);

DECLARE exit_loop BOOLEAN;         

DECLARE agency_cursor CURSOR FOR
SELECT agency FROM agency_codes;

DECLARE CONTINUE HANDLER FOR NOT FOUND SET exit_loop = TRUE;

OPEN agency_cursor;

agency_loop: LOOP
FETCH  agency_cursor INTO agency_name;

SET @sql_text = Concat("(select 'class_code','course_code','course_name','course_type','username','grade_type','score','letter_grade','is_passed','completion_date','completion_status','registration_date','registration_entry_status','registration_type','comment','first_name','last_name','class_name','agency') Union (select class_code,course_code,course_name,course_type,username,grade_type,score,letter_grade,is_passed,completion_date,completion_status,registration_date,registration_entry_status,registration_type,comment,first_name,last_name,class_name,agency from hrdis_oru where hrdis_oru.agency =", agency_name," into outfile 'C:/HDD/",agency_name,".csv' fields enclosed by '\"' terminated by ',' escaped by '\"' lines terminated by '\r\n')");

prepare s1 from @sql_text;
execute s1;
deallocate prepare s1; 


IF exit_loop THEN
     CLOSE agency_cursor;
     LEAVE agency_loop;
 END IF;
END LOOP agency_loop;
END $$
DELIMITER ;

我的代理表中的前几个值是：

  • 17
  • 303
  • ACB
  • 精算师
  • 农业
  • 目标

任何帮助都会很棒。谢谢。

4

1 回答 1

0

如果您考虑@sql_text每次迭代将保留哪些值，问题应该立即显而易见。为了清楚起见，添加一些空格：

(
  select 'class_code','course_code','course_name','course_type','username',
         'grade_type','score','letter_grade','is_passed','completion_date',
         'completion_status','registration_date','registration_entry_status',
         'registration_type','comment','first_name','last_name','class_name',
         'agency'
) Union (
  select class_code,course_code,course_name,course_type,username,
         grade_type,score,letter_grade,is_passed,completion_date,
         completion_status,registration_date,registration_entry_status,
         registration_type,comment,first_name,last_name,class_name,
         agency
  from   hrdis_oru
  where  hrdis_oru.agency =ACB
  into   outfile 'C:/HDD/ACB.csv'
  fields enclosed   by '\"'
         terminated by ','
         escaped    by '\"'
  lines  terminated by '\r\n'
)

特别注意，where htdis_oru.agency =ACB.

由于ACB没有被引用，MySQL 将其解析为模式对象标识符，并在找不到任何此类命名的对象时抱怨（这不是纯数字机构名称的情况，因为它们被解析为随后被强制转换为的整数表达式评估期间的字符串）。

为了让 MySQL 正确地将非数字值解析为字符串文字，它们必须被引用：

... where hrdis_oru.agency ='", agency_name, "' ...
                            ^                 ^

自然地，这提出了一个问题应该agency_name包含'字符串引号字符——任何这样的出现当然必须被转义。MySQL 很方便QUOTE()地为此目的提供了一个函数：

... where hrdis_oru.agency =", QUOTE(agency_name), " ...

然而，为了防止可能的 SQL 注入攻击，你真的应该参数化你准备好的语句：

FETCH agency_cursor INTO @agency_name;

（您不再需要DECLARE agency_name）；然后：

... where hrdis_oru.agency = ? into outfile CONCAT('C:/HDD/', ?, '.csv') ...

其次是：

PREPARE s1 FROM @sql_text;
EXECUTE s1 USING @agency_name, @agency_name;
DEALLOCATE PREPARE s1;

请注意，您现在还可以在进入循环之前PREPARE使用该语句，并将其（使用适当的值）简单地放入循环中——这应该会产生轻微的性能改进。记得退出循环后。EXECUTEDEALLOCATE

最后一点：您应该在命令之后exit_loop 立即FETCH检查- 否则您最终会尝试在SELECT ... INTO OUTFILE没有更多机构的最后一次执行该语句。

还可能值得注意的是，在这种情况下，您根本不需要使用准备好的语句。您可以简单地执行以下操作：

CREATE PROCEDURE export_csv() BEGIN
  DECLARE agency_name VARCHAR(255);
  DECLARE agency_cursor CURSOR FOR SELECT agency FROM agency_codes;
  DECLARE EXIT HANDLER FOR NOT FOUND CLOSE agency_cursor;

  OPEN agency_cursor;
  LOOP
    FETCH agency_cursor INTO agency_name;

      SELECT 'class_code','course_code','course_name','course_type','username',
             'grade_type','score','letter_grade','is_passed','completion_date',
             'completion_status','registration_date','registration_entry_status',
             'registration_type','comment','first_name','last_name','class_name'

    UNION ALL

      SELECT class_code,course_code,course_name,course_type,username,
             grade_type,score,letter_grade,is_passed,completion_date,
             completion_status,registration_date,registration_entry_status,
             registration_type,comment,first_name,last_name,class_name
      FROM   hrdis_oru
      WHERE  agency = agency_name

    INTO   OUTFILE CONCAT('C:/HDD/', agency_name, '.csv')
    FIELDS ENCLOSED   BY '"'
           TERMINATED BY ','
           ESCAPED    BY '"'
    LINES  TERMINATED BY '\r\n';

  END LOOP;
END
于 2015-03-04T19:25:38.233 回答