kopia lustrzana https://github.com/nextcloud/social
new table and migration of recipients
Signed-off-by: Maxence Lange <maxence@artificial-owl.com>pull/752/head
rodzic
bbc89060d0
commit
a2cd51b560
|
@ -0,0 +1,237 @@
|
|||
<?php
|
||||
declare(strict_types=1);
|
||||
|
||||
|
||||
/**
|
||||
* Nextcloud - Social Support
|
||||
*
|
||||
* This file is licensed under the Affero General Public License version 3 or
|
||||
* later. See the COPYING file.
|
||||
*
|
||||
* @author Maxence Lange <maxence@artificial-owl.com>
|
||||
* @copyright 2018, Maxence Lange <maxence@artificial-owl.com>
|
||||
* @license GNU AGPL version 3 or any later version
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of the
|
||||
* License, or (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
*/
|
||||
|
||||
|
||||
namespace OCA\Social\Migration;
|
||||
|
||||
|
||||
use Closure;
|
||||
use Doctrine\DBAL\Exception\UniqueConstraintViolationException;
|
||||
use Doctrine\DBAL\Schema\SchemaException;
|
||||
use Exception;
|
||||
use OCP\DB\ISchemaWrapper;
|
||||
use OCP\IDBConnection;
|
||||
use OCP\Migration\IOutput;
|
||||
use OCP\Migration\SimpleMigrationStep;
|
||||
|
||||
|
||||
/**
|
||||
* Class Version0002Date20190916000001
|
||||
*
|
||||
* @package OCA\Social\Migration
|
||||
*/
|
||||
class Version0002Date20190916000001 extends SimpleMigrationStep {
|
||||
|
||||
|
||||
/** @var IDBConnection */
|
||||
private $connection;
|
||||
|
||||
|
||||
/**
|
||||
* @param IDBConnection $connection
|
||||
*/
|
||||
public function __construct(IDBConnection $connection) {
|
||||
$this->connection = $connection;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param IOutput $output
|
||||
* @param Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
|
||||
* @param array $options
|
||||
*
|
||||
* @return ISchemaWrapper
|
||||
* @throws SchemaException
|
||||
*/
|
||||
public function changeSchema(IOutput $output, Closure $schemaClosure, array $options
|
||||
): ISchemaWrapper {
|
||||
/** @var ISchemaWrapper $schema */
|
||||
$schema = $schemaClosure();
|
||||
|
||||
$table = $schema->getTable('social_a2_follows');
|
||||
if (!$table->hasColumn('follow_id_prim')) {
|
||||
$table->addColumn(
|
||||
'follow_id_prim', 'string',
|
||||
[
|
||||
'notnull' => true,
|
||||
'length' => 128,
|
||||
]
|
||||
);
|
||||
}
|
||||
|
||||
if (!$schema->hasTable('social_a2_stream_dest')) {
|
||||
$table = $schema->createTable('social_a2_stream_dest');
|
||||
|
||||
$table->addColumn(
|
||||
'stream_id', 'string',
|
||||
[
|
||||
'notnull' => true,
|
||||
'length' => 128,
|
||||
]
|
||||
);
|
||||
$table->addColumn(
|
||||
'actor_id', 'string',
|
||||
[
|
||||
'notnull' => true,
|
||||
'length' => 128,
|
||||
]
|
||||
);
|
||||
$table->addColumn(
|
||||
'type', 'string',
|
||||
[
|
||||
'notnull' => false,
|
||||
'length' => 7,
|
||||
]
|
||||
);
|
||||
|
||||
$table->addUniqueIndex(['stream_id', 'actor_id', 'type'], 'recipient');
|
||||
$table->setPrimaryKey(['stream_id', 'actor_id', 'type']);
|
||||
}
|
||||
|
||||
return $schema;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param IOutput $output
|
||||
* @param Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
|
||||
* @param array $options
|
||||
*
|
||||
* @throws Exception
|
||||
*/
|
||||
public function postSchemaChange(IOutput $output, Closure $schemaClosure, array $options) {
|
||||
/** @var ISchemaWrapper $schema */
|
||||
$schema = $schemaClosure();
|
||||
|
||||
$this->updateTableFollows($schema);
|
||||
$this->fillTableStreamDest($schema);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param ISchemaWrapper $schema
|
||||
*/
|
||||
private function updateTableFollows(ISchemaWrapper $schema) {
|
||||
if (!$schema->hasTable('social_a2_follows')) {
|
||||
return;
|
||||
}
|
||||
|
||||
$qb = $this->connection->getQueryBuilder();
|
||||
$qb->select('*')
|
||||
->from('social_a2_follows');
|
||||
|
||||
$cursor = $qb->execute();
|
||||
while ($data = $cursor->fetch()) {
|
||||
$this->updateFollowPrim($data);
|
||||
}
|
||||
|
||||
$cursor->closeCursor();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param array $data
|
||||
*/
|
||||
private function updateFollowPrim(array $data) {
|
||||
if ($data['follow_id_prim'] !== '') {
|
||||
return;
|
||||
}
|
||||
|
||||
$update = $this->connection->getQueryBuilder();
|
||||
$update->update('social_a2_follows');
|
||||
$update->set('follow_id_prim', $update->createNamedParameter(hash('sha512', $data['follow_id'])));
|
||||
|
||||
$expr = $update->expr();
|
||||
$update->where($expr->eq('id_prim', $update->createNamedParameter($data['id_prim'])));
|
||||
|
||||
$update->execute();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param ISchemaWrapper $schema
|
||||
*/
|
||||
private function fillTableStreamDest(ISchemaWrapper $schema) {
|
||||
if (!$schema->hasTable('social_a2_stream')) {
|
||||
return;
|
||||
}
|
||||
|
||||
$start = 0;
|
||||
$limit = 1000;
|
||||
while (true) {
|
||||
$qb = $this->connection->getQueryBuilder();
|
||||
$qb->select('id_prim', 'to_array', 'cc', 'bcc')
|
||||
->from('social_a2_stream')
|
||||
->setMaxResults(1000)
|
||||
->setFirstResult($start);
|
||||
|
||||
$cursor = $qb->execute();
|
||||
$count = 0;
|
||||
while ($data = $cursor->fetch()) {
|
||||
$count++;
|
||||
|
||||
$this->insertStreamDest($data);
|
||||
}
|
||||
$cursor->closeCursor();
|
||||
|
||||
$start += $count;
|
||||
if ($count < $limit) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private function insertStreamDest($data) {
|
||||
$recipients = [];
|
||||
$recipients['to'] = json_decode($data['to_array'], true);
|
||||
$recipients['cc'] = json_decode($data['cc'], true);
|
||||
$recipients['bcc'] = json_decode($data['bcc'], true);
|
||||
|
||||
$streamId = $data['id_prim'];
|
||||
foreach (array_keys($recipients) as $dest) {
|
||||
$type = $dest;
|
||||
foreach ($recipients[$dest] as $actorId) {
|
||||
$insert = $this->connection->getQueryBuilder();
|
||||
$insert->insert('social_a2_stream_dest');
|
||||
|
||||
$insert->setValue('stream_id', $insert->createNamedParameter($streamId));
|
||||
$insert->setValue('actor_id', $insert->createNamedParameter(hash('sha512', $actorId)));
|
||||
$insert->setValue('type', $insert->createNamedParameter($type));
|
||||
|
||||
try {
|
||||
$insert->execute();
|
||||
} catch (UniqueConstraintViolationException $e) {
|
||||
\OC::$server->getLogger()
|
||||
->log(3, 'Social - Duplicate recipient on Stream ' . json_encode($data));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
|
@ -49,6 +49,9 @@ class Follow extends ACore implements JsonSerializable {
|
|||
/** @var string */
|
||||
private $followId = '';
|
||||
|
||||
/** @var string */
|
||||
private $followIdPrim = '';
|
||||
|
||||
/** @var bool */
|
||||
private $accepted = false;
|
||||
|
||||
|
@ -84,6 +87,25 @@ class Follow extends ACore implements JsonSerializable {
|
|||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return string
|
||||
*/
|
||||
public function getFollowIdPrim(): string {
|
||||
return $this->followIdPrim;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $followIdPrim
|
||||
*
|
||||
* @return Follow
|
||||
*/
|
||||
public function setFollowIdPrim(string $followIdPrim): Follow {
|
||||
$this->followIdPrim = $followIdPrim;
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return bool
|
||||
*/
|
||||
|
@ -119,6 +141,7 @@ class Follow extends ACore implements JsonSerializable {
|
|||
|
||||
$this->setAccepted(($this->getInt('accepted', $data, 0) === 1) ? true : false);
|
||||
$this->setFollowId($this->get('follow_id', $data, ''));
|
||||
$this->setFollowIdPrim($this->get('follow_id_prim', $data, ''));
|
||||
}
|
||||
|
||||
|
||||
|
@ -132,8 +155,9 @@ class Follow extends ACore implements JsonSerializable {
|
|||
$result = array_merge(
|
||||
$result,
|
||||
[
|
||||
'follow_id' => $this->getFollowId(),
|
||||
'accepted' => $this->isAccepted()
|
||||
'follow_id' => $this->getFollowId(),
|
||||
'follow_id_prim' => $this->getFollowIdPrim(),
|
||||
'accepted' => $this->isAccepted()
|
||||
]
|
||||
);
|
||||
}
|
||||
|
|
Ładowanie…
Reference in New Issue