forked from baron/baron-sso
62 lines
1.9 KiB
Go
62 lines
1.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"baron-sso-backend/internal/domain"
|
|
"context"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type KetoOutboxRepository interface {
|
|
Create(ctx context.Context, entry *domain.KetoOutbox) error
|
|
CreateWithTx(tx *gorm.DB, entry *domain.KetoOutbox) error
|
|
FindPending(ctx context.Context, limit int) ([]domain.KetoOutbox, error)
|
|
UpdateStatus(ctx context.Context, id string, status string, retryCount int, lastError string) error
|
|
MarkProcessed(ctx context.Context, id string) error
|
|
}
|
|
|
|
type ketoOutboxRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewKetoOutboxRepository(db *gorm.DB) KetoOutboxRepository {
|
|
return &ketoOutboxRepository{db: db}
|
|
}
|
|
|
|
func (r *ketoOutboxRepository) Create(ctx context.Context, entry *domain.KetoOutbox) error {
|
|
return r.db.WithContext(ctx).Create(entry).Error
|
|
}
|
|
|
|
func (r *ketoOutboxRepository) CreateWithTx(tx *gorm.DB, entry *domain.KetoOutbox) error {
|
|
return tx.Create(entry).Error
|
|
}
|
|
|
|
func (r *ketoOutboxRepository) FindPending(ctx context.Context, limit int) ([]domain.KetoOutbox, error) {
|
|
var entries []domain.KetoOutbox
|
|
err := r.db.WithContext(ctx).
|
|
Where("status = ?", domain.KetoOutboxStatusPending).
|
|
Order("created_at asc").
|
|
Limit(limit).
|
|
Find(&entries).Error
|
|
return entries, err
|
|
}
|
|
|
|
func (r *ketoOutboxRepository) UpdateStatus(ctx context.Context, id string, status string, retryCount int, lastError string) error {
|
|
return r.db.WithContext(ctx).Model(&domain.KetoOutbox{}).Where("id = ?", id).Updates(map[string]interface{}{
|
|
"status": status,
|
|
"retry_count": retryCount,
|
|
"last_error": lastError,
|
|
"updated_at": time.Now(),
|
|
}).Error
|
|
}
|
|
|
|
func (r *ketoOutboxRepository) MarkProcessed(ctx context.Context, id string) error {
|
|
now := time.Now()
|
|
return r.db.WithContext(ctx).Model(&domain.KetoOutbox{}).Where("id = ?", id).Updates(map[string]interface{}{
|
|
"status": domain.KetoOutboxStatusProcessed,
|
|
"processed_at": &now,
|
|
"updated_at": now,
|
|
}).Error
|
|
}
|