mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
77e123762f
1. add coordinator and proxy graceful stop timeout to 5s. 3. add other work node graceful stop timeout to 900s, and we should potentially change this to 600s when graceful stop is smooth 4. change the order of datacoord component while stop. 5. `LivenessCheck` do not perform graceful shutdown now. issue: https://github.com/milvus-io/milvus/issues/30310 pr: #30317 also see: https://github.com/milvus-io/milvus/pull/30306 --------- Signed-off-by: chyezh <chyezh@outlook.com>
39 lines
756 B
Go
39 lines
756 B
Go
package components
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cockroachdb/errors"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestExitWithTimeout(t *testing.T) {
|
|
// only normal path can be tested.
|
|
targetErr := errors.New("stop error")
|
|
err := exitWhenStopTimeout(func() error {
|
|
time.Sleep(1 * time.Second)
|
|
return targetErr
|
|
}, 5*time.Second)
|
|
assert.ErrorIs(t, err, targetErr)
|
|
}
|
|
|
|
func TestStopWithTimeout(t *testing.T) {
|
|
ch := make(chan struct{})
|
|
stop := func() error {
|
|
<-ch
|
|
return nil
|
|
}
|
|
|
|
err := stopWithTimeout(stop, 1*time.Second)
|
|
assert.ErrorIs(t, err, errStopTimeout)
|
|
|
|
targetErr := errors.New("stop error")
|
|
stop = func() error {
|
|
return targetErr
|
|
}
|
|
|
|
err = stopWithTimeout(stop, 1*time.Second)
|
|
assert.ErrorIs(t, err, targetErr)
|
|
}
|