用perl做数据库迁移，从MSSQL到MYSQL（二）自动建表

dcvb

　　话不多说，直接上代码。



#!/usr/bin/perl
use DBI;
use Switch;
#申请连接资源，因为Perl连接SQL Server只能用ODBC，如果还没部署好环境的，请参见小弟另一篇博文  【PERL连MSSQL】 http://www.cnblogs.com/davyfamer/archive/2012/05/31/2528818.html
$dbh=DBI->connect('dbi:ODBC:##隐去的ODBC数据库连接##','##隐去的MSSQL用户名##','##隐去的MSSQL数据库密码##');
#获取所有需要导入到MySQL的表
my $sth=$dbh->prepare("select name,object_id from sys.all_objects where type='U' and is_ms_shipped=0 and name <>'sysdiagrams'");
$sth->execute();
while (@data=$sth->fetchrow_array())
{
#@data=$sth->fetchrow_array();#测试时禁用循环，只看一张表
#获取列
get_columns($data[0],$data[1]);
}
#$dbh ->disconnect;   #添加了断开语句后，会报错，难道ODBC不需要断开？有点头晕。
##获取所有的列
sub get_columns
{
$dbh2=DBI->connect('dbi:ODBC:##隐去的ODBC数据库连接##','##隐去的MSSQL的用户名##','##隐去的MSSQL数据库的密码##');
my $sql="select col.name,tp.name,col.max_length,col.[precision],col.[scale],col.[is_nullable],col.[is_identity] from sys.all_columns col
inner join sys.types tp on col.system_type_id=tp.system_type_id  and col.user_type_id=tp.user_type_id
where object_id=$_[1]";
my $cols=$dbh2 -> prepare($sql);
$cols->execute();
my $cols_str = "";
while(@col= $cols->fetchrow_array())
{
($col_name,$type_name,$max_length,$precision,$scale,$is_nullable,$is_identity)=@col;
#做类型转换。。。
switch ( $type_name)
{
case "nchar"  { $type_name="char"; }
case "bit"  { $type_name="boolean"; }
case "ntext"  { $type_name="text"; }
case "nvarchar"  { $type_name="varchar"; }
else { $type_name=$type_name;}
}
if($cols_str ne "")
{
$cols_str = "$cols_str, \n";
}
#对hierarchyid 做特殊处理
if($type_name eq "hierarchyid")
{
if($cols_str eq "")
{
#$cols_str = "$cols_str `id` INT NOT NULL AUTO_INCREMENT PRIMARY KEY,`parent_id` INT";
$cols_str = "$cols_str `$col_name` varchar(31)";
}
else
{
$cols_str = "$cols_str `$col_name`varchar(31)";
}
}
else
{
$cols_str = "$cols_str `$col_name` $type_name";
switch($type_name)
{
case "varchar"
{
#sql server中的VARCHAR(MAX)
if($max_length == -1)
{
$cols_str = "$cols_str(65535)";
}
else
{
$cols_str = "$cols_str($max_length)";
}
}
case "numeric"
{
$cols_str = "$cols_str($precision,$scale)";
}
}
#是否能为空
if($is_nullable == 0)
{
$cols_str="$cols_str NOT NULL ";
}
#是否为标识符
if($is_identity == 1)
{
$cols_str="$cols_str AUTO_INCREMENT PRIMARY KEY";
}
}
}
#print "create table IF NOT EXISTS `$_[0]`($cols_str);";
exec_mysql( "create table IF NOT EXISTS `$_[0]`($cols_str);");
#$dbh2 ->disconnect;  #这一句加了会报错
}
#在MySQL中建表
sub exec_mysql
{
my $db_name="##隐去的MySQL目标数据库名##";
my $location="##隐去的MySQL数据库IP地址##";
my $port="##隐去的MySQL数据库端口##";
my $data_base = "DBI:mysql:$db_name:$location:$port";
my $db_user="##隐去的MySQL数据库用户名##";
my $db_pass="##隐去的MySQL数据库密码##";
my $dbh3=DBI -> connect($data_base,$db_user,$db_pass);
my $sth=$dbh3->prepare($_[0]);
$sth->execute() or die "ERROR::$_[0]::$dbh3->errstr";
$dbh3->disconnect;
}