2
0

20211218093500_articles_copy.php 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
  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_USER_ARTICLE_;#源数据库
  25. $src_table = _SQLITE_TABLE_ARTICLE_;#源表名
  26. $dest_db = _PG_DB_USER_ARTICLE_;#目标数据库
  27. $dest_table = _PG_TABLE_ARTICLE_;#目标表名
  28. fwrite(STDOUT,"migarate article".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. title,
  46. subtitle,
  47. summary,
  48. content,
  49. owner,
  50. owner_id,
  51. editor_id,
  52. setting,
  53. status,
  54. lang,
  55. create_time,
  56. modify_time,
  57. updated_at,
  58. created_at)
  59. VALUES (? , ? , ? , ?, ? , ? ,? , ? , ? , ? , ? , ? , ?,? ,?,?)";
  60. $stmtDEST = $PDO_DEST->prepare($queryInsert);
  61. $commitData = [];
  62. $allInsertCount = 0;
  63. $allSrcCount = 0;
  64. $count = 0;
  65. #从user数据表中读取
  66. $query = "SELECT id FROM ".$user_table." WHERE userid = ? ";
  67. $stmtUser = $PDO_USER->prepare($query);
  68. #从源数据表中读取
  69. $query = "SELECT * FROM ".$src_table;
  70. $stmtSrc = $PDO_SRC->prepare($query);
  71. $stmtSrc->execute();
  72. while($srcData = $stmtSrc->fetch(PDO::FETCH_ASSOC)){
  73. $allSrcCount++;
  74. if($srcData["owner"]=='test6'){
  75. $srcData["owner"] = 'f81c7140-64b4-4025-b58c-45a3b386324a';
  76. }
  77. if($srcData["owner"]=='test28'){
  78. $srcData["owner"] = 'df0ad9bc-c0cd-4cd9-af05-e43d23ed57f0';
  79. }
  80. if($srcData["owner"]=='290fd808-2f46-4b8c-b300-0367badd67ed'){
  81. $srcData["owner"] = 'f81c7140-64b4-4025-b58c-45a3b386324a';
  82. }
  83. if($srcData["owner"]=='BA837178-9ABD-4DD4-96A0-D2C21B756DC4'){
  84. $srcData["owner"] = 'ba5463f3-72d1-4410-858e-eadd10884713';
  85. }
  86. $stmtUser->execute(array($srcData["owner"]));
  87. $userId = $stmtUser->fetch(PDO::FETCH_ASSOC);
  88. if(!$userId){
  89. fwrite(STDERR,time()."error,no user id {$srcData["owner"]}".PHP_EOL);
  90. continue;
  91. }
  92. if(strlen($srcData["owner"])>36){
  93. fwrite(STDERR,time().",error,user id too long {$srcData["owner"]}".PHP_EOL);
  94. continue;
  95. }
  96. if(empty($srcData["lang"]) ){
  97. $srcData["lang"]='none';
  98. }
  99. //查询是否已经插入
  100. $queryExsit = "SELECT id FROM ".$dest_table." WHERE uid = ? ";
  101. $getExist = $PDO_DEST->prepare($queryExsit);
  102. $getExist->execute(array($srcData["id"]));
  103. $exist = $getExist->fetch(PDO::FETCH_ASSOC);
  104. if($exist){
  105. continue;
  106. }
  107. #插入目标表
  108. $created_at = date("Y-m-d H:i:s.",$srcData["create_time"]/1000).($srcData["create_time"]%1000)." UTC";
  109. $updated_at = date("Y-m-d H:i:s.",$srcData["modify_time"]/1000).($srcData["modify_time"]%1000)." UTC";
  110. $commitData = array(
  111. $snowflake->id(),
  112. $srcData["id"],
  113. $srcData["title"],
  114. $srcData["subtitle"],
  115. $srcData["summary"],
  116. $srcData["content"],
  117. $srcData["owner"],
  118. $userId["id"],
  119. $userId["id"],
  120. $srcData["setting"],
  121. $srcData["status"],
  122. $srcData["lang"],
  123. $srcData["create_time"],
  124. $srcData["modify_time"],
  125. $created_at,
  126. $updated_at
  127. );
  128. $stmtDEST->execute($commitData);
  129. $count++;
  130. $allInsertCount++;
  131. if($count ==10000){
  132. #10000行输出log 一次
  133. echo "finished $count".PHP_EOL;
  134. $count=0;
  135. }
  136. }
  137. fwrite(STDOUT,"insert done $allInsertCount in $allSrcCount ".PHP_EOL) ;
  138. fwrite(STDOUT, "all done in ".(time()-$start)."s".PHP_EOL);
  139. fclose($fpError);