use ContextValues's flag

b473f6f2 · AlexStocks · d338d5bd · b473f6f2
Commit b473f6f2 authored Apr 24, 2018 by AlexStocks
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

session.go session.go +6 -2

No files found.
--- a/session.go
+++ b/session.go
@@ -274,10 +274,14 @@ func (s *session) SetWaitTime(waitTime time.Duration) {

 // set attribute of key @session:key
 func (s *session) GetAttribute(key interface{}) interface{} {
-	var ret interface{}
 	s.lock.RLock()
-	ret = s.attrs.Get(key)
+	ret, flag := s.attrs.Get(key)
 	s.lock.RUnlock()
+
+	if !flag {
+		return nil
+	}
+
 	return ret
 }