Method graceShutdown

tars/application.go:493–564

()

}

func (a *application) graceShutdown() {
	var wg sync.WaitGroup

	atomic.StoreInt32(&a.isShutdowning, 1)
	pid := os.Getpid()

	var graceShutdownTimeout time.Duration
	if atomic.LoadInt32(&a.isShutdownByAdmin) == 1 {
		// shutdown by admin,we should need shorten the timeout
		graceShutdownTimeout = tools.ParseTimeOut(GracedownTimeout)
	} else {
		graceShutdownTimeout = a.svrCfg.GracedownTimeout
	}

	TLOG.Infof("grace shutdown start %d in %v", pid, graceShutdownTimeout)
	ctx, cancel := context.WithTimeout(context.Background(), graceShutdownTimeout)
	// deregister service
	wg.Add(1)
	go func() {
		defer wg.Done()
		a.deregisterAdapters(ctx)
	}()

	for _, obj := range a.destroyableObjs {
		wg.Add(1)
		go func(wg *sync.WaitGroup, obj destroyableImp) {
			defer wg.Done()
			obj.Destroy()
			TLOG.Infof("grace Destroy success %d", pid)
		}(&wg, obj)
	}

	for _, obj := range a.objRunList {
		if s, ok := a.httpSvrs[obj]; ok {
			wg.Add(1)
			go func(s *http.Server, ctx context.Context, wg *sync.WaitGroup, objstr string) {
				defer wg.Done()
				if err := s.Shutdown(ctx); err != nil {
					TLOG.Errorf("grace shutdown http %s failed within %v, err%v", objstr, graceShutdownTimeout, err)
				} else {
					TLOG.Infof("grace shutdown http %s success %d", objstr, pid)
				}
			}(s, ctx, &wg, obj)
		}

		if s, ok := a.goSvrs[obj]; ok {
			wg.Add(1)
			go func(s *transport.TarsServer, ctx context.Context, wg *sync.WaitGroup, objstr string) {
				defer wg.Done()
				if err := s.Shutdown(ctx); err != nil {
					TLOG.Errorf("grace shutdown tars %s failed within %v, err: %v", objstr, graceShutdownTimeout, err)
				} else {
					TLOG.Infof("grace shutdown tars %s success %d", objstr, pid)
				}
			}(s, ctx, &wg, obj)
		}
	}