Go每日一库之127:rk-boot(快速构建服务)

发布时间 2023-09-29 21:54:14作者: 阿瑞娜

rk-boot的介绍

通过 rk-boot,用户可以通过yaml格式的配置文件启动gRPC、gin、echo、GoFrame、prometheus客户端或自定义入口服务。很容易易编译、运行和调试你的 grpc 服务、grpc 网关、swagger UI 和 rk-tv Web UI。
该库是引用了rookie-ninja组织下面的gRPC、gin、echo、GoFrame和prometheus客户端定制boot包。

为什么我们需要它?

在我们公司项目中,各个部门的项目结构都不统一,甚至一个小组内都不会很统一,这种情况数不胜数(尽管用了框架规范)。

  • 公司中我们一般使用统一的项目布局来构建服务。
  • 也使用统一的logging, metrics, tracing, authorization来构建API。
  • 还可以让应用程序更快地替换掉核心依赖。
  • 可以节省我们编写一些流行框架和库的初始化程序的学习时间。
  • 我们还可以定义入口用于定制的服务。

下图是使用rk-boot的图示
每日一库之127:rk-boot(快速构建服务)-0
arch

Gin的示例

这里土拨鼠只举了关于gin的例子,跟其他组件例子(gRPC、echo、GoFrame、prometheus)相比,只是boot.yaml配置不同。这里gin会引用rk-gin的boot[1]包。
目录结构

├── boot.yaml
└── main.go

boot.yaml

---
gin:
  - name: greeter       # Required, Name of gin entry 服务名称
    port: 8080          # Required, Port of gin entry 端口号
    enabled: true       # Required, Enable gin entry 是否启用gin
    sw:
      enabled: true     # Optional, Enable swagger UI 是否启用swagger
    commonService:
      enabled: true     # Optional, Enable common service 是否启用commonService
    tv:
      enabled:  true    # Optional, Enable RK TV 是否启用RK TV

main.go

package main

import (
   "context"
   "github.com/rookie-ninja/rk-boot"
)

func main() {
   // 创建rkboot实例 -- Create a new boot instance.
   boot := rkboot.NewBoot()

   // 启动rkboot -- Bootstrap
   boot.Bootstrap(context.Background())

   // 等待接受关闭信号 -- Wait for shutdown sig
   boot.WaitForShutdownSig(context.Background())
}

运行&&请求

$ go run main.go
$ curl -X GET localhost:8080/rk/v1/healthy
{"healthy":true}

Swagger

可以通过 http://localhost:8080/sw 访问swagger页面
每日一库之127:rk-boot(快速构建服务)-1

TV

可以通过 http://localhost:8080/rk/v1/tv 访问TV页面
每日一库之127:rk-boot(快速构建服务)-2

开启中间件

其中rk-boot依赖的 rk-grpcrk-gin 集成了很多中间件,可以在boot.yaml中配置使用,下面是在gin的示例中开启了日志zap的使用,可以通过访问curl -X GET localhost:8080/rk/v1/healthy查看日志。

gin:
  - name: greeter                             # Required
    port: 8080                                # Required
    enabled: true                             # Required
    commonService:                            # Optional
      enabled: true                           # Optional, default: false
    interceptors:                             # Optional
      loggingZap:
        enabled: true  

另外还支持以下中间件,中间件源码包interceptor可以按需开启使用。对应的示例可以查看 example

  • logging middleware
  • prometheus metrics middleware
  • auth middleware
  • tracing middleware
  • panic middleware
  • metadata middleware
  • rate limit middleware
  • timeout middleware
  • gzip middleware
  • cors interceptor
  • jwt interceptor
  • secure interceptor
  • csrf interceptor

小结

rk-boot还支持gRPC代理目前还在试验阶段,相信后续会有更多功能支持。土拨鼠今天介绍这个库主要是觉得这种可配置化、定制的思想很值得在企业项目中借鉴和落地。还可以统一一些公共组件的使用,节省大量成本。如果你有相关看法,欢迎大家在留言区讨论交流。