20220205092400_group_member_copy.php 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170
  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_GROUP_;#源数据库
  25. $src_table = _SQLITE_TABLE_GROUP_MEMBER_;#源表名
  26. $dest_db = _PG_DB_GROUP_;#目标数据库
  27. $dest_table = _PG_TABLE_GROUP_MEMBER_;#目标表名
  28. fwrite(STDOUT,"migarate group member ".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. user_id,
  45. group_id,
  46. group_name,
  47. power,
  48. level,
  49. status,
  50. created_at,
  51. updated_at)
  52. VALUES (? , ? , ? , ?, ? , ? ,? , now() , now() )";
  53. $stmtDEST = $PDO_DEST->prepare($queryInsert);
  54. $commitData = [];
  55. $allInsertCount = 0;
  56. $allSrcCount = 0;
  57. $count = 0;
  58. #从user数据表中读取
  59. $query = "SELECT id FROM ".$user_table." WHERE userid = ? ";
  60. $stmtUser = $PDO_USER->prepare($query);
  61. #从源数据表中读取
  62. $query = "SELECT * FROM ".$src_table;
  63. $stmtSrc = $PDO_SRC->prepare($query);
  64. $stmtSrc->execute();
  65. while($srcData = $stmtSrc->fetch(PDO::FETCH_ASSOC)){
  66. $allSrcCount++;
  67. $queryExist = "SELECT * FROM ".$dest_table." where user_id=? and group_id=? ";
  68. $stmtExist = $PDO_DEST->prepare($queryExist);
  69. $stmtExist->execute([$srcData["user_id"],$srcData["group_id"]]);
  70. $isExist = $stmtExist->fetch(PDO::FETCH_ASSOC);
  71. if($isExist){
  72. echo "record is existed id=".$srcData['id'].PHP_EOL;
  73. continue;
  74. }
  75. {
  76. if($srcData["user_id"]=='visuddhinanda'){
  77. $srcData["user_id"] = 'ba5463f3-72d1-4410-858e-eadd10884713';
  78. }
  79. if($srcData["user_id"]=='test7'){
  80. $srcData["user_id"] = '6bd2f4d7-d970-419c-8ee5-f4bac42f4bc1';
  81. }
  82. if($srcData["user_id"]=='Dhammadassi'){
  83. $srcData["user_id"] = 'd8538ebd-d369-4777-b99a-3ccb1aff8bfc';
  84. }
  85. if($srcData["user_id"]=='pannava'){
  86. $srcData["user_id"] = '4db550c4-bc1b-43f2-a518-2740cb478f37';
  87. }
  88. if($srcData["user_id"]=='NST'){
  89. $srcData["user_id"] = '5c23e629-56a3-48e9-97c7-2af73b59c3b9';
  90. }
  91. if($srcData["user_id"]=='viranyani'){
  92. $srcData["user_id"] = 'C1AB2ABF-EAA8-4EEF-B4D9-3854321852B4';
  93. }
  94. if($srcData["user_id"]=='test6'){
  95. $srcData["user_id"] = 'f81c7140-64b4-4025-b58c-45a3b386324a';
  96. }
  97. if($srcData["user_id"]=='test28'){
  98. $srcData["user_id"] = 'df0ad9bc-c0cd-4cd9-af05-e43d23ed57f0';
  99. }
  100. if($srcData["user_id"]=='290fd808-2f46-4b8c-b300-0367badd67ed'){
  101. $srcData["user_id"] = 'f81c7140-64b4-4025-b58c-45a3b386324a';
  102. }
  103. if($srcData["user_id"]=='BA837178-9ABD-4DD4-96A0-D2C21B756DC4'){
  104. $srcData["user_id"] = 'ba5463f3-72d1-4410-858e-eadd10884713';
  105. }
  106. $stmtUser->execute(array($srcData["user_id"]));
  107. $userId = $stmtUser->fetch(PDO::FETCH_ASSOC);
  108. if(!$userId){
  109. fwrite(STDERR,time()."error,no user id {$srcData["user_id"]}".PHP_EOL);
  110. continue;
  111. }
  112. }
  113. if(strlen($srcData["user_id"])>36){
  114. fwrite(STDERR,time().",error,user id too long {$srcData["user_id"]}".PHP_EOL);
  115. continue;
  116. }
  117. #插入目标表
  118. $commitData = array(
  119. $snowflake->id(),
  120. $srcData["user_id"],
  121. $srcData["group_id"],
  122. $srcData["group_name"],
  123. $srcData["power"],
  124. $srcData["level"],
  125. $srcData["status"]
  126. );
  127. try{
  128. $stmtDEST->execute($commitData);
  129. }catch (Exception $e) {
  130. echo "Failed: " . $e->getMessage();
  131. }
  132. $count++;
  133. $allInsertCount++;
  134. if($count ==10000){
  135. #10000行输出log 一次
  136. echo "finished $count".PHP_EOL;
  137. $count=0;
  138. }
  139. }
  140. fwrite(STDOUT,"insert done $allInsertCount in $allSrcCount ".PHP_EOL) ;
  141. fwrite(STDOUT, "all done in ".(time()-$start)."s".PHP_EOL);
  142. fclose($fpError);