Merge pull request #116 from mattn/real-ip

use X-Real-Ip if presented

handlers.go

@@ -348,7 +348,13 @@ func (s *Server) HandleWebsocket(w http.ResponseWriter, r *http.Request) {
 	ticker := time.NewTicker(pingPeriod)
 	stop := make(chan struct{})
 
-	s.Log.Infof("connected from %s", conn.RemoteAddr().String())
+	ip := conn.RemoteAddr().String()
+	if realIP := r.Header.Get("X-Forwarded-For"); realIP != "" {
+		ip = realIP // possible to be multiple comma separated
+	} else if realIP := r.Header.Get("X-Real-Ip"); realIP != "" {
+		ip = realIP
+	}
+	s.Log.Infof("connected from %s", ip)
 
 	ws := challenge(conn)
 
@@ -372,7 +378,7 @@ func (s *Server) HandleWebsocket(w http.ResponseWriter, r *http.Request) {
 				removeListener(ws)
 			}
 			s.clientsMu.Unlock()
-			s.Log.Infof("disconnected from %s", conn.RemoteAddr().String())
+			s.Log.Infof("disconnected from %s", ip)
 
 			ctx.Done()
 		}()
@@ -438,7 +444,7 @@ func (s *Server) HandleWebsocket(w http.ResponseWriter, r *http.Request) {
 					s.Log.Errorf("error writing ping: %v; closing websocket", err)
 					return
 				}
-				s.Log.Infof("pinging for %s", conn.RemoteAddr().String())
+				s.Log.Infof("pinging for %s", ip)
 			case <-stop:
 				return
 			}