linka-cloud/grpc
2
0
Fork 0
mirror of https://github.com/linka-cloud/grpc.git synced 2024-11-21 18:36:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

add WithListener option

Browse Source 
Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
This commit is contained in:
Adphi 2022-09-05 15:32:33 +02:00
parent b230278441
commit 9bf4e691ce
Signed by: adphi
GPG Key ID: 46BE4062DB2397FF
2 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
service/options.go
View File

@ -127,6 +127,15 @@ func WithAddress(addr string) Option {
}
}
// WithListener specifies a listener for the service.
// It can be used to specify a custom listener.
// This will override the WithAddress and WithTLSConfig options
func WithListener(lis net.Listener) Option {
return func(o *options) {
o.lis = lis
}
}
func WithReflection(r bool) Option {
return func(o *options) {
o.reflection = r
@ -330,6 +339,7 @@ type options struct {
name string
version string
address string
lis net.Listener
reflection bool
health bool

8
service/service.go
View File

@ -151,6 +151,7 @@ func (s *service) run() error {
s.opts.address = strings.TrimPrefix(s.opts.address, "unix://")
}
if s.opts.lis == nil {
lis, err := net.Listen(network, s.opts.address)
if err != nil {
return err
@ -158,10 +159,13 @@ func (s *service) run() error {
if s.opts.tlsConfig != nil {
lis = tls.NewListener(lis, s.opts.tlsConfig)
}
s.opts.lis = lis
s.opts.address = lis.Addr().String()
} else {
s.opts.address = s.opts.lis.Addr().String()
}
mux := cmux.New(lis)
mux := cmux.New(s.opts.lis)
mux.SetReadTimeout(5 * time.Second)
gLis := mux.MatchWithWriters(cmux.HTTP2MatchHeaderFieldSendSettings("content-type", "application/grpc"))