redis_import_dict.php 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  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. while (($data = fgets($fp)) !== false) {
  45. $data1 = explode(",",trim($data));
  46. if(count($data1)>7){
  47. if($data1[2]==".comp." && $data1[1]===$data1[7]){
  48. continue;
  49. }
  50. $old = $redis->hGet($task->rediskey,$data1[$task->keycol]);
  51. $new = array();
  52. if($old){
  53. $new = json_decode($old,true);
  54. array_push($new,$data1);
  55. }
  56. else{
  57. $new[] = $data1;
  58. }
  59. $redis->hSet($task->rediskey,$data1[$task->keycol],json_encode($new, JSON_UNESCAPED_UNICODE));
  60. }
  61. $count++;
  62. if($count%50000==0){
  63. sleep(1);
  64. echo $count."\n";
  65. }
  66. }
  67. fclose($fp);
  68. sleep(1);
  69. echo "task : {$task->rediskey}:".$redis->hLen($task->rediskey)."\n";
  70. } else {
  71. echo "can not open csv file. ";
  72. }
  73. }
  74. }
  75. ?>