mediawiki-extensions-OATHAuth/OATHUserRepository.php
Aaron Schulz 28d625e6bc Avoid DB connections on OATHUserRepository construction
Bug: T136224
Change-Id: I405d2f544409d635e4fac1d5222c42e8036c6945
2016-05-31 13:24:18 -07:00

56 lines
1.4 KiB
PHP

<?php
class OATHUserRepository {
/** @var LoadBalancer */
protected $lb;
public function __construct( LoadBalancer $lb ) {
$this->lb = $lb;
}
public function findByUser( User $user ) {
$oathUser = new OATHUser( $user, null );
$uid = CentralIdLookup::factory()->centralIdFromLocalUser( $user );
$res = $this->getDB( DB_SLAVE )
->selectRow( 'oathauth_users', '*', array( 'id' => $uid ), __METHOD__ );
if ( $res ) {
$key = new OATHAuthKey( $res->secret, explode( ',', $res->scratch_tokens ) );
$oathUser->setKey( $key );
}
return $oathUser;
}
public function persist( OATHUser $user ) {
$this->getDB( DB_MASTER )->replace(
'oathauth_users',
array( 'id' ),
array(
'id' => CentralIdLookup::factory()->centralIdFromLocalUser( $user->getUser() ),
'secret' => $user->getKey()->getSecret(),
'scratch_tokens' => implode( ',', $user->getKey()->getScratchTokens() ),
),
__METHOD__
);
}
public function remove( OATHUser $user ) {
$this->getDB( DB_MASTER )->delete(
'oathauth_users',
array( 'id' => CentralIdLookup::factory()->centralIdFromLocalUser( $user->getUser() ) ),
__METHOD__
);
}
/**
* @param integer $index DB_MASTER/DB_SLAVE
* @return DBConnRef
*/
private function getDB( $index ) {
global $wgOATHAuthDatabase;
return $this->lb->getConnectionRef( $index, array(), $wgOATHAuthDatabase );
}
}