Go程序中怎么重啟服務(wù)器,很多新手對此不是很清楚,為了幫助大家解決這個難題,下面小編將為大家詳細(xì)講解,有這方面需求的人可以來學(xué)習(xí)下,希望你能有所收獲。
專注于為中小企業(yè)提供成都網(wǎng)站制作、做網(wǎng)站服務(wù),電腦端+手機(jī)端+微信端的三站合一,更高效的管理,為中小企業(yè)始興免費做網(wǎng)站提供優(yōu)質(zhì)的服務(wù)。我們立足成都,凝聚了一批互聯(lián)網(wǎng)行業(yè)人才,有力地推動了千余家企業(yè)的穩(wěn)健成長,幫助中小企業(yè)通過網(wǎng)站建設(shè)實現(xiàn)規(guī)模擴(kuò)充和轉(zhuǎn)變。
原理
在基于Unix的操作系統(tǒng)中,signal(信號)是與長時間運(yùn)行的進(jìn)程交互的常用方法.
SIGTERM: 優(yōu)雅地停止進(jìn)程
SIGHUP: 重啟/重新加載進(jìn)程 (例如: nginx, sshd, apache)
如果收到SIGHUP信號,優(yōu)雅地重啟進(jìn)程需要以下幾個步驟:
服務(wù)器要拒絕新的連接請求,但要保持已有的連接。
啟用新版本的進(jìn)程
將socket“交給”新進(jìn)程,新進(jìn)程開始接受新連接請求
舊進(jìn)程處理完畢后立即停止。
停止接受連接請求
服務(wù)器程序的共同點:持有一個死循環(huán)來接受連接請求:
復(fù)制代碼 代碼如下:
for {
conn, err := listener.Accept()
// Handle connection
}
跳出這個循環(huán)的最簡單方式是在socket監(jiān)聽器上設(shè)置一個超時,當(dāng)調(diào)用listener.SetTimeout(time.Now())后,listener.Accept()會立即返回一個timeout err,你可以捕獲并處理:
復(fù)制代碼 代碼如下:
for {
conn, err := listener.Accept()
if err != nil {
if nerr, ok := err.(net.Err); ok && nerr.Timeout() {
fmt.Println(“Stop accepting connections”)
return
}
}
}
注意這個操作與關(guān)閉listener有所不同。這樣進(jìn)程仍在監(jiān)聽服務(wù)器端口,但連接請求會被操作系統(tǒng)的網(wǎng)絡(luò)棧排隊,等待一個進(jìn)程接受它們。
啟動新進(jìn)程
Go提供了一個原始類型ForkExec來產(chǎn)生新進(jìn)程.你可以與這個新進(jìn)程共享某些消息,例如文件描述符或環(huán)境參數(shù)。
復(fù)制代碼 代碼如下:
execSpec := &syscall.ProcAttr{
Env: os.Environ(),
Files: []uintptr{os.Stdin.Fd(), os.Stdout.Fd(), os.Stderr.Fd()},
}
fork, err := syscall.ForkExec(os.Args[0], os.Args, execSpec)
[…]
你會發(fā)現(xiàn)這個進(jìn)程使用完全相同的參數(shù)os.Args啟動了一個新進(jìn)程。
發(fā)送socket到子進(jìn)程并恢復(fù)它
正如你先前看到的,你可以將文件描述符傳遞到新進(jìn)程,這需要一些UNIX魔法(一切都是文件),我們可以把socket發(fā)送到新進(jìn)程中,這樣新進(jìn)程就能夠使用它并接收及等待新的連接。
但fork-execed進(jìn)程需要知道它必須從文件中得到socket而不是新建一個(有些興許已經(jīng)在使用了,因為我們還沒斷開已有的監(jiān)聽)。你可以按任何你希望的方法來,最常見的是通過環(huán)境變量或命令行標(biāo)志。
復(fù)制代碼 代碼如下:
listenerFile, err := listener.File()
if err != nil {
log.Fatalln("Fail to get socket file descriptor:", err)
}
listenerFd := listenerFile.Fd()
// Set a flag for the new process start process
os.Setenv("_GRACEFUL_RESTART", "true")
execSpec := &syscall.ProcAttr{
Env: os.Environ(),
Files: []uintptr{os.Stdin.Fd(), os.Stdout.Fd(), os.Stderr.Fd(), listenerFd},
}
// Fork exec the new version of your server
fork, err := syscall.ForkExec(os.Args[0], os.Args, execSpec)
然后在程序的開始處:
復(fù)制代碼 代碼如下:
var listener *net.TCPListener
if os.Getenv("_GRACEFUL_RESTART") == "true" {
// The second argument should be the filename of the file descriptor
// however, a socker is not a named file but we should fit the interface
// of the os.NewFile function.
file := os.NewFile(3, "")
listener, err := net.FileListener(file)
if err != nil {
// handle
}
var bool ok
listener, ok = listener.(*net.TCPListener)
if !ok {
// handle
}
} else {
listener, err = newListenerWithPort(12345)
}
文件描述沒有被隨機(jī)的選擇為3,這是因為uintptr的切片已經(jīng)發(fā)送了fork,監(jiān)聽獲取了索引3。留意隱式聲明問題。
最后一步,等待舊服務(wù)連接停止
到此為止,就這樣,我們已經(jīng)將其傳到另一個正在正確運(yùn)行的進(jìn)程,對于舊服務(wù)器的最后操作是等其連接關(guān)閉。由于標(biāo)準(zhǔn)庫里提供了sync.WaitGroup結(jié)構(gòu)體,用go實現(xiàn)這個功能很簡單。
每次接收一個連接,在WaitGroup上加1,然后,我們在它完成時將計數(shù)器減一:
復(fù)制代碼 代碼如下:
for {
conn, err := listener.Accept()
wg.Add(1)
go func() {
handle(conn)
wg.Done()
}()
}
至于等待連接的結(jié)束,你僅需要wg.Wait(),因為沒有新的連接,我們等待wg.Done()已經(jīng)被所有正在運(yùn)行的handler調(diào)用。
Bonus: 不要無限制等待,給定限量的時間
復(fù)制代碼 代碼如下:
timeout := time.NewTimer(time.Minute)
wait := make(chan struct{})
go func() {
wg.Wait()
wait <- struct{}{}
}()
select {
case <-timeout.C:
return WaitTimeoutError
case <-wait:
return nil
}
看完上述內(nèi)容是否對您有幫助呢?如果還想對相關(guān)知識有進(jìn)一步的了解或閱讀更多相關(guān)文章,請關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道,感謝您對創(chuàng)新互聯(lián)的支持。
網(wǎng)站題目:Go程序中怎么重啟服務(wù)器
鏈接分享:http://www.rwnh.cn/article18/pcoidp.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供網(wǎng)站收錄、外貿(mào)網(wǎng)站建設(shè)、企業(yè)建站、網(wǎng)站排名、網(wǎng)站導(dǎo)航、關(guān)鍵詞優(yōu)化
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時需注明來源: 創(chuàng)新互聯(lián)