Page MenuHomeWickedGov Phorge
Files F4128607

MessageGroupSubscriptionStore.php
No OneTemporary
Actions

Size
2 KB
Referenced Files
None
Subscribers
None

MessageGroupSubscriptionStore.php
View Options

<?php
declare( strict_types = 1 );
namespace MediaWiki\Extension\Translate\MessageGroupProcessing;
use Wikimedia\Rdbms\IConnectionProvider;
use Wikimedia\Rdbms\IResultWrapper;
/**
* Store service for looking up and storing user subscriptions to message group
* @since 2024.04
* @license GPL-2.0-or-later
* @author Abijeet Patro
*/
class MessageGroupSubscriptionStore {
private const VIRTUAL_DOMAIN = 'virtual-translate';
private const TABLE_NAME = 'translate_message_group_subscriptions';
/** @var int Match field tmgs_group byte length */
private const MAX_GROUP_LENGTH = 200;
private IConnectionProvider $dbProvider;
public function __construct( IConnectionProvider $dbProvider ) {
$this->dbProvider = $dbProvider;
}
public function addSubscription( string $groupId, int $userId ): void {
$this->dbProvider->getPrimaryDatabase( self::VIRTUAL_DOMAIN )
->newReplaceQueryBuilder()
->replaceInto( self::TABLE_NAME )
->uniqueIndexFields( [ 'tmgs_group', 'tmgs_user_id' ] )
->row( [
'tmgs_group' => self::getGroupIdForDatabase( $groupId ),
'tmgs_user_id' => $userId,
] )
->caller( __METHOD__ )
->execute();
}
public function getSubscriptions( ?array $groupIds, ?int $userId ): IResultWrapper {
$queryBuilder = $this->dbProvider
->getReplicaDatabase( self::VIRTUAL_DOMAIN )
->newSelectQueryBuilder()
->select( [ 'tmgs_group', 'tmgs_user_id' ] )
->from( self::TABLE_NAME )
->caller( __METHOD__ );
if ( $groupIds !== null ) {
$dbGroupIds = [];
foreach ( $groupIds as $groupId ) {
$dbGroupIds[] = self::getGroupIdForDatabase( $groupId );
}
$queryBuilder->where( [ 'tmgs_group' => $dbGroupIds ] );
}
if ( $userId !== null ) {
$queryBuilder->andWhere( [ 'tmgs_user_id' => $userId ] );
}
return $queryBuilder->fetchResultSet();
}
public function getSubscriptionByGroupUnion( array $groupIds ): array {
$queryBuilder = $this->dbProvider
->getReplicaDatabase( self::VIRTUAL_DOMAIN )
->newSelectQueryBuilder()
->select( [ 'tmgs_user_id' ] )
->from( self::TABLE_NAME )
->where( [ 'tmgs_group' => $groupIds ] )
->having( 'COUNT(tmgs_group) = ' . count( $groupIds ) )
->caller( __METHOD__ );
return $queryBuilder->fetchFieldValues();
}
public function removeSubscriptions( string $groupId, int $userId ): void {
$this->dbProvider->getPrimaryDatabase( self::VIRTUAL_DOMAIN )
->newDeleteQueryBuilder()
->deleteFrom( self::TABLE_NAME )
->where( [
'tmgs_group' => $groupId,
'tmgs_user_id' => $userId
] )
->caller( __METHOD__ )
->execute();
}
private static function getGroupIdForDatabase( string $groupId ): string {
// Check if length is more than 200 bytes
if ( strlen( $groupId ) <= self::MAX_GROUP_LENGTH ) {
return $groupId;
}
$hash = hash( 'md5', $groupId );
// We take 160 bytes of the original string and append the md5 hash (32 bytes)
return mb_strcut( $groupId, 0, 160 ) . '||' . $hash;
}
}

File Metadata

Mime Type
text/x-php
Expires
Wed, Aug 19, 09:21 (2 w, 5 d ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
34/f5/f1001d2bf9a911324b114ffcdb4e
Default Alt Text
MessageGroupSubscriptionStore.php (2 KB)

Event Timeline