db_insert_bookword_from_csv_cli.php 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. <?php
  2. /*
  3. 生成 巴利原文段落表
  4. */
  5. require_once __DIR__."/../config.php";
  6. require_once __DIR__.'/../public/_pdo.php';
  7. echo "Insert Pali Text To DB".PHP_EOL;
  8. if ($argc != 3) {
  9. echo "help".PHP_EOL;
  10. echo $argv[0]." from to".PHP_EOL;
  11. echo "from = 1-217".PHP_EOL;
  12. echo "to = 1-217".PHP_EOL;
  13. exit;
  14. }
  15. $_from = (int) $argv[1];
  16. $_to = (int) $argv[2];
  17. if ($_to > 217) {
  18. $_to = 217;
  19. }
  20. $dirLog = _DIR_LOG_ . "/";
  21. $dirXmlBase = _DIR_PALI_CSV_ . "/";
  22. $filelist = array();
  23. $fileNums = 0;
  24. $log = "";
  25. //PostgreSQL
  26. define("_DB_", _DB_ENGIN_.":host="._DB_HOST_.";port="._DB_PORT_.";dbname="._DB_NAME_.";user="._DB_USERNAME_.";password="._DB_PASSWORD_.";");
  27. define("_TABLE_", "book_words");
  28. global $dbh_word_index;
  29. $dns = _DB_;
  30. $dbh_word_index = new PDO($dns, _DB_USERNAME_, _DB_PASSWORD_, array(PDO::ATTR_PERSISTENT => true));
  31. $dbh_word_index->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_WARNING);
  32. if (($handle = fopen(__DIR__."/filelist.csv", 'r')) !== false) {
  33. while (($filelist[$fileNums] = fgetcsv($handle, 0, ',')) !== false) {
  34. $fileNums++;
  35. }
  36. }
  37. if ($_to == 0 || $_to >= $fileNums) {
  38. $_to = $fileNums ;
  39. }
  40. for ($from=$_from-1; $from < $_to; $from++) {
  41. echo "doing ".($from+1).PHP_EOL;
  42. $bookword = array();
  43. if (($fpoutput = fopen(_DIR_CSV_PALI_CANON_WORD_ . "/{$from}_words.csv", "r")) !== false) {
  44. $count = 0;
  45. while (($data = fgetcsv($fpoutput, 0, ',')) !== false) {
  46. $book = $data[1];
  47. if (isset($bookword[$data[3]])) {
  48. $bookword[$data[3]]++;
  49. } else {
  50. $bookword[$data[3]] = 1;
  51. }
  52. $count++;
  53. }
  54. }
  55. #删除原来的数据
  56. $query = "DELETE FROM "._TABLE_." WHERE book = ?";
  57. $stmt = $dbh_word_index->prepare($query);
  58. $stmt->execute(array($book));
  59. if (!$stmt || ($stmt && $stmt->errorCode() != 0)) {
  60. $error = $dbh_word_index->errorInfo();
  61. echo "error - $error[2]".PHP_EOL;
  62. $log .= "$from, $FileName, error, $error[2] \r\n";
  63. }else{
  64. // 开始一个事务,关闭自动提交
  65. $dbh_word_index->beginTransaction();
  66. $query = "INSERT INTO "._TABLE_." (book , wordindex , count) VALUES ( ? , ? , ? )";
  67. $stmt = $dbh_word_index->prepare($query);
  68. foreach ($bookword as $key => $value) {
  69. $stmt->execute(array($book, $key, $value));
  70. }
  71. // 提交更改
  72. $dbh_word_index->commit();
  73. if (!$stmt || ($stmt && $stmt->errorCode() != 0)) {
  74. $error = $dbh_word_index->errorInfo();
  75. echo "error - $error[2]".PHP_EOL;
  76. $log .= "$from, $FileName, error, $error[2] \r\n";
  77. } else {
  78. echo "updata $count recorders.".PHP_EOL;
  79. $log .= "updata $count recorders.\r\n";
  80. }
  81. }
  82. /*
  83. $myLogFile = fopen($dirLog . "insert_index.log", "a");
  84. fwrite($myLogFile, $log);
  85. fclose($myLogFile);
  86. */
  87. }
  88. echo "齐活!功德无量!all done!".PHP_EOL;
  89. ?>