Skip to content

Commit

Permalink
Avoid using channel in GRPCServer
Browse files Browse the repository at this point in the history
Signed-off-by: James Ryans <james.ryans2012@gmail.com>
  • Loading branch information
james-ryans committed Apr 6, 2024
1 parent 5dc1b34 commit 4fb4fff
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 22 deletions.
4 changes: 1 addition & 3 deletions cmd/jaeger/internal/integration/grpc_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,7 @@ type GRPCStorageIntegration struct {
}

func (s *GRPCStorageIntegration) initialize(t *testing.T) {
var err error
s.server, err = integration.NewGRPCServer()
require.NoError(t, err)
s.server = integration.NewGRPCServer()
require.NoError(t, s.server.Start())

s.Refresh = func(_ *testing.T) {}
Expand Down
25 changes: 8 additions & 17 deletions plugin/storage/integration/grpc_server.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,13 @@ import (
)

type GRPCServer struct {
errChan chan error
server *googleGRPC.Server
wg sync.WaitGroup
server *googleGRPC.Server
serveErr error
wg sync.WaitGroup
}

func NewGRPCServer() (*GRPCServer, error) {
return &GRPCServer{errChan: make(chan error, 1)}, nil
func NewGRPCServer() *GRPCServer {
return &GRPCServer{}
}

func (s *GRPCServer) Start() error {
Expand Down Expand Up @@ -51,12 +51,7 @@ func (s *GRPCServer) Start() error {
s.wg.Add(1)
go func() {
defer s.wg.Done()
if err = s.server.Serve(listener); err != nil {
select {
case s.errChan <- err:
default:
}
}
s.serveErr = s.server.Serve(listener)
}()
return nil
}
Expand All @@ -69,10 +64,6 @@ func (s *GRPCServer) Close() error {
s.server.GracefulStop()
s.server = nil
s.wg.Wait()
select {
case err := <-s.errChan:
return err
default:
}
return nil

return s.serveErr
}
3 changes: 1 addition & 2 deletions plugin/storage/integration/grpc_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -133,8 +133,7 @@ func TestGRPCRemoteStorage(t *testing.T) {
"--grpc-storage.server=localhost:17271",
"--grpc-storage.tls.enabled=false",
}
server, err := NewGRPCServer()
require.NoError(t, err)
server := NewGRPCServer()

s := &GRPCStorageIntegrationTestSuite{
flags: flags,
Expand Down

0 comments on commit 4fb4fff

Please sign in to comment.