Merge pull request #865 from dickens7/feat-wirteerror

fix: handler mode write error
smallnest · Jul 18, 2024 · ff384af · ff384af
2 parents 9d62234 + d5ee82d
commit ff384af
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/server/server.go b/server/server.go
@@ -584,7 +584,12 @@ func (s *Server) processOneRequest(ctx *share.Context, req *protocol.Message, co
 		sctx := NewContext(ctx, conn, req, s.AsyncWrite)
 		err := handler(sctx)
 		if err != nil {
-			log.Errorf("[handler internal error]: servicepath: %s, servicemethod, err: %v", req.ServicePath, req.ServiceMethod, err)
+			if s.HandleServiceError != nil {
+				s.HandleServiceError(err)
+			} else {
+				log.Errorf("[handler internal error]: servicepath: %s, servicemethod, err: %v", req.ServicePath, req.ServiceMethod, err)
+			}
+			sctx.WriteError(err)
 		}
 
 		return