add context support

Author: LyricTian

README.md

@@ -18,6 +18,7 @@ $ go get -u -v gopkg.in/go-session/redis.v1
 package main
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 
@@ -36,7 +37,7 @@ func main() {
 	)
 
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-		store, err := session.Start(w, r)
+		store, err := session.Start(context.Background(), w, r)
 		if err != nil {
 			fmt.Fprint(w, err)
 			return
@@ -53,7 +54,7 @@ func main() {
 	})
 
 	http.HandleFunc("/foo", func(w http.ResponseWriter, r *http.Request) {
-		store, err := session.Start(w, r)
+		store, err := session.Start(context.Background(), w, r)
 		if err != nil {
 			fmt.Fprint(w, err)
 			return

redis.go

@@ -1,6 +1,7 @@
 package redis
 
 import (
+	"context"
 	"encoding/json"
 	"sync"
 	"time"
@@ -54,17 +55,17 @@ func (s *managerStore) parseValue(value string) (map[string]string, error) {
 	return values, nil
 }
 
-func (s *managerStore) Create(sid string, expired int64) (session.Store, error) {
+func (s *managerStore) Create(ctx context.Context, sid string, expired int64) (session.Store, error) {
 	values := make(map[string]string)
-	return &store{sid: sid, cli: s.cli, expired: expired, values: values}, nil
+	return &store{ctx: ctx, sid: sid, cli: s.cli, expired: expired, values: values}, nil
 }
 
-func (s *managerStore) Update(sid string, expired int64) (session.Store, error) {
+func (s *managerStore) Update(ctx context.Context, sid string, expired int64) (session.Store, error) {
 	value, err := s.getValue(sid)
 	if err != nil {
 		return nil, err
 	} else if value == "" {
-		return s.Create(sid, expired)
+		return s.Create(ctx, sid, expired)
 	}
 
 	cmd := s.cli.Set(sid, value, time.Duration(expired)*time.Second)
@@ -77,15 +78,15 @@ func (s *managerStore) Update(sid string, expired int64) (session.Store, error)
 		return nil, err
 	}
 
-	return &store{sid: sid, cli: s.cli, expired: expired, values: values}, nil
+	return &store{ctx: ctx, sid: sid, cli: s.cli, expired: expired, values: values}, nil
 }
 
-func (s *managerStore) Delete(sid string) error {
+func (s *managerStore) Delete(_ context.Context, sid string) error {
 	cmd := s.cli.Del(sid)
 	return cmd.Err()
 }
 
-func (s *managerStore) Check(sid string) (bool, error) {
+func (s *managerStore) Check(_ context.Context, sid string) (bool, error) {
 	cmd := s.cli.Get(sid)
 	if err := cmd.Err(); err != nil {
 		if err == redis.Nil {
@@ -106,6 +107,11 @@ type store struct {
 	expired int64
 	values  map[string]string
 	sync.RWMutex
+	ctx context.Context
+}
+
+func (s *store) Context() context.Context {
+	return s.ctx
 }
 
 func (s *store) SessionID() string {

redis_test.go

@@ -17,7 +17,7 @@ func TestStore(t *testing.T) {
 	})
 	defer mstore.Close()
 
-	store, err := mstore.Create("store", 2)
+	store, err := mstore.Create(nil, "store", 2)
 	if err != nil {
 		t.Error(err.Error())
 		return
@@ -88,7 +88,7 @@ func TestManagerStore(t *testing.T) {
 	defer mstore.Close()
 
 	sid := "manager"
-	store, err := mstore.Create(sid, 2)
+	store, err := mstore.Create(nil, sid, 2)
 	if err != nil {
 		t.Error(err.Error())
 		return
@@ -101,7 +101,7 @@ func TestManagerStore(t *testing.T) {
 		return
 	}
 
-	store, err = mstore.Update(sid, 2)
+	store, err = mstore.Update(nil, sid, 2)
 	if err != nil {
 		t.Error(err.Error())
 		return
@@ -113,13 +113,13 @@ func TestManagerStore(t *testing.T) {
 		return
 	}
 
-	err = mstore.Delete(sid)
+	err = mstore.Delete(nil, sid)
 	if err != nil {
 		t.Error(err.Error())
 		return
 	}
 
-	exists, err := mstore.Check(sid)
+	exists, err := mstore.Check(nil, sid)
 	if err != nil {
 		t.Error(err.Error())
 		return
@@ -136,7 +136,7 @@ func TestStoreWithExpired(t *testing.T) {
 	defer mstore.Close()
 
 	sid := "test_store_expired"
-	store, err := mstore.Create(sid, 1)
+	store, err := mstore.Create(nil, sid, 1)
 	if err != nil {
 		t.Error(err.Error())
 		return
@@ -157,7 +157,7 @@ func TestStoreWithExpired(t *testing.T) {
 
 	time.Sleep(time.Second * 2)
 
-	exists, err := mstore.Check(sid)
+	exists, err := mstore.Check(nil, sid)
 	if err != nil {
 		t.Error(err.Error())
 		return