正在显示
1 个修改的文件
包含
507 行增加
和
0 行删除
cmd/sync_to_es_v2b.php
0 → 100644
| 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 | + |
-
请 注册 或 登录 后发表评论