Merge pull request olahol#2 from huljas/generic_params

Changed session params to be generic type
skyfour · Nov 18, 2016 · 150502c · 150502c
2 parents 0d8427c + aae1ca8
commit 150502c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/melody.go b/melody.go
@@ -80,7 +80,7 @@ func (m *Melody) HandleRequest(w http.ResponseWriter, r *http.Request) {
 
 	session := &Session{
 		Request: r,
-		params:  make(map[string]string),
+		Params:  make(map[string]interface{}),
 		conn:    conn,
 		output:  make(chan *envelope, m.Config.MessageBufferSize),
 		melody:  m,

diff --git a/session.go b/session.go
@@ -10,7 +10,7 @@ import (
 // A melody session.
 type Session struct {
 	Request *http.Request
-	params  map[string]string
+	Params  map[string]interface{}
 	conn    *websocket.Conn
 	output  chan *envelope
 	melody  *Melody
@@ -118,13 +118,3 @@ 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]
-}