81 lines
2.3 KiB
PHP
81 lines
2.3 KiB
PHP
|
<?php
|
|||
|
|
/**
|
|||
|
|
* 需登录的功能必须继承该类
|
|||
|
|
*/
|
|||
|
|
namespace app\index\controller;
|
|||
|
|
|
|||
|
|
use app\common\controller\JsonReturn;
|
|||
|
|
use think\facade\Request;
|
|||
|
|
use think\facade\Cache;
|
|||
|
|
use app\common\model\User;
|
|||
|
|
use app\common\model\UserToken;
|
|||
|
|
use app\common\model\UserRelation;
|
|||
|
|
use app\common\exception\ServiceException;
|
|||
|
|
|
|||
|
|
class AuthBase extends Home
|
|||
|
|
{
|
|||
|
|
protected $user;
|
|||
|
|
|
|||
|
|
protected function initialize()
|
|||
|
|
{
|
|||
|
|
// 必须先执行父类的加载方法
|
|||
|
|
parent::initialize();
|
|||
|
|
|
|||
|
|
$token = Request::post('token', '', 'trim');
|
|||
|
|
if (empty($token)) {
|
|||
|
|
throw new ServiceException('请先登录！');
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$token_info = UserToken::where('token', $token)->find();
|
|||
|
|
if (empty($token_info)) {
|
|||
|
|
throw new ServiceException('请先登录！');
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if ($token_info->expire <= time()) {
|
|||
|
|
throw new ServiceException('token已失效，请重新登录！');
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$this->user = $this->getUserInfo($token_info);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取用户信息和好友列表
|
|||
|
|
* @param $token_info UserToken token信息
|
|||
|
|
* @return array|mixed|\PDOStatement|string|\think\Model|null
|
|||
|
|
*/
|
|||
|
|
protected function getUserInfo($token_info)
|
|||
|
|
{
|
|||
|
|
$info = Cache::get($token_info->token);
|
|||
|
|
if (!$info) {
|
|||
|
|
$info = User::where('id', $token_info->uid)->with(['userRelation' => function($query) {
|
|||
|
|
$query->where('status', UserRelation::STATUS_ACTIVE);
|
|||
|
|
}])->find();
|
|||
|
|
Cache::set($token_info->token, $info);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return $info;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取好友列表信息
|
|||
|
|
* @param string $field string 要查询的字段
|
|||
|
|
* @return array|mixed
|
|||
|
|
*/
|
|||
|
|
protected function getFriendList()
|
|||
|
|
{
|
|||
|
|
$list = Cache::get('user_friend_' . $this->user->id);
|
|||
|
|
if (!$list) {
|
|||
|
|
$user_relation = $this->user->userRelation;
|
|||
|
|
// 考虑到会出现缓存穿透的情况，如果返回为空也记录进缓存
|
|||
|
|
if ($user_relation) {
|
|||
|
|
$friend = array_column($user_relation->toArray(), 'friend_id');
|
|||
|
|
$list = User::whereIn('id', $friend)->select()->toArray();
|
|||
|
|
} else {
|
|||
|
|
$list = $user_relation;
|
|||
|
|
}
|
|||
|
|
Cache::set('user_friend_' . $this->user->id, $list, 3600);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return $list;
|
|||
|
|
}
|
|||
|
|
}
|