2
0

20220206143600_fileindex_copy.php 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. <?php
  2. /*
  3. 迁移 article 库
  4. 从旧数据表中提取数据插入到新的表
  5. 插入时用uuid判断是否曾经插入
  6. 曾经插入就不插入了
  7. */
  8. require_once __DIR__."/../../../public/app/config.php";
  9. require_once __DIR__."/../../../public/app/public/snowflakeid.php";
  10. set_exception_handler(function($e){
  11. fwrite(STDERR,"error-msg:".$e->getMessage().PHP_EOL);
  12. fwrite(STDERR,"error-file:".$e->getFile().PHP_EOL);
  13. fwrite(STDERR,"error-line:".$e->getLine().PHP_EOL);
  14. exit;
  15. });
  16. $start = time();
  17. # 雪花id
  18. $snowflake = new SnowFlakeId();
  19. $fpError = fopen(__DIR__.'/log/'.basename($_SERVER['PHP_SELF'],'.php').".err.data.csv",'w');
  20. #user info
  21. $user_db=_FILE_DB_USERINFO_;#user数据库
  22. $user_table=_TABLE_USER_INFO_;#user表名
  23. #
  24. $src_db = _SQLITE_DB_FILEINDEX_;#源数据库
  25. $src_table = _SQLITE_TABLE_FILEINDEX_;#源表名
  26. $dest_db = _PG_DB_FILEINDEX_;#目标数据库
  27. $dest_table = _PG_TABLE_FILEINDEX_;#目标表名
  28. fwrite(STDOUT,"migarate file index".PHP_EOL);
  29. #打开user数据库
  30. $PDO_USER = new PDO($user_db,_DB_USERNAME_,_DB_PASSWORD_,array(PDO::ATTR_PERSISTENT=>true));
  31. $PDO_USER->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
  32. fwrite(STDOUT,"open user table".PHP_EOL);
  33. #打开源数据库
  34. $PDO_SRC = new PDO($src_db,_DB_USERNAME_,_DB_PASSWORD_,array(PDO::ATTR_PERSISTENT=>true));
  35. $PDO_SRC->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
  36. fwrite(STDOUT,"open src table".PHP_EOL);
  37. #打开目标数据库
  38. $PDO_DEST = new PDO($dest_db,_DB_USERNAME_,_DB_PASSWORD_,array(PDO::ATTR_PERSISTENT=>true));
  39. $PDO_DEST->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
  40. fwrite(STDOUT,"open dest table".PHP_EOL);
  41. $queryInsert = "INSERT INTO ".$dest_table."
  42. (
  43. id,
  44. uid,
  45. parent_id,
  46. user_id,
  47. book,
  48. paragraph,
  49. channal,
  50. file_name,
  51. title,
  52. tag,
  53. status,
  54. file_size,
  55. share,
  56. doc_info,
  57. doc_block,
  58. create_time,
  59. modify_time,
  60. accese_time,
  61. accesed_at,
  62. updated_at,
  63. created_at)
  64. VALUES (? , ? , ? , ? , ?, ? ,? , ? , ? , ?, ? ,? , ? , ? , ?, ? ,? , ? , ? , ?, ? )";
  65. $stmtDEST = $PDO_DEST->prepare($queryInsert);
  66. $commitData = [];
  67. $allInsertCount = 0;
  68. $allSrcCount = 0;
  69. $count = 0;
  70. #从user数据表中读取
  71. $query = "SELECT id FROM ".$user_table." WHERE id = ? ";
  72. $stmtUser = $PDO_USER->prepare($query);
  73. #从源数据表中读取
  74. $query = "SELECT * FROM ".$src_table;
  75. $stmtSrc = $PDO_SRC->prepare($query);
  76. $stmtSrc->execute();
  77. while($srcData = $stmtSrc->fetch(PDO::FETCH_ASSOC)){
  78. $allSrcCount++;
  79. $queryExist = "SELECT * FROM ".$dest_table." where uid=? ";
  80. $stmtExist = $PDO_DEST->prepare($queryExist);
  81. $stmtExist->execute([$srcData["id"]]);
  82. $isExist = $stmtExist->fetch(PDO::FETCH_ASSOC);
  83. if($isExist){
  84. continue;
  85. }
  86. $stmtUser->execute(array($srcData["user_id"]));
  87. $userId = $stmtUser->fetch(PDO::FETCH_ASSOC);
  88. if(!$userId){
  89. fwrite(STDERR,time()."error,no user id {$srcData["user_id"]}".PHP_EOL);
  90. continue;
  91. }
  92. if(empty($srcData["doc_block"]) ){
  93. fwrite(STDERR,time().",error, doc_block is empty {$srcData["id"]}".PHP_EOL);
  94. fputcsv($fpError,$srcData);
  95. continue;
  96. }
  97. if(empty($srcData["book"]) ){
  98. $srcData["book"] = 0;
  99. fwrite(STDERR,time().",error, book is empty {$srcData["id"]}".PHP_EOL);
  100. fputcsv($fpError,array('book is empty',$srcData["id"]));
  101. }
  102. if(empty($srcData["paragraph"]) ){
  103. $srcData["paragraph"] = 0;
  104. fwrite(STDERR,time().",error, paragraph is empty {$srcData["id"]}".PHP_EOL);
  105. fputcsv($fpError,array('paragraph is empty',$srcData["id"]));
  106. }
  107. if(empty($srcData["modify_time"])){
  108. $srcData["modify_time"] = $srcData["create_time"];
  109. }
  110. if($srcData["create_time"] < 15987088320){
  111. $srcData["create_time"] *= 1000;
  112. }
  113. if($srcData["modify_time"] < 15987088320){
  114. $srcData["modify_time"] *= 1000;
  115. }
  116. //查询是否已经插入
  117. $queryExsit = "SELECT id FROM ".$dest_table." WHERE uid = ? ";
  118. $getExist = $PDO_DEST->prepare($queryExsit);
  119. $getExist->execute(array($srcData["id"]));
  120. $exist = $getExist->fetch(PDO::FETCH_ASSOC);
  121. if($exist){
  122. continue;
  123. }
  124. #插入目标表
  125. $created_at = date("Y-m-d H:i:s.",$srcData["create_time"]/1000).($srcData["create_time"]%1000)." UTC";
  126. $updated_at = date("Y-m-d H:i:s.",$srcData["modify_time"]/1000).($srcData["modify_time"]%1000)." UTC";
  127. $commitData = array(
  128. $snowflake->id(),
  129. $srcData["id"],
  130. $srcData["parent_id"],
  131. $srcData["user_id"],
  132. $srcData["book"],
  133. $srcData["paragraph"],
  134. $srcData["channal"],
  135. $srcData["file_name"],
  136. $srcData["title"],
  137. $srcData["tag"],
  138. $srcData["status"],
  139. $srcData["file_size"],
  140. $srcData["share"],
  141. $srcData["doc_info"],
  142. $srcData["doc_block"],
  143. $srcData["create_time"],
  144. $srcData["modify_time"],
  145. $srcData["accese_time"],
  146. $created_at,
  147. $updated_at,
  148. $updated_at
  149. );
  150. $stmtDEST->execute($commitData);
  151. $count++;
  152. $allInsertCount++;
  153. if($count ==10000){
  154. #10000行输出log 一次
  155. echo "finished $count".PHP_EOL;
  156. $count=0;
  157. }
  158. }
  159. fwrite(STDOUT,"insert done $allInsertCount in $allSrcCount ".PHP_EOL) ;
  160. fwrite(STDOUT, "all done in ".(time()-$start)."s".PHP_EOL);
  161. fclose($fpError);