runningwater 4 年之前
父節點
當前提交
0a63e480e6
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      controller.go

+ 6 - 0
controller.go

@@ -35,13 +35,19 @@ func FooControllerHandler(ctx *framework.Context) error {
     select {
         // 监听 panic
     case <-panicChan:
+        ctx.WriterMux().Lock()
+        defer ctx.WriterMux().Unlock()
         ctx.Json(500, "panic")
         // 监听结束
     case <-finish:
         fmt.Println("finish")
         // 监听超时
     case <-durationCtx.Done():
+        ctx.WriterMux().Lock()
+        defer ctx.WriterMux().Unlock()
         ctx.Json(500, "time out")
+        // 这里记得设置标记
+        ctx.SetHasTimeout()
     }
     return ctx.Json(200, map[string]interface{}{
         "code": 0,