View source
<?php
namespace Drupal\flag;
use Drupal\Component\Datetime\TimeInterface;
use Drupal\Core\Database\Connection;
use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Session\AccountInterface;
use Drupal\flag\Event\FlagEvents;
use Drupal\flag\Event\FlaggingEvent;
use Drupal\flag\Event\UnflaggingEvent;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;
class FlagCountManager implements FlagCountManagerInterface, EventSubscriberInterface {
protected $entityCounts = [];
protected $flagCounts = [];
protected $flagEntityCounts = [];
protected $userFlagCounts = [];
protected $connection;
protected $dateTime;
public function __construct(Connection $connection, TimeInterface $date_time) {
$this->connection = $connection;
$this->dateTime = $date_time;
}
public static function create(ContainerInterface $container) {
return new self($container
->get('database'), $container
->get('datetime.time'));
}
public function getEntityFlagCounts(EntityInterface $entity) {
$entity_type = $entity
->getEntityTypeId();
$entity_id = $entity
->id();
if (!isset($this->entityCounts[$entity_type][$entity_id])) {
$this->entityCounts[$entity_type][$entity_id] = [];
$query = $this->connection
->select('flag_counts', 'fc');
$result = $query
->fields('fc', [
'flag_id',
'count',
])
->condition('fc.entity_type', $entity_type)
->condition('fc.entity_id', $entity_id)
->execute();
foreach ($result as $row) {
$this->entityCounts[$entity_type][$entity_id][$row->flag_id] = $row->count;
}
}
return $this->entityCounts[$entity_type][$entity_id];
}
public function getFlagFlaggingCount(FlagInterface $flag) {
$flag_id = $flag
->id();
$entity_type = $flag
->getFlaggableEntityTypeId();
if (!isset($this->flagCounts[$flag_id][$entity_type])) {
$query = $this->connection
->select('flagging', 'f')
->condition('flag_id', $flag_id)
->condition('entity_type', $entity_type);
$query
->addExpression('COUNT(*)');
$this->flagCounts[$flag_id][$entity_type] = $query
->execute()
->fetchField();
}
return $this->flagCounts[$flag_id][$entity_type];
}
public function getFlagEntityCount(FlagInterface $flag) {
$flag_id = $flag
->id();
if (!isset($this->flagEntityCounts[$flag_id])) {
$query = $this->connection
->select('flag_counts', 'fc')
->condition('flag_id', $flag_id);
$query
->addExpression('COUNT(*)');
$this->flagEntityCounts[$flag_id] = $query
->execute()
->fetchField();
}
return $this->flagEntityCounts[$flag_id];
}
public function getUserFlagFlaggingCount(FlagInterface $flag, AccountInterface $user, $session_id = NULL) {
$flag_id = $flag
->id();
$uid = $user
->id();
$get_by_session_id = $user
->isAnonymous();
if ($get_by_session_id) {
if (is_null($session_id)) {
throw new \LogicException('Anonymous users must be identified by session_id');
}
if (isset($this->userFlagCounts[$flag_id][$uid][$session_id])) {
return $this->userFlagCounts[$flag_id][$uid][$session_id];
}
}
elseif (isset($this->userFlagCounts[$flag_id][$uid])) {
return $this->userFlagCounts[$flag_id][$uid];
}
$query = $this->connection
->select('flagging', 'f')
->condition('flag_id', $flag_id)
->condition('uid', $uid);
if ($get_by_session_id) {
$query
->condition('session_id', $session_id);
}
$query
->addExpression('COUNT(*)');
$result = $query
->execute()
->fetchField();
if ($get_by_session_id) {
$this->userFlagCounts[$flag_id][$uid][$session_id] = $result;
}
else {
$this->userFlagCounts[$flag_id][$uid] = $result;
}
return $result;
}
public function incrementFlagCounts(FlaggingEvent $event) {
$flagging = $event
->getFlagging();
$flag = $flagging
->getFlag();
$entity = $flagging
->getFlaggable();
$this->connection
->merge('flag_counts')
->key([
'flag_id' => $flag
->id(),
'entity_id' => $entity
->id(),
'entity_type' => $entity
->getEntityTypeId(),
])
->fields([
'last_updated' => $this->dateTime
->getRequestTime(),
'count' => 1,
])
->expression('count', 'count + :inc', [
':inc' => 1,
])
->execute();
$this
->resetLoadedCounts($entity, $flag);
}
public function decrementFlagCounts(UnflaggingEvent $event) {
$flaggings_count = [];
$flag_ids = [];
$entity_ids = [];
$flaggings = $event
->getFlaggings();
foreach ($flaggings as $flagging) {
$flag_id = $flagging
->getFlagId();
$entity_id = $flagging
->getFlaggableId();
$flag_ids[$flag_id] = $flag_id;
$entity_ids[$entity_id] = $entity_id;
if (!isset($flaggings_count[$flag_id][$entity_id])) {
$flaggings_count[$flag_id][$entity_id] = 1;
}
else {
$flaggings_count[$flag_id][$entity_id]++;
}
$this
->resetLoadedCounts($flagging
->getFlaggable(), $flagging
->getFlag());
}
$result = $this->connection
->select('flag_counts')
->fields('flag_counts', [
'flag_id',
'entity_type',
'entity_id',
'count',
])
->condition('flag_id', $flag_ids, 'IN')
->condition('entity_id', $entity_ids, 'IN')
->execute();
$to_delete = [];
foreach ($result as $row) {
if (!isset($flaggings_count[$row->flag_id][$row->entity_id])) {
continue;
}
if ($row->count <= $flaggings_count[$row->flag_id][$row->entity_id]) {
$to_delete[$row->flag_id][] = $row->entity_id;
}
else {
$this->connection
->update('flag_counts')
->expression('count', 'count - :decrement', [
':decrement' => $flaggings_count[$row->flag_id][$row->entity_id],
])
->condition('flag_id', $row->flag_id)
->condition('entity_id', $row->entity_id)
->execute();
}
}
foreach ($to_delete as $flag_id => $entity_ids) {
$this->connection
->delete('flag_counts')
->condition('flag_id', $flag_id)
->condition('entity_id', $entity_ids, 'IN')
->execute();
}
}
public static function getSubscribedEvents() {
$events = [];
$events[FlagEvents::ENTITY_FLAGGED][] = [
'incrementFlagCounts',
-100,
];
$events[FlagEvents::ENTITY_UNFLAGGED][] = [
'decrementFlagCounts',
-100,
];
return $events;
}
protected function resetLoadedCounts(EntityInterface $entity, FlagInterface $flag) {
unset($this->entityCounts[$entity
->getEntityTypeId()][$entity
->id()]);
unset($this->flagCounts[$flag
->id()]);
unset($this->flagEntityCounts[$flag
->id()]);
unset($this->userFlagCounts[$flag
->id()]);
}
}