浅谈MySQL load data local infile细节 -- 从源码层面

最新推荐文章于 2024-10-21 13:05:55 发布

转载最新推荐文章于 2024-10-21 13:05:55 发布 · 166 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/lispking/p/3619484.html

文章标签：

#数据库

本文详细介绍了MySQL中Load Data Local Infile的功能及其实现原理。从客户端和服务端两个角度出发，通过官方源码剖析了文件传输的具体流程，包括文件存在性的检查、文件路径的指定以及文件传输结束的标记。

相信大伙对mysql的load data local infile并不陌生，今天来巩固一下这里面隐藏的一些细节，对于想自己动手开发一个mysql客户端有哪些点需要注意的呢？

首先，了解一下流程：

3个点：

1、Is '<path>/<filename>' exists?对于客户端来说，在文件发送前是先检查一下文件是否存在；

2、filename前建议加上绝对路径；

3、空包，表示命令已执行完毕。

接下来，一起来学习下官方源码(版本5.5.36)是如何实现该流程？由于篇幅关系，只贴出关键部分，其他读者自行查阅源码。

首先，看看客户端的核心实现，源码在libmysql\libmysql.c的handle_local_infile函数：

my_bool handle_local_infile(MYSQL *mysql, const char *net_filename)
 {
   /* initialize local infile (open file, usually) */
   ((*options->local_infile_init)(&li_ptr, net_filename,
     options->local_infile_userdata));
 
   /* read blocks of data from local infile callback */
   while ((readcount =
       (*options->local_infile_read)(li_ptr, buf,
                     packet_length)) > 0);
 
   /* Send empty packet to mark end of file */
   (my_net_write(net, (const uchar*) "", 0))
 }

接下来，看看服务端的核心实现，源码在sql\sql_load.cc的mysql_load函数：

int mysql_load(THD *thd,sql_exchange *ex,TABLE_LIST *table_list,
             List<Item> &fields_vars, List<Item> &set_fields,
                 List<Item> &set_values,
                 enum enum_duplicates handle_duplicates, bool ignore,
                 bool read_file_from_client)
 {
     net_request_file(&thd->net,ex->file_name);
   
    // 接收客户端发过来的文件内容
    while (!read_info.next_line())
     ;

    /* ok to client sent only after binlog write and engine commit */
    my_ok(thd, info.copied + info.deleted, 0L, name);
 }

bool net_request_file(NET* net, const char* fname)
{
  DBUG_ENTER("net_request_file");
  DBUG_RETURN(net_write_command(net, 251, (uchar*) fname, strlen(fname),
                                (uchar*) "", 0)); // 251即上图的0xfb
}