feat: Add ent-based sqlite3 storage
Signed-off-by: m.nabokikh <maksim.nabokikh@flant.com>
This commit is contained in:
95
storage/ent/client/main.go
Normal file
95
storage/ent/client/main.go
Normal file
@@ -0,0 +1,95 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"context"
|
||||
"hash"
|
||||
"time"
|
||||
|
||||
"github.com/dexidp/dex/storage"
|
||||
"github.com/dexidp/dex/storage/ent/db"
|
||||
"github.com/dexidp/dex/storage/ent/db/authcode"
|
||||
"github.com/dexidp/dex/storage/ent/db/authrequest"
|
||||
"github.com/dexidp/dex/storage/ent/db/devicerequest"
|
||||
"github.com/dexidp/dex/storage/ent/db/devicetoken"
|
||||
"github.com/dexidp/dex/storage/ent/db/migrate"
|
||||
)
|
||||
|
||||
var _ storage.Storage = (*Database)(nil)
|
||||
|
||||
type Database struct {
|
||||
client *db.Client
|
||||
hasher func() hash.Hash
|
||||
}
|
||||
|
||||
// NewDatabase returns new database client with set options.
|
||||
func NewDatabase(opts ...func(*Database)) *Database {
|
||||
database := &Database{}
|
||||
for _, f := range opts {
|
||||
f(database)
|
||||
}
|
||||
return database
|
||||
}
|
||||
|
||||
// WithClient sets client option of a Database object.
|
||||
func WithClient(c *db.Client) func(*Database) {
|
||||
return func(s *Database) {
|
||||
s.client = c
|
||||
}
|
||||
}
|
||||
|
||||
// WithHasher sets client option of a Database object.
|
||||
func WithHasher(h func() hash.Hash) func(*Database) {
|
||||
return func(s *Database) {
|
||||
s.hasher = h
|
||||
}
|
||||
}
|
||||
|
||||
// Schema exposes migration schema to perform migrations.
|
||||
func (d *Database) Schema() *migrate.Schema {
|
||||
return d.client.Schema
|
||||
}
|
||||
|
||||
// Close calls the corresponding method of the ent database client.
|
||||
func (d *Database) Close() error {
|
||||
return d.client.Close()
|
||||
}
|
||||
|
||||
// GarbageCollect removes expired entities from the database.
|
||||
func (d *Database) GarbageCollect(now time.Time) (storage.GCResult, error) {
|
||||
result := storage.GCResult{}
|
||||
utcNow := now.UTC()
|
||||
|
||||
q, err := d.client.AuthRequest.Delete().
|
||||
Where(authrequest.ExpiryLT(utcNow)).
|
||||
Exec(context.TODO())
|
||||
if err != nil {
|
||||
return result, convertDBError("gc auth request: %w", err)
|
||||
}
|
||||
result.AuthRequests = int64(q)
|
||||
|
||||
q, err = d.client.AuthCode.Delete().
|
||||
Where(authcode.ExpiryLT(utcNow)).
|
||||
Exec(context.TODO())
|
||||
if err != nil {
|
||||
return result, convertDBError("gc auth code: %w", err)
|
||||
}
|
||||
result.AuthCodes = int64(q)
|
||||
|
||||
q, err = d.client.DeviceRequest.Delete().
|
||||
Where(devicerequest.ExpiryLT(utcNow)).
|
||||
Exec(context.TODO())
|
||||
if err != nil {
|
||||
return result, convertDBError("gc device request: %w", err)
|
||||
}
|
||||
result.DeviceRequests = int64(q)
|
||||
|
||||
q, err = d.client.DeviceToken.Delete().
|
||||
Where(devicetoken.ExpiryLT(utcNow)).
|
||||
Exec(context.TODO())
|
||||
if err != nil {
|
||||
return result, convertDBError("gc device token: %w", err)
|
||||
}
|
||||
result.DeviceTokens = int64(q)
|
||||
|
||||
return result, err
|
||||
}
|
Reference in New Issue
Block a user