50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
|
package permission
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
|
|||
|
|
"122.51.31.227/go-course/go18/devcloud/mcenter/apps/endpoint"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc/config/gorestful"
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc/config/log"
|
|||
|
|
"github.com/rs/zerolog"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
ioc.Api().Registry(&ApiRegister{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func GetApiRegister() *ApiRegister {
|
|||
|
|
return ioc.Api().Get("api_register").(*ApiRegister)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 接口注册模块: 扫描当前GoResuful Container下所有路径，并完成注册
|
|||
|
|
type ApiRegister struct {
|
|||
|
|
ioc.ObjectImpl
|
|||
|
|
|
|||
|
|
log *zerolog.Logger
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *ApiRegister) Name() string {
|
|||
|
|
return "api_register"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 这个Init一定要放到所有的路由都添加完成后进行
|
|||
|
|
func (i *ApiRegister) Priority() int {
|
|||
|
|
return -100
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (a *ApiRegister) Init() error {
|
|||
|
|
a.log = log.Sub(a.Name())
|
|||
|
|
// 注册认证中间件
|
|||
|
|
entries := endpoint.NewEntryFromRestfulContainer(gorestful.RootRouter())
|
|||
|
|
req := endpoint.NewRegistryEndpointRequest()
|
|||
|
|
req.AddItem(entries...)
|
|||
|
|
set, err := endpoint.GetService().RegistryEndpoint(context.Background(), req)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
a.log.Info().Msgf("registry endpoinst: %s", set.Items)
|
|||
|
|
return nil
|
|||
|
|
}
|