php守候進程-發送隊列郵件 (一)

在linux系統下運行

./demo.php

注：只能在liunx系統下運行

demo.conf

# demo.conf

# daemon mode
daemon yes

# child number
child_num 5

# child user and group
user nobody
group nogroup

# retry times
retry_times 3

# pid file
pid_file /tmp/demo.pid

# log file
log_file /tmp/demo.log

demo.php

#!/usr/bin/env php
<?php

error_reporting(0);

include "SendMail.php";

define("EXIT_SUCCESS", 		0);
define("EXIT_FAILURE", 		1);

define("DEMO_LOG_DEBUG", 	0);
define("DEMO_LOG_INFO",  	1);
define("DEMO_LOG_WARN",  	2);
define("DEMO_LOG_ERR",   	3);

// default config
$config = array(
	"daemon"		=> FALSE,
	"child_num"		=> 1,
	"user"			=> NULL,
	"group"			=> NULL,
	"retry_times"	=> 0,
	"pid_file"		=> NULL,
	"log"			=> STDERR,
	);

// parse config file
parse_config($config);

if ($argc > 1) {
	switch ($argv[1]) {

	case "help":
		exit("Usage: ./demo.php {start|restart|reload|quit}\n\n");

	case "start":
		break;

	case "restart":
	case "reload":
	case "quit":
		$fp = fopen($config["pid_file"], "r");
		if (!$fp) {
			exit("Can't open pid file '{$config["pid_file"]}'.\n");
		}

		$pid = fgets($fp, 6);
		if ($pid) {
			$pid = intval($pid);
		}
		fclose($fp);

		if ($argv[1] == "restart" || $argv[1] == "quit") {
			posix_kill($pid, SIGINT);
			if ($argv[1] == "restart") {
				unlink($config["pid_file"]);
			} else {
				exit(EXIT_SUCCESS);
			}

		} else {
			posix_kill($pid, SIGHUP);
			exit(EXIT_SUCCESS);
		}
		break;
	}
}

if (file_exists($config["pid_file"])) {
	exit("pid file '{$config["pid_file"]}' already exits.\n");
}

if ($config["daemon"]) {
	if (pcntl_fork() > 0) {
		// parent exit
		exit(EXIT_SUCCESS);
	}

	// set session id
	posix_setsid();

	// write pid file
	if ($config["pid_file"]) {
		$fp = fopen($config["pid_file"], "w");
		if (!$fp) {
			exit("Can't open pid file '{$config["pid_file"]}'.\n");
		}

		if (!fputs($fp, posix_getpid())) {
			exit("Can't write pid to file '{$config["pid_file"]}'.\n");
		}
		fclose($fp);
	}
}

write_log(DEMO_LOG_INFO, "main process started");

declare(ticks = 1);

pcntl_signal(SIGTERM, SIG_IGN);

pcntl_signal(SIGHUP,  "reload");
pcntl_signal(SIGINT,  "quit");
pcntl_signal(SIGCHLD, "handle_child");

$childs = array();

$reparse_config = FALSE;

$run = TRUE;
while ($run) {

	if ($reparse_config) {
		write_log(DEMO_LOG_INFO, "reparse config file");
		parse_config($config);
		posix_kill(0, SIGTERM);
		$reparse_config = FALSE;
	}

	for ($i = count($childs); $i < $config["child_num"]; ++$i) {
		if (($child_pid = pcntl_fork()) == 0) {
			do_child($config);
			exit(EXIT_SUCCESS);
		}

		write_log(DEMO_LOG_INFO, "start child process: %d", $child_pid);
		$childs[$child_pid] = 1;
	}

	sleep(1);
}

// kill child
posix_kill(0, SIGTERM);

// remove pid file
@unlink($config["pid_file"]);

write_log(DEMO_LOG_INFO, "main process exit");
exit(EXIT_SUCCESS);

function reload() {
	$GLOBALS["reparse_config"] = TRUE;
}

function quit() {
	$GLOBALS["run"] = FALSE;
}

function handle_child() {
	while (($child_pid = pcntl_waitpid(-1, $status, WNOHANG)) > 0) {
		write_log(DEMO_LOG_INFO, "exit child process: %d", $child_pid);
		unset($GLOBALS["childs"][$child_pid]);
	}
}

function parse_config(Array &$config) {
	$fp = fopen("demo.conf", "r");
	if (!$fp) {
		exit("Can't open config file.\n");
	}

	$lineno = 0;
	while (($line = fgets($fp, 1024)) !== FALSE) {
		++$lineno;

		$line = trim($line);
		if (!$line || $line[0] == "#") {
			continue;
		}

		$params = preg_split("/\s+/", $line);
		switch (strtolower($params[0])) {

		case "daemon":
			if ($params[1] == "yes") {
				$config["daemon"] = TRUE;
			} else if ($params[1] == "no") {
				$config["daemon"] = FALSE;
			} else {
				$err = "daemon value must be 'yes' or 'no'";
				goto parse_failed;
			}
			break;

		case "child_num":
			$child_num = intval($params[1]);
			if ($child_num < 0 || $child_num > 1024) {
				$err = "invalid child_num value '{$params[1]}'";
				goto parse_failed;
			}
			$config["child_num"] = $child_num;
			break;

		case "retry_times":
			$retry_times = intval($params[1]);
			if ($retry_times < 1 || $retry_times > 100) {
				$err = "invalid retry_times value '{$params[1]}'";
				goto parse_failed;
			}
			$config["retry_times"] = $retry_times;
			break;

		case "user":
			$user = posix_getpwnam($params[1]);
			if (!$user) {
				$err = "invalid user value '{$params[1]}'";
				goto parse_failed;
			}
			$config["user"] = $user["uid"];
			break;

		case "group":
			$group = posix_getgrnam($params[1]);
			if (!$group) {
				$err = "invalid group value '{$params[1]}'";
				goto parse_failed;
			}
			$config["group"] = $group["gid"];
			break;

		case "pid_file":
			$config["pid_file"] = $params[1];
			break;

		case "log_file":
			$log = fopen($params[1], "a");
			if (!$log) {
				$err = "Can't open log file '{$params[1]}'";
				goto parse_failed;
			}
			$config["log"] = $log;
			break;
		}

		continue;

	parse_failed:
		fprintf(STDERR, "\n*** FATAL CONFIG FILE ERROR ***\n");
		fprintf(STDERR, "Reading the configuration file, at line %d\n", $lineno);
		fprintf(STDERR, ">>> '%s'\n", $line);
		fprintf(STDERR, "%s\n", $err);
		exit(EXIT_FAILURE);
	}
	fclose($fp);
}

function write_log($level, $fmt) {
	$chars = ".-*#";

	if (func_num_args() > 2) {
		$args = func_get_args();
		$err  = vsprintf($args[1], array_slice($args, 2));

	} else {
		$err = $fmt;
	}

	fprintf($GLOBALS["config"]["log"], "[%d] [%s] %s %s\n", 
			posix_getpid(), date("Y-m-d H:i:s"), $chars[$level], $err);
}

function do_child() {
	global $run;

	$user 		 = $GLOBALS["config"]["user"];
	$group 		 = $GLOBALS["config"]["group"];
	$retry_times = $GLOBALS["config"]["retry_times"];

	if ($user) {
		posix_setuid($user);
	}

	if ($group) {
		posix_setuid($group);
	}

	pcntl_signal(SIGTERM, "quit");

	$redis = new Redis;
	$redis->pconnect("127.0.0.1", 6379);

	while ($run) {
		try {
			$email = $redis->lpop("email_queue");
			if ($email) {
				for ($i = 0; $i < $retry_times; ++$i) {
					if (do_sendmail($email)) {
						write_log(DEMO_LOG_INFO, "send mail to '%s' success", $email);
						break;
					}

					write_log(DEMO_LOG_ERR, "send mail to '%s' failed, try again", $email);
				}

				if ($i == $retry_times) {
					write_log(DEMO_LOG_ERR, "send mail to '%s' failed", $email);
				}
			}

		} catch (RedisException $e) {
			write_log(LOG_ERR, "receive message failed: %s", $e->getMessage());
			exit(EXIT_FAILURE);
		}

		sleep(1);
	}

	exit(EXIT_SUCCESS);
}

function do_sendmail($email) {
	$mail = new SendMail();
	$mail->setServer("smtp.exmail.qq.com", "dingpeilong@xywy.com", "PLDing1989.com", 465, 1);
	$mail->setFrom("dingpeilong@xywy.com");
	$mail->setReceiver("77676182@qq.com");
	$mail->setMail("test", "**hello world!**");
	return $mail->sendMail();
}

SendMail.php

<?php
/**
* 郵件發送類
* 支持發送純文本郵件和HTML格式的郵件，能夠多收件人，多抄送，多祕密抄送，帶附件(單個或多個附件),支持到服務器的ssl鏈接
* 須要的php擴展：sockets、Fileinfo和openssl。
* 編碼格式是UTF-8，傳輸編碼格式是base64
* @example
* $mail = new SendMail();
* $mail->setServer("smtp@126.com", "XXXXX@126.com", "XXXXX"); //設置smtp服務器，普通鏈接方式
* $mail->setServer("smtp.gmail.com", "XXXXX@gmail.com", "XXXXX", 465, true); //設置smtp服務器，到服務器的SSL鏈接
* $mail->setFrom("XXXXX"); //設置發件人
* $mail->setReceiver("XXXXX"); //設置收件人，多個收件人，調用屢次
* $mail->setCc("XXXX"); //設置抄送，多個抄送，調用屢次
* $mail->setBcc("XXXXX"); //設置祕密抄送，多個祕密抄送，調用屢次
* $mail->addAttachment("XXXX"); //添加附件，多個附件，調用屢次
* $mail->setMail("test", "**test**"); //設置郵件主題、內容
* $mail->sendMail(); //發送
*/
class SendMail {
    /**
    * @var string 郵件傳輸代理用戶名
    * @access protected
    */
    protected $_userName;

    /**
    * @var string 郵件傳輸代理密碼
    * @access protected
    */
    protected $_password;

    /**
    * @var string 郵件傳輸代理服務器地址
    * @access protected
    */
    protected $_sendServer;

    /**
    * @var int 郵件傳輸代理服務器端口
    * @access protected
    */
    protected $_port;

    /**
    * @var string 發件人
    * @access protected
    */
    protected $_from;

    /**
    * @var array 收件人
    * @access protected
    */
    protected $_to = array();

    /**
    * @var array 抄送
    * @access protected
    */
    protected $_cc = array();

    /**
    * @var array 祕密抄送
    * @access protected
    */
    protected $_bcc = array();

    /**
    * @var string 主題
    * @access protected
    */
    protected $_subject;

    /**
    * @var string 郵件正文
    * @access protected
    */
    protected $_body;

    /**
    * @var array 附件
    * @access protected
    */
    protected $_attachment = array();

    /**
    * @var reource socket資源
    * @access protected
    */
    protected $_socket;

    /**
    * @var reource 是不是安全鏈接
    * @access protected
    */
    protected $_isSecurity;

    /**
    * @var string 錯誤信息
    * @access protected
    */
    protected $_errorMessage;

    /**
    * 設置郵件傳輸代理，若是是能夠匿名發送有郵件的服務器，只需傳遞代理服務器地址就行
    * @access public
    * @param string $server 代理服務器的ip或者域名
    * @param string $username 認證帳號
    * @param string $password 認證密碼
    * @param int $port 代理服務器的端口，smtp默認25號端口
    * @param boolean $isSecurity 到服務器的鏈接是否爲安全鏈接，默認false
    * @return boolean
    */
    public function setServer($server, $username="", $password="", $port=25, $isSecurity=false) {
        $this->_sendServer = $server;
        $this->_port = $port;
        $this->_isSecurity = $isSecurity;
        $this->_userName = empty($username) ? "" : base64_encode($username);
        $this->_password = empty($password) ? "" : base64_encode($password);
        return true;
    }

    /**
    * 設置發件人
    * @access public
    * @param string $from 發件人地址
    * @return boolean
    */
    public function setFrom($from) {
        $this->_from = $from;
        return true;
    }

    /**
    * 設置收件人，多個收件人，調用屢次.
    * @access public
    * @param string $to 收件人地址
    * @return boolean
    */
    public function setReceiver($to) {
        $this->_to[] = $to;
        return true;
    }

    /**
    * 設置抄送，多個抄送，調用屢次.
    * @access public
    * @param string $cc 抄送地址
    * @return boolean
    */
    public function setCc($cc) {
        $this->_cc[] = $cc;
        return true;
    }

    /**
    * 設置祕密抄送，多個祕密抄送，調用屢次
    * @access public
    * @param string $bcc 祕密抄送地址
    * @return boolean
    */
    public function setBcc($bcc) {
        $this->_bcc[] = $bcc;
        return true;
    }

    /**
    * 設置郵件附件，多個附件，調用屢次
    * @access public
    * @param string $file 文件地址
    * @return boolean
    */
    public function addAttachment($file) {
        if(!file_exists($file)) {
            $this->_errorMessage = "file " . $file . " does not exist.";
            return false;
        }
        $this->_attachment[] = $file;
        return true;
    }

    /**
    * 設置郵件信息
    * @access public
    * @param string $body 郵件主題
    * @param string $subject 郵件主體內容，能夠是純文本，也但是是HTML文本
    * @return boolean
    */
    public function setMail($subject, $body) {
        $this->_subject = base64_encode($subject);
        $this->_body = base64_encode($body);
        return true;
    }

    /**
    * 發送郵件
    * @access public
    * @return boolean
    */
    public function sendMail() {
        $command = $this->getCommand();

        $this->_isSecurity ? $this->socketSecurity() : $this->socket();

        foreach ($command as $value) {
            $result = $this->_isSecurity ? $this->sendCommandSecurity($value[0], $value[1]) : $this->sendCommand($value[0], $value[1]);
            if($result) {
                continue;
            }
            else{
                return false;
            }
        }

        //其實這裏也不必關閉，smtp命令：QUIT發出以後，服務器就關閉了鏈接，本地的socket資源會自動釋放
        $this->_isSecurity ? $this->closeSecutity() : $this->close();
        return true;
    }

    /**
    * 返回錯誤信息
    * @return string
    */
    public function error(){
        if(!isset($this->_errorMessage)) {
            $this->_errorMessage = "";
        }
        return $this->_errorMessage;
    }

    /**
    * 返回mail命令
    * @access protected
    * @return array
    */
    protected function getCommand() {
        $separator = "----=_Part_" . md5($this->_from . time()) . uniqid(); //分隔符

        $command = array(
                array("HELO sendmail\r\n", 250)
            );
        if(!empty($this->_userName)){
            $command[] = array("AUTH LOGIN\r\n", 334);
            $command[] = array($this->_userName . "\r\n", 334);
            $command[] = array($this->_password . "\r\n", 235);
        }

        //設置發件人
        $command[] = array("MAIL FROM: <" . $this->_from . ">\r\n", 250);
        $header = "FROM: <" . $this->_from . ">\r\n";

        //設置收件人
        if(!empty($this->_to)) {
            $count = count($this->_to);
            if($count == 1){
                $command[] = array("RCPT TO: <" . $this->_to[0] . ">\r\n", 250);
                $header .= "TO: <" . $this->_to[0] .">\r\n";
            }
            else{
                for($i=0; $i<$count; $i++){
                    $command[] = array("RCPT TO: <" . $this->_to[$i] . ">\r\n", 250);
                    if($i == 0){
                        $header .= "TO: <" . $this->_to[$i] .">";
                    }
                    elseif($i + 1 == $count){
                        $header .= ",<" . $this->_to[$i] .">\r\n";
                    }
                    else{
                        $header .= ",<" . $this->_to[$i] .">";
                    }
                }
            }
        }

        //設置抄送
        if(!empty($this->_cc)) {
            $count = count($this->_cc);
            if($count == 1){
                $command[] = array("RCPT TO: <" . $this->_cc[0] . ">\r\n", 250);
                $header .= "CC: <" . $this->_cc[0] .">\r\n";
            }
            else{
                for($i=0; $i<$count; $i++){
                    $command[] = array("RCPT TO: <" . $this->_cc[$i] . ">\r\n", 250);
                    if($i == 0){
                    $header .= "CC: <" . $this->_cc[$i] .">";
                    }
                    elseif($i + 1 == $count){
                        $header .= ",<" . $this->_cc[$i] .">\r\n";
                    }
                    else{
                        $header .= ",<" . $this->_cc[$i] .">";
                    }
                }
            }
        }

        //設置祕密抄送
        if(!empty($this->_bcc)) {
            $count = count($this->_bcc);
            if($count == 1) {
                $command[] = array("RCPT TO: <" . $this->_bcc[0] . ">\r\n", 250);
                $header .= "BCC: <" . $this->_bcc[0] .">\r\n";
            }
            else{
                for($i=0; $i<$count; $i++){
                    $command[] = array("RCPT TO: <" . $this->_bcc[$i] . ">\r\n", 250);
                    if($i == 0){
                    $header .= "BCC: <" . $this->_bcc[$i] .">";
                    }
                    elseif($i + 1 == $count){
                        $header .= ",<" . $this->_bcc[$i] .">\r\n";
                    }
                    else{
                        $header .= ",<" . $this->_bcc[$i] .">";
                    }
                }
            }
        }

        //主題
        $header .= "Subject: =?UTF-8?B?" . $this->_subject ."?=\r\n";
        if(isset($this->_attachment)) {
            //含有附件的郵件頭須要聲明成這個
            $header .= "Content-Type: multipart/mixed;\r\n";
        }
        elseif(false){
            //郵件體含有圖片資源的,且包含的圖片在郵件內部時聲明成這個，若是是引用的遠程圖片，就不須要了
            $header .= "Content-Type: multipart/related;\r\n";
        }
        else{
            //html或者純文本的郵件聲明成這個
            $header .= "Content-Type: multipart/alternative;\r\n";
        }

        //郵件頭分隔符
        $header .= "\t" . 'boundary="' . $separator . '"';

        $header .= "\r\nMIME-Version: 1.0\r\n";

        //這裏開始是郵件的body部分，body部分分紅幾段發送
        $header .= "\r\n--" . $separator . "\r\n";
        $header .= "Content-Type:text/html; charset=utf-8\r\n";
        $header .= "Content-Transfer-Encoding: base64\r\n\r\n";
        $header .= $this->_body . "\r\n";
        $header .= "--" . $separator . "\r\n";

        //加入附件
        if(!empty($this->_attachment)){
            $count = count($this->_attachment);
            for($i=0; $i<$count; $i++){
                $header .= "\r\n--" . $separator . "\r\n";
                $header .= "Content-Type: " . $this->getMIMEType($this->_attachment[$i]) . '; name="=?UTF-8?B?' . base64_encode( basename($this->_attachment[$i]) ) . '?="' . "\r\n";
                $header .= "Content-Transfer-Encoding: base64\r\n";
                $header .= 'Content-Disposition: attachment; filename="=?UTF-8?B?' . base64_encode( basename($this->_attachment[$i]) ) . '?="' . "\r\n";
                $header .= "\r\n";
                $header .= $this->readFile($this->_attachment[$i]);
                $header .= "\r\n--" . $separator . "\r\n";
            }
        }

        //結束郵件數據發送
        $header .= "\r\n.\r\n";

        $command[] = array("DATA\r\n", 354);
        $command[] = array($header, 250);
        $command[] = array("QUIT\r\n", 221);

        return $command;
    }

    /**
    * 發送命令
    * @access protected
    * @param string $command 發送到服務器的smtp命令
    * @param int $code 指望服務器返回的響應嗎
    * @return boolean
    */
    protected function sendCommand($command, $code) {
        //發送命令給服務器
        try{
            if(@socket_write($this->_socket, $command, strlen($command))){

                //當郵件內容分屢次發送時，沒有$code，服務器沒有返回
                if(empty($code))  {
                    return true;
                }

                //讀取服務器返回
                $data = trim(socket_read($this->_socket, 1024));

                if($data) {
                    $pattern = "/^".$code."+?/";
                    if(preg_match($pattern, $data)) {
                        return true;
                    }
                    else{
                        $this->_errorMessage = "Error:" . $data . "|**| command:";
                        return false;
                    }
                }
                else{
                    $this->_errorMessage = "Error:" . socket_strerror(socket_last_error());
                    return false;
                }
            }
            else{
                $this->_errorMessage = "Error:" . socket_strerror(socket_last_error());
                return false;
            }
        }catch(Exception $e) {
            $this->_errorMessage = "Error:" . $e->getMessage();
        }
    }

    /**
    * 安全鏈接發送命令
    * @access protected
    * @param string $command 發送到服務器的smtp命令
    * @param int $code 指望服務器返回的響應嗎
    * @return boolean
    */
    protected function sendCommandSecurity($command, $code) {
        try {
            if(fwrite($this->_socket, $command)){
                //當郵件內容分屢次發送時，沒有$code，服務器沒有返回
                if(empty($code))  {
                    return true;
                }
                //讀取服務器返回
                $data = trim(fread($this->_socket, 1024));

                if($data) {
                    $pattern = "/^".$code."+?/";
                    if(preg_match($pattern, $data)) {
                        return true;
                    }
                    else{
                        $this->_errorMessage = "Error:" . $data . "|**| command:";
                        return false;
                    }
                }
                else{
                    return false;
                }
            }
            else{
                $this->_errorMessage = "Error: " . $command . " send failed";
                return false;
            }
        }catch(Exception $e) {
            $this->_errorMessage = "Error:" . $e->getMessage();
        }
    }

    /**
    * 讀取附件文件內容，返回base64編碼後的文件內容
    * @access protected
    * @param string $file 文件
    * @return mixed
    */
    protected function readFile($file) {
        if(file_exists($file)) {
            $file_obj = file_get_contents($file);
            return base64_encode($file_obj);
        }
        else {
            $this->_errorMessage = "file " . $file . " dose not exist";
            return false;
        }
    }

    /**
    * 獲取附件MIME類型
    * @access protected
    * @param string $file 文件
    * @return mixed
    */
    protected function getMIMEType($file) {
        if(file_exists($file)) {
            $mime = mime_content_type($file);
            /*if(! preg_match("/gif|jpg|png|jpeg/", $mime)){
                $mime = "application/octet-stream";
            }*/
            return $mime;
        }
        else {
            return false;
        }
    }

    /**
    * 創建到服務器的網絡鏈接
    * @access protected
    * @return boolean
    */
    protected function socket() {
        //建立socket資源
        $this->_socket = socket_create(AF_INET, SOCK_STREAM, getprotobyname('tcp'));

        if(!$this->_socket) {
            $this->_errorMessage = socket_strerror(socket_last_error());
            return false;
        }

        socket_set_block($this->_socket);//設置阻塞模式

        //鏈接服務器
        if(!@socket_connect($this->_socket, $this->_sendServer, $this->_port)) {
            $this->_errorMessage = socket_strerror(socket_last_error());
            return false;
        }
        $str = socket_read($this->_socket, 1024);
        if(!preg_match("/220+?/", $str)){
            $this->_errorMessage = $str;
            return false;
        }

        return true;
    }

    /**
    * 創建到服務器的SSL網絡鏈接
    * @access protected
    * @return boolean
    */
    protected function socketSecurity() {
        $remoteAddr = "tcp://" . $this->_sendServer . ":" . $this->_port;
        $this->_socket = stream_socket_client($remoteAddr, $errno, $errstr, 30);
        if(!$this->_socket){
            $this->_errorMessage = $errstr;
            return false;
        }

        //設置加密鏈接，默認是ssl，若是須要tls鏈接，能夠查看php手冊stream_socket_enable_crypto函數的解釋
        @stream_socket_enable_crypto($this->_socket, true, STREAM_CRYPTO_METHOD_SSLv23_CLIENT);

        stream_set_blocking($this->_socket, 1); //設置阻塞模式
        $str = fread($this->_socket, 1024);
        if(!preg_match("/220+?/", $str)){
            $this->_errorMessage = $str;
            return false;
        }

        return true;
    }

    /**
    * 關閉socket
    * @access protected
    * @return boolean
    */
    protected function close() {
        if(isset($this->_socket) && is_object($this->_socket)) {
            $this->_socket->close();
            return true;
        }
        $this->_errorMessage = "No resource can to be close";
        return false;
    }

    /**
    * 關閉安全socket
    * @access protected
    * @return boolean
    */
    protected function closeSecutity() {
        if(isset($this->_socket) && is_object($this->_socket)) {
            stream_socket_shutdown($this->_socket, STREAM_SHUT_WR);
            return true;
        }
        $this->_errorMessage = "No resource can to be close";
        return false;
    }
}