Commit 8f9f4d99 authored by rui.zheng's avatar rui.zheng

fix runtime error

parent 2a81d2c5
...@@ -98,6 +98,7 @@ func (conn *HttpClientConn) Write(b []byte) (n int, err error) { ...@@ -98,6 +98,7 @@ func (conn *HttpClientConn) Write(b []byte) (n int, err error) {
resp.Body.Close() resp.Body.Close()
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
log.Println(resp.Status)
return 0, errors.New(resp.Status) return 0, errors.New(resp.Status)
} }
//log.Println("http w:", len(b)) //log.Println("http w:", len(b))
...@@ -202,7 +203,7 @@ type HttpServer struct { ...@@ -202,7 +203,7 @@ type HttpServer struct {
func (s *HttpServer) s2c(w http.ResponseWriter, r *http.Request) { func (s *HttpServer) s2c(w http.ResponseWriter, r *http.Request) {
token := uuid.New() token := uuid.New()
ch := make(chan []byte, 1) ch := make(chan []byte, 8)
conn := NewHttpServerConn(w, ch) conn := NewHttpServerConn(w, ch)
if _, err := conn.Write([]byte(token)); err != nil { if _, err := conn.Write([]byte(token)); err != nil {
...@@ -216,6 +217,12 @@ func (s *HttpServer) s2c(w http.ResponseWriter, r *http.Request) { ...@@ -216,6 +217,12 @@ func (s *HttpServer) s2c(w http.ResponseWriter, r *http.Request) {
} }
func (s *HttpServer) c2s(w http.ResponseWriter, r *http.Request) { func (s *HttpServer) c2s(w http.ResponseWriter, r *http.Request) {
defer func() {
if err := recover(); err != nil {
log.Println(err)
}
}()
if r.Method != "POST" { if r.Method != "POST" {
w.WriteHeader(http.StatusMethodNotAllowed) w.WriteHeader(http.StatusMethodNotAllowed)
return return
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment