mongodb용 세션 추가

session/provider.go

@@ -6,10 +6,16 @@ import (
 
 	"github.com/go-redis/redis/v8"
 	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo/options"
 	"repositories.action2quare.com/ayo/gocommon"
 )
 
-type Provider struct {
+type Provider interface {
+	Update(key string, input *Authorization) error
+	Delete(key string) error
+}
+
+type provider_redis struct {
 	redisClient   *redis.Client
 	updateChannel string
 	deleteChannel string
@@ -17,13 +23,38 @@ type Provider struct {
 	ctx           context.Context
 }
 
-func NewProvider(ctx context.Context, redisUrl string, ttl time.Duration) (*Provider, error) {
+type provider_mongo struct {
+	mongoClient gocommon.MongoClient
+}
+
+func NewProviderWithMongo(ctx context.Context, mongoUrl string, dbname string, ttl time.Duration) (Provider, error) {
+	mc, err := gocommon.NewMongoClient(ctx, mongoUrl, dbname)
+	if err != nil {
+		return nil, err
+	}
+
+	if err = mc.MakeUniqueIndices(session_collection_name, map[string]bson.D{
+		"key": {{Key: "key", Value: 1}},
+	}); err != nil {
+		return nil, err
+	}
+
+	if err := mc.MakeExpireIndex(session_collection_name, int32(ttl.Seconds())); err != nil {
+		return nil, err
+	}
+
+	return &provider_mongo{
+		mongoClient: mc,
+	}, nil
+}
+
+func NewProviderWithRedis(ctx context.Context, redisUrl string, ttl time.Duration) (Provider, error) {
 	redisClient, err := gocommon.NewRedisClient(redisUrl)
 	if err != nil {
 		return nil, err
 	}
 
-	return &Provider{
+	return &provider_redis{
 		redisClient:   redisClient,
 		updateChannel: communication_channel_name_prefix + "_u",
 		deleteChannel: communication_channel_name_prefix + "_d",
@@ -32,7 +63,7 @@ func NewProvider(ctx context.Context, redisUrl string, ttl time.Duration) (*Prov
 	}, nil
 }
 
-func (p *Provider) Update(key string, input *Authorization) error {
+func (p *provider_redis) Update(key string, input *Authorization) error {
 	bt, err := bson.Marshal(input)
 	if err != nil {
 		return err
@@ -47,7 +78,7 @@ func (p *Provider) Update(key string, input *Authorization) error {
 	return err
 }
 
-func (p *Provider) Delete(key string) error {
+func (p *provider_redis) Delete(key string) error {
 	cnt, err := p.redisClient.Del(p.ctx, key).Result()
 	if err != nil {
 		return err
@@ -59,3 +90,23 @@ func (p *Provider) Delete(key string) error {
 
 	return err
 }
+
+func (p *provider_mongo) Update(key string, input *Authorization) error {
+	_, _, err := p.mongoClient.Update(session_collection_name, bson.M{
+		"key": key,
+	}, bson.M{
+		"$set": input,
+		"$currentDate": bson.M{
+			"_ts": bson.M{"$type": "date"},
+		},
+	}, options.Update().SetUpsert(true))
+
+	return err
+}
+
+func (p *provider_mongo) Delete(key string) error {
+	_, err := p.mongoClient.Delete(session_collection_name, bson.M{
+		"key": key,
+	})
+	return err
+}