首页 > 代码库 > Yii2框架RESTful API教程(二) - 格式化响应,授权认证和速率限制

Yii2框架RESTful API教程(二) - 格式化响应,授权认证和速率限制

之前写过一篇Yii2框架RESTful API教程(一) - 快速入门,今天接着来探究一下Yii2 RESTful的格式化响应,授权认证和速率限制三个部分

一、目录结构

先列出需要改动的文件。目录如下:

 
web ├─ common │      └─ models  │              └ User.php └─ frontend        ├─ config        │   └ main.php        └─ controllers            └ BookController.php
 

二、格式化响应

Yii2 RESTful支持JSON和XML格式,如果想指定返回数据的格式,需要配置yii\filters\ContentNegotiator::formats属性。例如,要返回JSON格式,修改frontend/controllers/BookController.php,加入红色标记代码:

 
namespace frontend\controllers;use yii\rest\ActiveController;use yii\web\Response;class BookController extends ActiveController{    public $modelClass = ‘frontend\models\Book‘;    public function behaviors() {        $behaviors = parent::behaviors();        $behaviors[‘contentNegotiator‘][‘formats‘][‘text/html‘] = Response::FORMAT_JSON;        return $behaviors;    }}
 

返回XML格式:FORMAT_XML。formats属性的keys支持MIME类型,而values必须在yii\web\Response::formatters中支持被响应格式名称。

三、授权认证

RESTful APIs通常是无状态的,因此每个请求应附带某种授权凭证,即每个请求都发送一个access token来认证用户。

1.配置user应用组件(不是必要的,但是推荐配置):
  设置yii\web\User::enableSession属性为false(因为RESTful APIs为无状态的,当yii\web\User::enableSession为false,请求中的用户认证状态就不能通过session来保持)
  设置yii\web\User::loginUrl属性为null(显示一个HTTP 403 错误而不是跳转到登录界面)
具体方法,修改frontend/config/main.php,加入红色标记代码:

 
‘components‘ => [    ...    ‘user‘ => [        ‘identityClass‘ => ‘common\models\User‘,        ‘enableAutoLogin‘ => true,                ‘enableSession‘ => false,        ‘loginUrl‘ => null,            ],    ...]
 

2.在控制器类中配置authenticator行为来指定使用哪种认证方式,修改frontend/controllers/BookController.php,加入红色标记代码:

 
namespace frontend\controllers;use yii\rest\ActiveController;use yii\web\Response;use yii\filters\auth\CompositeAuth;use yii\filters\auth\QueryParamAuth;class BookController extends ActiveController{    public $modelClass = ‘frontend\models\Book‘;    public function behaviors() {        $behaviors = parent::behaviors();            $behaviors[‘authenticator‘] = [            ‘class‘ => CompositeAuth::className(),            ‘authMethods‘ => [                /*下面是三种验证access_token方式*/                //1.HTTP 基本认证: access token 当作用户名发送,应用在access token可安全存在API使用端的场景,例如,API使用端是运行在一台服务器上的程序。                //HttpBasicAuth::className(),                //2.OAuth 2: 使用者从认证服务器上获取基于OAuth2协议的access token,然后通过 HTTP Bearer Tokens 发送到API 服务器。                //HttpBearerAuth::className(),                //3.请求参数: access token 当作API URL请求参数发送,这种方式应主要用于JSONP请求,因为它不能使用HTTP头来发送access token                //http://localhost/user/index/index?access-token=123                QueryParamAuth::className(),            ],        ];                $behaviors[‘contentNegotiator‘][‘formats‘][‘text/html‘] = Response::FORMAT_JSON;        return $behaviors;    }}
 

3.创建一张user表

 
-- ------------------------------ Table structure for user-- ----------------------------DROP TABLE IF EXISTS `user`;CREATE TABLE `user` (  `id` int(10) unsigned NOT NULL AUTO_INCREMENT,  `username` varchar(20) NOT NULL DEFAULT ‘‘ COMMENT ‘用户名‘,  `password_hash` varchar(100) NOT NULL DEFAULT ‘‘ COMMENT ‘密码‘,  `password_reset_token` varchar(50) NOT NULL DEFAULT ‘‘ COMMENT ‘密码token‘,  `email` varchar(20) NOT NULL DEFAULT ‘‘ COMMENT ‘邮箱‘,  `auth_key` varchar(50) NOT NULL DEFAULT ‘‘,  `status` tinyint(3) unsigned NOT NULL DEFAULT ‘0‘ COMMENT ‘状态‘,  `created_at` int(10) unsigned NOT NULL DEFAULT ‘0‘ COMMENT ‘创建时间‘,  `updated_at` int(10) unsigned NOT NULL DEFAULT ‘0‘ COMMENT ‘更新时间‘,  `access_token` varchar(50) NOT NULL DEFAULT ‘‘ COMMENT ‘restful请求token‘,  `allowance` int(10) unsigned NOT NULL DEFAULT ‘0‘ COMMENT ‘restful剩余的允许的请求数‘,  `allowance_updated_at` int(10) unsigned NOT NULL DEFAULT ‘0‘ COMMENT ‘restful请求的UNIX时间戳数‘,  PRIMARY KEY (`id`),  UNIQUE KEY `username` (`username`),  UNIQUE KEY `access_token` (`access_token`)) ENGINE=InnoDB DEFAULT CHARSET=utf8;-- ------------------------------ Records of user-- ----------------------------INSERT INTO `user` VALUES (‘1‘, ‘admin‘, ‘$2y$13$1KWwchqGvxDeORDt5pRW.OJarf06PjNYxe2vEGVs7e5amD3wnEX.i‘, ‘‘, ‘‘, ‘z3sM2KZvXdk6mNXXrz25D3JoZlGXoJMC‘, ‘10‘, ‘1478686493‘, ‘1478686493‘, ‘123‘, ‘4‘, ‘1478686493‘);
 

在common/models/User.php类中实现 yii\web\IdentityInterface::findIdentityByAccessToken()方法。修改common/models/User.php,加入红色标记代码::

 
public static function findIdentityByAccessToken($token, $type = null){    //findIdentityByAccessToken()方法的实现是系统定义的    //例如,一个简单的场景,当每个用户只有一个access token, 可存储access token 到user表的access_token列中, 方法可在User类中简单实现,如下所示:    return static::findOne([‘access_token‘ => $token]);    //throw new NotSupportedException(‘"findIdentityByAccessToken" is not implemented.‘);}
 

四、速率限制

为防止滥用,可以增加速率限制。例如,限制每个用户的API的使用是在60秒内最多10次的API调用,如果一个用户同一个时间段内太多的请求被接收,将返回响应状态代码 429 (这意味着过多的请求)。

1.Yii会自动使用yii\filters\RateLimiter为yii\rest\Controller配置一个行为过滤器来执行速率限制检查。如果速度超出限制,该速率限制器将抛出一个yii\web\TooManyRequestsHttpException。
修改frontend/controllers/BookController.php,加入红色标记代码:

 
namespace frontend\controllers;use yii\rest\ActiveController;use yii\web\Response;use yii\filters\auth\CompositeAuth;use yii\filters\auth\QueryParamAuth;use yii\filters\RateLimiter;class BookController extends ActiveController{    public $modelClass = ‘frontend\models\Book‘;    public function behaviors() {        $behaviors = parent::behaviors();                $behaviors[‘rateLimiter‘] = [            ‘class‘ => RateLimiter::className(),            ‘enableRateLimitHeaders‘ => true,        ];                $behaviors[‘authenticator‘] = [            ‘class‘ => CompositeAuth::className(),            ‘authMethods‘ => [                /*下面是三种验证access_token方式*/                //1.HTTP 基本认证: access token 当作用户名发送,应用在access token可安全存在API使用端的场景,例如,API使用端是运行在一台服务器上的程序。                //HttpBasicAuth::className(),                //2.OAuth 2: 使用者从认证服务器上获取基于OAuth2协议的access token,然后通过 HTTP Bearer Tokens 发送到API 服务器。                //HttpBearerAuth::className(),                //3.请求参数: access token 当作API URL请求参数发送,这种方式应主要用于JSONP请求,因为它不能使用HTTP头来发送access token                //http://localhost/user/index/index?access-token=123                QueryParamAuth::className(),            ],        ];        $behaviors[‘contentNegotiator‘][‘formats‘][‘text/html‘] = Response::FORMAT_JSON;        return $behaviors;    }}
 

2.在user表中使用两列来记录容差和时间戳信息。为了提高性能,可以考虑使用缓存或NoSQL存储这些信息。
修改common/models/User.php,加入红色标记代码:

 
namespace common\models;use Yii;use yii\base\NotSupportedException;use yii\behaviors\TimestampBehavior;use yii\db\ActiveRecord;use yii\web\IdentityInterface;use yii\filters\RateLimitInterface;class User extends ActiveRecord implements IdentityInterface, RateLimitInterface{    ....    // 返回在单位时间内允许的请求的最大数目,例如,[10, 60] 表示在60秒内最多请求10次。    public function getRateLimit($request, $action)    {        return [5, 10];    }    // 返回剩余的允许的请求数。    public function loadAllowance($request, $action)    {        return [$this->allowance, $this->allowance_updated_at];    }    // 保存请求时的UNIX时间戳。    public function saveAllowance($request, $action, $allowance, $timestamp)    {        $this->allowance = $allowance;        $this->allowance_updated_at = $timestamp;        $this->save();    }        ....        public static function findIdentityByAccessToken($token, $type = null)    {        //throw new NotSupportedException(‘"findIdentityByAccessToken" is not implemented.‘);        //findIdentityByAccessToken()方法的实现是系统定义的        //例如,一个简单的场景,当每个用户只有一个access token, 可存储access token 到user表的access_token列中, 方法可在User类中简单实现,如下所示:        return static::findOne([‘access_token‘ => $token]);    }        ....}
 

 PHP的RESTful实例: PHP实现RESTful风格的API实例

 

Yii2框架RESTful API教程(二) - 格式化响应,授权认证和速率限制