多应用+插件架构,代码干净,二开方便,首家独创一键云编译技术,文档视频完善,免费商用码云13.8K 广告
# 简单API Server框架 我们实现一个最最简单的数学计算:加、减、乘、除,给大家演示如何搭建简单的 API Server。 按照我们前面几章的写法,我们先来看看加法、减法示例代码: ~~~ worker_processes 1; #nginx worker 数量 error_log logs/error.log; #指定错误日志文件路径 events { worker_connections 1024; } http { server { listen 6699; # 加法 location /addition { content_by_lua_block { local args = ngx.req.get_uri_args() ngx.say(args.a + args.b) } } # 减法 location /subtraction { content_by_lua_block { local args = ngx.req.get_uri_args() ngx.say(args.a - args.b) } } # 乘法 location /multiplication { content_by_lua_block { local args = ngx.req.get_uri_args() ngx.say(args.a * args.b) } } # 除法 location /division { content_by_lua_block { local args = ngx.req.get_uri_args() ngx.say(args.a / args.b) } } } } ~~~ 代码写多了我们一眼就可以看出来,这么简单的加减乘除,居然写了这么长,而且我们还要对每个 API 都写一个 location ,这明显是让人不爽的。如果我们的某个API接口比较冗长,这样写岂不是直接会撑爆 nginx.conf 文件。要知道即使你喜欢这样写,nginx的配置文件对字符串最大长度有限制,不能超过4K。而且代码中如果需要出现单引号等字符,都需要进行转义,这些都是比较痛苦的。 - 首先就是把这些 location 合并为一个; - 其次是这些接口的实现放到独立文件中,保持 nginx 配置文件的简洁; 基于这两点要求,我们可以改成下面的版本,看上去有那么几分模样: > nginx.conf 内容: ~~~ worker_processes 1; #nginx worker 数量 error_log logs/error.log; #指定错误日志文件路径 events { worker_connections 1024; } http { # 设置默认 lua 搜索路径,添加 lua 路径 lua_package_path 'lua/?.lua;/blah/?.lua;;'; # 对于开发研究,我们可以对代码 cache 进行关闭,这样我们不必每次都重新加载 nginx。 lua_code_cache off; server { listen 6699; # 在代码路径中使用nginx变量 # 注意: nginx var 的变量一定要谨慎,否则将会带来非常大的风险 location ~ ^/api/([-_a-zA-Z0-9/]+) { # 准入阶段完成参数验证 access_by_lua_file lua/access_check.lua; #内容生成阶段 content_by_lua_file lua/$1.lua; } } } ~~~ > 其他文件内容: ~~~ --========== {$prefix}/lua/addition.lua local args = ngx.req.get_uri_args() ngx.say(args.a + args.b) --========== {$prefix}/lua/subtraction.lua local args = ngx.req.get_uri_args() ngx.say(args.a - args.b) --========== {$prefix}/lua/multiplication.lua local args = ngx.req.get_uri_args() ngx.say(args.a * args.b) --========== {$prefix}/lua/division.lua local args = ngx.req.get_uri_args() ngx.say(args.a / args.b) ~~~ 既然我们对外提供的是 API Server,作为一个服务端程序员,怎么可以容忍输入参数不检查呢?万一对方送过来的不是数字或者为空,这些都要过滤掉嘛。参数检查过滤的方法是统一,在这几个 API 中如何共享这个方法呢?这时候就需要 Lua 中的模块来完成了。 - 使用统一的公共模块,完成参数验证; - 对本示例,参数验证的方式方法是统一的,我们可以把它们集中在一处完成; > nginx.conf 内容: ~~~ worker_processes 1; #nginx worker 数量 error_log logs/error.log; #指定错误日志文件路径 events { worker_connections 1024; } http { server { listen 6699; # 在代码路径中使用nginx变量 # 注意: nginx var 的变量一定要谨慎,否则将会带来非常大的风险 location ~ ^/api/([-_a-zA-Z0-9/]+) { access_by_lua_file /path/to/lua/app/root/access_check.lua; content_by_lua_file /path/to/lua/app/root/$1.lua; } } } ~~~ > 新增文件内容: ~~~ --========== {$prefix}/lua/comm/param.lua local _M = {} -- 对输入参数逐个进行校验,只要有一个不是数字类型,则返回 false function _M.is_number(n, ...) local arg = {...} if n ~= #arg then return false end local num for _,v in ipairs(arg) do num = tonumber(v) if nil == num then return false end end return true end return _M --========== {$prefix}/lua/access_check.lua local param= require("comm.param") local args = ngx.req.get_uri_args() if not param.is_number(args.a, args.b) then ngx.exit(ngx.HTTP_BAD_REQUEST) return end ~~~ 看看curl测试结果吧: ~~~ $ nginx curl '127.0.0.1:6699/api/addition?a=1' 400 Bad Request 400 Bad Request openresty/1.9.3.1 $ nginx curl '127.0.0.1:6699/api/addition?a=1&b=3' 4 ~~~ 基本是按照我们预期执行的。参数不全、错误时,会提示400错误。正常处理,可以返回预期结果。 我们来整体看一下目前的目录关系: ~~~ . ├── conf │ ├── nginx.conf ├── logs │ ├── error.log │ └── nginx.pid ├── lua │ ├── access_check.lua │ ├── addition.lua │ ├── subtraction.lua │ ├── multiplication.lua │ ├── division.lua │ └── comm │ └── param.lua └── sbin └── nginx ~~~ 怎么样,有点 magic 的味道不?其实你的接口越是规范,有固定规律可寻,那么 OpenResty 就总是很容易能找到适合你的位置。当然这里你也可以把 `access_check.lua` 内容分别复制到加、减、乘、除实现的四个 Lua 文件中,肯定也是能用的。这里只是为了给大家提供更多的玩法,偶尔需要的时候我们可以有更多的选择。 本章目的是搭建一个简单API Server,记住这绝对不是终极版本。这里面还有很多需要我们进一步去考虑的地方,但是作为最基本的框架已经有了。