136 lines
3.0 KiB
PHP
136 lines
3.0 KiB
PHP
<?php
|
|
|
|
namespace app\common\library\token\driver;
|
|
|
|
use app\common\library\token\Driver;
|
|
|
|
/**
|
|
* Token操作类.
|
|
*/
|
|
class Mysql extends Driver
|
|
{
|
|
/**
|
|
* 默认配置.
|
|
*
|
|
* @var array
|
|
*/
|
|
protected $options = [
|
|
'table' => 'user_token',
|
|
'expire' => 2592000,
|
|
'connection' => [],
|
|
];
|
|
|
|
/**
|
|
* 构造函数.
|
|
*
|
|
* @param array $options 参数
|
|
*/
|
|
public function __construct($options = [])
|
|
{
|
|
if (! empty($options)) {
|
|
$this->options = array_merge($this->options, $options);
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* @return \think\db\BaseQuery|\think\facade\Db
|
|
*/
|
|
private function getHandler(){
|
|
if ($this->options['connection']) {
|
|
return\think\facade\Db::connect($this->options['connection'])->name($this->options['table']);
|
|
} else {
|
|
return \think\facade\Db::name($this->options['table']);
|
|
}
|
|
}
|
|
/**
|
|
* 存储Token.
|
|
*
|
|
* @param string $token Token
|
|
* @param int $user_id 会员ID
|
|
* @param int $expire 过期时长,0表示无限,单位秒
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function set($token, $user_id, $expire = null)
|
|
{
|
|
$expiretime = ! is_null($expire) && $expire !== 0 ? time() + $expire : 0;
|
|
$token = $this->getEncryptedToken($token);
|
|
$this->getHandler()->insert([
|
|
'token' => $token, 'user_id' => $user_id, 'createtime' => time(),
|
|
'expiretime' => $expiretime,
|
|
]);
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 获取Token内的信息.
|
|
*
|
|
* @param string $token
|
|
*
|
|
* @return array
|
|
*/
|
|
public function get($token)
|
|
{
|
|
$data = $this->getHandler()->where('token', $this->getEncryptedToken($token))->find();
|
|
if ($data) {
|
|
if (! $data['expiretime'] || $data['expiretime'] > time()) {
|
|
//返回未加密的token给客户端使用
|
|
$data['token'] = $token;
|
|
//返回剩余有效时间
|
|
$data['expires_in'] = $this->getExpiredIn($data['expiretime']);
|
|
|
|
return $data;
|
|
} else {
|
|
self::delete($token);
|
|
}
|
|
}
|
|
|
|
return [];
|
|
}
|
|
|
|
/**
|
|
* 判断Token是否可用.
|
|
*
|
|
* @param string $token Token
|
|
* @param int $user_id 会员ID
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function check($token, $user_id)
|
|
{
|
|
$data = $this->get($token);
|
|
|
|
return $data && $data['user_id'] == $user_id ? true : false;
|
|
}
|
|
|
|
/**
|
|
* 删除Token.
|
|
*
|
|
* @param string $token
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function delete($token)
|
|
{
|
|
$this->getHandler()->where('token', $this->getEncryptedToken($token))->delete();
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* 删除指定用户的所有Token.
|
|
*
|
|
* @param int $user_id
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function clear($user_id)
|
|
{
|
|
$this->getHandler()->where('user_id', $user_id)->delete();
|
|
|
|
return true;
|
|
}
|
|
}
|