2
0

redis_import_dict.php 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. <?php
  2. require_once "../path.php";
  3. require_once "../install/filelist.php";
  4. require_once "../redis/function.php";
  5. if (PHP_SAPI == "cli") {
  6. if ($argc >= 2) {
  7. $list = $argv[1];
  8. if(isset($argv[2])){
  9. $tableNum = (int)$argv[2];
  10. }
  11. else{
  12. $tableNum = -1;
  13. }
  14. $redis = redis_connect();
  15. if ($redis == false) {
  16. echo "no redis connect\n";
  17. exit;
  18. }
  19. $taskList = json_decode(file_get_contents(__DIR__."/".$list));
  20. $dir = dirname(__DIR__."/".$list);
  21. if($tableNum<0){
  22. foreach ($taskList as $key => $task) {
  23. # code...
  24. runTask($redis,$task,$dir);
  25. }
  26. }
  27. else{
  28. if($tableNum<count($taskList)){
  29. runTask($redis,$taskList[$tableNum],$dir);
  30. }
  31. else{
  32. echo "wrong task number task length is ".count($taskList);
  33. }
  34. }
  35. }
  36. }
  37. function runTask($redis,$task,$dir){
  38. $count=0;
  39. $redis->del($task->rediskey);
  40. foreach ($task->csv as $csv) {
  41. $csvfile = $dir."/".$csv;
  42. if (($fp = fopen($csvfile, "r")) !== false) {
  43. echo "单词表load {$csvfile}\n";
  44. $row=0;
  45. while (($data = fgets($fp)) !== false) {
  46. $row++;
  47. $data1 = explode(",",trim($data));
  48. if(count($data1)>7){
  49. if($data1[2]==".comp." && $data1[1]===$data1[7]){
  50. continue;
  51. }
  52. $old = $redis->hGet($task->rediskey,$data1[$task->keycol]);
  53. $new = array();
  54. if($old){
  55. $new = json_decode($old,true);
  56. array_push($new,$data1);
  57. }
  58. else{
  59. $new[] = $data1;
  60. }
  61. $redis->hSet($task->rediskey,$data1[$task->keycol],json_encode($new, JSON_UNESCAPED_UNICODE));
  62. }
  63. else{
  64. echo "列不足够:行:{$row} 列:".count($data1)." 数据:{$data} \n";
  65. }
  66. $count++;
  67. if($count%50000==0){
  68. sleep(1);
  69. echo $count."\n";
  70. }
  71. }
  72. fclose($fp);
  73. sleep(1);
  74. echo "task : {$task->rediskey}:".$redis->hLen($task->rediskey)."\n";
  75. } else {
  76. echo "can not open csv file. ";
  77. }
  78. }
  79. }
  80. ?>