从前边的程序的运行情况来看,程序是可以运行的,但速度太扯了,在读写1000W条之前速度还是可以的(大概2000条/秒左右),但过了1000W之后(变成400条/秒左右),当然这个与SQL SERVER读取,网络还有服务器等性能都是有关系的,但,这速度,不晓得有测试过的朋友受不受不了,我反正是受不了的,于是想了下,单线程慢,咱得改吧。改成多线程,多进程嘛。

另外再啰嗦一句,经小弟实测,改之后,效率真是快很多。。。。

不再啰嗦,直接上代码吧。

  1 #!/usr/bin/perl
  2 use DBI;
  3 use Switch;
  4 use strict;
  5 use Net::HandlerSocket;
  6 use threads;
  7 use Time::HiRes 'time';
  8 
  9 my $source_name = "##ODBC配置##";
 10 my $source_user_name = "##隐去的MSSQL数据库用户名##";
 11 my $source_user_psd = '##隐去的MSSQL数据库密码##';
 12 
 13 my $aim_ip='##隐去的目标MySQL数据库IP##';
 14 my $aim_db_name = "##隐去的目标MySQL数据库名##";
 15 ##关于HandlerSocket的安装,配置神马的,小编不想啰嗦,网上很多现成的东西。只是很纠结,以前HandlerSocket有一个很好的官网,不晓得怎么回事,被干了~~
 16 my $hs_port = 9999;
 17 
 18 my $dbh=DBI->connect("dbi:ODBC:$source_name",$source_user_name,$source_user_psd);
 19 #获取所有的用户表,不导有地理字段的表
 20 #地理字段的表,数据量不大,使用前边那个单线程的东西就OK了
 21 my $sth=$dbh->prepare("select name,object_id from sys.all_objects ao where type='U' and not exists(
 22 select 1 from  sys.all_columns col where col.object_id=ao.object_id and system_type_id=240)");
 23 $sth->execute();
 24 
 25 ##线程数。。。这个很纠结,小弟的服务器,在导的时候,5个线程以上,服务器会挂起~~~~
 26 ##于是想想,把线程改成进程,都不行~~哎!看来我这基础知识还差啊。
 27 my $threads_cnt=(not defined $ARGV[0])?5:$ARGV[0];   
 28 ##每次导的记录数
 29 my $per_records=(not defined $ARGV[1])?3000:$ARGV[1];
 30 
 31 my @data;
 32 while (@data=$sth->fetchrow_array())
 33 {
 34         ##测试时用
 35 #       @data=$sth->fetchrow_array();
 36         my ($select_columns,$insert_columns,$column_count,$sort_column,$column_types);
 37         #获取某个表的列,并构建 查询,插入,列总数,列类型
 38         ##输入参数如下:
 39         ###data[0]:表名,data[1]:对像ID
 40         ##返回参数描述如下:
 41         ###$select_columns:构建SELECT的时候,列字符串
 42         ###$insert_columns:构建insert的时候,列字符串。之所以要把这两分开,因为有些类型在select的时候,会用到列属性方法,例如geometry.STAsText()
 43         ###$column_count:列数,其实可以从@$column_types得到,但@$columns_types是后边加的,此参数也就没有去掉
 44         ###$sort_column:用来排序的字段,因为总结了一下,一般第一个字段都是标识字段,主键,因此,这里只取的第一个字段
 45         ###$columns_types:列的类型列表,一个数组。因为sql server里边的某些类型的值,在进mysql的时候,需要做处理,例如geometry
 46         #######另外再啰嗦一句,很少见有能同时返回多个值的东东(当然,可能是我把C#忘得差不多了)
 47         ($select_columns,$insert_columns,$column_count,$sort_column,$column_types)=get_columns($data[0],$data[1]);
 48         #查询结果。如果是导入失败,会返回False,否则为空
 49         ##传入参数?说明请参照上边的输出参数
 50         my $relt = export_data_in ($select_columns,$insert_columns,$column_count,$sort_column,$data[0],$column_types);
 51         
 52 }
 53 
 54 
 55 
 56 $dbh->disconnect;
 57 
 58 
 59 #插入数据
 60 sub export_data_in
 61 {
 62         #构建SQL
 63         my($select_columns,$insert_columns,$columns_count,$sort_column,$table_name,$column_types) = @_;
 64         my $rows_count=0;
 65         my $dbh2=DBI->connect("dbi:ODBC:$source_name",$source_user_name,$source_user_psd);
 66         my $sth_sc=$dbh2->prepare("select count(1) from $table_name");
 67         $sth_sc->execute();
 68         my @data_count=$sth_sc->fetchrow_array();
 69         my $begin_cnt = 0;
 70         ##这里不-1,会报21
 71         my $end_cnt = $per_records - 1;
 72         while($begin_cnt <= @data_count[0])
 73         {
 74                 my @threads;
 75                 ##循环开启导数据线程
 76                 for(my $count=1;$count<=$threads_cnt;$count++)
 77                 {
 78                         ##基本,下边的SQL语句成了本程序最大的性能瓶颈了。小弟的测试中,前1000W条数据还好,但,在1000W条之后,此SQL语句的查询性能急剧下降,当然,小弟是在远程测试上边导的,(当然,我的表没分区的,有MSSQL优化经验的知道,表是可以分区的)
 79                         ##与没有对MSSQL数据库进行性能优化也有很大关系。。
 80                         my $sql_select="select *
 81                         FROM
 82                         (
 83                                 SELECT $select_columns,ROW_NUMBER() OVER (ORDER BY $sort_column) AS RowNum
 84                                 FROM $table_name
 85                         ) as t
 86                         where t.RowNum  BETWEEN $begin_cnt and $end_cnt";
 87 
 88                         print "exporting data $table_name;total:@data_count[0];now:$begin_cnt \n";
 89                         ##开线程。参数请参照上边的描述
 90                         my $res0=threads->new(\&export_data,   $table_name,$sql_select,$insert_columns,$columns_count,$column_types);
 91                         push(@threads,$res0);
 92                         $begin_cnt = $begin_cnt + $per_records;
 93                         $end_cnt = $end_cnt + $per_records;
 94                 }
 95                 ##回收
 96                 foreach(@threads)
 97                 {
 98                         $_->join;
 99                 }
100         }
101 #       $dbh2->disconnect;
102 }
103 
104 sub export_data
105 {
106         my $startTime=time;
107         my ($table_name,$sql_select,$insert_columns,$columns_count,$column_types)=@_;
108         my $dbh_mssql=DBI->connect("dbi:ODBC:$source_name",$source_user_name,$source_user_psd);
109 
110         my $sth_select=$dbh_mssql->prepare($sql_select);
111         $sth_select->execute();
112         $sth_select->{LongTruncOk}=1;
113         ##生成标识ID
114         my $gid=rand(3200);
115         my $data_str="";
116 
117         my $select_data;
118         ##还是改成fetchrow_arrayref(),小弟测试了下,这个的速度,真不是之前fetchrow_array能比的
119         ##另外,我这种拼handlerSocket多条插入语句的方式,应该还是有点问题。我记得有一种更优化的方式是:insert,但由于找不到资料,只好作罢。
120         while($select_data=$sth_select->fetchrow_arrayref())
121         {
122                 if($data_str ne "")
123                 {
124                         $data_str="$data_str,";
125                 }
126 
127                 $data_str=$data_str."[$gid,'+',['".join("','",@{$select_data})."']]";
128 
129         }
130         printf("读出时间%.1f seconds.\n",time-$startTime);
131         $startTime=time;
132         ##测试的时候,查看数据的语句。
133 #       print "\n",$data_str,"\n";
134         if($data_str ne "")
135         {
136             $data_str="[$data_str]";
137             my $args = { host => $aim_ip, port => $hs_port };
138             my $hs = new Net::HandlerSocket($args);
139             my $res = $hs->open_index($gid, $aim_db_name, $table_name, 'PRIMARY', "$insert_columns");
140             die $hs->get_error() if $res != 0;
141             ##这里不加EVAL不行的,不信?您试
142             $res = $hs->execute_multi(eval($data_str));
143             die $hs->get_error() if $hs->get_error() != 0;
144             $hs->close();
145         }
146         undef $data_str;
147         printf("写入时间%.1f seconds.\n",time-$startTime);
148         
149         ##这里啰嗦一下,也给大家展示一下我的结果  ^-^
150             # exporting data t_p_areagroup_plate_userdiy_l;total:42758121;now:12825000
151             # exporting data t_p_areagroup_plate_userdiy_l;total:42758121;now:12830000
152             # exporting data t_p_areagroup_plate_userdiy_l;total:42758121;now:12835000
153             # exporting data t_p_areagroup_plate_userdiy_l;total:42758121;now:12840000
154             # exporting data t_p_areagroup_plate_userdiy_l;total:42758121;now:12845000
155             # 读出时间18.9 seconds.
156             # 写入时间1.3 seconds.
157             # 读出时间23.3 seconds.
158             # 写入时间1.4 seconds.
159             # 读出时间23.7 seconds.
160             # 写入时间1.1 seconds.
161             # 读出时间25.6 seconds.
162             # 写入时间0.6 seconds.
163             # 读出时间25.6 seconds.
164             # 写入时间0.9 seconds.
165         ##怎么样,写的速度够快吧,这就是TMD  HandlerSocket,而且,还不用去考滤锁。
166 }
167 
168 sub get_columns
169 {
170         print "loading columns of $_[0] \n";
171         my $sql="select col.name,tp.name from sys.all_columns col
172                                         inner join sys.types tp on col.system_type_id=tp.system_type_id  and col.user_type_id=tp.user_type_id
173                                         where object_id=$_[1]";
174         my $dbh2=DBI->connect("dbi:ODBC:$source_name",$source_user_name,$source_user_psd);
175         my $cols=$dbh2 -> prepare($sql);
176         $cols->execute();
177         my $cols_insert = "";
178         my $cols_select = "";
179         my $cols_count = 0;
180         my $sort_column="";
181         my @cols_types;
182         my @col;
183         while(@col= $cols->fetchrow_array())
184         {
185                 my ($col_name,$type_name)=@col;
186                 @cols_types[$cols_count]=$type_name;
187                 if($cols_count>0)
188                 {
189                         $cols_insert="$cols_insert,";
190                         $cols_select="$cols_select ,";
191                 }
192                 else
193                 {
194                         $sort_column="[$col_name]";
195                 }
196                 if($type_name eq "hierarchyid")
197                 {
198                         $cols_select = "$cols_select [$col_name].ToString() as $col_name";
199                         $cols_insert = "$cols_insert$col_name";
200                 }
201                 else
202                 {
203                         $cols_select="$cols_select [$col_name]";
204                         $cols_insert = "$cols_insert$col_name";
205                 }
206                 $cols_count++;
207         }
208         $dbh2->disconnect;
209         ($cols_select,$cols_insert,$cols_count,$sort_column,\@cols_types);
210 }
211 
212 

调用方法(将运行结果放到out.log):

1 nohup perl export_data_muti_thread_v0.5.pl 10 5000 > out.log &

 

 

另外再啰嗦一句。。。cnblogs的回复真不多,哪怕是拍砖也好呀。别这么死气沉沉的。

posted on 2012-06-14 22:55  davyfamer  阅读(1617)  评论(4编辑  收藏  举报