70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"net/http"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc/config/datasource"
|
|||
|
|
ioc_gin "github.com/infraboard/mcube/v2/ioc/config/gin"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc/server"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
|
|||
|
|
// 开启Health健康检查
|
|||
|
|
_ "github.com/infraboard/mcube/v2/ioc/apps/health/gin"
|
|||
|
|
// 开启Metric
|
|||
|
|
_ "github.com/infraboard/mcube/v2/ioc/apps/metric/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
// 注册HTTP接口类
|
|||
|
|
ioc.Api().Registry(&ApiHandler{})
|
|||
|
|
|
|||
|
|
// 开启配置文件读取配置
|
|||
|
|
// server 是一个特殊的对象，在ioc 提供的 服务器
|
|||
|
|
server.DefaultConfig.ConfigFile.Enabled = true
|
|||
|
|
server.DefaultConfig.ConfigFile.Paths = []string{"etc/application.toml"}
|
|||
|
|
|
|||
|
|
// 启动应用
|
|||
|
|
err := server.Run(context.Background())
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ApiHandler struct {
|
|||
|
|
// 继承自Ioc对象
|
|||
|
|
ioc.ObjectImpl
|
|||
|
|
|
|||
|
|
// mysql db依赖
|
|||
|
|
db *gorm.DB
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 覆写对象的名称, 该名称名称会体现在API的路径前缀里面
|
|||
|
|
// 比如: /simple/api/v1/module_a/db_stats
|
|||
|
|
// 其中/simple/api/v1/module_a 就是对象API前缀, 命名规则如下:
|
|||
|
|
//
|
|||
|
|
// <service_name>/<path_prefix>/<object_version>/<object_name>
|
|||
|
|
func (h *ApiHandler) Name() string {
|
|||
|
|
return "module_a"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 初始化db属性, 从ioc的配置区域获取共用工具 gorm db对象
|
|||
|
|
func (h *ApiHandler) Init() error {
|
|||
|
|
h.db = datasource.DB()
|
|||
|
|
|
|||
|
|
// 进行业务暴露, router 通过ioc
|
|||
|
|
router := ioc_gin.RootRouter()
|
|||
|
|
router.GET("/db_stats", h.GetDbStats)
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 业务功能
|
|||
|
|
func (h *ApiHandler) GetDbStats(ctx *gin.Context) {
|
|||
|
|
db, _ := h.db.DB()
|
|||
|
|
ctx.JSON(http.StatusOK, gin.H{
|
|||
|
|
"data": db.Stats(),
|
|||
|
|
})
|
|||
|
|
}
|