作者 邓超

优化 发送邮件脚本

正在显示 1 个修改的文件 包含 204 行增加147 行删除
... ... @@ -3,185 +3,242 @@
include_once "../vendor/autoload.php";
// 这里试试不用多进程模式,用多协程模式
function start(){
_echo('启动邮件群发任务');
// 删除key
redis()->delete('send_job_is_stop');
// 开启协程
\Co\run(function (){
$cNum = 0;//协程运行的数量
$maxRunNum = 500;
while ($maxRunNum){
$maxRunNum--;
if(!$maxRunNum){
break;
}
try {
// 是否要停止
if(redis()->get('send_job_is_stop')=='stop'){
break;
}
$lists = db()->all(\Model\sendJobsSql::sendList(500));
$lists = $lists?$lists:[];
// 循环
foreach ($lists as $list){
// 占用 id
if(redis()->add('send_job_run_id_'.$list['id'],$list['id'],3600)){
go(function ($data) use (&$cNum){
$cNum++; // 协程数+1
// 表单数据
$data['maildata'] = json_decode($data['maildata'],true);
// 查询邮箱
$email = db()->first(\Model\emailSql::first($data['email_id']));
// 更新状态
\Model\sendJobsSql::upStatus($data['id'],1,db());
// 是否是单发送
if($data['maildata']['massSuit']??0){
$tos = $data['maildata']['tos'];
foreach ($tos as $to){
// 是否暂停
$dst = db()->first(\Model\sendJobsSql::isStatus($data['id']));
if($dst && $dst['status'] === 3){
break;
}
// 是否已发送过了
if(db()->count(\Model\sendJobStatusSql::count($data['id'],$to['email']))){
continue;
}
// 每个收件人单独发送
$data['maildata']['tos'] = [$to];
//替换邮件内容中的指定字段为客户名字
$data['maildata']['body'] = str_replace('{customer_name}', $to['name'], $data['maildata']['body']);
$result = \Lib\Mail\MailFun::sendEmail($data['maildata'],$email);
// 插入紫薯精
db()->insert(\Model\sendJobStatusSql::$table,[
'job_id' => $data['id'],
'to_email' => $to['email'],
'status' => $result[0] ? 1 : 0,
'error' => $result[0] ? $result[1] : ''
]);
// 时间距离下次的时间
if($data['maildata']['masssuit_interval_send']??[]){
$time = rand($data['maildata']['masssuit_interval_send']['start'],$data['maildata']['masssuit_interval_send']['end']);
if($time){
$block = false;
while (true){
// 没5秒循环一次
if(redis()->get('send_job_is_stop')=='stop'){
$block = true;
break;
}
$time-=5;
co::sleep(5);
// 执行下一次了
if (!$time){
$block = true;
break;
}
}
if($block){
break;
}
}
}
class SendJob {
}
public $cnum = 0;
}else{
$result = \Lib\Mail\MailFun::sendEmail($data['maildata'],$email);
// 更新状态
db()->update(\Model\sendJobsSql::$table,[
'status' => 2,
'success' => $result[0] ? $data['total'] : 0,
'error' => $result[0] ? 0 : $data['total'],
],dbWhere(['id'=>$data['id']]));
// 插入紫薯精
db()->insert(\Model\sendJobStatusSql::$table,[
'job_id' => $data['id'],
'to_email' => 'all',
'status' => $result[0] ? 1 : 0,
'error' => $result[0] ? $result[1] : ''
]);
}
/**
* 是否停止
* @return bool
* @author:dc
* @time 2024/4/10 9:12
*/
private function isStop(){
return redis()->get('send_job_is_stop') == 'stop';
}
// 协程结束后
co::defer(function ($id) use(&$cNum,$data){
$cNum--;
// 验证是否完成
if($data['maildata']['massSuit']??0){
$total = db()->first(\Model\sendJobStatusSql::countSum($data['id']));
if($total){
// 更新状态
db()->update(\Model\sendJobsSql::$table,[
'status' => $total['t'] == $data['total'] ? 2 : 0,
'success' => $total['s'],
'error' => $total['e'],
],dbWhere(['id'=>$data['id']]));
}
/**
* 休眠
* @param float $sleep
* @return bool
* @author:dc
* @time 2024/4/10 9:12
*/
private function s_sleep(float $sleep):bool {
if($sleep > 0){
$t = microtime(1);
while (!$this->isStop()){
co::sleep(0.1);
if($sleep - (microtime(1)-$t) <= 0){
break;
}
}
}
return true;
}
}
// 写入日志
\Lib\Log::getInstance()->write();
public function start(){
_echo('启动邮件群发任务 '.getmypid());
// 删除key
redis()->delete('send_job_is_stop');
// 结束后要关闭数据库链接,不然链接一直暂用
db()->close();
// 删除占用
redis()->delete('send_job_run_id_'.$data['id']);
redis()->close();
});
while (1){
// 是否要停止
if($this->isStop()){
break;
}
},$list);
}
$lists = db()->all(\Model\sendJobsSql::sendList(500));
$lists = $lists?$lists:[];
if($lists){
foreach ($lists as $list){
$this->go_($list);
}
}catch (Throwable $e){
logs($e->getMessage().$e->getTraceAsString());
}else{
// 休眠30秒
$this->s_sleep(30);
}
\Lib\Log::getInstance()->write();
// 暂停5秒
co::sleep(5);
}
// 这个是等待所有协程退出
while (true){
if(!$cNum){
_echo('等待协程退出...');
if(!$this->cnum){
break;
}
co::sleep(0.5);
co::sleep(1);
}
}
});
}
/**
* @param $list
* @throws \Lib\Err
* @throws \PHPMailer\PHPMailer\Exception
* @author:dc
* @time 2024/4/10 9:25
*/
public function go_($list){
// 占用 id
if(redis()->add('send_job_run_id_'.$list['id'],$list['id'],600)){
go(function ($data) {
_echo('正在执行任务 '.$data['id']);
$this->cnum++; // 协程数+1
// 表单数据
$data['maildata'] = json_decode($data['maildata'],true);
// 查询邮箱
$email = db()->first(\Model\emailSql::first($data['email_id']));
// 更新状态
\Model\sendJobsSql::upStatus($data['id'],1,db());
// 是否是单发送
if($data['maildata']['massSuit']??0){
$tos = $data['maildata']['tos'];
foreach ($tos as $to){
_echo('正在执行任务 发送邮件 '.$to['email']);
// 续时间
redis()->set('send_job_run_id_'.$data['id'],$data['id'],600);
// 是否暂停
$dst = db()->first(\Model\sendJobsSql::isStatus($data['id']));
if($dst && $dst['status'] === 3){
break;
}
// 是否已发送过了
if(db()->count(\Model\sendJobStatusSql::count($data['id'],$to['email']))){
continue;
}
// 每个收件人单独发送
$data['maildata']['tos'] = [$to];
//替换邮件内容中的指定字段为客户名字
$data['maildata']['body'] = str_replace('{customer_name}', $to['name'], $data['maildata']['body']);
$result = \Lib\Mail\MailFun::sendEmail($data['maildata'],$email);
// 插入紫薯精
db()->insert(\Model\sendJobStatusSql::$table,[
'job_id' => $data['id'],
'to_email' => $to['email'],
'status' => $result[0] ? 1 : 0,
'error' => $result[0] ? $result[1] : ''
]);
// 时间距离下次的时间
if($data['maildata']['masssuit_interval_send']??[]){
$time = rand($data['maildata']['masssuit_interval_send']['start'],$data['maildata']['masssuit_interval_send']['end']);
if($time){
$block = false;
while (true){
// 没5秒循环一次
if($this->isStop()){
$block = true;
break;
}
$time-=5;
$this->s_sleep(5);
// 执行下一次了
if (!$time){
$block = true;
break;
}
}
if($block){
break;
}
}
}
}
}
else{
$result = \Lib\Mail\MailFun::sendEmail($data['maildata'],$email);
// 更新状态
db()->update(\Model\sendJobsSql::$table,[
'status' => 2,
'success' => $result[0] ? $data['total'] : 0,
'error' => $result[0] ? 0 : $data['total'],
],dbWhere(['id'=>$data['id']]));
// 插入紫薯精
db()->insert(\Model\sendJobStatusSql::$table,[
'job_id' => $data['id'],
'to_email' => 'all',
'status' => $result[0] ? 1 : 0,
'error' => $result[0] ? $result[1] : ''
]);
}
// 协程结束后
co::defer(function ($id) use($data){
$this->cnum--;
// 验证是否完成
if($data['maildata']['massSuit']??0){
$total = db()->first(\Model\sendJobStatusSql::countSum($data['id']));
if($total){
// 更新状态
db()->update(\Model\sendJobsSql::$table,[
'status' => $total['t'] == $data['total'] ? 2 : 0,
'success' => $total['s'],
'error' => $total['e'],
],dbWhere(['id'=>$data['id']]));
}
}
// 写入日志
\Lib\Log::getInstance()->write();
// 结束后要关闭数据库链接,不然链接一直暂用
db()->close();
// 删除占用
redis()->delete('send_job_run_id_'.$data['id']);
redis()->close();
});
},$list);
}
}
}
$ps = "ps -ef | grep \"send_job.php start\" | grep -v grep | wc -l";
switch ($argv[1]??0){
case 'start':{
// $num = exec($ps);
// if($num){
// echo '正则运行,请勿重复运行';
// }else{
start();
// }
// 开启协程
\Co\run(function (){
$handler = function ($signal){
// 可以处理其他程序
redis()->set('send_job_is_stop','stop');
_echo('收到退出信号 '.$signal);
};
\Swoole\Process::signal(SIGTERM,$handler);
\Swoole\Process::signal(SIGINT,$handler);
(new SendJob)->start();
_echo('进程已退出');
});
break;
}
case 'stop':{
... ...