mirror of
				https://github.com/MariaDB/server.git
				synced 2025-11-03 14:33:32 +03:00 
			
		
		
		
	into bodhi.local:/opt/local/work/mysql-5.1-runtime-merge BitKeeper/etc/ignore: auto-union BitKeeper/deleted/.del-im_options_set.result~59278f56be61d921: Auto merged BitKeeper/deleted/.del-mysqld.dsp~ffdbf2d234e23e56: Auto merged BitKeeper/deleted/.del-mysys.dsp~32695fee91189326: Auto merged BitKeeper/deleted/.del-im_options_set.imtest~b53d9d60e5684833: Auto merged BitKeeper/deleted/.del-im_options_unset.imtest~768eb186b51d0048: Auto merged BitKeeper/deleted/.del-im_options_unset.result~20a4790cd3c70a4f: Auto merged client/mysql.cc: Auto merged client/mysqlbinlog.cc: Auto merged client/mysqlcheck.c: Auto merged client/mysqldump.c: Auto merged client/mysqltest.c: Auto merged dbug/dbug.c: Auto merged extra/perror.c: Auto merged extra/yassl/src/yassl_imp.cpp: Auto merged extra/yassl/src/yassl_int.cpp: Auto merged include/mysql.h: Auto merged include/mysql_com.h: Auto merged libmysql/libmysql.c: Auto merged mysql-test/r/cast.result: Auto merged mysql-test/r/date_formats.result: Auto merged mysql-test/r/federated.result: Auto merged mysql-test/r/func_compress.result: Auto merged mysql-test/r/func_group.result: Auto merged mysql-test/r/func_time.result: Auto merged mysql-test/r/gis-rtree.result: Auto merged mysql-test/r/gis.result: Auto merged mysql-test/r/im_daemon_life_cycle.result: Auto merged mysql-test/r/im_utils.result: Auto merged mysql-test/r/join_outer.result: Auto merged mysql-test/r/mysqlcheck.result: Auto merged mysql-test/r/rpl_sp.result: Auto merged mysql-test/r/rpl_trigger.result: Auto merged mysql-test/r/sp-code.result: Auto merged mysql-test/r/sp-security.result: Auto merged mysql-test/r/strict.result: Auto merged mysql-test/r/type_blob.result: Auto merged mysql-test/r/type_datetime.result: Auto merged mysql-test/r/type_ranges.result: Auto merged mysql-test/r/udf.result: Auto merged mysql-test/r/user_var.result: Auto merged mysql-test/t/cast.test: Auto merged mysql-test/t/disabled.def: Auto merged mysql-test/t/func_group.test: Auto merged mysql-test/t/func_time.test: Auto merged mysql-test/t/im_daemon_life_cycle.imtest: Auto merged mysql-test/t/im_life_cycle.imtest: Auto merged mysql-test/t/im_utils.imtest: Auto merged mysql-test/t/mysql.test: Auto merged mysql-test/t/mysqlbinlog.test: Auto merged mysql-test/t/mysqlcheck.test: Auto merged mysql-test/t/ps.test: Auto merged mysql-test/t/rpl_trigger.test: Auto merged mysql-test/t/sp-security.test: Auto merged mysql-test/t/strict.test: Auto merged mysql-test/t/udf.test: Auto merged sql/field.cc: Auto merged sql/item.cc: Auto merged sql/item_func.cc: Auto merged sql/item_func.h: Auto merged sql/item_strfunc.cc: Auto merged sql/item_strfunc.h: Auto merged sql/item_subselect.cc: Auto merged sql/item_sum.h: Auto merged sql/item_timefunc.cc: Auto merged sql/mysqld.cc: Auto merged sql/protocol.cc: Auto merged sql/slave.cc: Auto merged sql/sp.cc: Auto merged sql/sp_head.h: Auto merged sql/sql_base.cc: Auto merged sql/sql_class.h: Auto merged sql/sql_lex.cc: Auto merged sql/sql_parse.cc: Auto merged sql/sql_prepare.cc: Auto merged sql/sql_udf.cc: Auto merged sql/sql_view.cc: Auto merged sql/table.cc: Auto merged sql-common/client.c: Auto merged sql-common/my_time.c: Auto merged sql/table.h: Auto merged storage/ndb/src/kernel/error/ndbd_exit_codes.c: Auto merged storage/ndb/src/mgmsrv/ConfigInfo.cpp: Auto merged mysql-test/r/im_life_cycle.result: e use local mysql-test/r/ps.result: use local client/Makefile.am: Manual merge. client/mysqlimport.c: Manual merge. configure.in: Manual merge. mysql-test/mysql-test-run.pl: Manual merge. mysql-test/r/mysqldump.result: Manual merge. mysql-test/r/mysqltest.result: Manual merge. mysql-test/r/ndb_basic.result: Manual merge. mysql-test/r/rpl_view.result: Manual merge. mysql-test/r/show_check.result: Manual merge. mysql-test/r/sp-error.result: Manual merge. mysql-test/r/sp.result: Manual merge. mysql-test/r/union.result: Manual merge. mysql-test/t/mysqldump.test: Manual merge. mysql-test/t/mysqltest.test: Manual merge. mysql-test/t/ndb_basic.test: Manual merge. mysql-test/t/rpl_sp.test: Manual merge. mysql-test/t/rpl_view.test: Manual merge. mysql-test/t/show_check.test: Manual merge. mysql-test/t/sp-error.test: Manual merge. mysql-test/t/sp.test: Manual merge. sql/item_sum.cc: Manual merge. sql/mysql_priv.h: Manual merge. sql/sp_head.cc: Manual merge. sql/sql_db.cc: Manual merge. sql/sql_delete.cc: Manual merge. sql/sql_lex.h: Manual merge. sql/sql_show.cc: Manual merge. sql/sql_table.cc: Manual merge. sql/sql_trigger.cc: Manual merge. sql/sql_yacc.yy: Manual merge. tests/mysql_client_test.c: Manual merge.
		
			
				
	
	
		
			669 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			669 lines
		
	
	
		
			20 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* Copyright (C) 2000 MySQL AB
 | 
						|
 | 
						|
   This program is free software; you can redistribute it and/or modify
 | 
						|
   it under the terms of the GNU General Public License as published by
 | 
						|
   the Free Software Foundation; either version 2 of the License, or
 | 
						|
   (at your option) any later version.
 | 
						|
 | 
						|
   This program is distributed in the hope that it will be useful,
 | 
						|
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
   GNU General Public License for more details.
 | 
						|
 | 
						|
   You should have received a copy of the GNU General Public License
 | 
						|
   along with this program; if not, write to the Free Software
 | 
						|
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
 | 
						|
 | 
						|
/*
 | 
						|
**	   mysqlimport.c  - Imports all given files
 | 
						|
**			    into a table(s).
 | 
						|
**
 | 
						|
**			   *************************
 | 
						|
**			   *			   *
 | 
						|
**			   * AUTHOR: Monty & Jani  *
 | 
						|
**			   * DATE:   June 24, 1997 *
 | 
						|
**			   *			   *
 | 
						|
**			   *************************
 | 
						|
*/
 | 
						|
#define IMPORT_VERSION "3.5"
 | 
						|
 | 
						|
#include "client_priv.h"
 | 
						|
#include "mysql_version.h"
 | 
						|
#ifdef HAVE_LIBPTHREAD
 | 
						|
#include <my_pthread.h>
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
/* Global Thread counter */
 | 
						|
int counter;
 | 
						|
#ifdef HAVE_LIBPTHREAD
 | 
						|
pthread_mutex_t counter_mutex;
 | 
						|
pthread_cond_t count_threshhold;
 | 
						|
#endif
 | 
						|
 | 
						|
static void db_error_with_table(MYSQL *mysql, char *table);
 | 
						|
static void db_error(MYSQL *mysql);
 | 
						|
static char *field_escape(char *to,const char *from,uint length);
 | 
						|
static char *add_load_option(char *ptr,const char *object,
 | 
						|
			     const char *statement);
 | 
						|
 | 
						|
static my_bool	verbose=0,lock_tables=0,ignore_errors=0,opt_delete=0,
 | 
						|
		replace=0,silent=0,ignore=0,opt_compress=0,
 | 
						|
                opt_low_priority= 0, tty_password= 0;
 | 
						|
static my_bool opt_use_threads= 0;
 | 
						|
static uint     opt_local_file=0;
 | 
						|
static char	*opt_password=0, *current_user=0,
 | 
						|
		*current_host=0, *current_db=0, *fields_terminated=0,
 | 
						|
		*lines_terminated=0, *enclosed=0, *opt_enclosed=0,
 | 
						|
		*escaped=0, *opt_columns=0, 
 | 
						|
		*default_charset= (char*) MYSQL_DEFAULT_CHARSET_NAME;
 | 
						|
static uint     opt_mysql_port= 0, opt_protocol= 0;
 | 
						|
static my_string opt_mysql_unix_port=0;
 | 
						|
static longlong opt_ignore_lines= -1;
 | 
						|
static CHARSET_INFO *charset_info= &my_charset_latin1;
 | 
						|
#include <sslopt-vars.h>
 | 
						|
 | 
						|
#ifdef HAVE_SMEM
 | 
						|
static char *shared_memory_base_name=0;
 | 
						|
#endif
 | 
						|
 | 
						|
static struct my_option my_long_options[] =
 | 
						|
{
 | 
						|
#ifdef __NETWARE__
 | 
						|
  {"autoclose", OPT_AUTO_CLOSE, "Auto close the screen on exit for Netware.",
 | 
						|
   0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#endif
 | 
						|
  {"character-sets-dir", OPT_CHARSETS_DIR,
 | 
						|
   "Directory where character sets are.", (gptr*) &charsets_dir,
 | 
						|
   (gptr*) &charsets_dir, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"default-character-set", OPT_DEFAULT_CHARSET,
 | 
						|
   "Set the default character set.", (gptr*) &default_charset,
 | 
						|
   (gptr*) &default_charset, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"columns", 'c',
 | 
						|
   "Use only these columns to import the data to. Give the column names in a comma separated list. This is same as giving columns to LOAD DATA INFILE.",
 | 
						|
   (gptr*) &opt_columns, (gptr*) &opt_columns, 0, GET_STR, REQUIRED_ARG, 0, 0, 0,
 | 
						|
   0, 0, 0},
 | 
						|
  {"compress", 'C', "Use compression in server/client protocol.",
 | 
						|
   (gptr*) &opt_compress, (gptr*) &opt_compress, 0, GET_BOOL, NO_ARG, 0, 0, 0,
 | 
						|
   0, 0, 0},
 | 
						|
  {"debug",'#', "Output debug log. Often this is 'd:t:o,filename'.", 0, 0, 0,
 | 
						|
   GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"delete", 'd', "First delete all rows from table.", (gptr*) &opt_delete,
 | 
						|
   (gptr*) &opt_delete, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"fields-terminated-by", OPT_FTB,
 | 
						|
   "Fields in the textfile are terminated by ...", (gptr*) &fields_terminated,
 | 
						|
   (gptr*) &fields_terminated, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"fields-enclosed-by", OPT_ENC,
 | 
						|
   "Fields in the importfile are enclosed by ...", (gptr*) &enclosed,
 | 
						|
   (gptr*) &enclosed, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"fields-optionally-enclosed-by", OPT_O_ENC,
 | 
						|
   "Fields in the i.file are opt. enclosed by ...", (gptr*) &opt_enclosed,
 | 
						|
   (gptr*) &opt_enclosed, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"fields-escaped-by", OPT_ESC, "Fields in the i.file are escaped by ...",
 | 
						|
   (gptr*) &escaped, (gptr*) &escaped, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0,
 | 
						|
   0, 0},
 | 
						|
  {"force", 'f', "Continue even if we get an sql-error.",
 | 
						|
   (gptr*) &ignore_errors, (gptr*) &ignore_errors, 0, GET_BOOL, NO_ARG, 0, 0,
 | 
						|
   0, 0, 0, 0},
 | 
						|
  {"help", '?', "Displays this help and exits.", 0, 0, 0, GET_NO_ARG, NO_ARG,
 | 
						|
   0, 0, 0, 0, 0, 0},
 | 
						|
  {"host", 'h', "Connect to host.", (gptr*) ¤t_host,
 | 
						|
   (gptr*) ¤t_host, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"ignore", 'i', "If duplicate unique key was found, keep old row.",
 | 
						|
   (gptr*) &ignore, (gptr*) &ignore, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"ignore-lines", OPT_IGN_LINES, "Ignore first n lines of data infile.",
 | 
						|
   (gptr*) &opt_ignore_lines, (gptr*) &opt_ignore_lines, 0, GET_LL,
 | 
						|
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"lines-terminated-by", OPT_LTB, "Lines in the i.file are terminated by ...",
 | 
						|
   (gptr*) &lines_terminated, (gptr*) &lines_terminated, 0, GET_STR,
 | 
						|
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"local", 'L', "Read all files through the client.", (gptr*) &opt_local_file,
 | 
						|
   (gptr*) &opt_local_file, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"lock-tables", 'l', "Lock all tables for write (this disables threads).",
 | 
						|
    (gptr*) &lock_tables, (gptr*) &lock_tables, 0, GET_BOOL, NO_ARG, 
 | 
						|
    0, 0, 0, 0, 0, 0},
 | 
						|
  {"low-priority", OPT_LOW_PRIORITY,
 | 
						|
   "Use LOW_PRIORITY when updating the table.", (gptr*) &opt_low_priority,
 | 
						|
   (gptr*) &opt_low_priority, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"password", 'p',
 | 
						|
   "Password to use when connecting to server. If password is not given it's asked from the tty.",
 | 
						|
   0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#ifdef __WIN__
 | 
						|
  {"pipe", 'W', "Use named pipes to connect to server.", 0, 0, 0, GET_NO_ARG,
 | 
						|
   NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#endif
 | 
						|
  {"port", 'P', "Port number to use for connection.", (gptr*) &opt_mysql_port,
 | 
						|
   (gptr*) &opt_mysql_port, 0, GET_UINT, REQUIRED_ARG, 0, 0, 0, 0, 0,
 | 
						|
   0},
 | 
						|
  {"protocol", OPT_MYSQL_PROTOCOL, "The protocol of connection (tcp,socket,pipe,memory).",
 | 
						|
   0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"replace", 'r', "If duplicate unique key was found, replace old row.",
 | 
						|
   (gptr*) &replace, (gptr*) &replace, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#ifdef HAVE_SMEM
 | 
						|
  {"shared-memory-base-name", OPT_SHARED_MEMORY_BASE_NAME,
 | 
						|
   "Base name of shared memory.", (gptr*) &shared_memory_base_name, (gptr*) &shared_memory_base_name,
 | 
						|
   0, GET_STR_ALLOC, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#endif
 | 
						|
  {"silent", 's', "Be more silent.", (gptr*) &silent, (gptr*) &silent, 0,
 | 
						|
   GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"socket", 'S', "Socket file to use for connection.",
 | 
						|
   (gptr*) &opt_mysql_unix_port, (gptr*) &opt_mysql_unix_port, 0, GET_STR,
 | 
						|
   REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#include <sslopt-longopts.h>
 | 
						|
  {"use-threads", OPT_USE_THREADS,
 | 
						|
   "Load files in parallel. The argument is the number "
 | 
						|
   "of threads to use for loading data.",
 | 
						|
   (gptr*) &opt_use_threads, (gptr*) &opt_use_threads, 0, 
 | 
						|
   GET_UINT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#ifndef DONT_ALLOW_USER_CHANGE
 | 
						|
  {"user", 'u', "User for login if not current user.", (gptr*) ¤t_user,
 | 
						|
   (gptr*) ¤t_user, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
#endif
 | 
						|
  {"verbose", 'v', "Print info about the various stages.", (gptr*) &verbose,
 | 
						|
   (gptr*) &verbose, 0, GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  {"version", 'V', "Output version information and exit.", 0, 0, 0, GET_NO_ARG,
 | 
						|
   NO_ARG, 0, 0, 0, 0, 0, 0},
 | 
						|
  { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
static const char *load_default_groups[]= { "mysqlimport","client",0 };
 | 
						|
 | 
						|
#include <help_start.h>
 | 
						|
 | 
						|
static void print_version(void)
 | 
						|
{
 | 
						|
  printf("%s  Ver %s Distrib %s, for %s (%s)\n" ,my_progname,
 | 
						|
	  IMPORT_VERSION, MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);
 | 
						|
  NETWARE_SET_SCREEN_MODE(1);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static void usage(void)
 | 
						|
{
 | 
						|
  print_version();
 | 
						|
  puts("Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB");
 | 
						|
  puts("This software comes with ABSOLUTELY NO WARRANTY. This is free software,\nand you are welcome to modify and redistribute it under the GPL license\n");
 | 
						|
  printf("\
 | 
						|
Loads tables from text files in various formats.  The base name of the\n\
 | 
						|
text file must be the name of the table that should be used.\n\
 | 
						|
If one uses sockets to connect to the MySQL server, the server will open and\n\
 | 
						|
read the text file directly. In other cases the client will open the text\n\
 | 
						|
file. The SQL command 'LOAD DATA INFILE' is used to import the rows.\n");
 | 
						|
 | 
						|
  printf("\nUsage: %s [OPTIONS] database textfile...",my_progname);
 | 
						|
  print_defaults("my",load_default_groups);
 | 
						|
  my_print_help(my_long_options);
 | 
						|
  my_print_variables(my_long_options);
 | 
						|
}
 | 
						|
 | 
						|
#include <help_end.h>
 | 
						|
 | 
						|
static my_bool
 | 
						|
get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
 | 
						|
	       char *argument)
 | 
						|
{
 | 
						|
  switch(optid) {
 | 
						|
#ifdef __NETWARE__
 | 
						|
  case OPT_AUTO_CLOSE:
 | 
						|
    setscreenmode(SCR_AUTOCLOSE_ON_EXIT);
 | 
						|
    break;
 | 
						|
#endif
 | 
						|
  case 'p':
 | 
						|
    if (argument)
 | 
						|
    {
 | 
						|
      char *start=argument;
 | 
						|
      my_free(opt_password,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
      opt_password=my_strdup(argument,MYF(MY_FAE));
 | 
						|
      while (*argument) *argument++= 'x';		/* Destroy argument */
 | 
						|
      if (*start)
 | 
						|
	start[1]=0;				/* Cut length of argument */
 | 
						|
      tty_password= 0;
 | 
						|
    }
 | 
						|
    else
 | 
						|
      tty_password= 1;
 | 
						|
    break;
 | 
						|
#ifdef __WIN__
 | 
						|
  case 'W':
 | 
						|
    opt_protocol = MYSQL_PROTOCOL_PIPE;
 | 
						|
    opt_local_file=1;
 | 
						|
    break;
 | 
						|
#endif
 | 
						|
  case OPT_MYSQL_PROTOCOL:
 | 
						|
  {
 | 
						|
    if ((opt_protocol= find_type(argument, &sql_protocol_typelib,0)) <= 0)
 | 
						|
    {
 | 
						|
      fprintf(stderr, "Unknown option to protocol: %s\n", argument);
 | 
						|
      exit(1);
 | 
						|
    }
 | 
						|
    break;
 | 
						|
  }
 | 
						|
  case '#':
 | 
						|
    DBUG_PUSH(argument ? argument : "d:t:o");
 | 
						|
    break;
 | 
						|
#include <sslopt-case.h>
 | 
						|
  case 'V': print_version(); exit(0);
 | 
						|
  case 'I':
 | 
						|
  case '?':
 | 
						|
    usage();
 | 
						|
    exit(0);
 | 
						|
  }
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static int get_options(int *argc, char ***argv)
 | 
						|
{
 | 
						|
  int ho_error;
 | 
						|
 | 
						|
  if ((ho_error=handle_options(argc, argv, my_long_options, get_one_option)))
 | 
						|
    exit(ho_error);
 | 
						|
 | 
						|
  if (enclosed && opt_enclosed)
 | 
						|
  {
 | 
						|
    fprintf(stderr, "You can't use ..enclosed.. and ..optionally-enclosed.. at the same time.\n");
 | 
						|
    return(1);
 | 
						|
  }
 | 
						|
  if (replace && ignore)
 | 
						|
  {
 | 
						|
    fprintf(stderr, "You can't use --ignore (-i) and --replace (-r) at the same time.\n");
 | 
						|
    return(1);
 | 
						|
  }
 | 
						|
  if (strcmp(default_charset, charset_info->csname) &&
 | 
						|
      !(charset_info= get_charset_by_csname(default_charset,
 | 
						|
  					    MY_CS_PRIMARY, MYF(MY_WME))))
 | 
						|
    exit(1);
 | 
						|
  if (*argc < 2)
 | 
						|
  {
 | 
						|
    usage();
 | 
						|
    return 1;
 | 
						|
  }
 | 
						|
  current_db= *((*argv)++);
 | 
						|
  (*argc)--;
 | 
						|
  if (tty_password)
 | 
						|
    opt_password=get_tty_password(NullS);
 | 
						|
  return(0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static int write_to_table(char *filename, MYSQL *mysql)
 | 
						|
{
 | 
						|
  char tablename[FN_REFLEN], hard_path[FN_REFLEN],
 | 
						|
       sql_statement[FN_REFLEN*16+256], *end;
 | 
						|
  DBUG_ENTER("write_to_table");
 | 
						|
  DBUG_PRINT("enter",("filename: %s",filename));
 | 
						|
 | 
						|
  fn_format(tablename, filename, "", "", 1 | 2); /* removes path & ext. */
 | 
						|
  if (!opt_local_file)
 | 
						|
    strmov(hard_path,filename);
 | 
						|
  else
 | 
						|
    my_load_path(hard_path, filename, NULL); /* filename includes the path */
 | 
						|
 | 
						|
  if (opt_delete)
 | 
						|
  {
 | 
						|
    if (verbose)
 | 
						|
      fprintf(stdout, "Deleting the old data from table %s\n", tablename);
 | 
						|
#ifdef HAVE_SNPRINTF
 | 
						|
    snprintf(sql_statement, FN_REFLEN*16+256, "DELETE FROM %s", tablename);
 | 
						|
#else
 | 
						|
    sprintf(sql_statement, "DELETE FROM %s", tablename);
 | 
						|
#endif
 | 
						|
    if (mysql_query(mysql, sql_statement))
 | 
						|
    {
 | 
						|
      db_error_with_table(mysql, tablename);
 | 
						|
      DBUG_RETURN(1);
 | 
						|
    }
 | 
						|
  }
 | 
						|
  to_unix_path(hard_path);
 | 
						|
  if (verbose)
 | 
						|
  {
 | 
						|
    if (opt_local_file)
 | 
						|
      fprintf(stdout, "Loading data from LOCAL file: %s into %s\n",
 | 
						|
	      hard_path, tablename);
 | 
						|
    else
 | 
						|
      fprintf(stdout, "Loading data from SERVER file: %s into %s\n",
 | 
						|
	      hard_path, tablename);
 | 
						|
  }
 | 
						|
  sprintf(sql_statement, "LOAD DATA %s %s INFILE '%s'",
 | 
						|
	  opt_low_priority ? "LOW_PRIORITY" : "",
 | 
						|
	  opt_local_file ? "LOCAL" : "", hard_path);
 | 
						|
  end= strend(sql_statement);
 | 
						|
  if (replace)
 | 
						|
    end= strmov(end, " REPLACE");
 | 
						|
  if (ignore)
 | 
						|
    end= strmov(end, " IGNORE");
 | 
						|
  end= strmov(strmov(end, " INTO TABLE "), tablename);
 | 
						|
 | 
						|
  if (fields_terminated || enclosed || opt_enclosed || escaped)
 | 
						|
      end= strmov(end, " FIELDS");
 | 
						|
  end= add_load_option(end, fields_terminated, " TERMINATED BY");
 | 
						|
  end= add_load_option(end, enclosed, " ENCLOSED BY");
 | 
						|
  end= add_load_option(end, opt_enclosed,
 | 
						|
		       " OPTIONALLY ENCLOSED BY");
 | 
						|
  end= add_load_option(end, escaped, " ESCAPED BY");
 | 
						|
  end= add_load_option(end, lines_terminated, " LINES TERMINATED BY");
 | 
						|
  if (opt_ignore_lines >= 0)
 | 
						|
    end= strmov(longlong10_to_str(opt_ignore_lines, 
 | 
						|
				  strmov(end, " IGNORE "),10), " LINES");
 | 
						|
  if (opt_columns)
 | 
						|
    end= strmov(strmov(strmov(end, " ("), opt_columns), ")");
 | 
						|
  *end= '\0';
 | 
						|
 | 
						|
  if (mysql_query(mysql, sql_statement))
 | 
						|
  {
 | 
						|
    db_error_with_table(mysql, tablename);
 | 
						|
    DBUG_RETURN(1);
 | 
						|
  }
 | 
						|
  if (!silent)
 | 
						|
  {
 | 
						|
    if (mysql_info(mysql)) /* If NULL-pointer, print nothing */
 | 
						|
    {
 | 
						|
      fprintf(stdout, "%s.%s: %s\n", current_db, tablename,
 | 
						|
	      mysql_info(mysql));
 | 
						|
    }
 | 
						|
  }
 | 
						|
  DBUG_RETURN(0);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static void lock_table(MYSQL *mysql, int tablecount, char **raw_tablename)
 | 
						|
{
 | 
						|
  DYNAMIC_STRING query;
 | 
						|
  int i;
 | 
						|
  char tablename[FN_REFLEN];
 | 
						|
 | 
						|
  if (verbose)
 | 
						|
    fprintf(stdout, "Locking tables for write\n");
 | 
						|
  init_dynamic_string(&query, "LOCK TABLES ", 256, 1024);
 | 
						|
  for (i=0 ; i < tablecount ; i++)
 | 
						|
  {
 | 
						|
    fn_format(tablename, raw_tablename[i], "", "", 1 | 2);
 | 
						|
    dynstr_append(&query, tablename);
 | 
						|
    dynstr_append(&query, " WRITE,");
 | 
						|
  }
 | 
						|
  if (mysql_real_query(mysql, query.str, query.length-1))
 | 
						|
    db_error(mysql); /* We shall countinue here, if --force was given */
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static MYSQL *db_connect(char *host, char *database,
 | 
						|
                         char *user, char *passwd)
 | 
						|
{
 | 
						|
  MYSQL *mysql;
 | 
						|
  if (verbose)
 | 
						|
    fprintf(stdout, "Connecting to %s\n", host ? host : "localhost");
 | 
						|
  if (!(mysql= mysql_init(NULL)))
 | 
						|
    return 0;
 | 
						|
  if (opt_compress)
 | 
						|
    mysql_options(mysql,MYSQL_OPT_COMPRESS,NullS);
 | 
						|
  if (opt_local_file)
 | 
						|
    mysql_options(mysql,MYSQL_OPT_LOCAL_INFILE,
 | 
						|
		  (char*) &opt_local_file);
 | 
						|
#ifdef HAVE_OPENSSL
 | 
						|
  if (opt_use_ssl)
 | 
						|
    mysql_ssl_set(mysql, opt_ssl_key, opt_ssl_cert, opt_ssl_ca,
 | 
						|
		  opt_ssl_capath, opt_ssl_cipher);
 | 
						|
  mysql_options(mysql,MYSQL_OPT_SSL_VERIFY_SERVER_CERT,
 | 
						|
                (char*)&opt_ssl_verify_server_cert);
 | 
						|
#endif
 | 
						|
  if (opt_protocol)
 | 
						|
    mysql_options(mysql,MYSQL_OPT_PROTOCOL,(char*)&opt_protocol);
 | 
						|
#ifdef HAVE_SMEM
 | 
						|
  if (shared_memory_base_name)
 | 
						|
    mysql_options(mysql,MYSQL_SHARED_MEMORY_BASE_NAME,shared_memory_base_name);
 | 
						|
#endif
 | 
						|
  if (!(mysql_real_connect(mysql,host,user,passwd,
 | 
						|
                           database,opt_mysql_port,opt_mysql_unix_port,
 | 
						|
                           0)))
 | 
						|
  {
 | 
						|
    ignore_errors=0;	  /* NO RETURN FROM db_error */
 | 
						|
    db_error(mysql);
 | 
						|
  }
 | 
						|
  mysql->reconnect= 0;
 | 
						|
  if (verbose)
 | 
						|
    fprintf(stdout, "Selecting database %s\n", database);
 | 
						|
  if (mysql_select_db(mysql, database))
 | 
						|
  {
 | 
						|
    ignore_errors=0;
 | 
						|
    db_error(mysql);
 | 
						|
  }
 | 
						|
  return mysql;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static void db_disconnect(char *host, MYSQL *mysql)
 | 
						|
{
 | 
						|
  if (verbose)
 | 
						|
    fprintf(stdout, "Disconnecting from %s\n", host ? host : "localhost");
 | 
						|
  mysql_close(mysql);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static void safe_exit(int error, MYSQL *mysql)
 | 
						|
{
 | 
						|
  if (ignore_errors)
 | 
						|
    return;
 | 
						|
  if (mysql)
 | 
						|
    mysql_close(mysql);
 | 
						|
  exit(error);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static void db_error_with_table(MYSQL *mysql, char *table)
 | 
						|
{
 | 
						|
  my_printf_error(0,"Error: %d, %s, when using table: %s",
 | 
						|
		  MYF(0), mysql_errno(mysql), mysql_error(mysql), table);
 | 
						|
  safe_exit(1, mysql);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static void db_error(MYSQL *mysql)
 | 
						|
{
 | 
						|
  my_printf_error(0,"Error: %d %s", MYF(0), mysql_errno(mysql), mysql_error(mysql));
 | 
						|
  safe_exit(1, mysql);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static char *add_load_option(char *ptr, const char *object,
 | 
						|
			     const char *statement)
 | 
						|
{
 | 
						|
  if (object)
 | 
						|
  {
 | 
						|
    /* Don't escape hex constants */
 | 
						|
    if (object[0] == '0' && (object[1] == 'x' || object[1] == 'X'))
 | 
						|
      ptr= strxmov(ptr," ",statement," ",object,NullS);
 | 
						|
    else
 | 
						|
    {
 | 
						|
      /* char constant; escape */
 | 
						|
      ptr= strxmov(ptr," ",statement," '",NullS);
 | 
						|
      ptr= field_escape(ptr,object,(uint) strlen(object));
 | 
						|
      *ptr++= '\'';
 | 
						|
    }
 | 
						|
  }
 | 
						|
  return ptr;
 | 
						|
}
 | 
						|
 | 
						|
/*
 | 
						|
** Allow the user to specify field terminator strings like:
 | 
						|
** "'", "\", "\\" (escaped backslash), "\t" (tab), "\n" (newline)
 | 
						|
** This is done by doubleing ' and add a end -\ if needed to avoid
 | 
						|
** syntax errors from the SQL parser.
 | 
						|
*/ 
 | 
						|
 | 
						|
static char *field_escape(char *to,const char *from,uint length)
 | 
						|
{
 | 
						|
  const char *end;
 | 
						|
  uint end_backslashes=0; 
 | 
						|
 | 
						|
  for (end= from+length; from != end; from++)
 | 
						|
  {
 | 
						|
    *to++= *from;
 | 
						|
    if (*from == '\\')
 | 
						|
      end_backslashes^=1;    /* find odd number of backslashes */
 | 
						|
    else 
 | 
						|
    {
 | 
						|
      if (*from == '\'' && !end_backslashes)
 | 
						|
	*to++= *from;      /* We want a dublicate of "'" for MySQL */
 | 
						|
      end_backslashes=0;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  /* Add missing backslashes if user has specified odd number of backs.*/
 | 
						|
  if (end_backslashes)
 | 
						|
    *to++= '\\';          
 | 
						|
  return to;
 | 
						|
}
 | 
						|
 | 
						|
int exitcode= 0;
 | 
						|
 | 
						|
#ifdef HAVE_LIBPTHREAD
 | 
						|
pthread_handler_t worker_thread(void *arg)
 | 
						|
{
 | 
						|
  int error;
 | 
						|
  char *raw_table_name= (char *)arg;
 | 
						|
  MYSQL *mysql= 0;
 | 
						|
 | 
						|
  if (mysql_thread_init())
 | 
						|
    goto error;
 | 
						|
  
 | 
						|
  if (!(mysql= db_connect(current_host,current_db,current_user,opt_password)))
 | 
						|
  {
 | 
						|
    goto error;
 | 
						|
  }
 | 
						|
 | 
						|
  if (mysql_query(mysql, "/*!40101 set @@character_set_database=binary */;"))
 | 
						|
  {
 | 
						|
    db_error(mysql); /* We shall countinue here, if --force was given */
 | 
						|
    goto error;
 | 
						|
  }
 | 
						|
 | 
						|
  /*
 | 
						|
    We are not currently catching the error here.
 | 
						|
  */
 | 
						|
  if((error= write_to_table(raw_table_name, mysql)))
 | 
						|
    if (exitcode == 0)
 | 
						|
      exitcode= error;
 | 
						|
 | 
						|
error:
 | 
						|
  if (mysql)
 | 
						|
    db_disconnect(current_host, mysql);
 | 
						|
 | 
						|
  pthread_mutex_lock(&counter_mutex);
 | 
						|
  counter--;
 | 
						|
  pthread_cond_signal(&count_threshhold);
 | 
						|
  pthread_mutex_unlock(&counter_mutex);
 | 
						|
  my_thread_end();
 | 
						|
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
  int error=0;
 | 
						|
  char **argv_to_free;
 | 
						|
  MY_INIT(argv[0]);
 | 
						|
 | 
						|
  load_defaults("my",load_default_groups,&argc,&argv);
 | 
						|
  /* argv is changed in the program */
 | 
						|
  argv_to_free= argv;
 | 
						|
  if (get_options(&argc, &argv))
 | 
						|
  {
 | 
						|
    free_defaults(argv_to_free);
 | 
						|
    return(1);
 | 
						|
  }
 | 
						|
 | 
						|
#ifdef HAVE_LIBPTHREAD
 | 
						|
  if (opt_use_threads && !lock_tables)
 | 
						|
  {
 | 
						|
    pthread_t mainthread;            /* Thread descriptor */
 | 
						|
    pthread_attr_t attr;          /* Thread attributes */
 | 
						|
    pthread_attr_init(&attr);
 | 
						|
    pthread_attr_setdetachstate(&attr,
 | 
						|
                                PTHREAD_CREATE_DETACHED);
 | 
						|
 | 
						|
    VOID(pthread_mutex_init(&counter_mutex, NULL));
 | 
						|
    VOID(pthread_cond_init(&count_threshhold, NULL));
 | 
						|
 | 
						|
    for (counter= 0; *argv != NULL; argv++) /* Loop through tables */
 | 
						|
    {
 | 
						|
      pthread_mutex_lock(&counter_mutex);
 | 
						|
      while (counter == opt_use_threads)
 | 
						|
      {
 | 
						|
        struct timespec abstime;
 | 
						|
 | 
						|
        set_timespec(abstime, 3);
 | 
						|
        pthread_cond_timedwait(&count_threshhold, &counter_mutex, &abstime);
 | 
						|
      }
 | 
						|
      /* Before exiting the lock we set ourselves up for the next thread */
 | 
						|
      counter++;
 | 
						|
      pthread_mutex_unlock(&counter_mutex);
 | 
						|
      /* now create the thread */
 | 
						|
      if (pthread_create(&mainthread, &attr, worker_thread, 
 | 
						|
                         (void *)*argv) != 0)
 | 
						|
      {
 | 
						|
        pthread_mutex_lock(&counter_mutex);
 | 
						|
        counter--;
 | 
						|
        pthread_mutex_unlock(&counter_mutex);
 | 
						|
        fprintf(stderr,"%s: Could not create thread\n",
 | 
						|
                my_progname);
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    /*
 | 
						|
      We loop until we know that all children have cleaned up.
 | 
						|
    */
 | 
						|
    pthread_mutex_lock(&counter_mutex);
 | 
						|
    while (counter)
 | 
						|
    {
 | 
						|
      struct timespec abstime;
 | 
						|
 | 
						|
      set_timespec(abstime, 3);
 | 
						|
      pthread_cond_timedwait(&count_threshhold, &counter_mutex, &abstime);
 | 
						|
    }
 | 
						|
    pthread_mutex_unlock(&counter_mutex);
 | 
						|
    VOID(pthread_mutex_destroy(&counter_mutex));
 | 
						|
    VOID(pthread_cond_destroy(&count_threshhold));
 | 
						|
    pthread_attr_destroy(&attr);
 | 
						|
  }
 | 
						|
  else
 | 
						|
#endif
 | 
						|
  {
 | 
						|
    MYSQL *mysql= 0;
 | 
						|
    if (!(mysql= db_connect(current_host,current_db,current_user,opt_password)))
 | 
						|
    {
 | 
						|
      free_defaults(argv_to_free);
 | 
						|
      return(1); /* purecov: deadcode */
 | 
						|
    }
 | 
						|
 | 
						|
    if (mysql_query(mysql, "/*!40101 set @@character_set_database=binary */;"))
 | 
						|
    {
 | 
						|
      db_error(mysql); /* We shall countinue here, if --force was given */
 | 
						|
      return(1);
 | 
						|
    }
 | 
						|
 | 
						|
    if (lock_tables)
 | 
						|
      lock_table(mysql, argc, argv);
 | 
						|
    for (; *argv != NULL; argv++)
 | 
						|
      if ((error= write_to_table(*argv, mysql)))
 | 
						|
        if (exitcode == 0)
 | 
						|
          exitcode= error;
 | 
						|
    db_disconnect(current_host, mysql);
 | 
						|
  }
 | 
						|
  my_free(opt_password,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
#ifdef HAVE_SMEM
 | 
						|
  my_free(shared_memory_base_name,MYF(MY_ALLOW_ZERO_PTR));
 | 
						|
#endif
 | 
						|
  free_defaults(argv_to_free);
 | 
						|
  my_end(0);
 | 
						|
  return(exitcode);
 | 
						|
}
 |