2020-07-01 19:06:59 +03:00
|
|
|
package util
|
|
|
|
|
|
|
|
import (
|
2020-07-27 15:22:00 +03:00
|
|
|
"github.com/micro/go-micro/v3/registry"
|
|
|
|
"github.com/micro/go-micro/v3/router"
|
2020-07-01 19:06:59 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
// Router is a hack for API routing
|
|
|
|
func Router(srvs []*registry.Service) router.Router {
|
|
|
|
var routes []router.Route
|
|
|
|
|
|
|
|
for _, srv := range srvs {
|
|
|
|
for _, n := range srv.Nodes {
|
|
|
|
routes = append(routes, router.Route{Address: n.Address, Metadata: n.Metadata})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return &apiRouter{routes: routes}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *apiRouter) Lookup(...router.QueryOption) ([]router.Route, error) {
|
|
|
|
return r.routes, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
type apiRouter struct {
|
|
|
|
routes []router.Route
|
|
|
|
router.Router
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *apiRouter) String() string {
|
|
|
|
return "api"
|
|
|
|
}
|