sqler sql 转rest api 源码解析（三） rest协议

rest 服务说明

rest 协议主要是将配置文件中的宏暴露为rest 接口，使用了labstack/echo web 框架，同时基于context 模型
进行宏管理对象的共享，同时进行了一些中间件的注册 cors RemoveTrailingSlash gzip Recover

rest 启动

• 中间件注册

 

  e.Pre(middleware.RemoveTrailingSlash())

  e.Use(middleware.CORS())

  e.Use(middleware.GzipWithConfig(middleware.GzipConfig{Level: 9}))

  e.Use(middleware.Recover())

• 路由配置
  默认配置，以及指定宏的路由

 

  e.GET("/", routeIndex)

  e.Any("/:macro", routeExecMacro, middlewareAuthorize)

 

 

• 宏执行路由
  包含了一个授权的中间件，同时在这个中间件中传递共享变量（macro）

 

func middlewareAuthorize(next echo.HandlerFunc) echo.HandlerFunc {

  return func(c echo.Context) error {

    if strings.HasPrefix(c.Param("macro"), "_") {

      return c.JSON(403, map[string]interface{}{

        "success": false,

        "error": "access not allowed",

      })

    }

​

    macro := macrosManager.Get(c.Param("macro"))

    if macro == nil {

      return c.JSON(404, map[string]interface{}{

        "success": false,

        "error": "resource not found",

      })

    }

​

    if len(macro.Methods) < 1 {

      macro.Methods = []string{c.Request().Method}

    }

​

    methodIsAllowed := false

    for _, method := range macro.Methods {

      method = strings.ToUpper(method)

      if c.Request().Method == method {

        methodIsAllowed = true

        break

      }

    }

​

    if !methodIsAllowed {

      return c.JSON(405, map[string]interface{}{

        "success": false,

        "error": "method not allowed",

      })

    }

​

    // for _, endpoint := range macro.Authorizers {

    // parts := strings.SplitN(endpoint, " ", 2)

    // if len(parts) < 2 {

    // return c.JSON(500, map[string]interface{}{

    // "success": false,

    // "error": fmt.Sprintf("authorizer: %s is invalid", endpoint),

    // })

    // }

    // resp, err := resty.R().SetHeaders(map[string]string{

    // "Authorization": c.Request().Header.Get("Authorization"),

    // }).Execute(parts[0], parts[1])

    // if err != nil {

    // return c.JSON(500, map[string]interface{}{

    // "success": false,

    // "error": err.Error(),

    // })

    // }

    // if resp.StatusCode() >= 400 {

    // return c.JSON(resp.StatusCode(), map[string]interface{}{

    // "success": false,

    // "error": resp.Status(),

    // })

    // }

    // }

   // 路由共享变量

    c.Set("macro", macro)

​

    return next(c)

  }

}

 

 

• routeExecMacro
  routes.go 解析请求参数，解析宏，返回json 格式数据

 

// routeExecMacro - execute the requested macro

func routeExecMacro(c echo.Context) error {

// 获取请求参数，转换为支持宏绑定的map 对象

  macro := c.Get("macro").(*Macro)

  input := make(map[string]interface{})

  body := make(map[string]interface{})

​

  c.Bind(&body)

​

  for k := range c.QueryParams() {

    input[k] = c.QueryParam(k)

  }

​

  for k, v := range body {

    input[k] = v

  }

​

  headers := c.Request().Header

  for k, v := range headers {

    input["http_"+strings.Replace(strings.ToLower(k), "-", "_", -1)] = v[0]

  }

​

  out, err := macro.Call(input)

  if err != nil {

    code := errStatusCodeMap[err]

    if code < 1 {

      code = 500

    }

    return c.JSON(code, map[string]interface{}{

      "success": false,

      "error": err.Error(),

      "data": out,

    })

  }

​

  return c.JSON(200, map[string]interface{}{

    "success": true,

    "data": out,

  })

}

 

参考资料

https://github.com/labstack/echo
https://github.com/alash3al/sqler/blob/master/server_rest.go
https://github.com/alash3al/sqler/blob/master/routes.go