From cf8236b93857bbfb2f70139971facf600f367817 Mon Sep 17 00:00:00 2001
From: Heikki Uljas <huljas@reforgedstudios.com>
Date: Wed, 13 Jan 2016 08:03:00 +0200
Subject: [PATCH] Added params to session

---
 melody.go  |  1 +
 session.go | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/melody.go b/melody.go
index acf8c45..0b636c5 100644
--- a/melody.go
+++ b/melody.go
@@ -80,6 +80,7 @@ func (m *Melody) HandleRequest(w http.ResponseWriter, r *http.Request) {
 
 	session := &Session{
 		Request: r,
+		params:  make(map[string]string),
 		conn:    conn,
 		output:  make(chan *envelope, m.Config.MessageBufferSize),
 		melody:  m,
diff --git a/session.go b/session.go
index d66d4b4..35dc223 100644
--- a/session.go
+++ b/session.go
@@ -10,6 +10,7 @@ import (
 // A melody session.
 type Session struct {
 	Request *http.Request
+	params  map[string]string
 	conn    *websocket.Conn
 	output  chan *envelope
 	melody  *Melody
@@ -117,3 +118,13 @@ func (s *Session) WriteBinary(msg []byte) {
 func (s *Session) Close() {
 	s.writeMessage(&envelope{t: websocket.CloseMessage, msg: []byte{}})
 }
+
+// Set session param
+func (s *Session) SetParam(key string, value string) {
+	s.params[key] = value
+}
+
+// Get session param
+func (s *Session) GetParam(key string) string {
+	return s.params[key]
+}