diff options
author | Sami Hiltunen <shiltunen@gitlab.com> | 2021-05-07 14:52:37 +0300 |
---|---|---|
committer | Sami Hiltunen <shiltunen@gitlab.com> | 2021-05-10 10:17:13 +0300 |
commit | 2b1d3870d63b725788ba408ce95f45177ee5d78d (patch) | |
tree | be84aa45279a2f5a2912e7272caf9a1be7253133 | |
parent | ba534213898d3e4f91cad972b79b0c47e526f5fc (diff) |
Remove the security type split in GitalyServerFactory
GitalyServerFactory maintains the secure and insecure servers separately.
There appears to be no need to do so, so let's just combine the servers
into the same slice to simplify the code.
-rw-r--r-- | internal/gitaly/server/server_factory.go | 23 |
1 files changed, 7 insertions, 16 deletions
diff --git a/internal/gitaly/server/server_factory.go b/internal/gitaly/server/server_factory.go index d37171fe7..6fe0139e1 100644 --- a/internal/gitaly/server/server_factory.go +++ b/internal/gitaly/server/server_factory.go @@ -21,9 +21,9 @@ import ( // GitalyServerFactory is a factory of gitaly grpc servers type GitalyServerFactory struct { - registry *backchannel.Registry - cfg config.Cfg - secure, insecure []*grpc.Server + registry *backchannel.Registry + cfg config.Cfg + servers []*grpc.Server } // NewGitalyServerFactory allows to create and start secure/insecure 'grpc.Server'-s with gitaly-ruby @@ -87,7 +87,7 @@ func (s *GitalyServerFactory) StartWorkers(ctx context.Context, l logrus.FieldLo // Stop stops all servers started by calling Serve and the gitaly-ruby server. func (s *GitalyServerFactory) Stop() { - for _, srv := range s.all() { + for _, srv := range s.servers { srv.Stop() } } @@ -96,7 +96,7 @@ func (s *GitalyServerFactory) Stop() { func (s *GitalyServerFactory) GracefulStop() { wg := sync.WaitGroup{} - for _, srv := range s.all() { + for _, srv := range s.servers { wg.Add(1) go func(s *grpc.Server) { @@ -115,15 +115,6 @@ func (s *GitalyServerFactory) Create(secure bool) (*grpc.Server, error) { return nil, err } - if secure { - s.secure = append(s.secure, server) - return s.secure[len(s.secure)-1], nil - } - - s.insecure = append(s.insecure, server) - return s.insecure[len(s.insecure)-1], nil -} - -func (s *GitalyServerFactory) all() []*grpc.Server { - return append(s.secure[:], s.insecure...) + s.servers = append(s.servers, server) + return server, nil } |