3 * Table Definition for core_notify_server
5 class_exists('DB_DataObject') ? '' : require_once 'DB/DataObject.php';
7 class Pman_Core_DataObjects_Core_notify_server extends DB_DataObject
10 /* the code below is auto generated do not remove the above tag */
12 public $__table = 'core_notify_server'; // table name
13 public $id; // int(11) not_null primary_key auto_increment
22 function applyFilters($q, $au, $roo)
24 if (isset($q['_with_queue_size'])) {
25 $this->addQueueSize();
30 function addQueueSize()
32 // look for database links for server_id (which should find core_notify + others..)
33 $cn = get_class(DB_DataObject::factory('core_notify'));
35 foreach($this->databaseLinks() as $tbl => $kv) {
36 foreach($kv as $k=>$v) {
37 if ($v != 'core_notify_server:id') {
41 $test = DB_DAtaObject::factory($tbl);
42 if (!is_a($test, $cn)) {
49 die("OOPS - no tables for notify_server references");
52 foreach($tables as $t) {
59 server_id = core_notify_server.id
61 sent < '1970-01-01' OR sent IS NULL
67 $this->selectAdd("(" . implode(" + ", $totals) . ") as in_queue ");
76 // most services should call this first..
78 function getCurrent($notify, $force = false)
80 static $current = false;;
82 if ($current !== false) {
86 $ns = DB_DataObject::factory('core_notify_server');
88 $ns->poolname = $notify->poolname;
90 $ns->hostname = gethostname();
92 if ($ns->find(true)) {
97 $notify->jerr("Server not found for this server " . gethostname() . " in core_notify_server" );
99 // fallback to any server - if we are using force. (this is so helo will work...)
101 $ns = DB_DataObject::factory('core_notify_server');
103 $ns->hostname = gethostname();
104 if (!$ns->find(true)) {
105 $notify->jerr("Server not found for this server " . gethostname() . " in core_notify_server" );
112 function isFirstServer()
114 $servers = $this->availableServers();
115 if (empty($servers)) {
118 // only run this on the first server...
119 return $this->id == $servers[0]->id;
123 // called on current server.
124 function assignQueues($notify)
128 $servers = $this->availableServers();
130 foreach($servers as $s) {
136 $notify->jerr("no configured servers in core_notify_server for poolname = {$notify->poolname}");
140 // only run this on the first server...
141 if ($this->id != $ids[0]) {
144 foreach($ids as $rn) {
148 $num_servers = count($ids);
150 if ($num_servers == 1) {
151 $p = DB_DataObject::factory($notify->table);
156 server_id = {$ids[0]}
162 act_start < NOW() + INTERVAL 3 HOUR
164 server_id != {$ids[0]}
171 $p = DB_DataObject::factory($notify->table);
177 act_start < NOW() + INTERVAL 3 HOUR
179 server_id NOT IN (" . implode(",", $ids) . ")
181 $p->orderBy('act_when asc'); //?
182 $total_add = $p->count();
183 if ($total_add < 1) {
187 $to_add = $p->fetchAll('id');
189 $p = DB_DataObject::factory($notify->table);
196 server_id IN (" . implode(",", $ids) . ")
199 $p->selectAdd('server_id, count(id) as n');
200 $p->groupBy('server_id');
201 $in_q = $p->fetchAll('server_id', 'n');
203 // if queue is empty it will not get allocated anything.
204 foreach($ids as $sid) {
205 if (!isset($in_q[$sid])) {
210 foreach($in_q as $sid => $n) {
216 $target_len = floor( ($totalq + $total_add) / $num_servers );
218 foreach($in_q as $sid => $cq) {
219 if ( $target_len - $cq < 1) {
222 $up[ $sid ] = array_slice($to_add, 0, $target_len - $cq);
224 foreach($to_add as $i) {
225 $up[ $ids[0] ][] = $i;
228 // distribution needs to go to ones that have the shortest queues. - so to balance out the queues
232 foreach($up as $sid => $nids) {
236 $p = DB_DataObject::factory($notify->table);
243 id IN (". implode(',', $nids). ')'
247 DB_DataObject::factory("core_notify_blacklist")->prune();
250 // called on current server.
252 function availableServers()
254 $ns = DB_DataObject::factory('core_notify_server');
255 $ns->poolname = $this->poolname;
257 $ns->orderBy('id ASC');
258 return $ns->fetchAll();
262 function updateNotifyToNextServer( $cn , $when = false, $allow_same = false)
264 // fixme - this should take into account blacklisted - and return false if no more servers are available
265 $email = empty($cn->to_email) ? ($cn->person() ? $cn->person()->email : $cn->to_email) : $cn->to_email;
267 $w = DB_DataObject::factory($cn->tableName());
270 $servers = $this->availableServers();
272 foreach($servers as $i => $s) {
273 if ($s->id == $this->id) {
278 $offset = ($start + 1) % count($servers);
280 while ($offset != $start) {
281 $s = $servers[$offset];
282 if (!$s->isBlacklisted($email)) {
286 $offset = ($offset + 1) % count($servers);
289 if ($good == false && $allow_same) {
293 if ($good == false) {
300 $w->server_id = $good->id;
301 $w->act_when = $when === false ? $w->sqlValue('NOW() + INTERVAL 1 MINUTE') : $when;
307 function isBlacklisted($email)
309 // return current server id..
310 static $cache = array();
312 $ea = explode('@',$email);
313 $dom = strtolower(array_pop($ea));
314 if (isset( $cache[$this->id . '-'. $dom])) {
315 return $cache[$this->id . '-'. $dom];
318 $cd = DB_DataObject::factory('core_domain')->loadOrCreate($dom);
320 $bl = DB_DataObject::factory('core_notify_blacklist');
321 $bl->server_id = $this->id;
322 $bl->domain_id = $cd->id;
324 $cache[$this->id . '-'. $dom] = true;
332 $ff = HTML_FlexyFramework::get();
333 $ff->Mail['helo'] = $this->helo;
336 function checkSmtpResponse($errmsg, $core_domain)
338 $bl = DB_DataObject::factory('core_notify_blacklist');
339 $bl->server_id = $this->id;
340 $bl->domain_id = $core_domain->id;
344 // is it a blacklist message
345 if (!$bl->messageIsBlacklisted($errmsg)) {
348 $bl->error_str = $errmsg;
349 $bl->added_dt = $bl->sqlValue("NOW()");