c60cc7ca75
Some checks failed
CI / Run tests and lints (push) Failing after 29s
CI / Build CLI docs (push) Has been skipped
CI / Run Playwright end-to-end tests (mysql) (push) Failing after 3s
CI / Run Playwright end-to-end tests (postgres) (push) Failing after 2s
CI / Run Playwright end-to-end tests (sqlite) (push) Failing after 3s
CodeQL / Analyze (go) (push) Failing after 4s
CodeQL / Analyze (javascript) (push) Failing after 3s
Docker Image Scanners / scanners (push) Failing after 4s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Run end-to-end tests (postgres) (push) Failing after 19s
CI / Run end-to-end tests (sqlite) (push) Failing after 2s
CI / Run Playwright end-to-end tests (cockroach) (push) Failing after 2s
CI / Draft newsletter (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
// Copyright © 2023 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package sql
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/gofrs/uuid"
|
|
|
|
"github.com/ory/x/otelx"
|
|
"github.com/ory/x/sqlcon"
|
|
|
|
"github.com/ory/kratos/persistence/sql/update"
|
|
"github.com/ory/kratos/selfservice/flow/registration"
|
|
)
|
|
|
|
func (p *Persister) CreateRegistrationFlow(ctx context.Context, r *registration.Flow) (err error) {
|
|
ctx, span := p.r.Tracer(ctx).Tracer().Start(ctx, "persistence.sql.CreateRegistrationFlow")
|
|
defer otelx.End(span, &err)
|
|
|
|
r.NID = p.NetworkID(ctx)
|
|
r.EnsureInternalContext()
|
|
return p.GetConnection(ctx).Create(r)
|
|
}
|
|
|
|
func (p *Persister) UpdateRegistrationFlow(ctx context.Context, r *registration.Flow) (err error) {
|
|
ctx, span := p.r.Tracer(ctx).Tracer().Start(ctx, "persistence.sql.UpdateRegistrationFlow")
|
|
defer otelx.End(span, &err)
|
|
|
|
r.EnsureInternalContext()
|
|
cp := *r
|
|
cp.NID = p.NetworkID(ctx)
|
|
return update.Generic(ctx, p.GetConnection(ctx), p.r.Tracer(ctx).Tracer(), cp)
|
|
}
|
|
|
|
func (p *Persister) GetRegistrationFlow(ctx context.Context, id uuid.UUID) (_ *registration.Flow, err error) {
|
|
ctx, span := p.r.Tracer(ctx).Tracer().Start(ctx, "persistence.sql.GetRegistrationFlow")
|
|
defer otelx.End(span, &err)
|
|
|
|
var r registration.Flow
|
|
if err := p.GetConnection(ctx).Where("id = ? AND nid = ?",
|
|
id, p.NetworkID(ctx)).First(&r); err != nil {
|
|
return nil, sqlcon.HandleError(err)
|
|
}
|
|
|
|
return &r, nil
|
|
}
|
|
|
|
func (p *Persister) DeleteExpiredRegistrationFlows(ctx context.Context, expiresAt time.Time, limit int) (err error) {
|
|
ctx, span := p.r.Tracer(ctx).Tracer().Start(ctx, "persistence.sql.DeleteExpiredRegistrationFlows")
|
|
defer otelx.End(span, &err)
|
|
//#nosec G201 -- TableName is static
|
|
err = p.GetConnection(ctx).RawQuery(fmt.Sprintf(
|
|
"DELETE FROM %[1]s WHERE id in (SELECT id FROM (SELECT id FROM %[1]s c WHERE expires_at <= ? and nid = ? ORDER BY expires_at ASC LIMIT ?) AS s)",
|
|
registration.Flow{}.TableName(),
|
|
),
|
|
expiresAt,
|
|
p.NetworkID(ctx),
|
|
limit,
|
|
).Exec()
|
|
|
|
return sqlcon.HandleError(err)
|
|
}
|