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

话不多说，直接上代码。

#!/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;

}