mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
56b916a2b0
* begin init checks for projections * first projection checks * debug notification providers with query fixes * more projections and first index * more projections * more projections * finish projections * fix tests (remove db name) * create tables in setup * fix logging / error handling * add tenant to views * rename tenant to instance_id * add instance_id to all projections * add instance_id to all queries * correct instance_id on projections * add instance_id to failed_events * use separate context for instance * implement features projection * implement features projection * remove unique constraint from setup when migration failed * add error to failed setup event * add instance_id to primary keys * fix IAM projection * remove old migrations folder * fix keysFromYAML test
93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package crdb
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/caos/logging"
|
|
|
|
"github.com/caos/zitadel/internal/errors"
|
|
"github.com/caos/zitadel/internal/id"
|
|
)
|
|
|
|
const (
|
|
lockStmtFormat = "INSERT INTO %[1]s" +
|
|
" (locker_id, locked_until, projection_name) VALUES ($1, now()+$2::INTERVAL, $3)" +
|
|
" ON CONFLICT (projection_name)" +
|
|
" DO UPDATE SET locker_id = $1, locked_until = now()+$2::INTERVAL" +
|
|
" WHERE %[1]s.projection_name = $3 AND (%[1]s.locker_id = $1 OR %[1]s.locked_until < now())"
|
|
)
|
|
|
|
type Locker interface {
|
|
Lock(ctx context.Context, lockDuration time.Duration) <-chan error
|
|
Unlock() error
|
|
}
|
|
|
|
type locker struct {
|
|
client *sql.DB
|
|
lockStmt string
|
|
workerName string
|
|
projectionName string
|
|
}
|
|
|
|
func NewLocker(client *sql.DB, lockTable, projectionName string) Locker {
|
|
workerName, err := os.Hostname()
|
|
if err != nil || workerName == "" {
|
|
workerName, err = id.SonyFlakeGenerator.Next()
|
|
logging.OnError(err).Panic("unable to generate lockID")
|
|
}
|
|
return &locker{
|
|
client: client,
|
|
lockStmt: fmt.Sprintf(lockStmtFormat, lockTable),
|
|
workerName: workerName,
|
|
projectionName: projectionName,
|
|
}
|
|
}
|
|
|
|
func (h *locker) Lock(ctx context.Context, lockDuration time.Duration) <-chan error {
|
|
errs := make(chan error)
|
|
go h.handleLock(ctx, errs, lockDuration)
|
|
return errs
|
|
}
|
|
|
|
func (h *locker) handleLock(ctx context.Context, errs chan error, lockDuration time.Duration) {
|
|
renewLock := time.NewTimer(0)
|
|
for {
|
|
select {
|
|
case <-renewLock.C:
|
|
errs <- h.renewLock(ctx, lockDuration)
|
|
//refresh the lock 500ms before it times out. 500ms should be enough for one transaction
|
|
renewLock.Reset(lockDuration - (500 * time.Millisecond))
|
|
case <-ctx.Done():
|
|
close(errs)
|
|
renewLock.Stop()
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (h *locker) renewLock(ctx context.Context, lockDuration time.Duration) error {
|
|
//the unit of crdb interval is seconds (https://www.cockroachlabs.com/docs/stable/interval.html).
|
|
res, err := h.client.Exec(h.lockStmt, h.workerName, lockDuration.Seconds(), h.projectionName)
|
|
if err != nil {
|
|
return errors.ThrowInternal(err, "CRDB-uaDoR", "unable to execute lock")
|
|
}
|
|
|
|
if rows, _ := res.RowsAffected(); rows == 0 {
|
|
return errors.ThrowAlreadyExists(nil, "CRDB-mmi4J", "projection already locked")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (h *locker) Unlock() error {
|
|
_, err := h.client.Exec(h.lockStmt, h.workerName, float64(0), h.projectionName)
|
|
if err != nil {
|
|
return errors.ThrowUnknown(err, "CRDB-JjfwO", "unlock failed")
|
|
}
|
|
return nil
|
|
}
|