golang的httpserver優(yōu)雅重啟方法詳解
前言
去年在做golangserver的時(shí)候,內(nèi)部比較頭疼的就是在線服務(wù)發(fā)布的時(shí)候,大量用戶的請(qǐng)求在發(fā)布時(shí)候會(huì)被重連,在那時(shí)候也想了n多的方法,最后還是落在一個(gè)github上的項(xiàng)目,facebook的一個(gè)golang項(xiàng)目grace,那時(shí)候簡(jiǎn)單研究測(cè)試了一下可以就直接在內(nèi)部使用了起來,這段時(shí)間突然想起來,又想仔細(xì)研究一下這個(gè)項(xiàng)目了。
從原理上來說是這樣一個(gè)過程:
1)發(fā)布新的bin文件去覆蓋老的bin文件
2)發(fā)送一個(gè)信號(hào)量,告訴正在運(yùn)行的進(jìn)程,進(jìn)行重啟
3)正在運(yùn)行的進(jìn)程收到信號(hào)后,會(huì)以子進(jìn)程的方式啟動(dòng)新的bin文件
4)新進(jìn)程接受新請(qǐng)求,并處理
5)老進(jìn)程不再接受請(qǐng)求,但是要等正在處理的請(qǐng)求處理完成,所有在處理的請(qǐng)求處理完之后,便自動(dòng)退出
6)新進(jìn)程在老進(jìn)程退出之后,由init進(jìn)程收養(yǎng),但是會(huì)繼續(xù)服務(wù)。
所以一步一步來看,關(guān)鍵是從第2步開始之后怎么做,所以我們先來看看第2步的實(shí)現(xiàn),這個(gè)應(yīng)該說很簡(jiǎn)單,發(fā)送信號(hào)量到一個(gè)進(jìn)程,使用kill命令即可,在facebook這個(gè)項(xiàng)目中發(fā)送的信號(hào)量有3個(gè):SIGINT,SIGTERM,SIGUSR2,前面兩個(gè)信號(hào)收到后程序會(huì)直接退出,后面一個(gè)信號(hào)SIGUSR2才會(huì)執(zhí)行所謂的優(yōu)雅重啟。
第3步,正在運(yùn)行的進(jìn)程收到SIGUSR2信號(hào)后,會(huì)以子進(jìn)程的方式啟動(dòng)新的bin文件。先直接上代碼看:https://github.com/facebookgo/grace/blob/master/gracehttp/http.go
func (a *app) signalHandler(wg *sync.WaitGroup) { ch := make(chan os.Signal, 10) signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM, syscall.SIGUSR2) for { sig := <-ch switch sig { case syscall.SIGINT, syscall.SIGTERM: // this ensures a subsequent INT/TERM will trigger standard go behaviour of // terminating. 執(zhí)行標(biāo)準(zhǔn)的go終止行為,程序就結(jié)束了 signal.Stop(ch) a.term(wg) return case syscall.SIGUSR2: // 這里開始執(zhí)行優(yōu)雅重啟 err := a.preStartProcess() // 這個(gè)函數(shù)在源代碼中沒有具體實(shí)現(xiàn)功能,只是預(yù)留了一個(gè)鉤子函數(shù),用戶可以注冊(cè)自己的函數(shù),可以在重啟之前做些自定義的事情。一般情況下也沒有什么可以做的,除非有些特殊的服務(wù)環(huán)境或是狀態(tài)保存之類的,至少目前,我們的server還沒有遇到 if err != nil { a.errors <- err } // we only return here if there's an error, otherwise the new process // will send us a TERM when it's ready to trigger the actual shutdown. if _, err := a.net.StartProcess(); err != nil { // 這里開始正式所謂的優(yōu)雅重啟 a.errors <- err } } } }
a.net.StartProcess的過程我們來看看基本過程:
func (n *Net) StartProcess() (int, error) { listeners, err := n.activeListeners() // 獲取目前在監(jiān)聽的端口,這塊也是重點(diǎn),下面重點(diǎn)介紹 if err != nil { return 0, err } // Extract the fds from the listeners. 從監(jiān)聽端口中把文件描述符取出來 files := make([]*os.File, len(listeners)) for i, l := range listeners { files[i], err = l.(filer).File() if err != nil { return 0, err } defer files[i].Close() } // Use the original binary location. This works with symlinks such that if // the file it points to has been changed we will use the updated symlink. // 獲取可執(zhí)行bin文件的路勁,也可以是鏈接路勁,會(huì)使用最新的鏈接路徑作為啟動(dòng)文件路勁的 argv0, err := exec.LookPath(os.Args[0]) if err != nil { return 0, err } // Pass on the environment and replace the old count key with the new one. // 獲取 LISTEN_FDS 換進(jìn)變量值 var env []string for _, v := range os.Environ() { if !strings.HasPrefix(v, envCountKeyPrefix) { env = append(env, v) } } env = append(env, fmt.Sprintf("%s%d", envCountKeyPrefix, len(listeners))) allFiles := append([]*os.File{os.Stdin, os.Stdout, os.Stderr}, files...) // 這里調(diào)用一個(gè)golang底層的進(jìn)程啟動(dòng)函數(shù),來指定,上面獲取的參數(shù)來啟動(dòng)進(jìn)程 process, err := os.StartProcess(argv0, os.Args, &os.ProcAttr{ Dir: originalWD, Env: env, Files: allFiles, }) if err != nil { return 0, err } // 返回新進(jìn)程id。 return process.Pid, nil }
以上是啟動(dòng)新進(jìn)程,并且接管監(jiān)聽端口的過程, 一般情況下端口是不可以重復(fù)監(jiān)聽的,所以這里就要需要使用比較特別的辦法,從上面的代碼來看就是讀取監(jiān)聽端口的文件描述符,并且把監(jiān)聽端口的文件描述符傳遞給子進(jìn)程,子進(jìn)程里從這個(gè)文件描述符實(shí)現(xiàn)對(duì)端口的監(jiān)聽
另外還有一個(gè)比較特別的地方就是老的接口怎么關(guān)閉的問題,關(guān)閉必須要把已經(jīng)收到的請(qǐng)求處理完成之后再關(guān)閉。為此facebook的同學(xué)另外開了一個(gè)項(xiàng)目httpdown,繼承了原始的httpserver,但是多了對(duì)各種鏈接狀態(tài)的維護(hù)和處理,這部分后面在分析。
總結(jié)
以上就是這篇文章的全部?jī)?nèi)容了,希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,如果有疑問大家可以留言交流,謝謝大家對(duì)腳本之家的支持。
相關(guān)文章
gin框架Context如何獲取Get?Query?Param函數(shù)數(shù)據(jù)
這篇文章主要為大家介紹了gin框架Context?Get?Query?Param函數(shù)獲取數(shù)據(jù),有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-03-03Go語言字符串及strings和strconv包使用實(shí)例
字符串是工作中最常用的,值得我們專門的練習(xí)一下,下面這篇文章主要給大家介紹了關(guān)于Go語言字符串及strings和strconv包使用的相關(guān)資料,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-06-06Go中的 panic / recover 簡(jiǎn)介與實(shí)踐記錄
這篇文章主要介紹了Go中的 panic / recover 簡(jiǎn)介與實(shí)踐,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2023-04-04Go實(shí)現(xiàn)后臺(tái)任務(wù)調(diào)度系統(tǒng)的實(shí)例代碼
平常我們?cè)陂_發(fā)API的時(shí)候,前端傳遞過來的大批數(shù)據(jù)需要經(jīng)過后端處理,如果后端處理的速度快,前端響應(yīng)就快,反之則很慢,影響用戶體驗(yàn),為了解決這一問題,需要我們自己實(shí)現(xiàn)后臺(tái)任務(wù)調(diào)度系統(tǒng),本文將介紹如何用Go語言實(shí)現(xiàn)后臺(tái)任務(wù)調(diào)度系統(tǒng),需要的朋友可以參考下2023-06-06golang協(xié)程池模擬實(shí)現(xiàn)群發(fā)郵件功能
這篇文章主要介紹了golang協(xié)程池模擬實(shí)現(xiàn)群發(fā)郵件功能,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2021-05-05