62 lines
1.7 KiB
PHP
62 lines
1.7 KiB
PHP
<?php
|
|
namespace MailPoet\Models;
|
|
|
|
if(!defined('ABSPATH')) exit;
|
|
|
|
class ScheduledTaskSubscriber extends Model {
|
|
const STATUS_TO_PROCESS = 0;
|
|
const STATUS_PROCESSED = 1;
|
|
|
|
public static $_table = MP_SCHEDULED_TASK_SUBSCRIBERS_TABLE;
|
|
public static $_id_column = array('task_id', 'subscriber_id');
|
|
|
|
static function createOrUpdate($data = array()) {
|
|
if(!is_array($data) || empty($data['task_id']) || empty($data['subscriber_id'])) return;
|
|
|
|
$task_subscriber = self::where('subscriber_id', $data['subscriber_id'])
|
|
->where('task_id', $data['task_id'])
|
|
->findOne();
|
|
|
|
if(empty($task_subscriber)) $task_subscriber = self::create();
|
|
|
|
$task_subscriber->task_id = $data['task_id'];
|
|
$task_subscriber->subscriber_id = $data['subscriber_id'];
|
|
$task_subscriber->processed = self::STATUS_TO_PROCESS;
|
|
$task_subscriber->save();
|
|
|
|
return $task_subscriber;
|
|
}
|
|
|
|
static function addSubscribers($task_id, array $subscriber_ids) {
|
|
foreach($subscriber_ids as $subscriber_id) {
|
|
self::createOrUpdate(array(
|
|
'task_id' => $task_id,
|
|
'subscriber_id' => $subscriber_id
|
|
));
|
|
}
|
|
}
|
|
|
|
static function getToProcessCount($task_id) {
|
|
return self::getCount($task_id, false);
|
|
}
|
|
|
|
static function getProcessedCount($task_id) {
|
|
return self::getCount($task_id, true);
|
|
}
|
|
|
|
static function getTotalCount($task_id) {
|
|
return self::getCount($task_id);
|
|
}
|
|
|
|
private static function getCount($task_id, $processed = null) {
|
|
$orm = self::where('task_id', $task_id);
|
|
if(!is_null($processed)) {
|
|
$orm->where(
|
|
'processed',
|
|
($processed) ? self::STATUS_PROCESSED : self::STATUS_TO_PROCESS
|
|
);
|
|
}
|
|
return $orm->count();
|
|
}
|
|
}
|