fix(gateway): qualify concurrent fair allocation
This commit is contained in:
+6
-2
@@ -449,11 +449,12 @@ type fakeSession struct {
|
||||
}
|
||||
|
||||
func (s *fakeSession) Ready(ctx context.Context) error {
|
||||
s.mu.Lock()
|
||||
if s.failure == FakeFailureReadinessTimeout {
|
||||
s.mu.Unlock()
|
||||
<-ctx.Done()
|
||||
return ctx.Err()
|
||||
}
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if s.state.State == ProviderStateDisconnected {
|
||||
return ErrProviderDisconnected
|
||||
@@ -585,7 +586,10 @@ func (s *fakeSession) ReleaseAll(_ context.Context) error {
|
||||
}
|
||||
|
||||
func (s *fakeSession) Terminate(ctx context.Context) error {
|
||||
if s.failure == FakeFailureTerminationTimeout {
|
||||
s.mu.Lock()
|
||||
terminationTimeout := s.failure == FakeFailureTerminationTimeout
|
||||
s.mu.Unlock()
|
||||
if terminationTimeout {
|
||||
<-ctx.Done()
|
||||
s.mu.Lock()
|
||||
s.state.State = ProviderStateCleanup
|
||||
|
||||
Reference in New Issue
Block a user