redis_upgrade_pali_sent.php 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. <?php
  2. /*
  3. 从巴利句子列表数据库中提取数据填充redis
  4. 每个句子包含
  5. pali
  6. id
  7. sim_count
  8. */
  9. require_once "../path.php";
  10. require_once "../redis/function.php";
  11. if (isset($argv[1])) {
  12. if ($argv[1] == "del") {
  13. $redis = new redis();
  14. $r_conn = $redis->connect('127.0.0.1', 6379);
  15. $keys = $redis->keys('pali_sent_*');
  16. $count=0;
  17. foreach ($keys as $key => $value) {
  18. # code...
  19. $deleted = $redis->del($value);
  20. $count += $deleted;
  21. }
  22. echo "delete ok ".$count;
  23. }
  24. } else {
  25. $dbh = new PDO(_FILE_DB_PALI_SENTENCE_, "", "", array(PDO::ATTR_PERSISTENT => true));
  26. $dbh->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_WARNING);
  27. $db_pali_sent_sim = new PDO(_FILE_DB_PALI_SENTENCE_SIM_, "", "", array(PDO::ATTR_PERSISTENT => true));
  28. $db_pali_sent_sim->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_WARNING);
  29. $query = "SELECT id, book,paragraph, begin,end ,html FROM pali_sent WHERE 1 ";
  30. $stmt = $dbh->prepare($query);
  31. $stmt->execute();
  32. $redis = redis_connect();
  33. $stringSize = 0;
  34. $count = 0;
  35. if ($redis) {
  36. while ($sent = $stmt->fetch(PDO::FETCH_ASSOC)) {
  37. $count++;
  38. $stringSize += strlen($sent["html"]);
  39. if ($stringSize > 50000000) {
  40. sleep(1);
  41. $stringSize = 0;
  42. }
  43. if($count%10000==0){
  44. echo $count . "-".$sent["book"] . "_" . $sent["paragraph"] . "\n";
  45. }
  46. $result = $redis->hSet('pali://sent/' . $sent["book"] . "_" . $sent["paragraph"] . "_" . $sent["begin"] . "_" . $sent["end"], "pali", $sent["html"]);
  47. if($result===FALSE){
  48. echo "hSet error \n";
  49. }
  50. $result = $redis->hSet('pali://sent/' . $sent["book"] . "_" . $sent["paragraph"] . "_" . $sent["begin"] . "_" . $sent["end"], "id", $sent["id"]);
  51. $query = "SELECT count FROM 'sent_sim_index' WHERE sent_id = ? ";
  52. $sth = $db_pali_sent_sim->prepare($query);
  53. $sth->execute(array($sent["id"]));
  54. $row = $sth->fetch(PDO::FETCH_ASSOC);
  55. if ($row) {
  56. $pali_sim = $row["count"];
  57. } else {
  58. $pali_sim = 0;
  59. }
  60. $result = $redis->hSet('pali://sent/' . $sent["book"] . "_" . $sent["paragraph"] . "_" . $sent["begin"] . "_" . $sent["end"], "sim_count", $pali_sim);
  61. }
  62. echo "完成 ". count($redis->keys("pali://sent/*"));
  63. } else {
  64. echo "连接redis失败";
  65. }
  66. }