1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166
|
<?php
/*
* This file is part of SwiftMailer.
* (c) 2004-2009 Chris Corbyn
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/
/**
* Redundantly and rotationally uses several Transports when sending.
*
* @author Chris Corbyn
*/
class Swift_Transport_LoadBalancedTransport implements Swift_Transport
{
/**
* Transports which are deemed useless.
*
* @var Swift_Transport[]
*/
private $_deadTransports = array();
/**
* The Transports which are used in rotation.
*
* @var Swift_Transport[]
*/
protected $_transports = array();
/**
* Creates a new LoadBalancedTransport.
*/
public function __construct()
{
}
/**
* Set $transports to delegate to.
*
* @param Swift_Transport[] $transports
*/
public function setTransports(array $transports)
{
$this->_transports = $transports;
$this->_deadTransports = array();
}
/**
* Get $transports to delegate to.
*
* @return Swift_Transport[]
*/
public function getTransports()
{
return array_merge($this->_transports, $this->_deadTransports);
}
/**
* Test if this Transport mechanism has started.
*
* @return bool
*/
public function isStarted()
{
return count($this->_transports) > 0;
}
/**
* Start this Transport mechanism.
*/
public function start()
{
$this->_transports = array_merge($this->_transports, $this->_deadTransports);
}
/**
* Stop this Transport mechanism.
*/
public function stop()
{
foreach ($this->_transports as $transport) {
$transport->stop();
}
}
/**
* Send the given Message.
*
* Recipient/sender data will be retrieved from the Message API.
* The return value is the number of recipients who were accepted for delivery.
*
* @param Swift_Mime_Message $message
* @param string[] $failedRecipients An array of failures by-reference
*
* @return int
*/
public function send(Swift_Mime_Message $message, &$failedRecipients = null)
{
$maxTransports = count($this->_transports);
$sent = 0;
for ($i = 0; $i < $maxTransports
&& $transport = $this->_getNextTransport(); ++$i) {
try {
if (!$transport->isStarted()) {
$transport->start();
}
if ($sent = $transport->send($message, $failedRecipients)) {
break;
}
} catch (Swift_TransportException $e) {
$this->_killCurrentTransport();
}
}
if (count($this->_transports) == 0) {
throw new Swift_TransportException(
'All Transports in LoadBalancedTransport failed, or no Transports available'
);
}
return $sent;
}
/**
* Register a plugin.
*
* @param Swift_Events_EventListener $plugin
*/
public function registerPlugin(Swift_Events_EventListener $plugin)
{
foreach ($this->_transports as $transport) {
$transport->registerPlugin($plugin);
}
}
/**
* Rotates the transport list around and returns the first instance.
*
* @return Swift_Transport
*/
protected function _getNextTransport()
{
if ($next = array_shift($this->_transports)) {
$this->_transports[] = $next;
}
return $next;
}
/**
* Tag the currently used (top of stack) transport as dead/useless.
*/
protected function _killCurrentTransport()
{
if ($transport = array_pop($this->_transports)) {
try {
$transport->stop();
} catch (Exception $e) {
}
$this->_deadTransports[] = $transport;
}
}
}
|