20211218132000_collections_copy.php 4.4 KB

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