79 lines
2.2 KiB
PHP
79 lines
2.2 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Skobkin\Bundle\PointToolsBundle\DataFixtures\ORM;
|
||
|
|
|
||
|
|
use Doctrine\Common\DataFixtures\AbstractFixture;
|
||
|
|
use Doctrine\Common\DataFixtures\OrderedFixtureInterface;
|
||
|
|
use Doctrine\Common\Persistence\ObjectManager;
|
||
|
|
use Skobkin\Bundle\PointToolsBundle\Entity\Subscription;
|
||
|
|
use Skobkin\Bundle\PointToolsBundle\Entity\SubscriptionEvent;
|
||
|
|
use Skobkin\Bundle\PointToolsBundle\Entity\User;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Load user subscriptions
|
||
|
|
*/
|
||
|
|
class LoadSubscribersData extends AbstractFixture implements OrderedFixtureInterface
|
||
|
|
{
|
||
|
|
public function load(ObjectManager $om)
|
||
|
|
{
|
||
|
|
/** @var User[] $users */
|
||
|
|
$users = [
|
||
|
|
$this->getReference('test_user_99999'),
|
||
|
|
$this->getReference('test_user_99998'),
|
||
|
|
$this->getReference('test_user_99997'),
|
||
|
|
$this->getReference('test_user_99996'),
|
||
|
|
$this->getReference('test_user_99995'),
|
||
|
|
];
|
||
|
|
|
||
|
|
foreach ($users as $key => $user) {
|
||
|
|
// At least 2 subscribers for first user in the list
|
||
|
|
if (0 === $key) {
|
||
|
|
$minimum = 2;
|
||
|
|
} else {
|
||
|
|
$minimum = mt_rand(0, count($users));
|
||
|
|
}
|
||
|
|
|
||
|
|
$subscribers = $this->getRandomSubscribers($users, $minimum);
|
||
|
|
|
||
|
|
foreach ($subscribers as $subscriber) {
|
||
|
|
$subscription = new Subscription($user,$subscriber);
|
||
|
|
$subscriptionEvent = new SubscriptionEvent($user, $subscriber, SubscriptionEvent::ACTION_SUBSCRIBE);
|
||
|
|
$om->persist($subscription);
|
||
|
|
$om->persist($subscriptionEvent);
|
||
|
|
$user->addSubscriber($subscription);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
$om->flush();
|
||
|
|
}
|
||
|
|
|
||
|
|
public function getOrder()
|
||
|
|
{
|
||
|
|
return 4;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Returns array with random users from given users array
|
||
|
|
*
|
||
|
|
* @param User[] $users
|
||
|
|
* @param int $min
|
||
|
|
*
|
||
|
|
* @return User[]
|
||
|
|
*/
|
||
|
|
private function getRandomSubscribers($users, $min = 0)
|
||
|
|
{
|
||
|
|
if (0 === $number = mt_rand($min, count($users))) {
|
||
|
|
return [];
|
||
|
|
}
|
||
|
|
|
||
|
|
$keys = array_rand($users, $number);
|
||
|
|
|
||
|
|
$result = [];
|
||
|
|
|
||
|
|
foreach ($keys as $key) {
|
||
|
|
$result[] = $users[$key];
|
||
|
|
}
|
||
|
|
|
||
|
|
return $result;
|
||
|
|
}
|
||
|
|
}
|