用perl写的一个导库程序

litdong

刚学perl,google了些网上的sample code组合了一下,觉得很乱,帮忙看看如何改进,大致是这样,数据库被导出成xml格式,需要perl解析xml并写到数据库中

1. 
   
2. #!/usr/bin/perl -w
   
3. use strict;
   
4. use XML::LibXML;
   
5. use XML::LibXML::Common;
   
6. use DBI;
   
7. 
   
8. 
   
9. # 处理参数传入,如果没有参数,则显示提示帮助信息给用户
   
10. if ( @ARGV < 1 ) {
    
11.         print STDERR "\n == INTRODUCTION ==\n";
    
12.         print STDERR "\n == PARAMETERS ==\n";
    
13.         print STDERR "  -f <mandatory input> xml file you want to load\n";
    
14.         print STDERR "  -h <mandatory input> database host name  \n";
    
15.         print STDERR "  -u <mandatory input> database user name  \n";
    
16.         print STDERR "  -p <optional input> database password,default is empty \n";
    
17.         print STDERR "  -d <mandatory input> database name  \n";
    
18.         print STDERR "\n == EXAMPLE ==\n";
    
19.         print STDERR "$0 -f /usr/demo.xml -h 135.252.234.35 -u mysql -p -d fmbb \n";
    
20.         exit;
    
21. }
    
22. 
    
23. my ( $temp, $xmlfile, $hostname, $username, $password, $databasename, );
    
24. 
    
25. 
    
26. # 密码缺省可设置为空
    
27. $password = "";
    
28. 
    
29. 
    
30. #遍历参数,赋予相应变量
    
31. while (@ARGV) {
    
32.         $temp = shift;
    
33.         if    ( $temp eq "-f" ) { $xmlfile      = shift; }
    
34.         elsif ( $temp eq "-h" ) { $hostname     = shift; }
    
35.         elsif ( $temp eq "-u" ) { $username     = shift; }
    
36.         elsif ( $temp eq "-p" ) { $password     = shift; }
    
37.         elsif ( $temp eq "-d" ) { $databasename = shift; }
    
38.         else {
    
39.                 print STDERR "Warning: argument $temp igorned.\n";
    
40.         }
    
41. }
    
42. 
    
43. #检查各个传入的参数值是否为空
    
44. if ( !defined($xmlfile) ) {
    
45.         print STDERR "Error: argument '-f' is needed.\n";
    
46.         exit;
    
47. }
    
48. 
    
49. if ( !defined($hostname) ) {
    
50.         print STDERR "Error: argument '-h' is needed.\n";
    
51.         exit;
    
52. }
    
53. 
    
54. if ( !defined($databasename) ) {
    
55.         print STDERR "Error: argument '-d' is needed.\n";
    
56.         exit;
    
57. }
    
58. 
    
59. if ( !defined($username) ) {
    
60.         print STDERR "Error: argument '-u' is needed.\n";
    
61.         exit;
    
62. }
    
63. 
    
64. #根据传入参数,拼接连接数据库的字符串
    
65. my $connectionstr = "DBI:mysql:" . $databasename . ";host=" . $hostname;
    
66. 
    
67. print "Conect to database " . $hostname . " " . "$databasename" . "... \n";
    
68. 
    
69. #连接数据库
    
70. my $dbh =
    
71.   DBI->connect( $connectionstr, $username, $password,
    
72.         { RaiseError => 1, AutoCommit => 0 } )
    
73.   || die "Could not connect to database: $DBI::errstr";
    
74. 
    
75. print "Start to parse source file ... \n";
    
76. 
    
77. #载入xml
    
78. my $parser = XML::LibXML->new();
    
79. my $doc    = $parser->parse_file($xmlfile);
    
80. 
    
81. print "Parse complete, ready to dump into database \n";
    
82. 
    
83. 
    
84. #定义xml读出结果的变量
    
85. my ( $a, $b, $c, $d, $e, $f, $g );
    
86. 
    
87. #定义执行语句的结果
    
88. my $success = 1;
    
89. 
    
90. #设置一个计数器
    
91. my $count   = 0;
    
92. 
    
93. #开始遍历xml,把读出的值放在标量中,并打印导出内容到屏幕上
    
94. my $root                     = $doc->getDocumentElement();
    
95. my @acknowledgedAlarms_nodes = $root->childNodes;
    
96. foreach my $acknowledgedAlarm (@acknowledgedAlarms_nodes) {
    
97.         print "############\n";
    
98.         if ( $acknowledgedAlarm->nodeType == ELEMENT_NODE ) {
    
99.                 foreach my $values ( $acknowledgedAlarm->childNodes ) {
    
100. 
     
101.                         if ( $values->nodeType != XML_TEXT_NODE ) {
     
102.                                 if ( $values->nodeName() eq "a" ) {
     
103.                                         print $values->nodeName() . ": "
     
104.                                           . $values->textContent . "\n";
     
105.                                         $a = $values->textContent;
     
106.                                 }
     
107.                                 elsif ( $values->nodeName() eq "b" ) {
     
108.                                         print $values->nodeName() . ": "
     
109.                                           . $values->textContent . "\n";
     
110.                                         $b = $values->textContent;
     
111.                                 }
     
112.                                 elsif ( $values->nodeName() eq "c" ) {
     
113.                                         print $values->nodeName() . ": "
     
114.                                           . $values->textContent . "\n";
     
115.                                         $c = $values->textContent;
     
116.                                 }
     
117.                                 elsif ( $values->nodeName() eq "d" ) {
     
118.                                         print $values->nodeName() . ": "
     
119.                                           . $values->textContent . "\n";
     
120.                                         $d = $values->textContent;
     
121.                                 }
     
122.                                 elsif ( $values->nodeName() eq "e" ) {
     
123.                                         print $values->nodeName() . ": "
     
124.                                           . $values->textContent . "\n";
     
125.                                         $e = $values->textContent;
     
126.                                 }
     
127.                                 elsif ( $values->nodeName() eq "f" ) {
     
128.                                         print $values->nodeName() . ": "
     
129.                                           . $values->textContent . "\n";
     
130.                                         $f = $values->textContent;
     
131.                                 }
     
132.                                 elsif ( $values->nodeName() eq "g" ) {
     
133.                                         print $values->nodeName() . ": "
     
134.                                           . $values->textContent . "\n";
     
135.                                         $g = $values->textContent;
     
136.                                 }
     
137.                                 else {
     
138.                                         print $values->nodeName() . ": " . $values->textContent;
     
139.                                         print "\n";
     
140.                                 }
     
141.                         }
     
142.                 }
     
143. 
     
144.                 #计数器加1
     
145.                 $count++;
     
146.                 print "############\n";
     
147. 
     
148.                 #执行,当然最后commit才会放到数据库中
     
149.                 my $insert_handle =
     
150.                   $dbh->prepare_cached('INSERT INTO ack_alarms VALUES (?,?,?,?,?,?,?)');
     
151.                 $success &&=
     
152.                   $insert_handle->execute( $a, $b, $c,
     
153.                         $d, $e, $f, $g );
     
154. 
     
155.         }
     
156. }
     
157. 
     
158. #commit到数据库中,有问题则回滚
     
159. my $result = ( $success ? $dbh->commit : $dbh->rollback );
     
160. 
     
161. #写库失败的话终止
     
162. unless ($result) {
     
163.         $count=0;
     
164.         $dbh->disconnect();
     
165.         die "Couldn't finish transaction: " . $dbh->errstr;
     
166. }
     
167. 
     
168. #断开数据库链接
     
169. $dbh->disconnect();
     
170. 
     
171. print "Totally import " . $count . " rows to database \n";
     
172. 
     

复制代码

shileiadmin

1、能把注释加上么？
2、你具体要改进什么？

ynchnluiti

第90 95 100 105 110 115 120 125 行的print可以合成一行

另外，没看出对字母a,b,...,g的判断及$a,...$g的用处，或许可以简化

Mr-Summer

我记得mysql cookbook后面有一章专门讲这东西的。lz可以去翻一下