root/gost_software
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix runtime error

Browse Source
This commit is contained in:
rui.zheng 2015-06-05 17:02:08 +08:00
parent 2a81d2c59b
commit 8f9f4d99ec
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
http.go
View File

@ -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 {
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) {
} }
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