65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package refreshtokensdb
|
|
|
|
import (
|
|
"github.com/tech/sendico/pkg/db/repository"
|
|
ri "github.com/tech/sendico/pkg/db/repository/index"
|
|
"github.com/tech/sendico/pkg/db/template"
|
|
"github.com/tech/sendico/pkg/mlogger"
|
|
"github.com/tech/sendico/pkg/model"
|
|
"github.com/tech/sendico/pkg/mservice"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type RefreshTokenDB struct {
|
|
template.DBImp[*model.RefreshToken]
|
|
clients repository.Repository
|
|
}
|
|
|
|
func Create(logger mlogger.Logger, db *mongo.Database) (*RefreshTokenDB, error) {
|
|
p := &RefreshTokenDB{
|
|
DBImp: *template.Create[*model.RefreshToken](logger, mservice.RefreshTokens, db),
|
|
clients: repository.CreateMongoRepository(db, mservice.Clients),
|
|
}
|
|
|
|
if err := p.Repository.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{{Field: "token", Sort: ri.Asc}},
|
|
Unique: true,
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create unique token index", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
// Add unique constraint on account/client/device combination
|
|
if err := p.Repository.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{
|
|
{Field: "accountRef", Sort: ri.Asc},
|
|
{Field: "clientId", Sort: ri.Asc},
|
|
{Field: "deviceId", Sort: ri.Asc},
|
|
},
|
|
Unique: true,
|
|
Name: "unique_active_session",
|
|
PartialFilter: repository.Filter(IsRevokedField, false),
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create unique account/client/device index", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
if err := p.Repository.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{{Field: IsRevokedField, Sort: ri.Asc}},
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create unique token revokation status index", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
if err := p.clients.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{{Field: "clientId", Sort: ri.Asc}},
|
|
Unique: true,
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create unique client identifier index", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
return p, nil
|
|
}
|