Make Read/Write Buffer Size configurable

configuration.go

@@ -40,3 +40,12 @@ func SetReadingWait(seconds int) DialerConfig {
 		c.readingWait = time.Duration(seconds) * time.Second
 	}
 }
+
+
+//SetBufferSize sets the read/write buffer size
+func SetBufferSize(readBufferSize int, writeBufferSize int) DialerConfig {
+	return func(c *Ws) {
+		c.readBufSize = readBufferSize
+                c.writeBufSize = writeBufferSize
+	}
+}

connection.go

@@ -38,6 +38,8 @@ type Ws struct {
 	writingWait  time.Duration
 	readingWait  time.Duration
 	timeout      time.Duration
+        readBufSize  int
+        writeBufSize int
 	quit         chan struct{}
 	sync.RWMutex
 }
@@ -50,8 +52,8 @@ type auth struct {
 
 func (ws *Ws) connect() (err error) {
 	d := websocket.Dialer{
-		WriteBufferSize:  8192,
-		ReadBufferSize:   8192,
+		WriteBufferSize:  ws.writeBufSize,
+		ReadBufferSize:   ws.readBufSize,
 		HandshakeTimeout: ws.timeout, // Timeout or else we'll hang forever and never fail on bad hosts.
 	}
 	ws.conn, _, err = d.Dial(ws.host, http.Header{})