fix #1539

Former-commit-id: f2f277cd5cbe781ce596adc7840a1b1bc3b3bfc6

sessions/sessiondb/badger/database.go

@@ -156,6 +156,12 @@ func (db *Database) Get(sid string, key string) (value interface{}) {
 	return
 }
 
+// validSessionItem reports whether the current iterator's item key
+// is a value of the session id "prefix".
+func validSessionItem(key, prefix []byte) bool {
+	return len(key) > len(prefix) && bytes.Equal(key[0:len(prefix)], prefix)
+}
+
 // Visit loops through all session keys and values.
 func (db *Database) Visit(sid string, cb func(key string, value interface{})) {
 	prefix := makePrefix(sid)
@@ -166,30 +172,28 @@ func (db *Database) Visit(sid string, cb func(key string, value interface{})) {
 	iter := txn.NewIterator(badger.DefaultIteratorOptions)
 	defer iter.Close()
 
-	for iter.Rewind(); iter.ValidForPrefix(prefix); iter.Next() {
+	for iter.Rewind(); ; iter.Next() {
+		if !iter.Valid() {
+			break
+		}
+
 		item := iter.Item()
+		key := item.Key()
+		if !validSessionItem(key, prefix) {
+			continue
+		}
+
 		var value interface{}
 
-		// err := item.Value(func(valueBytes []byte) {
-		// 	if err := sessions.DefaultTranscoder.Unmarshal(valueBytes, &value); err != nil {
-		// 		golog.Error(err)
-		// 	}
-		// })
-
-		// if err != nil {
-		// 	golog.Error(err)
-		// 	continue
-		// }
-
 		err := item.Value(func(valueBytes []byte) error {
 			return sessions.DefaultTranscoder.Unmarshal(valueBytes, &value)
 		})
 		if err != nil {
-			golog.Error(err)
+			golog.Errorf("[sessionsdb.badger.Visit] %v", err)
 			continue
 		}
 
-		cb(string(bytes.TrimPrefix(item.Key(), prefix)), value)
+		cb(string(bytes.TrimPrefix(key, prefix)), value)
 	}
 }
 
@@ -207,8 +211,14 @@ func (db *Database) Len(sid string) (n int) {
 	txn := db.Service.NewTransaction(false)
 	iter := txn.NewIterator(iterOptionsNoValues)
 
-	for iter.Rewind(); iter.ValidForPrefix(prefix); iter.Next() {
-		n++
+	for iter.Rewind(); ; iter.Next() {
+		if !iter.Valid() {
+			break
+		}
+
+		if validSessionItem(iter.Item().Key(), prefix) {
+			n++
+		}
 	}
 
 	iter.Close()

sessions/sessiondb/redis/driver_radix.go

@@ -263,7 +263,12 @@ func (r *RadixDriver) getKeys(cursor, prefix string) ([]string, error) {
 		return nil, err
 	}
 
-	keys := res.keys[0:]
+	if len(res.keys) <= 1 {
+		return nil, nil
+	}
+
+	keys := res.keys[1:] // the first one is always the session id itself.
+
 	if res.cur != "0" {
 		moreKeys, err := r.getKeys(res.cur, prefix)
 		if err != nil {

sessions/sessiondb/redis/driver_redigo.go

@@ -205,10 +205,19 @@ func (r *RedigoDriver) getKeysConn(c redis.Conn, cursor interface{}, prefix stri
 		if len(replies) == 2 {
 			// take the second, it must contain the slice of keys.
 			if keysSliceAsBytes, ok := replies[1].([]interface{}); ok {
-				keys := make([]string, len(keysSliceAsBytes))
+				n := len(keysSliceAsBytes) - 1 // scan match returns the session id key too.
+				if n <= 0 {
+					return nil, nil
+				}
+
+				keys := make([]string, n)
 
 				for i, k := range keysSliceAsBytes {
-					keys[i] = fmt.Sprintf("%s", k)[len(r.Config.Prefix):]
+					key := fmt.Sprintf("%s", k)[len(r.Config.Prefix):]
+					if key == prefix {
+						continue // it's the session id itself.
+					}
+					keys[i] = key
 				}
 
 				if cur := fmt.Sprintf("%s", replies[0]); cur != "0" {

sessions/sessions.go

@@ -83,6 +83,13 @@ func (s *Sessions) Start(ctx context.Context, cookieOptions ...context.CookieOpt
 		sid := s.config.SessionIDGenerator(ctx)
 
 		sess := s.provider.Init(s, sid, s.config.Expires)
+		// n := s.provider.db.Len(sid)
+		// fmt.Printf("db.Len(%s) = %d\n", sid, n)
+		// if n > 0 {
+		// 	s.provider.db.Visit(sid, func(key string, value interface{}) {
+		// 		fmt.Printf("%s=%s\n", key, value)
+		// 	})
+		// }
 		sess.isNew = s.provider.db.Len(sid) == 0
 
 		s.updateCookie(ctx, sid, s.config.Expires, cookieOptions...)