我正在開發(fā)一個帶有網(wǎng)絡服務器的 Go 應用程序。我試圖添加超時并遇到問題。這是我用來重現(xiàn)它的示例代碼，因為發(fā)布實際代碼是不可能的：package mainimport ( "fmt" "html/template" "net/http" "time")var layout *template.Templatefunc main() { router := http.NewServeMux() server := &http.Server{ Addr: ":8888", Handler: router, ReadTimeout: 5 * time.Second, WriteTimeout: 1 * time.Second, IdleTimeout: 15 * time.Second, } router.HandleFunc("/", home) var err error layout, err = template.ParseFiles("./layout.html") if err != nil { fmt.Printf("Error1: %+v\n", err) } server.ListenAndServe()}func home(w http.ResponseWriter, r *http.Request) { fmt.Println("responding") err := layout.Execute(w, template.HTML(`World`)) if err != nil { fmt.Printf("Error2: %+v\n", err) } time.Sleep(5 * time.Second)}布局.html： Hello {{.}}!當我運行它并訪問 127.0.0.1:8888 時，瀏覽器保持加載狀態(tài)，并且觸發(fā)超時home()，重新開始，它在停止之前執(zhí)行了 10 次，并且瀏覽器顯示連接重置錯誤。我期望超時后，函數(shù)將立即結(jié)束，連接關(guān)閉，瀏覽器停止加載并顯示錯誤。我怎樣才能實現(xiàn)這個目標？
1 回答
楊__羊羊
TA貢獻1943條經(jīng)驗 獲得超7個贊
立即響應使用 goroutine 和上下文超時
package main
import (
"context"
"fmt"
"html/template"
"net/http"
"time"
)
var layout *template.Template
var WriteTimeout = 1 * time.Second
func main() {
router := http.NewServeMux()
server := &http.Server{
Addr: ":8889",
Handler: router,
ReadTimeout: 5 * time.Second,
WriteTimeout: WriteTimeout + 10*time.Millisecond, //10ms Redundant time
IdleTimeout: 15 * time.Second,
}
router.HandleFunc("/", home)
server.ListenAndServe()
}
func home(w http.ResponseWriter, r *http.Request) {
fmt.Printf("responding\n")
ctx, _ := context.WithTimeout(context.Background(), WriteTimeout)
worker, cancel := context.WithCancel(context.Background())
var buffer string
go func() {
// do something
time.Sleep(2 * time.Second)
buffer = "ready all response\n"
//do another
time.Sleep(2 * time.Second)
cancel()
fmt.Printf("worker finish\n")
}()
select {
case <-ctx.Done():
//add more friendly tips
w.WriteHeader(http.StatusInternalServerError)
return
case <-worker.Done():
w.Write([]byte(buffer))
fmt.Printf("writed\n")
return
}
}
- 1 回答
- 0 關(guān)注
- 181 瀏覽
添加回答
舉報
0/150
提交
取消