正在显示
1 个修改的文件
包含
0 行增加
和
507 行删除
cmd/sync_to_es_v2b.php
已删除
100644 → 0
| 1 | -<?php | ||
| 2 | - | ||
| 3 | -//error_reporting(); | ||
| 4 | - | ||
| 5 | -require_once "../vendor/autoload.php"; | ||
| 6 | - | ||
| 7 | - | ||
| 8 | -/** | ||
| 9 | - * 把mysql的数据同步到es | ||
| 10 | - * @author:dc | ||
| 11 | - * @time 2025/3/4 10:19 | ||
| 12 | - * Class SyncToEsCmd | ||
| 13 | - */ | ||
| 14 | -class SyncToEsCmd { | ||
| 15 | - | ||
| 16 | - public $isStop = false; | ||
| 17 | - | ||
| 18 | - /** | ||
| 19 | - * 文件夹 | ||
| 20 | - * @var array | ||
| 21 | - */ | ||
| 22 | - public $folders = []; | ||
| 23 | - | ||
| 24 | - /** | ||
| 25 | - * @var \Lib\Db | ||
| 26 | - */ | ||
| 27 | - public $fob_db; | ||
| 28 | - | ||
| 29 | - /** | ||
| 30 | - * @var \Lib\Db | ||
| 31 | - */ | ||
| 32 | - public $db; | ||
| 33 | - | ||
| 34 | - | ||
| 35 | - public $startTime = 0; | ||
| 36 | - | ||
| 37 | - /** | ||
| 38 | - * SyncToEsCmd constructor. | ||
| 39 | - */ | ||
| 40 | - public function __construct() | ||
| 41 | - { | ||
| 42 | - $this->db = db(); | ||
| 43 | - $this->fob_db = fob_mysql(); | ||
| 44 | - | ||
| 45 | - $handler = function ($signal){ | ||
| 46 | - _echo('收到进程信号 '. $signal); | ||
| 47 | - // 可以处理其他程序 | ||
| 48 | - $this->isStop = true; | ||
| 49 | - }; | ||
| 50 | - pcntl_signal(SIGTERM, $handler); // 这个是kill | ||
| 51 | - pcntl_signal(SIGINT, $handler); // 这个是 ctrl+c | ||
| 52 | - | ||
| 53 | - $this->startTime = time(); | ||
| 54 | - } | ||
| 55 | - | ||
| 56 | - /** | ||
| 57 | - * 是否是自动回复 | ||
| 58 | - * @author:dc | ||
| 59 | - * @time 2025/6/6 11:07 | ||
| 60 | - */ | ||
| 61 | - public function isAuto($data){ | ||
| 62 | - // 没有发件人的直接标记 | ||
| 63 | - if(!$data['from']){ | ||
| 64 | - return 1; | ||
| 65 | - } | ||
| 66 | - | ||
| 67 | - $is_auto = $this->db->count('select count(*) from `lists_auto` where `list_id` = '.$data['id']) ? 1 : 0; | ||
| 68 | - | ||
| 69 | - if($is_auto) return 1; | ||
| 70 | - | ||
| 71 | - return isAiAutoMail($data['from'],$data['subject'],$data['body']??'') === 1 ? 1 : 0; | ||
| 72 | - | ||
| 73 | - } | ||
| 74 | - | ||
| 75 | - /** | ||
| 76 | - * @return bool | ||
| 77 | - */ | ||
| 78 | - public function isStop(): bool | ||
| 79 | - { | ||
| 80 | - // 检查是否接收到信号 | ||
| 81 | - pcntl_signal_dispatch(); | ||
| 82 | - | ||
| 83 | - // 是否超过来最大执行时间 | ||
| 84 | - if(time()-43200 > $this->startTime){ | ||
| 85 | - return true; | ||
| 86 | - } | ||
| 87 | - | ||
| 88 | - return $this->isStop; | ||
| 89 | - } | ||
| 90 | - | ||
| 91 | - /** | ||
| 92 | - * es链接 | ||
| 93 | - * @var \Lib\Es\Es | ||
| 94 | - */ | ||
| 95 | - public $es; | ||
| 96 | - | ||
| 97 | - /** | ||
| 98 | - * @var \Lib\Es\BulkData | ||
| 99 | - */ | ||
| 100 | - protected $bulkData; | ||
| 101 | - | ||
| 102 | - | ||
| 103 | - public function handler($id){ | ||
| 104 | - | ||
| 105 | - $this->es = es('email_lists_copy'); // 第二个库 新 | ||
| 106 | - | ||
| 107 | - $this->bulkData = new \Lib\Es\BulkData(); | ||
| 108 | - | ||
| 109 | - $this->bulkData($id); | ||
| 110 | - $this->toDataEs(true); | ||
| 111 | - | ||
| 112 | - } | ||
| 113 | - | ||
| 114 | - /** | ||
| 115 | - * 批量处理数据并存储到ES | ||
| 116 | - * | ||
| 117 | - * @param string $id 数据ID,如果包含点号则只取点号前的部分 | ||
| 118 | - * @return void | ||
| 119 | - */ | ||
| 120 | - public function bulkData($id){ | ||
| 121 | - $is_check_body = false; | ||
| 122 | - if(str_contains($id, '.')){ | ||
| 123 | - $id = explode('.',$id)[0]; | ||
| 124 | - $is_check_body = true; | ||
| 125 | - } | ||
| 126 | - | ||
| 127 | - $data = $this->getDataByEs($id,$is_check_body); | ||
| 128 | - if($data){ | ||
| 129 | - list($doc_id,$data) = $data; | ||
| 130 | - // 主库 | ||
| 131 | - $this->bulkData->add('email_lists_copy',$doc_id,$data); | ||
| 132 | - if($data['postid']){ | ||
| 133 | - // 分库 | ||
| 134 | - $this->bulkData->add('email_lists_branch_'.$data['postid'],$doc_id,$data); | ||
| 135 | - } | ||
| 136 | - | ||
| 137 | - // 个人邮箱的情况 | ||
| 138 | - $postids = $this->getPostids($data['email_id']); | ||
| 139 | - print_r($postids); | ||
| 140 | - if($postids){ | ||
| 141 | - foreach ($postids as $postid){ | ||
| 142 | - $data['postid'] = $postid; | ||
| 143 | - $data['source'] = 1; | ||
| 144 | - // 分库 个人邮箱 | ||
| 145 | - $this->bulkData ->add('email_lists_branch_'.$postid,$doc_id,$data); | ||
| 146 | - } | ||
| 147 | - } | ||
| 148 | - // 其他非fob邮件数据 | ||
| 149 | - if (!$data['postid']){ | ||
| 150 | - // 分库 其他 非fob数据源 | ||
| 151 | - $this->bulkData->add('email_lists_branch_'.$data['postid'],$doc_id,$data); | ||
| 152 | - } | ||
| 153 | - | ||
| 154 | - } | ||
| 155 | - } | ||
| 156 | - | ||
| 157 | - | ||
| 158 | - /** | ||
| 159 | - * 个人邮箱情况 | ||
| 160 | - * @param $email_id | ||
| 161 | - * @author:dc | ||
| 162 | - * @time 2025/8/5 14:53 | ||
| 163 | - */ | ||
| 164 | - private function getPostids($email_id){ | ||
| 165 | - $postids = $this->fob_db->throw()->cache(1800)->all("select `post_id` from `e_mail_binds` where `source` = 1 and `email_id` = {$email_id} and `deleted_at` is null"); | ||
| 166 | - | ||
| 167 | - if($postids){ | ||
| 168 | - return array_column($postids,'post_id'); | ||
| 169 | - } | ||
| 170 | - | ||
| 171 | - return []; | ||
| 172 | - | ||
| 173 | - } | ||
| 174 | - | ||
| 175 | - | ||
| 176 | - /** | ||
| 177 | - * @param $id | ||
| 178 | - * @param $is_check_body | ||
| 179 | - * @return array|false | ||
| 180 | - * @author:dc | ||
| 181 | - * @time 2025/8/5 10:21 | ||
| 182 | - */ | ||
| 183 | - public function getDataByEs($id,$is_check_body) { | ||
| 184 | - try { | ||
| 185 | - $data = $this->db->throw()->first(\Model\listsSql::first('`id` = '.$id)); | ||
| 186 | - if(!$data){ | ||
| 187 | - $data = $this->db->throw()->first(\Model\listsSql::firstHot('`id` = '.$id)); | ||
| 188 | - } | ||
| 189 | - }catch (Throwable $e){ | ||
| 190 | - $this->log([$id]); | ||
| 191 | -// redis()->rPush('sync_to_es',$origin_id); | ||
| 192 | - _echo('sync to es '.$id.":".$e->getMessage()); | ||
| 193 | - return false; | ||
| 194 | - } | ||
| 195 | - | ||
| 196 | - if($data){ | ||
| 197 | - try { | ||
| 198 | - // 文件夹 | ||
| 199 | - if(empty($this->folders[$data['folder_id']])){ | ||
| 200 | - $this->folders[$data['folder_id']] = $this->db->throw()->value(\Model\folderSql::first($data['folder_id'],'folder')); | ||
| 201 | - } | ||
| 202 | - | ||
| 203 | - // 为文件夹打标 方便查询 | ||
| 204 | - $data['folder_as_int'] = folder2int($this->folders[$data['folder_id']]); | ||
| 205 | - $data['is_auto'] = 0; | ||
| 206 | - // 是否是自动回复 | ||
| 207 | - if($data['folder_as_int'] === 1){ | ||
| 208 | - // 是否检查body | ||
| 209 | - if($is_check_body){ | ||
| 210 | - $body = getMailBody($data['id'],$this->db); | ||
| 211 | - if($body){ | ||
| 212 | - $data['body'] = getBodyHtml($body); | ||
| 213 | - } | ||
| 214 | - } | ||
| 215 | - | ||
| 216 | - $data['is_auto'] = $this->isAuto($data); | ||
| 217 | - | ||
| 218 | - unset($data['body']); | ||
| 219 | - } | ||
| 220 | - | ||
| 221 | - // postid ai邮箱要用 这个是查询黑格 | ||
| 222 | - list($data['postid'],$data['source']) = $this->getPostid($data['email_id'],$data['udate']); | ||
| 223 | - | ||
| 224 | - }catch (Throwable $e){ | ||
| 225 | - $this->log([$id]); | ||
| 226 | -// redis()->rPush('sync_to_es',$origin_id); | ||
| 227 | - _echo('sync to es '.$id.":".$e->getMessage()); | ||
| 228 | - return false; | ||
| 229 | - } | ||
| 230 | - | ||
| 231 | - $data = $this->getEsData($data); | ||
| 232 | - | ||
| 233 | - $doc_id = $data['email_id'].'_'.$data['folder_id'].'_'.$data['uid']; | ||
| 234 | - | ||
| 235 | - return [$doc_id,$data]; | ||
| 236 | - | ||
| 237 | - } | ||
| 238 | - return false; | ||
| 239 | - } | ||
| 240 | - | ||
| 241 | - | ||
| 242 | - public $checkEsIndex = []; | ||
| 243 | - | ||
| 244 | - | ||
| 245 | - public function setEsMap($index){ | ||
| 246 | - $this->es->setIndex($index); | ||
| 247 | - if($this->es->getMapping()){ | ||
| 248 | - return 9; | ||
| 249 | - } | ||
| 250 | - | ||
| 251 | - if(redis()->add('setmaplock:'.$index,1,20)){ | ||
| 252 | - return $this->es->putMapping([ | ||
| 253 | - 'properties' => [ | ||
| 254 | - 'subject' => ['type' => 'text'], | ||
| 255 | - 'from' => [ | ||
| 256 | - 'type' => 'object', // 定义 from 字段为对象 | ||
| 257 | - 'properties' => [ | ||
| 258 | - 'name' => [ | ||
| 259 | - 'type' => 'keyword' // 或者 'keyword',根据需求选择 | ||
| 260 | - ], | ||
| 261 | - 'email' => [ | ||
| 262 | - 'type' => 'text' // email 通常使用 keyword 类型 | ||
| 263 | - ] | ||
| 264 | - ] | ||
| 265 | - ], | ||
| 266 | - 'to' => ['type' => 'text'], | ||
| 267 | - 'cc' => ['type' => 'keyword'], | ||
| 268 | - 'bcc' => ['type' => 'keyword'], | ||
| 269 | - 'uid' => ['type' => 'integer'], | ||
| 270 | - 'udate' => ['type' => 'integer'], | ||
| 271 | - 'folder_id' => ['type' => 'integer'], | ||
| 272 | - 'email_id' => ['type' => 'integer'], | ||
| 273 | - 'size' => ['type' => 'integer'], | ||
| 274 | - 'recent' => ['type' => 'integer'], | ||
| 275 | - 'flagged' => ['type' => 'integer'], | ||
| 276 | - 'deleted' => ['type' => 'integer'], | ||
| 277 | - 'seen' => ['type' => 'integer'], | ||
| 278 | - 'draft' => ['type' => 'integer'], | ||
| 279 | - 'is_file' => ['type' => 'integer'], | ||
| 280 | - 'is_hots' => ['type' => 'integer'], | ||
| 281 | - 'is_auto' => ['type' => 'integer'], | ||
| 282 | - 'folder_as_int' => ['type' => 'integer'], | ||
| 283 | - 'postid' => ['type' => 'integer'], | ||
| 284 | - 'source' => ['type' => 'integer'], | ||
| 285 | - 'created_at' => ['type' => 'date'], | ||
| 286 | - 'updated_at' => ['type' => 'date'], | ||
| 287 | - 'description' => ['type' => 'keyword'], | ||
| 288 | - 'references' => ['type' => 'keyword'] | ||
| 289 | - ] | ||
| 290 | - ],$index == 'email_lists_copy' ? [ | ||
| 291 | - 'number_of_shards' => 21, // 设置分片数 | ||
| 292 | - 'number_of_replicas' => 1, // 设置副本数 暂用内存 主片+副片* | ||
| 293 | - ]:[ | ||
| 294 | - 'number_of_shards' => 1, // 设置分片数 | ||
| 295 | - 'number_of_replicas' => 0, // 设置副本数 暂用内存 主片+副片* | ||
| 296 | - ] | ||
| 297 | - ); | ||
| 298 | - } | ||
| 299 | - // 暂停1秒在试 | ||
| 300 | - sleep(1); | ||
| 301 | - return $this->setEsMap($index); | ||
| 302 | - } | ||
| 303 | - | ||
| 304 | - /** | ||
| 305 | - * @var int | ||
| 306 | - */ | ||
| 307 | - protected $pervSubmitTime = 0; | ||
| 308 | - | ||
| 309 | - /** | ||
| 310 | - * 同步数据到es | ||
| 311 | - * @param bool $nowSubmit 是否立即提交 | ||
| 312 | - * @return bool | ||
| 313 | - * @throws \Elastic\Elasticsearch\Exception\ClientResponseException | ||
| 314 | - * @throws \Elastic\Elasticsearch\Exception\ServerResponseException | ||
| 315 | - * @author:dc | ||
| 316 | - * @time 2025/8/7 10:29 | ||
| 317 | - */ | ||
| 318 | - public function toDataEs(bool $nowSubmit){ | ||
| 319 | - // 不立即提交 | ||
| 320 | - if (!$nowSubmit){ | ||
| 321 | - if($this->bulkData->total() < 20){ | ||
| 322 | - // 不足20条时 满足2秒也提交 | ||
| 323 | - if ($this->pervSubmitTime + 2 > time()){ | ||
| 324 | - return true; | ||
| 325 | - } | ||
| 326 | - } | ||
| 327 | - } | ||
| 328 | - // 为空不提交 | ||
| 329 | - if($this->bulkData->isEmpty()){ | ||
| 330 | - return true; | ||
| 331 | - } | ||
| 332 | - | ||
| 333 | - // 上一次提交的时间 | ||
| 334 | - $this->pervSubmitTime = time(); | ||
| 335 | - | ||
| 336 | - foreach ($this->bulkData->getIndexs() as $index){ | ||
| 337 | - $this->es->setIndex($index); | ||
| 338 | - // 检查数据库是否存在 | ||
| 339 | - if(empty($this->checkEsIndex[$index]) && $index != 'email_lists_copy'){ | ||
| 340 | - if(!redis()->has('esmapcheck:'.$index)){ | ||
| 341 | - $m = $this->setEsMap($index); | ||
| 342 | - if($m !== 9) _echo("{$index} 创建索引 ".$m); | ||
| 343 | - redis()->set('esmapcheck:'.$index,1,86400); | ||
| 344 | - } | ||
| 345 | - } | ||
| 346 | - // 下次不在检查 | ||
| 347 | - $this->checkEsIndex[$index] = 1; | ||
| 348 | - } | ||
| 349 | - // 批量提交数据的 | ||
| 350 | - $ret = $this->es->bulk($this->bulkData); | ||
| 351 | - | ||
| 352 | - if(!empty($ret['errors'])){ | ||
| 353 | - @file_put_contents(LOG_PATH.'/sync_es_fail.error.log',print_r($ret['errors'],1)."\n",FILE_APPEND|LOCK_EX); | ||
| 354 | - } | ||
| 355 | - // 清空 | ||
| 356 | - $this->bulkData->clear(); | ||
| 357 | - // 为空表示提交成功 | ||
| 358 | - return empty($ret['errors']); | ||
| 359 | - } | ||
| 360 | - | ||
| 361 | - /** | ||
| 362 | - * 记录日志 | ||
| 363 | - * @param array $ids | ||
| 364 | - * @param string $index | ||
| 365 | - * @author:dc | ||
| 366 | - * @time 2025/8/5 10:17 | ||
| 367 | - */ | ||
| 368 | - public function log(array $ids){ | ||
| 369 | - file_put_contents(LOG_PATH.'/sync_es_fail.log',implode("\n",$ids)."\n",FILE_APPEND|LOCK_EX); | ||
| 370 | - } | ||
| 371 | - | ||
| 372 | - | ||
| 373 | - | ||
| 374 | - /** | ||
| 375 | - * 项目id | ||
| 376 | - * @author:dc | ||
| 377 | - * @time 2025/5/20 15:44 | ||
| 378 | - */ | ||
| 379 | - public function getPostid($email_id,$udate){ | ||
| 380 | - | ||
| 381 | - //每60秒验证一次 | ||
| 382 | - if(redis()->add('fob_bind_mail_times_check:'.$email_id,1,60)){ | ||
| 383 | - $lastpostid = $this->fob_db->throw()->value("select `post_id` from `e_mail_binds_log` where `source` = 2 and `email_id` = '{$email_id}' order by `id` desc limit 1"); | ||
| 384 | - $thelast = $this->fob_db->throw()->first("select * from `e_mail_binds` where `source` = 2 and `email_id` = '{$email_id}' and `deleted_at` is null order by `id` desc limit 1"); | ||
| 385 | - if(!$thelast){ | ||
| 386 | - return [0,0]; | ||
| 387 | - } | ||
| 388 | - | ||
| 389 | - $thelastpostid = $thelast['post_id']; | ||
| 390 | - | ||
| 391 | - // 说明变了 | ||
| 392 | - if($lastpostid!=$thelastpostid){ | ||
| 393 | - $ret = $this->fob_db->throw()->insert('e_mail_binds_log',[ | ||
| 394 | - 'post_id' => $thelastpostid, | ||
| 395 | - 'bind_time' => date('Y-m-d H:i:s',time()-2), | ||
| 396 | - 'source' => $thelast['source'], | ||
| 397 | - 'email' => $thelast['email'], | ||
| 398 | - 'email_id' => $thelast['email_id'], | ||
| 399 | - ],false); | ||
| 400 | - _echo("邮箱异常分配 ".$email_id.' -- '.$lastpostid.' == '.$thelastpostid.' === '.$ret); | ||
| 401 | - if(!$ret){ | ||
| 402 | - throw new Exception('新增失败'); | ||
| 403 | - } | ||
| 404 | - } | ||
| 405 | - | ||
| 406 | - } | ||
| 407 | - | ||
| 408 | - | ||
| 409 | - // 查询历史记录 | ||
| 410 | - $times = redis()->getSet('fob_bind_mail_times3:'.$email_id,300,function ($email_id){ | ||
| 411 | - | ||
| 412 | - $times = $this->fob_db->throw()->all("select `post_id`,`bind_time`,`source` from `e_mail_binds_log` where `source` = 2 and `email_id` = {$email_id} order by `bind_time` desc "); | ||
| 413 | - if(!$times){ | ||
| 414 | - return []; | ||
| 415 | - } | ||
| 416 | - | ||
| 417 | - return $times; | ||
| 418 | - | ||
| 419 | - },$email_id); | ||
| 420 | - | ||
| 421 | - if(is_array($times) && $times){ | ||
| 422 | - foreach ($times as $time){ | ||
| 423 | - $t = strtotime($time['bind_time']); | ||
| 424 | - // 邮件收到的时间是否大于绑定时间 | ||
| 425 | - if($udate > $t){ | ||
| 426 | - $data = $time; | ||
| 427 | - break; | ||
| 428 | - } | ||
| 429 | - } | ||
| 430 | - } | ||
| 431 | - | ||
| 432 | - // 没有找到历史,就找绑定表 | ||
| 433 | - if(empty($data)){ | ||
| 434 | - $data = redis()->getSet('fob_bind_mail3:'.$email_id,300,function ($email_id){ | ||
| 435 | - return $this->fob_db->throw()->first("select `post_id`,`source` from `e_mail_binds` where `source` = 2 and `email_id` = '{$email_id}' and `deleted_at` is null order by `id` desc limit 1"); | ||
| 436 | - },$email_id); | ||
| 437 | - | ||
| 438 | - } | ||
| 439 | - | ||
| 440 | - return [ | ||
| 441 | - $data['post_id']??0, | ||
| 442 | - $data['source']??0, | ||
| 443 | - ]; | ||
| 444 | - } | ||
| 445 | - | ||
| 446 | - | ||
| 447 | - public function getEsData($data){ | ||
| 448 | - if(!empty($data['id'])){ | ||
| 449 | - $data['uuid'] = $data['id']; | ||
| 450 | - unset($data['id']); | ||
| 451 | - } | ||
| 452 | - | ||
| 453 | - $data['from'] = [ | ||
| 454 | - 'email' => $data['from'], | ||
| 455 | - 'name' => $data['from_name']??'' | ||
| 456 | - ]; | ||
| 457 | - unset($data['from_name']); | ||
| 458 | - unset($data['date']); | ||
| 459 | - | ||
| 460 | - $data['to_name'] = $this->trimEmail($data['to_name']??[]); | ||
| 461 | - $data['cc'] = $this->trimEmail($data['cc']??[]); | ||
| 462 | - $data['bcc'] = $this->trimEmail($data['bcc']??[]); | ||
| 463 | - | ||
| 464 | - $data['description'] = str_replace(["\n"],"",$data['description']); | ||
| 465 | - // unset($data['to_name']); | ||
| 466 | - | ||
| 467 | - if(!empty($data['created_at'])){ | ||
| 468 | - $data['created_at'] = date('Y-m-d\TH:i:s',strtotime($data['created_at'])); | ||
| 469 | - } | ||
| 470 | - if(!empty($data['updated_at'])){ | ||
| 471 | - $data['updated_at'] = date('Y-m-d\TH:i:s',strtotime($data['updated_at'])); | ||
| 472 | - } | ||
| 473 | - $data['references'] = empty($data['references']) ? '' : $data['references']; | ||
| 474 | - return $data; | ||
| 475 | - } | ||
| 476 | - | ||
| 477 | - | ||
| 478 | - private function trimEmail($emails){ | ||
| 479 | - if($emails){ | ||
| 480 | - $emails = is_array($emails) ? $emails : @json_decode($emails,true); | ||
| 481 | - if(is_array($emails) && count($emails)>100){ | ||
| 482 | - | ||
| 483 | - $emails = array_map(function ($v){ | ||
| 484 | - $v['name'] = ''; | ||
| 485 | - return $v; | ||
| 486 | - },$emails); | ||
| 487 | - $emails = array_slice($emails,0,100); | ||
| 488 | - return json_encode($emails); | ||
| 489 | - } | ||
| 490 | - } | ||
| 491 | - return ''; | ||
| 492 | - } | ||
| 493 | - | ||
| 494 | -} | ||
| 495 | - | ||
| 496 | -(new SyncToEsCmd())->handler($argv[1]); | ||
| 497 | - | ||
| 498 | -return 1; | ||
| 499 | - | ||
| 500 | - | ||
| 501 | - | ||
| 502 | - | ||
| 503 | - | ||
| 504 | - | ||
| 505 | - | ||
| 506 | - | ||
| 507 | - | 
- 
请 注册 或 登录 后发表评论