세션 provider, consumer 인터페이스 변경

session/common.go

@@ -49,7 +49,7 @@ func MakeAuthrizationFromStringMap(src map[string]string) Authorization {
 
 type Provider interface {
 	New(*Authorization) (string, error)
-	Invalidate(primitive.ObjectID) error
+	RevokeAll(primitive.ObjectID) error
 	Query(string) (Authorization, error)
 	Touch(string) (bool, error)
 }
@@ -57,7 +57,8 @@ type Provider interface {
 type Consumer interface {
 	Query(string) (Authorization, error)
 	Touch(string) (Authorization, error)
-	IsInvalidated(primitive.ObjectID) bool
+	IsRevoked(primitive.ObjectID) bool
+	Revoke(string)
 	RegisterOnSessionInvalidated(func(primitive.ObjectID))
 }
 

session/impl_mongo.go

@@ -64,7 +64,7 @@ func (p *provider_mongo) New(input *Authorization) (string, error) {
 	return string(storagekey_to_publickey(sk)), err
 }
 
-func (p *provider_mongo) Invalidate(acc primitive.ObjectID) error {
+func (p *provider_mongo) RevokeAll(acc primitive.ObjectID) error {
 	_, err := p.mongoClient.Delete(session_collection_name, bson.M{
 		"_id": acc,
 	})
@@ -338,7 +338,23 @@ func (c *consumer_mongo) Touch(pk string) (Authorization, error) {
 	return *si.Auth, nil
 }
 
-func (c *consumer_mongo) IsInvalidated(id primitive.ObjectID) bool {
+func (c *consumer_mongo) Revoke(pk string) {
+	sk := publickey_to_storagekey(publickey(pk))
+	_, err := c.mongoClient.Delete(session_collection_name, bson.M{
+		"key": sk,
+	})
+
+	if err == nil {
+		for id, v := range c.ids {
+			if v == sk {
+				delete(c.ids, id)
+				break
+			}
+		}
+	}
+}
+
+func (c *consumer_mongo) IsRevoked(id primitive.ObjectID) bool {
 	_, ok := c.ids[id]
 	return !ok
 }

session/impl_redis.go

@@ -81,7 +81,7 @@ func (p *provider_redis) New(input *Authorization) (string, error) {
 	return string(pk), err
 }
 
-func (p *provider_redis) Invalidate(account primitive.ObjectID) error {
+func (p *provider_redis) RevokeAll(account primitive.ObjectID) error {
 	prefix := account.Hex()
 	sks, err := p.redisClient.Keys(p.ctx, prefix+"*").Result()
 	if err != nil {
@@ -130,7 +130,8 @@ func (p *provider_redis) Touch(pk string) (bool, error) {
 
 type consumer_redis struct {
 	consumer_common[*sessionRedis]
-	redisClient *redis.Client
+	redisClient   *redis.Client
+	deleteChannel string
 }
 
 func newConsumerWithRedis(ctx context.Context, redisUrl string, ttl time.Duration) (Consumer, error) {
@@ -139,6 +140,9 @@ func newConsumerWithRedis(ctx context.Context, redisUrl string, ttl time.Duratio
 		return nil, err
 	}
 
+	deleteChannel := fmt.Sprintf("%s_%d_d", communication_channel_name_prefix, redisClient.Options().DB)
+	sub := redisClient.Subscribe(ctx, deleteChannel)
+
 	consumer := &consumer_redis{
 		consumer_common: consumer_common[*sessionRedis]{
 			ttl:       ttl,
@@ -146,12 +150,10 @@ func newConsumerWithRedis(ctx context.Context, redisUrl string, ttl time.Duratio
 			stages:    [2]*cache_stage[*sessionRedis]{make_cache_stage[*sessionRedis](), make_cache_stage[*sessionRedis]()},
 			startTime: time.Now(),
 		},
-		redisClient: redisClient,
+		redisClient:   redisClient,
+		deleteChannel: deleteChannel,
 	}
 
-	deleteChannel := fmt.Sprintf("%s_%d_d", communication_channel_name_prefix, redisClient.Options().DB)
-	sub := redisClient.Subscribe(ctx, deleteChannel)
-
 	go func() {
 		stageswitch := time.Now().Add(ttl)
 		tickTimer := time.After(ttl)
@@ -317,7 +319,14 @@ func (c *consumer_redis) Touch(pk string) (Authorization, error) {
 	return Authorization{}, nil
 }
 
-func (c *consumer_redis) IsInvalidated(accid primitive.ObjectID) bool {
+func (c *consumer_redis) Revoke(pk string) {
+	sk := publickey_to_storagekey(publickey(pk))
+
+	c.redisClient.Del(c.ctx, string(sk))
+	c.redisClient.Publish(c.ctx, c.deleteChannel, string(sk)).Result()
+}
+
+func (c *consumer_redis) IsRevoked(accid primitive.ObjectID) bool {
 	sk := make_storagekey(accid)
 
 	c.lock.Lock()

session/session_test.go

@@ -75,7 +75,7 @@ func TestExpTable(t *testing.T) {
 	time.Sleep(2 * time.Second)
 
 	time.Sleep(2 * time.Second)
-	pv.Invalidate(au1.Account)
+	pv.RevokeAll(au1.Account)
 
 	cs.Touch(sk1)
 	time.Sleep(2 * time.Second)