anyproto/any-sync
1
0
Fork 0
mirror of https://github.com/anyproto/any-sync.git synced 2025-06-08 05:57:03 +09:00
Code Activity

secure service test

Browse source
This commit is contained in:
Sergey Cherepanov 2023-02-15 22:09:09 +03:00 • committed by Mikhail Iudin
parent e93812cdcc
commit ba1cc69c01
No known key found for this signature in database
GPG key ID: FAAAA8BAABDFF1C0
5 changed files with 122 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

11
net/rpc/server/drpcserver.go
View file

@ -5,9 +5,10 @@ import (
"github.com/anytypeio/any-sync/app"
"github.com/anytypeio/any-sync/app/logger"
"github.com/anytypeio/any-sync/metric"
"github.com/anytypeio/any-sync/net"
anyNet "github.com/anytypeio/any-sync/net"
"github.com/anytypeio/any-sync/net/secureservice"
"github.com/prometheus/client_golang/prometheus"
"net"
"storj.io/drpc"
)
@ -25,14 +26,14 @@ type DRPCServer interface {
}
type drpcServer struct {
config net.Config
config anyNet.Config
metric metric.Metric
transport secureservice.SecureService
*BaseDrpcServer
}
func (s *drpcServer) Init(a *app.App) (err error) {
s.config = a.MustComponent("config").(net.ConfigGetter).GetNet()
s.config = a.MustComponent("config").(anyNet.ConfigGetter).GetNet()
s.metric = a.MustComponent(metric.CName).(metric.Metric)
s.transport = a.MustComponent(secureservice.CName).(secureservice.SecureService)
return nil
@ -67,7 +68,9 @@ func (s *drpcServer) Run(ctx context.Context) (err error) {
SummaryVec: histVec,
}
},
Converter: s.transport.TLSListener,
Converter: func(listener net.Listener, timeoutMillis int) secureservice.ContextListener {
return s.transport.TLSListener(listener, timeoutMillis, s.config.Server.IdentityHandshake)
},
}
return s.BaseDrpcServer.Run(ctx, params)
}