server. store connection header. (for X-Forwarded-For, and so on)

Author: Gennadii Kovalev

loop.go

@@ -5,6 +5,7 @@ import (
 "errors"
 "github.com/graarh/golang-socketio/protocol"
 "github.com/graarh/golang-socketio/transport"
+"net/http"
 "sync"
 "time"
 )
@@ -47,8 +48,9 @@ type Channel struct {
 
 ack ackProcessor
 
-server *Server
-ip string
+server *Server
+ip string
+requestHeader http.Header
 }
 
 /**
@@ -150,7 +152,7 @@ outgoing messages loop, sends messages from channel to socket
 func outLoop(c *Channel, m *methods) error {
 for {
 outBufferLen := len(c.out)
-if outBufferLen >= queueBufferSize - 1 {
+if outBufferLen >= queueBufferSize-1 {
 return CloseChannel(c, m, ErrorSocketOverflood)
 } else if outBufferLen > int(queueBufferSize/2) {
 overfloodedLock.Lock()

server.go

@@ -39,11 +39,18 @@ type Server struct {
 
 /**
 Get ip of socket client
- */
+*/
 func (c *Channel) Ip() string {
 return c.ip
 }
 
+/**
+Get request header of this connection
+*/
+func (c *Channel) RequestHeader() http.Header {
+return c.requestHeader
+}
+
 /**
 Get channel by it's sid
 */
@@ -279,7 +286,9 @@ func (s *Server) SendOpenSequence(c *Channel) {
 /**
 Setup event loop for given connection
 */
-func (s *Server) SetupEventLoop(conn transport.Connection, remoteAddr string) {
+func (s *Server) SetupEventLoop(conn transport.Connection, remoteAddr string,
+requestHeader http.Header) {
+
 interval, timeout := conn.PingParams()
 hdr := Header{
 Sid: generateNewId(remoteAddr),
@@ -291,6 +300,7 @@ func (s *Server) SetupEventLoop(conn transport.Connection, remoteAddr string) {
 c := &Channel{}
 c.conn = conn
 c.ip = remoteAddr
+c.requestHeader = requestHeader
 c.initChannel()
 
 c.server = s
@@ -313,7 +323,7 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 return
 }
 
-s.SetupEventLoop(conn, r.RemoteAddr)
+s.SetupEventLoop(conn, r.RemoteAddr, r.Header)
 s.tr.Serve(w, r)
 }