blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-02 18:34:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix the timeout

Browse Source
This commit is contained in:
luhaoling 2024-02-05 10:48:58 +08:00
parent 482227552e
commit 38d3064399
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/common/startrpc/start.go
View File

@ -60,7 +60,7 @@ func Start(
rpcTcpAddr,
)
if err != nil {
return errs.Wrap(err, rpcTcpAddr)
return errs.Wrap(err, "rpc start err", rpcTcpAddr)
}
defer listener.Close()
@ -118,7 +118,7 @@ func Start(
// Create a HTTP server for prometheus.
httpServer = &http.Server{Handler: promhttp.HandlerFor(reg, promhttp.HandlerOpts{}), Addr: fmt.Sprintf("0.0.0.0:%d", prometheusPort)}
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
netErr = errs.Wrap(err, "prometheus start err: ", httpServer.Addr)
netErr = errs.Wrap(err, "prometheus start err", httpServer.Addr)
close(netDone)
}
}
@ -134,11 +134,12 @@ func Start(
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGUSR1)
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
select {
case <-sigs:
print("receive process terminal SIGUSR1 exit")
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
if err := gracefulStopWithCtx(ctx, srv.GracefulStop); err != nil {
return err
}
@ -162,7 +163,7 @@ func gracefulStopWithCtx(ctx context.Context, f func()) error {
}()
select {
case <-ctx.Done():
return errs.Wrap(errors.New("timeout"), "ctx graceful stop")
return errs.Wrap(errors.New("timeout, ctx graceful stop"))
case <-done:
return nil
}