add resty http client

Author: leafo

lapis-dev-1.rockspec

@@ -98,6 +98,7 @@ build = {
     ["lapis.nginx.context"] = "lapis/nginx/context.lua",
     ["lapis.nginx.http"] = "lapis/nginx/http.lua",
     ["lapis.nginx.postgres"] = "lapis/nginx/postgres.lua",
+    ["lapis.nginx.resty_http"] = "lapis/nginx/resty_http.lua",
     ["lapis.request"] = "lapis/request.lua",
     ["lapis.router"] = "lapis/router.lua",
     ["lapis.server"] = "lapis/server.lua",

lapis/nginx/resty_http.lua

@@ -0,0 +1,57 @@
+local lapis_config = require("lapis.config")
+local increment_perf
+increment_perf = require("lapis.nginx.context").increment_perf
+local request
+request = function(url, str_body)
+  local http = require("resty.http")
+  local ltn12 = require("ltn12")
+  local config = lapis_config.get()
+  local start_time
+  if config.measure_performance then
+    ngx.update_time()
+    start_time = ngx.now()
+  end
+  local return_res_body
+  local req
+  if type(url) == "table" then
+    req = url
+  else
+    return_res_body = true
+    req = {
+      url = url,
+      source = str_body and ltn12.source.string(str_body),
+      headers = str_body and {
+        ["Content-type"] = "application/x-www-form-urlencoded"
+      }
+    }
+  end
+  req.method = req.method or (req.source and "POST" or "GET")
+  local httpc = http.new()
+  local res, err = httpc:request_uri(req.url, {
+    method = req.method,
+    headers = req.headers,
+    body = req.source,
+    ssl_verify = true
+  })
+  if not (res) then
+    error("resty.http request failed: " .. tostring(err))
+  end
+  local out
+  if return_res_body then
+    out = res.body
+  else
+    if req.sink then
+      ltn12.pump.all(ltn12.source.string(res.body), req.sink)
+    end
+    out = 1
+  end
+  if start_time then
+    ngx.update_time()
+    increment_perf("http_count", 1)
+    increment_perf("http_time", ngx.now() - start_time)
+  end
+  return out, res.status, res.headers
+end
+return {
+  request = request
+}

lapis/nginx/resty_http.moon

@@ -0,0 +1,69 @@
+
+-- This implements a basic LuaSocket's http.request interface using lua-resty-http.
+--
+-- Usage:
+--   http = require "lapis.nginx.resty_http"
+--   body, status, headers = http.request("http://example.com")
+
+lapis_config = require "lapis.config"
+
+import increment_perf from require "lapis.nginx.context"
+
+---Make an HTTP request using lua-resty-http
+---@param url string|table URL string or request table with url, method, source, sink, headers fields
+---@param str_body string? Request body for simple POST requests
+---@return string|number body Response body (or 1 if sink was provided)
+---@return number status HTTP status code
+---@return table headers Response headers
+request = (url, str_body) ->
+  http = require "resty.http"
+  ltn12 = require "ltn12"
+
+  config = lapis_config.get!
+  start_time = if config.measure_performance
+    ngx.update_time!
+    ngx.now!
+
+  local return_res_body
+  req = if type(url) == "table"
+    url
+  else
+    return_res_body = true
+    {
+      :url
+      source: str_body and ltn12.source.string str_body
+      headers: str_body and {
+        "Content-type": "application/x-www-form-urlencoded"
+      }
+    }
+
+  req.method or= req.source and "POST" or "GET"
+
+  httpc = http.new!
+  -- TODO: if source iterator returns an error, resty.http treats it as end of stream and does not surface the error
+  res, err = httpc\request_uri req.url, {
+    method: req.method
+    headers: req.headers
+    body: req.source -- ltn12 source is compatible as body iterator; caller provides Content-Length header if needed
+    ssl_verify: true
+  }
+
+  unless res
+    error "resty.http request failed: #{err}"
+
+  out = if return_res_body
+    res.body
+  else
+    -- TODO: support streaming response body to sink, would need to use lower level API
+    if req.sink
+      ltn12.pump.all ltn12.source.string(res.body), req.sink
+    1
+
+  if start_time
+    ngx.update_time!
+    increment_perf "http_count", 1
+    increment_perf "http_time", ngx.now! - start_time
+
+  out, res.status, res.headers
+
+{ :request }