go/go-micro
1
0
Fork 0
mirror of https://github.com/go-micro/go-micro.git synced 2024-11-24 08:02:32 +02:00
Code Issues Releases Activity

fix(api): add WithRegistry option for api (#2618)

Browse Source
This commit is contained in:
Hellis 2023-03-08 00:05:25 +08:00 committed by GitHub
parent 521e6b644c
commit e337eb2c3d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/api.go
View File

@ -8,6 +8,7 @@ import (
"strings" "strings"
"go-micro.dev/v4/api/router" "go-micro.dev/v4/api/router"
"go-micro.dev/v4/client"
"go-micro.dev/v4/registry" "go-micro.dev/v4/registry"
"go-micro.dev/v4/server" "go-micro.dev/v4/server"
) )
@ -35,6 +36,8 @@ type Options struct {
Address string Address string
// Router for resolving routes // Router for resolving routes
Router router.Router Router router.Router
// Client to use for RPC
Client client.Client
} }
// Option type are API option args. // Option type are API option args.

12
api/options.go
View File

@ -2,6 +2,9 @@ package api
import ( import (
"go-micro.dev/v4/api/router" "go-micro.dev/v4/api/router"
registry2 "go-micro.dev/v4/api/router/registry"
"go-micro.dev/v4/client"
"go-micro.dev/v4/registry"
) )
func NewOptions(opts ...Option) Options { func NewOptions(opts ...Option) Options {
@ -23,3 +26,12 @@ func WithRouter(r router.Router) Option {
return nil return nil
} }
} }
// WithRegistry sets the api's client and router to use registry.
func WithRegistry(r registry.Registry) Option {
return func(o *Options) error {
o.Client = client.NewClient(client.Registry(r))
o.Router = registry2.NewRouter(router.WithRegistry(r))
return nil
}
}