2
0

20211215214400_channel_copy.php 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. <?php
  2. /*
  3. 迁移 sentence pr 库
  4. 从旧数据表中提取数据插入到新的表
  5. 插入时用uuid判断是否曾经插入
  6. 曾经插入就不插入了
  7. */
  8. require_once __DIR__.'/../../../public/vendor/autoload.php';
  9. require_once __DIR__."/../../../public/app/config.php";
  10. require_once __DIR__."/../../../public/app/public/snowflakeid.php";
  11. set_exception_handler(function($e){
  12. fwrite(STDERR,"error-msg:".$e->getMessage().PHP_EOL);
  13. fwrite(STDERR,"error-file:".$e->getFile().PHP_EOL);
  14. fwrite(STDERR,"error-line:".$e->getLine().PHP_EOL);
  15. exit;
  16. });
  17. $start = time();
  18. # 雪花id
  19. $snowflake = new SnowFlakeId();
  20. #user info
  21. $user_db=_FILE_DB_USERINFO_;#user数据库
  22. $user_table=_TABLE_USER_INFO_;#user表名
  23. # 更新索引表
  24. $src_db = _SQLITE_DB_CHANNAL_;#源数据库
  25. $src_table = _SQLITE_TABLE_CHANNEL_;#源表名
  26. $dest_db = _PG_DB_CHANNAL_;#目标数据库
  27. $dest_table = _PG_TABLE_CHANNEL_;#目标表名
  28. fwrite(STDOUT,"migarate channel".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. owner_uid,
  46. editor_id,
  47. name,
  48. summary,
  49. status,
  50. lang,
  51. setting,
  52. create_time,
  53. modify_time,
  54. updated_at,
  55. created_at)
  56. VALUES ( ? ,? , ? , ?, ? , ? ,? , ? , ? , ? , ? , to_timestamp(?), to_timestamp(?))";
  57. $stmtDEST = $PDO_DEST->prepare($queryInsert);
  58. $commitData = [];
  59. $allInsertCount = 0;
  60. $allSrcCount = 0;
  61. $count = 0;
  62. #从user数据表中读取
  63. $query = "SELECT id ,userid FROM ".$user_table." WHERE userid = ? or username = ? ";
  64. $stmtUser = $PDO_USER->prepare($query);
  65. #从源数据表中读取
  66. $query = "SELECT * FROM ".$src_table;
  67. $stmtSrc = $PDO_SRC->prepare($query);
  68. $stmtSrc->execute();
  69. while($srcData = $stmtSrc->fetch(PDO::FETCH_ASSOC)){
  70. $allSrcCount++;
  71. if($srcData["owner"]=='test6'){
  72. $srcData["owner"] = 'f81c7140-64b4-4025-b58c-45a3b386324a';
  73. }
  74. if($srcData["owner"]=='test28'){
  75. $srcData["owner"] = 'df0ad9bc-c0cd-4cd9-af05-e43d23ed57f0';
  76. }
  77. if($srcData["owner"]=='290fd808-2f46-4b8c-b300-0367badd67ed'){
  78. $srcData["owner"] = 'f81c7140-64b4-4025-b58c-45a3b386324a';
  79. }
  80. if($srcData["owner"]=='BA837178-9ABD-4DD4-96A0-D2C21B756DC4'){
  81. $srcData["owner"] = 'ba5463f3-72d1-4410-858e-eadd10884713';
  82. }
  83. $stmtUser->execute(array($srcData["owner"],$srcData["owner"]));
  84. $userId = $stmtUser->fetch(PDO::FETCH_ASSOC);
  85. if(!$userId){
  86. fwrite(STDERR,"no user id {$srcData["owner"]}".PHP_EOL);
  87. continue;
  88. }
  89. if(strlen($srcData["owner"])>36){
  90. fwrite(STDERR,"user id too long {$srcData["owner"]}".PHP_EOL);
  91. continue;
  92. }
  93. $uuid = $srcData["id"];
  94. #查询目标表中是否有相同数据
  95. $queryExsit = "SELECT id FROM ".$dest_table." WHERE uid = ? ";
  96. $getExist = $PDO_DEST->prepare($queryExsit);
  97. $getExist->execute(array($uuid));
  98. $exist = $getExist->fetch(PDO::FETCH_ASSOC);
  99. if($exist){
  100. #有相同数据
  101. continue;
  102. }
  103. #插入目标表
  104. $commitData = array(
  105. $snowflake->id(),
  106. $srcData["id"],
  107. $userId["userid"],
  108. $userId["id"],
  109. $srcData["name"],
  110. $srcData["summary"],
  111. $srcData["status"],
  112. $srcData["lang"],
  113. '',
  114. $srcData["create_time"],
  115. $srcData["modify_time"],
  116. $srcData["create_time"]/1000,
  117. $srcData["modify_time"]/1000
  118. );
  119. $stmtDEST->execute($commitData);
  120. $count++;
  121. $allInsertCount++;
  122. if($count ==10000){
  123. #10000行插入一次
  124. echo "finished $count".PHP_EOL;
  125. $count=0;
  126. }
  127. }
  128. fwrite(STDOUT,"insert done $allInsertCount in $allSrcCount ".PHP_EOL) ;
  129. fwrite(STDOUT,"all done".PHP_EOL);